131 lines
4.2 KiB
Go
131 lines
4.2 KiB
Go
package controller
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/QuantumNous/new-api/common"
|
|
"github.com/QuantumNous/new-api/logger"
|
|
"github.com/QuantumNous/new-api/model"
|
|
"github.com/QuantumNous/new-api/service"
|
|
"github.com/QuantumNous/new-api/setting"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/shopspring/decimal"
|
|
"github.com/thanhpk/randstr"
|
|
)
|
|
|
|
type SubscriptionWaffoPancakePayRequest struct {
|
|
PlanId int `json:"plan_id"`
|
|
}
|
|
|
|
func SubscriptionRequestWaffoPancakePay(c *gin.Context) {
|
|
if !requirePaymentCompliance(c) {
|
|
return
|
|
}
|
|
|
|
var req SubscriptionWaffoPancakePayRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil || req.PlanId <= 0 {
|
|
common.ApiErrorT(c, "Invalid parameters")
|
|
return
|
|
}
|
|
|
|
plan, err := model.GetSubscriptionPlanById(req.PlanId)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
if !plan.Enabled {
|
|
common.ApiErrorT(c, "Subscription plan is not enabled")
|
|
return
|
|
}
|
|
if strings.TrimSpace(plan.WaffoPancakeProductId) == "" {
|
|
common.ApiErrorT(c, "WaffoPancakeProductId is not configured for this plan")
|
|
return
|
|
}
|
|
// Plan targets its own Pancake product, so we only require credentials
|
|
// here — not the gateway-level WaffoPancakeProductID.
|
|
if strings.TrimSpace(setting.WaffoPancakeMerchantID) == "" ||
|
|
strings.TrimSpace(setting.WaffoPancakePrivateKey) == "" {
|
|
common.ApiErrorT(c, "Waffo Pancake is not configured or key is invalid")
|
|
return
|
|
}
|
|
|
|
userId := c.GetInt("id")
|
|
user, err := model.GetUserById(userId, false)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
if user == nil {
|
|
common.ApiErrorT(c, "User does not exist")
|
|
return
|
|
}
|
|
|
|
if plan.MaxPurchasePerUser > 0 {
|
|
count, err := model.CountUserSubscriptionsByPlan(userId, plan.Id)
|
|
if err != nil {
|
|
common.ApiError(c, err)
|
|
return
|
|
}
|
|
if count >= int64(plan.MaxPurchasePerUser) {
|
|
common.ApiErrorT(c, "Purchase limit for this plan has been reached")
|
|
return
|
|
}
|
|
}
|
|
|
|
// WAFFO_PANCAKE_SUB- prefix (vs. wallet's WAFFO_PANCAKE-) drives webhook
|
|
// dispatch in WaffoPancakeWebhook.
|
|
tradeNo := fmt.Sprintf("WAFFO_PANCAKE_SUB-%d-%d-%s", userId, time.Now().UnixMilli(), randstr.String(6))
|
|
|
|
order := &model.SubscriptionOrder{
|
|
UserId: userId,
|
|
PlanId: plan.Id,
|
|
Money: plan.PriceAmount,
|
|
TradeNo: tradeNo,
|
|
PaymentMethod: model.PaymentMethodWaffoPancake,
|
|
PaymentProvider: model.PaymentProviderWaffoPancake,
|
|
CreateTime: time.Now().Unix(),
|
|
Status: common.TopUpStatusPending,
|
|
}
|
|
if err := order.Insert(); err != nil {
|
|
logger.LogError(c.Request.Context(), common.LogText("Waffo Pancake failed to create subscription order user_id=%d plan_id=%d trade_no=%s error=%q", userId, plan.Id, tradeNo, err.Error()))
|
|
paymentError(c, common.NewMessage("Failed to create order"))
|
|
return
|
|
}
|
|
|
|
expiresInSeconds := 45 * 60
|
|
session, err := service.CreateWaffoPancakeCheckoutSession(c.Request.Context(), &service.WaffoPancakeCreateSessionParams{
|
|
ProductID: plan.WaffoPancakeProductId,
|
|
BuyerIdentity: service.WaffoPancakeBuyerIdentityFromUserID(user.Id),
|
|
PriceSnapshot: &service.WaffoPancakePriceSnapshot{
|
|
Amount: decimal.NewFromFloat(plan.PriceAmount).StringFixed(2),
|
|
TaxCategory: "saas",
|
|
},
|
|
BuyerEmail: getWaffoPancakeBuyerEmail(user),
|
|
ExpiresInSeconds: &expiresInSeconds,
|
|
OrderMerchantExternalID: tradeNo,
|
|
})
|
|
if err != nil {
|
|
logger.LogError(c.Request.Context(), common.LogText("Waffo Pancake failed to create subscription checkout session user_id=%d plan_id=%d trade_no=%s error=%q", userId, plan.Id, tradeNo, err.Error()))
|
|
order.Status = common.TopUpStatusFailed
|
|
_ = order.Update()
|
|
paymentError(c, common.NewMessage("Failed to start payment"))
|
|
return
|
|
}
|
|
logger.LogInfo(c.Request.Context(), common.LogText("Waffo Pancake subscription order created user_id=%d plan_id=%d trade_no=%s session_id=%s money=%.2f", userId, plan.Id, tradeNo, session.SessionID, plan.PriceAmount))
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"message": "success",
|
|
"data": gin.H{
|
|
"checkout_url": session.CheckoutURL,
|
|
"session_id": session.SessionID,
|
|
"expires_at": session.ExpiresAt,
|
|
"order_id": tradeNo,
|
|
"token": session.Token,
|
|
"token_expires_at": session.TokenExpiresAt,
|
|
},
|
|
})
|
|
}
|