add epay

controller/topup.go

@@ -0,0 +1,173 @@
+package controller
+
+import (
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"github.com/samber/lo"
+	epay "github.com/star-horizon/go-epay"
+	"log"
+	"net/url"
+	"one-api/common"
+	"one-api/model"
+	"strconv"
+	"time"
+)
+
+type EpayRequest struct {
+	Amount        int    `json:"amount"`
+	PaymentMethod string `json:"payment_method"`
+	TopUpCode     string `json:"top_up_code"`
+}
+
+type AmountRequest struct {
+	Amount    int    `json:"amount"`
+	TopUpCode string `json:"top_up_code"`
+}
+
+var client, _ = epay.NewClientWithUrl(&epay.Config{
+	PartnerID: "1096",
+	Key:       "n08V9LpE8JffA3NPP893689u8p39NV9J",
+}, "https://api.lempay.org")
+
+func GetAmount(id int, count float64, topUpCode string) float64 {
+	amount := count * 1.5
+	if topUpCode != "" {
+		if topUpCode == "nekoapi" {
+			if id == 89 {
+				amount = count * 1
+			} else if id == 98 || id == 105 || id == 107 {
+				amount = count * 1.2
+			} else if id == 1 {
+				amount = count * 1
+			}
+		}
+	}
+	return amount
+}
+
+func RequestEpay(c *gin.Context) {
+	var req EpayRequest
+	err := c.ShouldBindJSON(&req)
+	if err != nil {
+		c.JSON(200, gin.H{"message": err.Error(), "data": 10})
+		return
+	}
+	id := c.GetInt("id")
+	amount := GetAmount(id, float64(req.Amount), req.TopUpCode)
+	if id != 1 {
+		if req.Amount < 10 {
+			c.JSON(200, gin.H{"message": "最小充值10元", "data": amount, "count": 10})
+			return
+		}
+	}
+	if req.PaymentMethod == "zfb" {
+		if amount > 400 {
+			c.JSON(200, gin.H{"message": "支付宝最大充值400元", "data": amount, "count": 400})
+			return
+		}
+		req.PaymentMethod = "alipay"
+	}
+	if req.PaymentMethod == "wx" {
+		if amount > 600 {
+			c.JSON(200, gin.H{"message": "微信最大充值600元", "data": amount, "count": 600})
+			return
+		}
+		req.PaymentMethod = "wxpay"
+	}
+
+	returnUrl, _ := url.Parse("https://nekoapi.com/log")
+	notifyUrl, _ := url.Parse("https://nekoapi.com/api/user/epay/notify")
+	tradeNo := strconv.FormatInt(time.Now().Unix(), 10)
+	uri, params, err := client.Purchase(&epay.PurchaseArgs{
+		Type:           epay.PurchaseType(req.PaymentMethod),
+		ServiceTradeNo: "A" + tradeNo,
+		Name:           "B" + tradeNo,
+		Money:          strconv.FormatFloat(amount*0.99, 'f', 2, 64),
+		Device:         epay.PC,
+		NotifyUrl:      notifyUrl,
+		ReturnUrl:      returnUrl,
+	})
+	if err != nil {
+		c.JSON(200, gin.H{"message": "error", "data": "拉起支付失败"})
+		return
+	}
+	topUp := &model.TopUp{
+		UserId:     id,
+		Amount:     req.Amount,
+		Money:      int(amount),
+		TradeNo:    "A" + tradeNo,
+		CreateTime: time.Now().Unix(),
+		Status:     "pending",
+	}
+	err = topUp.Insert()
+	if err != nil {
+		c.JSON(200, gin.H{"message": "error", "data": "创建订单失败"})
+		return
+	}
+	c.JSON(200, gin.H{"message": "success", "data": params, "url": uri})
+}
+
+func EpayNotify(c *gin.Context) {
+	params := lo.Reduce(lo.Keys(c.Request.URL.Query()), func(r map[string]string, t string, i int) map[string]string {
+		r[t] = c.Request.URL.Query().Get(t)
+		return r
+	}, map[string]string{})
+	verifyInfo, err := client.Verify(params)
+	if err == nil && verifyInfo.VerifyStatus {
+		_, err := c.Writer.Write([]byte("success"))
+		if err != nil {
+			log.Println("易支付回调写入失败")
+		}
+	} else {
+		_, err := c.Writer.Write([]byte("fail"))
+		if err != nil {
+			log.Println("易支付回调写入失败")
+		}
+	}
+
+	if verifyInfo.TradeStatus == epay.StatusTradeSuccess {
+		log.Println(verifyInfo)
+		topUp := model.GetTopUpByTradeNo(verifyInfo.ServiceTradeNo)
+		if topUp.Status == "pending" {
+			topUp.Status = "success"
+			err := topUp.Update()
+			if err != nil {
+				log.Printf("易支付回调更新订单失败: %v", topUp)
+				return
+			}
+			//user, _ := model.GetUserById(topUp.UserId, false)
+			//user.Quota += topUp.Amount * 500000
+			err = model.IncreaseUserQuota(topUp.UserId, topUp.Amount*500000)
+			if err != nil {
+				log.Printf("易支付回调更新用户失败: %v", topUp)
+				return
+			}
+			log.Printf("易支付回调更新用户成功 %v", topUp)
+			model.RecordLog(topUp.UserId, model.LogTypeTopup, fmt.Sprintf("使用在线充值成功，充值金额: %v", common.LogQuota(topUp.Amount*500000)))
+		}
+	} else {
+		log.Printf("易支付异常回调: %v", verifyInfo)
+	}
+}
+
+func RequestAmount(c *gin.Context) {
+	var req AmountRequest
+	err := c.ShouldBindJSON(&req)
+	if err != nil {
+		c.JSON(200, gin.H{"message": err.Error(), "data": 10})
+		return
+	}
+	id := c.GetInt("id")
+	if id != 1 {
+		if req.Amount < 10 {
+			c.JSON(200, gin.H{"message": "最小充值10刀", "data": GetAmount(id, 10, req.TopUpCode), "count": 10})
+			return
+		}
+		if req.Amount > 400 {
+			c.JSON(200, gin.H{"message": "最大充值400刀", "data": GetAmount(id, 400, req.TopUpCode), "count": 400})
+			return
+		}
+	}
+
+	c.JSON(200, gin.H{"message": "success", "data": GetAmount(id, float64(req.Amount), req.TopUpCode)})
+}