Initial commit (code only without large binaries)

EdgeAPI/internal/payments/utils.go

@@ -0,0 +1,57 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package payments
+
+import (
+	"crypto/sha256"
+	"fmt"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models/accounts"
+	"github.com/TeaOSLab/EdgeAPI/internal/errors"
+	"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
+	"github.com/iwind/TeaGo/types"
+	"net/url"
+	"sort"
+	"strings"
+	"time"
+)
+
+// GeneratePayURL PayURL 构造支付URL
+func GeneratePayURL(order *accounts.UserOrder, method *accounts.OrderMethod) (payURL string, err error) {
+	if method == nil || !method.IsOn {
+		return "", errors.New("invalid method with id '" + types.String(method.Id) + "'")
+	}
+
+	// 内置支付方式
+	switch method.ParentCode {
+	case userconfigs.PayMethodAlipay: // 支付宝
+		return NewAlipayPayment().GeneratePayURL(order, method)
+	}
+
+	// 自定义
+	var args = []string{}
+	args = append(args, "EdgeOrderMethod="+url.QueryEscape(method.Code))
+	args = append(args, "EdgeOrderCode="+url.QueryEscape(order.Code))
+	args = append(args, "EdgeOrderTimestamp="+types.String(time.Now().Unix()))
+	args = append(args, "EdgeOrderAmount="+fmt.Sprintf("%.2f", order.Amount))
+
+	sort.Strings(args)
+
+	var signArgs = append([]string{}, args...)
+	signArgs = append(signArgs, method.Secret)
+	var sign = fmt.Sprintf("%x", sha256.Sum256([]byte(strings.Join(signArgs, "&"))))
+	args = append(args, "EdgeOrderSign="+sign)
+
+	if strings.Contains(method.Url, "?") {
+		return method.Url + "&" + strings.Join(args, "&"), nil
+	}
+	return method.Url + "?" + strings.Join(args, "&"), nil
+}
+
+func Verify(payMethod userconfigs.PayMethod, formValues url.Values) (orderId int64, err error) {
+	switch payMethod {
+	case userconfigs.PayMethodAlipay: // 支付宝
+		return NewAlipayPayment().Verify(formValues)
+	}
+
+	return 0, errors.New("invalid payment method '" + payMethod + "'")
+}