package pay

import (
	"aedata-server/app/main/api/internal/service"
	"aedata-server/pkg/lzkit/lzUtils"
	"context"
	"net/http"
	"time"

	"aedata-server/app/main/api/internal/svc"

	"github.com/wechatpay-apiv3/wechatpay-go/services/payments"
	"github.com/zeromicro/go-zero/core/logx"
)

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

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

func (l *WechatPayCallbackLogic) WechatPayCallback(w http.ResponseWriter, r *http.Request) error {
	notification, err := l.svcCtx.WechatPayService.HandleWechatPayNotification(l.ctx, r)
	if err != nil {
		logx.Errorf("微信支付回调，%v", err)
		return nil
	}

	// 兼容旧订单，假设没有前缀的是查询订单
	return l.handleQueryOrderPayment(w, notification)
}

// 处理查询订单支付
func (l *WechatPayCallbackLogic) handleQueryOrderPayment(w http.ResponseWriter, notification *payments.Transaction) error {
	order, findOrderErr := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, *notification.OutTradeNo)
	if findOrderErr != nil {
		logx.Errorf("微信支付回调，查找订单信息失败: %+v", findOrderErr)
		return nil
	}

	amount := lzUtils.ToWechatAmount(order.Amount)
	if amount != *notification.Amount.Total {
		logx.Errorf("微信支付回调，金额不一致")
		return nil
	}

	if order.Status != "pending" {
		w.WriteHeader(http.StatusOK)
		_, _ = w.Write([]byte("success"))
		return nil
	}

	switch *notification.TradeState {
	case service.TradeStateSuccess:
		order.Status = "paid"
		order.PayTime = lzUtils.TimeToNullTime(time.Now())
	case service.TradeStateClosed:
		order.Status = "closed"
		order.CloseTime = lzUtils.TimeToNullTime(time.Now())
	case service.TradeStateRevoked:
		order.Status = "failed"
	default:
		return nil
	}

	order.PlatformOrderId = lzUtils.StringToNullString(*notification.TransactionId)
	if updateErr := l.svcCtx.OrderModel.UpdateWithVersion(l.ctx, nil, order); updateErr != nil {
		logx.Errorf("微信支付回调，更新订单失败%+v", updateErr)
		return nil
	}

	if order.Status == "paid" {
		if asyncErr := l.svcCtx.AsynqService.SendQueryTask(order.Id); asyncErr != nil {
			logx.Errorf("异步任务调度失败: %v", asyncErr)
			return asyncErr
		}
	}

	w.WriteHeader(http.StatusOK)
	_, _ = w.Write([]byte("success"))
	return nil
}