geekai/api/service/payment/alipay_service.go

package payment

// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
// * Copyright 2023 The Geek-AI Authors. All rights reserved.
// * Use of this source code is governed by a Apache-2.0 license
// * that can be found in the LICENSE file.
// * @Author yangjian102621@163.com
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

import (
	"context"
	"fmt"
	"geekai/core/types"
	logger2 "geekai/logger"
	"github.com/go-pay/gopay"
	"github.com/go-pay/gopay/alipay"
	"net/http"
	"os"
)

type AlipayService struct {
	config *types.AlipayConfig
	client *alipay.Client
}

var logger = logger2.GetLogger()

func NewAlipayService(appConfig *types.AppConfig) (*AlipayService, error) {
	config := appConfig.AlipayConfig
	if !config.Enabled {
		logger.Info("Disabled Alipay service")
		return nil, nil
	}
	priKey, err := readKey(config.PrivateKey)
	if err != nil {
		return nil, fmt.Errorf("error with read App Private key: %v", err)
	}

	client, err := alipay.NewClient(config.AppId, priKey, !config.SandBox)
	if err != nil {
		return nil, fmt.Errorf("error with initialize alipay service: %v", err)
	}

	client.DebugSwitch = gopay.DebugOn

	client.SetLocation(alipay.LocationShanghai). // 设置时区，不设置或出错均为默认服务器时间
		SetCharset(alipay.UTF8). // 设置字符编码，不设置默认 utf-8
		SetSignType(alipay.RSA2). // 设置签名类型，不设置默认 RSA2
		SetReturnUrl(config.ReturnURL). // 设置返回URL
		SetNotifyUrl(config.NotifyURL)

	if err = client.SetCertSnByPath(config.PublicKey, config.RootCert, config.AlipayPublicKey); err != nil {
		return nil, fmt.Errorf("error with load payment public key: %v", err)
	}

	return &AlipayService{config: &config, client: client}, nil
}

func (s *AlipayService) PayUrlMobile(outTradeNo string, Amount string, subject string) (string, error) {
	bm := make(gopay.BodyMap)
	bm.Set("subject", subject)
	bm.Set("out_trade_no", outTradeNo)
	bm.Set("quit_url", s.config.ReturnURL)
	bm.Set("total_amount", Amount)
	bm.Set("product_code", "QUICK_WAP_WAY")
	return s.client.TradeWapPay(context.Background(), bm)
}

func (s *AlipayService) PayUrlPc(outTradeNo string, amount string, subject string) (string, error) {
	bm := make(gopay.BodyMap)
	bm.Set("subject", subject)
	bm.Set("out_trade_no", outTradeNo)
	bm.Set("total_amount", amount)
	bm.Set("product_code", "FAST_INSTANT_TRADE_PAY")
	return s.client.TradePagePay(context.Background(), bm)
}

// TradeVerify 交易验证
func (s *AlipayService) TradeVerify(request *http.Request) NotifyVo {
	notifyReq, err := alipay.ParseNotifyToBodyMap(request) // c.Request 是 gin 框架的写法
	if err != nil {
		return NotifyVo{
			Status:  0,
			Message: "error with parse notify request: " + err.Error(),
		}
	}

	_, err = alipay.VerifySignWithCert(s.config.AlipayPublicKey, notifyReq)
	if err != nil {
		return NotifyVo{
			Status:  0,
			Message: "error with verify sign: " + err.Error(),
		}
	}

	return s.TradeQuery(request.Form.Get("out_trade_no"))
}

func (s *AlipayService) TradeQuery(outTradeNo string) NotifyVo {
	bm := make(gopay.BodyMap)
	bm.Set("out_trade_no", outTradeNo)

	//查询订单
	rsp, err := s.client.TradeQuery(context.Background(), bm)
	if err != nil {
		return NotifyVo{
			Status:  0,
			Message: "异步查询验证订单信息发生错误" + outTradeNo + err.Error(),
		}
	}

	if rsp.Response.TradeStatus == "TRADE_SUCCESS" {
		return NotifyVo{
			Status:     1,
			OutTradeNo: rsp.Response.OutTradeNo,
			TradeNo:    rsp.Response.TradeNo,
			Amount:     rsp.Response.TotalAmount,
			Subject:    rsp.Response.Subject,
			Message:    "OK",
		}
	} else {
		return NotifyVo{
			Status:  0,
			Message: "异步查询验证订单信息发生错误" + outTradeNo,
		}
	}
}

func readKey(filename string) (string, error) {
	data, err := os.ReadFile(filename)
	if err != nil {
		return "", err
	}
	return string(data), nil
}

type NotifyVo struct {
	Status     int
	OutTradeNo string
	TradeNo    string
	Amount     string
	Message    string
	Subject    string
}

func (v NotifyVo) Success() bool {
	return v.Status == 1
}