package pay

import (
	"context"
	"strings"

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

	"github.com/pkg/errors"
	"github.com/zeromicro/go-zero/core/logx"
)

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

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

func (l *PaymentCheckLogic) PaymentCheck(req *types.PaymentCheckReq) (resp *types.PaymentCheckResp, err error) {
	// 根据订单号前缀判断订单类型
	// 白名单下架订单：前缀 W_
	if strings.HasPrefix(req.OrderNo, "W_") {
		whitelistOrder, err := l.svcCtx.WhitelistOrderModel.FindOneByOrderNo(l.ctx, req.OrderNo)
		if err != nil {
			return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询白名单订单失败: %v", err)
		}

		status := "pending"
		switch whitelistOrder.Status {
		case 1:
			status = "pending"
		case 2:
			status = "paid"
		case 3:
			status = "closed"
		default:
			status = "pending"
		}

		return &types.PaymentCheckResp{
			Type:   "whitelist",
			Status: status,
		}, nil
	}

	// 升级订单：前缀 U_
	if strings.HasPrefix(req.OrderNo, "U_") {
		order, err := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, req.OrderNo)
		if err != nil {
			return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询升级订单失败: %v", err)
		}
		return &types.PaymentCheckResp{
			Type:   "agent_upgrade",
			Status: order.Status,
		}, nil
	}

	// 其他视为常规查询订单（包括代理推广查询）
	order, err := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, req.OrderNo)
	if err != nil {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询订单失败: %v", err)
	}
	return &types.PaymentCheckResp{
		Type:   "query",
		Status: order.Status,
	}, nil
}