mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-27 13:46:38 +08:00
116 lines
4.0 KiB
Go
116 lines
4.0 KiB
Go
package payment
|
||
|
||
import (
|
||
"chatplus/core/types"
|
||
"chatplus/store/model"
|
||
chatPlusUtils "chatplus/utils"
|
||
"context"
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/wechatpay-apiv3/wechatpay-go/core/auth/verifiers"
|
||
"github.com/wechatpay-apiv3/wechatpay-go/core/downloader"
|
||
"github.com/wechatpay-apiv3/wechatpay-go/core/notify"
|
||
"github.com/wechatpay-apiv3/wechatpay-go/services/payments"
|
||
"github.com/wechatpay-apiv3/wechatpay-go/services/payments/jsapi"
|
||
"log"
|
||
"time"
|
||
|
||
"github.com/wechatpay-apiv3/wechatpay-go/core"
|
||
"github.com/wechatpay-apiv3/wechatpay-go/core/option"
|
||
"github.com/wechatpay-apiv3/wechatpay-go/utils"
|
||
)
|
||
|
||
type WxpayService struct {
|
||
config *types.WxpayConfig
|
||
client *core.Client
|
||
certificateVisitor core.CertificateGetter
|
||
}
|
||
|
||
func NewWxpayService(appConfig *types.AppConfig) *WxpayService {
|
||
config := appConfig.WxpayConfig
|
||
if !config.Enabled {
|
||
logger.Info("Disabled Wxpay service")
|
||
return nil
|
||
}
|
||
// 使用 utils 提供的函数从本地文件中加载商户私钥，商户私钥会用来生成请求的签名
|
||
mchPrivateKey, err := utils.LoadPrivateKeyWithPath(config.PrivateKey)
|
||
if err != nil {
|
||
log.Print("load merchant private key error")
|
||
return nil
|
||
}
|
||
|
||
ctx := context.Background()
|
||
// 使用商户私钥等初始化 client，并使它具有自动定时获取微信支付平台证书的能力
|
||
opts := []core.ClientOption{
|
||
option.WithWechatPayAutoAuthCipher(config.MchId, config.CertificateSerialNo, mchPrivateKey, config.MchKey),
|
||
}
|
||
client, err := core.NewClient(ctx, opts...)
|
||
if err != nil {
|
||
return nil
|
||
}
|
||
// 1. 使用 `RegisterDownloaderWithPrivateKey` 注册下载器
|
||
err2 := downloader.MgrInstance().RegisterDownloaderWithPrivateKey(ctx, mchPrivateKey, config.CertificateSerialNo, config.MchId, config.MchKey)
|
||
if err2 != nil {
|
||
logger.Error("支付回调校验失败，请检查应用私钥配置文件")
|
||
return nil
|
||
}
|
||
// 2. 获取商户号对应的微信支付平台证书访问器
|
||
certificateVisitor := downloader.MgrInstance().GetCertificateVisitor(config.MchId)
|
||
return &WxpayService{&config, client, certificateVisitor}
|
||
}
|
||
|
||
func (s *WxpayService) Pay(user model.User, order model.Order) (resp *jsapi.PrepayWithRequestPaymentResponse, err error) {
|
||
return s.PayUrlMobile(user, order)
|
||
}
|
||
|
||
func (s *WxpayService) PayUrlMobile(user model.User, order model.Order) (resp *jsapi.PrepayWithRequestPaymentResponse, err error) {
|
||
svc := jsapi.JsapiApiService{Client: s.client}
|
||
var outTradeNo = chatPlusUtils.RandString(16)
|
||
resp, result, err := svc.PrepayWithRequestPayment(context.Background(),
|
||
jsapi.PrepayRequest{
|
||
Appid: core.String(s.config.AppId),
|
||
Mchid: core.String(s.config.MchId),
|
||
Description: core.String(order.Subject),
|
||
OutTradeNo: core.String(outTradeNo),
|
||
TimeExpire: core.Time(time.Now()),
|
||
Attach: core.String(order.OrderNo),
|
||
NotifyUrl: core.String(s.config.NotifyURL),
|
||
SupportFapiao: core.Bool(false),
|
||
Amount: &jsapi.Amount{
|
||
Currency: core.String("CNY"),
|
||
Total: core.Int64(int64(order.Amount * 100)),
|
||
},
|
||
Payer: &jsapi.Payer{
|
||
Openid: core.String(user.OfficialOpenid),
|
||
},
|
||
SceneInfo: &jsapi.SceneInfo{
|
||
PayerClientIp: core.String("127.0.0.1"),
|
||
},
|
||
SettleInfo: &jsapi.SettleInfo{
|
||
ProfitSharing: core.Bool(false),
|
||
},
|
||
},
|
||
)
|
||
|
||
if err != nil {
|
||
// 处理错误
|
||
log.Printf("call Prepay err:%s", err)
|
||
} else {
|
||
// 处理返回结果
|
||
log.Printf("status=%d resp=%s", result.Response.StatusCode, resp)
|
||
}
|
||
return resp, err
|
||
}
|
||
|
||
func (s *WxpayService) TradeVerify(c *gin.Context) (orderNo string, tradeNo string, code int) {
|
||
// 3. 使用证书访问器初始化 `notify.Handler`
|
||
handler, _ := notify.NewRSANotifyHandler(s.config.MchKey, verifiers.NewSHA256WithRSAVerifier(s.certificateVisitor))
|
||
// 2. 获取商户号对应的微信支付平台证书访问器
|
||
transaction := new(payments.Transaction)
|
||
_, err3 := handler.ParseNotifyRequest(context.Background(), c.Request, transaction)
|
||
// 如果验签未通过，或者解密失败
|
||
if err3 != nil {
|
||
return "0", "0", 401
|
||
}
|
||
return *transaction.Attach, *transaction.OutTradeNo, 200
|
||
}
|