package admin_agent

import (
	"context"
	"errors"
	"fmt"

	"tydata-server/app/main/api/internal/svc"
	"tydata-server/app/main/api/internal/types"
	"tydata-server/app/main/model"
	"tydata-server/common/xerr"

	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

type AdminUpdateAgentWalletBalanceLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewAdminUpdateAgentWalletBalanceLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateAgentWalletBalanceLogic {
	return &AdminUpdateAgentWalletBalanceLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *AdminUpdateAgentWalletBalanceLogic) AdminUpdateAgentWalletBalance(req *types.AdminUpdateAgentWalletBalanceReq) (resp *types.AdminUpdateAgentWalletBalanceResp, err error) {
	// 参数校验
	if req.AgentId <= 0 {
		return nil, xerr.NewErrMsg("代理ID无效")
	}
	if req.Amount == 0 {
		return nil, xerr.NewErrMsg("修改金额不能为0")
	}

	// 查询代理钱包信息
	wallet, err := l.svcCtx.AgentWalletModel.FindOneByAgentId(l.ctx, req.AgentId)
	if err != nil {
		if errors.Is(err, model.ErrNotFound) {
			return nil, xerr.NewErrMsg("代理钱包不存在")
		}
		return nil, err
	}

	// 计算新余额
	newBalance := wallet.Balance + req.Amount

	// 校验余额不能为负数
	if newBalance < 0 {
		return nil, xerr.NewErrMsg(fmt.Sprintf("操作后余额不能为负数，当前余额: %.2f，操作金额: %.2f", wallet.Balance, req.Amount))
	}

	// 更新余额
	updateErr := l.svcCtx.AgentWalletModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error {
		// 使用版本号更新
		wallet.Balance = newBalance
		err := l.svcCtx.AgentWalletModel.UpdateWithVersion(transCtx, session, wallet)
		if err != nil {
			return err
		}

		// TODO: 这里可以添加记录余额变更日志的逻辑
		l.Logger.Infof("代理钱包余额变更 - AgentId: %d, 原余额: %.2f, 变更金额: %.2f, 新余额: %.2f",
			req.AgentId, wallet.Balance-req.Amount, req.Amount, newBalance)

		return nil
	})

	if updateErr != nil {
		l.Logger.Errorf("更新代理钱包余额失败: %+v", updateErr)
		return nil, xerr.NewErrMsg("更新余额失败")
	}

	resp = &types.AdminUpdateAgentWalletBalanceResp{
		Success: true,
		Balance: newBalance,
	}
	return
}