// Copyright 2025 The OpenAgent Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package object

import (
	"fmt"
	"strings"

	"github.com/beego/beego/logs"
	"github.com/robfig/cron/v3"
	"github.com/the-open-agent/openagent/auth"
	"github.com/the-open-agent/openagent/conf"
	"github.com/the-open-agent/openagent/util"
)

var OpenAgentHost = ""

// createTransactionFromMessage creates a transaction object from a message.
// This is a helper function to reduce code duplication.
func createTransactionFromMessage(message *Message) *auth.Transaction {
	transaction := &auth.Transaction{
		Owner:       conf.GetConfigString("casdoorOrganization"),
		CreatedTime: message.CreatedTime,
		Application: conf.GetConfigString("casdoorApplication"),
		Domain:      OpenAgentHost,
		Category:    "OpenAgent Chat",
		Type:        message.Chat,
		Subtype:     message.Name,
		Provider:    message.ModelProvider,
		User:        message.User,
		Tag:         "User",
		Amount:      -message.Price,
		Currency:    message.Currency,
		Payment:     "",
		State:       "Paid",
	}

	if util.IsAnonymousUserByUsername(message.User) {
		transaction.Tag = "Organization"
	}

	return transaction
}

// ValidateTransactionForMessage validates a transaction in dry run mode before committing it.
// This checks if the user has sufficient balance without actually creating the transaction.
func ValidateTransactionForMessage(message *Message) error {
	if conf.GetConfigString("casdoorEndpoint") == "" {
		return nil
	}

	site, err := GetBuiltInSite()
	if err != nil || site == nil || !site.CheckUserBalance {
		return nil
	}

	// Only validate transaction if message has a price
	if message.Price <= 0 {
		return nil
	}

	// Create transaction object
	transaction := createTransactionFromMessage(message)

	// Validate transaction via Casdoor SDK with dry run mode
	_, _, err = auth.AddTransactionWithDryRun(transaction, true)
	if err != nil {
		return fmt.Errorf("failed to validate transaction: %s", err.Error())
	}

	return nil
}

// AddTransactionForMessage creates a transaction in Casdoor for a message with price,
// sets the message's TransactionId, and if transaction creation fails, updates the message's ErrorText field in the database and returns an error to the caller.
func AddTransactionForMessage(message *Message) error {
	if conf.GetConfigString("casdoorEndpoint") == "" {
		return nil
	}

	// Only create transaction if message has a price
	if message.Price <= 0 {
		return nil
	}

	// Create transaction object
	transaction := createTransactionFromMessage(message)

	// Add transaction via Casdoor SDK
	_, transactionName, err := auth.AddTransaction(transaction)
	if err != nil {
		message.ErrorText = fmt.Sprintf("failed to add transaction: %s", err.Error())

		_, errUpdate := UpdateMessage(message.GetId(), message, false)
		if errUpdate != nil {
			return fmt.Errorf("failed to update message: %s", errUpdate.Error())
		}

		return fmt.Errorf("failed to add transaction: %s", err.Error())
	}

	message.TransactionId = util.GetId(transaction.Owner, transactionName)

	return nil
}

func retryFailedTransaction() error {
	if conf.GetConfigString("casdoorEndpoint") == "" {
		return nil
	}

	messages, err := GetGlobalFailMessages()
	if err != nil {
		return err
	}

	for _, message := range messages {
		if strings.HasPrefix(message.ErrorText, "failed to add transaction") {
			err = AddTransactionForMessage(message)
			if err != nil {
				return err
			}

			message.ErrorText = ""
			_, err = UpdateMessage(message.GetId(), message, false)
			if err != nil {
				return fmt.Errorf("failed to update message: %s", err.Error())
			}
		}
	}

	return nil
}

func retryFailedTransactionNoError() {
	err := retryFailedTransaction()
	if err != nil {
		logs.Error("retryFailedTransactionNoError() error: %s", err.Error())
	}
}

func InitMessageTransactionRetry() {
	cronJob := cron.New()
	schedule := "@every 5m"
	_, err := cronJob.AddFunc(schedule, retryFailedTransactionNoError)
	if err != nil {
		panic(err)
	}

	cronJob.Start()
}
