package weChat

import (
	"context"
	"errors"
	"github.com/wechatpay-apiv3/wechatpay-go/core"
	"github.com/wechatpay-apiv3/wechatpay-go/core/option"
	"github.com/wechatpay-apiv3/wechatpay-go/services/payments"
	"github.com/wechatpay-apiv3/wechatpay-go/utils"
	"hepa-calc-api/config"
	"time"
)

// 创建客户端
func createClient() (*core.Client, error) {
	mchId := config.C.Wechat.Pay1281030301.MchId                                           // 商户号
	mchCertificateSerialNumber := config.C.Wechat.Pay1281030301.MchCertificateSerialNumber // 商户证书序列号
	v3ApiSecret := config.C.Wechat.Pay1281030301.V3ApiSecret                               // 商户APIv3密钥
	privateKeyPath := config.C.Wechat.Pay1281030301.PrivateKey                             // 商户私钥文件地址

	if mchId == "" {
		return nil, errors.New("商户号错误")
	}

	mchPrivateKey, err := utils.LoadPrivateKeyWithPath(privateKeyPath)
	if err != nil {
		return nil, errors.New("微信支付生成失败")
	}

	ctx := context.Background()
	// 使用商户私钥等初始化 client，并使它具有自动定时获取微信支付平台证书的能力
	opts := []core.ClientOption{
		option.WithWechatPayAutoAuthCipher(mchId, mchCertificateSerialNumber, mchPrivateKey, v3ApiSecret),
	}

	client, err := core.NewClient(ctx, opts...)
	if err != nil {
		return nil, errors.New(err.Error())
	}

	return client, nil
}

type WxPayResult struct {
	OrderStatus *int       `json:"order_status"` // 订单状态（1:待支付 2:已完成 3:已取消）
	PayStatus   *int       `json:"pay_status"`   // 支付状态（1:未支付 2:已支付 3:支付中 4:支付失败 5:支付超时 6:支付关闭 7:已撤销 8:转入退款）
	PayTime     *time.Time `json:"pay_time"`     // 支付时间
}

// HandlePayStatus 处理支付状态
func HandlePayStatus(t *payments.Transaction) (w *WxPayResult, err error) {
	// 支付成功
	if *t.TradeState == "SUCCESS" {
		orderStatus := 2
		w.OrderStatus = &orderStatus

		payStatus := 2
		w.PayStatus = &payStatus

		parse, err := time.Parse("2006-01-02T15:04:05+07:00", *t.SuccessTime)
		if err != nil {
			return nil, errors.New("支付时间处理错误")
		}

		w.PayTime = &parse
		if err != nil {
			return nil, errors.New("支付时间错误")
		}
	}

	switch *t.TradeState {
	case "SUCCESS": // 支付成功
		orderStatus := 2
		w.OrderStatus = &orderStatus

		payStatus := 2
		w.PayStatus = &payStatus

		parse, err := time.Parse("2006-01-02T15:04:05+07:00", *t.SuccessTime)
		if err != nil {
			return nil, errors.New("支付时间处理错误")
		}

		w.PayTime = &parse
		if err != nil {
			return nil, errors.New("支付时间错误")
		}
	case "CLOSED": // 已关闭
		payStatus := 6
		w.PayStatus = &payStatus
	case "REVOKED": // 已撤销（付款码支付）
		payStatus := 7
		w.PayStatus = &payStatus
	case "USERPAYING": // 用户支付中（付款码支付）
		payStatus := 3
		w.PayStatus = &payStatus
	case "PAYERROR": // 支付失败(其他原因，如银行返回失败)
		payStatus := 4
		w.PayStatus = &payStatus
	default:
		return nil, errors.New("未知支付状态")
	}

	return w, nil
}