mirror of
https://github.com/yangjian102621/geekai.git
synced 2026-04-21 10:34:26 +08:00
218 lines
6.6 KiB
Go
218 lines
6.6 KiB
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"
|
|
"geekai/utils"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/go-pay/gopay"
|
|
"github.com/go-pay/gopay/wechat/v3"
|
|
)
|
|
|
|
type WxPayService struct {
|
|
config *types.WxPayConfig
|
|
client *wechat.ClientV3
|
|
}
|
|
|
|
func NewWxpayService(sysConfig *types.SystemConfig) (*WxPayService, error) {
|
|
config := sysConfig.Payment.WxPay
|
|
if !config.Enabled {
|
|
logger.Debug("Disabled WechatPay service")
|
|
}
|
|
|
|
service := &WxPayService{config: &config}
|
|
if config.Enabled {
|
|
err := service.UpdateConfig(&config)
|
|
if err != nil {
|
|
logger.Errorf("微信支付服务初始化失败: %v", err)
|
|
}
|
|
}
|
|
|
|
return service, nil
|
|
}
|
|
|
|
func (s *WxPayService) UpdateConfig(config *types.WxPayConfig) error {
|
|
client, err := wechat.NewClientV3(config.MchId, config.SerialNo, config.ApiV3Key, config.PrivateKey)
|
|
if err != nil {
|
|
return fmt.Errorf("error with initialize WechatPay service: %v", err)
|
|
}
|
|
err = client.AutoVerifySign()
|
|
if err != nil {
|
|
return fmt.Errorf("error with autoVerifySign: %v", err)
|
|
}
|
|
s.client = client
|
|
if os.Getenv("GEEKAI_DEBUG") == "true" {
|
|
logger.Info("WechatPay Debug mode is enabled")
|
|
client.DebugSwitch = gopay.DebugOn
|
|
}
|
|
s.config = config
|
|
return nil
|
|
}
|
|
|
|
func (s *WxPayService) Pay(params PayRequest) (string, error) {
|
|
expire := time.Now().Add(10 * time.Minute).Format(time.RFC3339)
|
|
// 初始化 BodyMap
|
|
bm := make(gopay.BodyMap)
|
|
bm.Set("appid", s.config.AppId).
|
|
Set("mchid", s.config.MchId).
|
|
Set("description", params.Subject).
|
|
Set("out_trade_no", params.OutTradeNo).
|
|
Set("time_expire", expire).
|
|
Set("notify_url", params.NotifyURL).
|
|
SetBodyMap("amount", func(bm gopay.BodyMap) {
|
|
bm.Set("total", utils.IntValue(params.TotalFee, 0)).
|
|
Set("currency", "CNY")
|
|
})
|
|
logger.Debugf("wxpay params: %+v", bm)
|
|
if params.Device == "mobile" {
|
|
bm.SetBodyMap("scene_info", func(bm gopay.BodyMap) {
|
|
bm.Set("payer_client_ip", params.ClientIP)
|
|
}).SetBodyMap("payer", func(bm gopay.BodyMap) {
|
|
bm.Set("openid", params.OpenID)
|
|
})
|
|
wxRsp, err := s.client.V3TransactionJsapi(context.Background(), bm)
|
|
if err != nil {
|
|
return "", fmt.Errorf("error with client v3 transaction Jsapi: %v", err)
|
|
}
|
|
if wxRsp.Code != wechat.Success {
|
|
return "", fmt.Errorf("error status with generating pay url: %v", wxRsp.Error)
|
|
}
|
|
return wxRsp.Response.PrepayId, nil
|
|
} else if params.Device == "pc" {
|
|
wxRsp, err := s.client.V3TransactionNative(context.Background(), bm)
|
|
if err != nil {
|
|
return "", fmt.Errorf("error with client v3 transaction Native: %v", err)
|
|
}
|
|
if wxRsp.Code != wechat.Success {
|
|
return "", fmt.Errorf("error status with generating pay url: %v", wxRsp.Error)
|
|
}
|
|
return wxRsp.Response.CodeUrl, nil
|
|
|
|
}
|
|
return "", nil
|
|
}
|
|
|
|
func (s *WxPayService) Query(outTradeNo string) (OrderInfo, error) {
|
|
wxRsp, err := s.client.V3TransactionQueryOrder(context.Background(), wechat.OutTradeNo, outTradeNo)
|
|
if err != nil {
|
|
return OrderInfo{}, fmt.Errorf("error with client v3 transaction query: %v", err)
|
|
}
|
|
|
|
if wxRsp.Code != wechat.Success {
|
|
return OrderInfo{}, fmt.Errorf("error status with querying order: %v", wxRsp.Error)
|
|
}
|
|
|
|
if wxRsp.Response.TradeState == "CLOSED" {
|
|
return OrderInfo{Status: Closed}, nil
|
|
}
|
|
|
|
orderInfo := OrderInfo{
|
|
OutTradeNo: wxRsp.Response.OutTradeNo,
|
|
TradeId: wxRsp.Response.TransactionId,
|
|
Amount: fmt.Sprintf("%d", wxRsp.Response.Amount.Total/100),
|
|
PayTime: wxRsp.Response.SuccessTime,
|
|
}
|
|
if wxRsp.Response.TradeState == "SUCCESS" {
|
|
orderInfo.Status = Success
|
|
} else {
|
|
orderInfo.Status = Failure
|
|
}
|
|
return orderInfo, nil
|
|
}
|
|
|
|
// TradeVerify 交易验证
|
|
func (s *WxPayService) TradeVerify(request *http.Request) (OrderInfo, error) {
|
|
notifyReq, err := wechat.V3ParseNotify(request)
|
|
if err != nil {
|
|
return OrderInfo{}, fmt.Errorf("error with client v3 parse notify: %v", err)
|
|
}
|
|
|
|
// 解密支付密文，验证订单信息
|
|
result, err := notifyReq.DecryptPayCipherText(s.config.ApiV3Key)
|
|
if err != nil {
|
|
return OrderInfo{}, fmt.Errorf("error with client v3 decrypt: %v", err)
|
|
}
|
|
|
|
return OrderInfo{
|
|
Status: Success,
|
|
OutTradeNo: result.OutTradeNo,
|
|
TradeId: result.TransactionId,
|
|
Amount: fmt.Sprintf("%.2f", float64(result.Amount.Total)/100),
|
|
PayTime: result.SuccessTime,
|
|
}, nil
|
|
}
|
|
|
|
// func (s *WechatPayService) PayUrlNative(params WechatPayParams) (string, error) {
|
|
// expire := time.Now().Add(10 * time.Minute).Format(time.RFC3339)
|
|
// // 初始化 BodyMap
|
|
// bm := make(gopay.BodyMap)
|
|
// bm.Set("appid", s.config.AppId).
|
|
// Set("mchid", s.config.MchId).
|
|
// Set("description", params.Subject).
|
|
// Set("out_trade_no", params.OutTradeNo).
|
|
// Set("time_expire", expire).
|
|
// Set("notify_url", params.NotifyURL).
|
|
// SetBodyMap("amount", func(bm gopay.BodyMap) {
|
|
// bm.Set("total", params.TotalFee).
|
|
// Set("currency", "CNY")
|
|
// })
|
|
|
|
// wxRsp, err := s.client.V3TransactionNative(context.Background(), bm)
|
|
// if err != nil {
|
|
// return "", fmt.Errorf("error with client v3 transaction Native: %v", err)
|
|
// }
|
|
// if wxRsp.Code != wechat.Success {
|
|
// return "", fmt.Errorf("error status with generating pay url: %v", wxRsp.Error)
|
|
// }
|
|
// return wxRsp.Response.CodeUrl, nil
|
|
// }
|
|
|
|
// func (s *WechatPayService) PayUrlH5(params WechatPayParams) (string, error) {
|
|
// expire := time.Now().Add(10 * time.Minute).Format(time.RFC3339)
|
|
// // 初始化 BodyMap
|
|
// bm := make(gopay.BodyMap)
|
|
// bm.Set("appid", s.config.AppId).
|
|
// Set("mchid", s.config.MchId).
|
|
// Set("description", params.Subject).
|
|
// Set("out_trade_no", params.OutTradeNo).
|
|
// Set("time_expire", expire).
|
|
// Set("notify_url", params.NotifyURL).
|
|
// SetBodyMap("amount", func(bm gopay.BodyMap) {
|
|
// bm.Set("total", params.TotalFee).
|
|
// Set("currency", "CNY")
|
|
// }).
|
|
// SetBodyMap("scene_info", func(bm gopay.BodyMap) {
|
|
// bm.Set("payer_client_ip", params.ClientIP).
|
|
// SetBodyMap("h5_info", func(bm gopay.BodyMap) {
|
|
// bm.Set("type", "Wap")
|
|
// })
|
|
// })
|
|
|
|
// wxRsp, err := s.client.V3TransactionH5(context.Background(), bm)
|
|
// if err != nil {
|
|
// return "", fmt.Errorf("error with client v3 transaction H5: %v", err)
|
|
// }
|
|
// if wxRsp.Code != wechat.Success {
|
|
// return "", fmt.Errorf("error with generating pay url: %v", wxRsp.Error)
|
|
// }
|
|
// return wxRsp.Response.H5Url, nil
|
|
// }
|
|
|
|
// type NotifyResponse struct {
|
|
// Code string `json:"code"`
|
|
// Message string `xml:"message"`
|
|
// }
|
|
|
|
var _ PayService = (*WxPayService)(nil)
|