361 lines
8.1 KiB
Go
361 lines
8.1 KiB
Go
package handlers
|
|
|
|
import (
|
|
"platform/pkg/env"
|
|
"platform/pkg/u"
|
|
"platform/web/auth"
|
|
"platform/web/common"
|
|
g "platform/web/globals"
|
|
q "platform/web/queries"
|
|
s "platform/web/services"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/smartwalle/alipay/v3"
|
|
"github.com/wechatpay-apiv3/wechatpay-go/services/payments/native"
|
|
)
|
|
|
|
// region ListResourcePss
|
|
|
|
type ListResourcePssReq struct {
|
|
common.PageReq
|
|
ResourceNo *string `json:"resource_no"`
|
|
Active *bool `json:"active"`
|
|
Type *int `json:"type"`
|
|
CreateAfter *time.Time `json:"create_after"`
|
|
CreateBefore *time.Time `json:"create_before"`
|
|
ExpireAfter *time.Time `json:"expire_after"`
|
|
ExpireBefore *time.Time `json:"expire_before"`
|
|
}
|
|
|
|
// ListResourcePss 获取套餐列表
|
|
func ListResourcePss(c *fiber.Ctx) error {
|
|
// 检查权限
|
|
authContext, err := auth.Protect(c, []s.PayloadType{s.PayloadUser}, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 解析请求参数
|
|
req := new(ListResourcePssReq)
|
|
if err := c.BodyParser(req); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 查询套餐列表
|
|
do := q.Resource.
|
|
Joins(q.Resource.Pss).
|
|
Where(q.Resource.UserID.Eq(authContext.Payload.Id))
|
|
if req.ResourceNo != nil && *req.ResourceNo != "" {
|
|
do = do.Where(q.Resource.ResourceNo.Eq(*req.ResourceNo))
|
|
}
|
|
if req.Active != nil {
|
|
do = do.Where(q.Resource.Active.Is(*req.Active))
|
|
}
|
|
if req.Type != nil {
|
|
do = do.Where(q.ResourcePss.As(q.Resource.Pss.Name()).Type.Eq(int32(*req.Type)))
|
|
}
|
|
if req.CreateAfter != nil {
|
|
do = do.Where(q.Resource.CreatedAt.Gte(common.LocalDateTime(*req.CreateAfter)))
|
|
}
|
|
if req.CreateBefore != nil {
|
|
do = do.Where(q.Resource.CreatedAt.Lte(common.LocalDateTime(*req.CreateBefore)))
|
|
}
|
|
if req.ExpireAfter != nil {
|
|
do = do.Where(q.ResourcePss.As(q.Resource.Pss.Name()).Expire.Gte(common.LocalDateTime(*req.ExpireAfter)))
|
|
}
|
|
if req.ExpireBefore != nil {
|
|
do = do.Where(q.ResourcePss.As(q.Resource.Pss.Name()).Expire.Lte(common.LocalDateTime(*req.ExpireBefore)))
|
|
}
|
|
|
|
resource, err := do.Debug().
|
|
Order(q.Resource.CreatedAt.Desc()).
|
|
Offset(req.GetOffset()).
|
|
Limit(req.GetLimit()).
|
|
Find()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var total int64
|
|
if len(resource) < req.GetLimit() {
|
|
total = int64(len(resource) + req.GetOffset())
|
|
} else {
|
|
total, err = do.Count()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return c.JSON(common.PageResp{
|
|
Total: int(total),
|
|
Page: req.GetPage(),
|
|
Size: req.GetSize(),
|
|
List: resource,
|
|
})
|
|
}
|
|
|
|
// endregion
|
|
|
|
// region AllResource
|
|
|
|
type AllResourceReq struct {
|
|
}
|
|
|
|
func AllResource(c *fiber.Ctx) error {
|
|
// 检查权限
|
|
authContext, err := auth.Protect(c, []s.PayloadType{s.PayloadUser}, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 查询套餐列表
|
|
pss := q.ResourcePss.As(q.Resource.Pss.Name())
|
|
do := q.Resource.Debug().
|
|
Joins(q.Resource.Pss).
|
|
Where(
|
|
q.Resource.UserID.Eq(authContext.Payload.Id),
|
|
q.Resource.Active.Is(true),
|
|
q.Resource.Where(
|
|
pss.Type.Eq(1),
|
|
pss.Expire.Gte(common.LocalDateTime(time.Now())),
|
|
).Or(
|
|
pss.Type.Eq(2),
|
|
pss.Quota.GtCol(pss.Used),
|
|
),
|
|
q.Resource.Where(
|
|
pss.DailyLast.Lt(common.LocalDateTime(u.Today())),
|
|
).Or(
|
|
pss.DailyUsed.LtCol(pss.DailyLimit),
|
|
),
|
|
)
|
|
|
|
resources, err := do.Debug().
|
|
Order(q.Resource.CreatedAt.Desc()).
|
|
Find()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return c.JSON(resources)
|
|
}
|
|
|
|
// endregion
|
|
|
|
// region CreateResourcePrepared
|
|
|
|
type CreateResourceReq struct {
|
|
s.CreateResourceData
|
|
}
|
|
|
|
type CreateResourceResp struct {
|
|
TradeNo string `json:"trade_no"`
|
|
PayURL string `json:"pay_url"`
|
|
}
|
|
|
|
type PaidCreateResourceReq struct {
|
|
TradeNo string `json:"trade_no" validate:"required"`
|
|
}
|
|
|
|
func PrepareResourceByAlipay(c *fiber.Ctx) error {
|
|
|
|
// 检查权限
|
|
authContext, err := auth.Protect(c, []s.PayloadType{s.PayloadUser}, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 解析请求参数
|
|
req := new(CreateResourceReq)
|
|
if err := c.BodyParser(req); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 生成订单
|
|
tradeNo, err := s.ID.GenSerial(c.Context())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 调用外部接口
|
|
alipayResp, err := g.Alipay.TradePagePay(alipay.TradePagePay{
|
|
QRPayMode: "4",
|
|
Trade: alipay.Trade{
|
|
OutTradeNo: tradeNo,
|
|
TotalAmount: strconv.FormatFloat(req.GetPrice(), 'f', 2, 64),
|
|
Subject: "购买套餐 - " + req.GetName(),
|
|
ProductCode: "FAST_INSTANT_TRADE_PAY",
|
|
TimeExpire: time.Now().Add(30 * time.Minute).Format("2006-01-02 15:04:05"),
|
|
},
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 保存交易信息
|
|
err = s.Resource.PrepareResource(c.Context(), &req.CreateResourceData, authContext.Payload.Id, tradeNo, 1)
|
|
|
|
// 返回结果
|
|
return c.JSON(CreateResourceResp{
|
|
TradeNo: tradeNo,
|
|
PayURL: alipayResp.String(),
|
|
})
|
|
}
|
|
|
|
func PrepareResourceByWechat(c *fiber.Ctx) error {
|
|
|
|
// 检查权限
|
|
authContext, err := auth.Protect(c, []s.PayloadType{s.PayloadUser}, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 解析请求参数
|
|
req := new(CreateResourceReq)
|
|
if err := c.BodyParser(req); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 生成订单号
|
|
tradeNo, err := s.ID.GenSerial(c.Context())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 调用外部接口
|
|
wechatPayResp, _, err := g.WechatPay.Native.Prepay(c.Context(), native.PrepayRequest{
|
|
Mchid: &env.WechatPayMchId,
|
|
Appid: &env.WechatPayAppId,
|
|
Description: u.P("购买套餐 - " + req.GetName()),
|
|
OutTradeNo: &tradeNo,
|
|
TimeExpire: u.P(time.Now().Add(30 * time.Minute)),
|
|
NotifyUrl: &env.WechatPayCallbackUrl,
|
|
Amount: &native.Amount{
|
|
Total: u.P(int64(req.GetPrice() * 100)),
|
|
},
|
|
})
|
|
|
|
// 保存交易信息
|
|
err = s.Resource.PrepareResource(c.Context(), &req.CreateResourceData, authContext.Payload.Id, tradeNo, 2)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 返回结果
|
|
return c.JSON(CreateResourceResp{
|
|
TradeNo: tradeNo,
|
|
PayURL: *wechatPayResp.CodeUrl,
|
|
})
|
|
}
|
|
|
|
func CreateResourceByAlipay(c *fiber.Ctx) error {
|
|
// 检查权限
|
|
_, err := auth.Protect(c, []s.PayloadType{s.PayloadUser}, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 解析请求参数
|
|
req := new(PaidCreateResourceReq)
|
|
if err := c.BodyParser(req); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 验证支付结果
|
|
alipayResp, err := g.Alipay.TradeQuery(c.Context(), alipay.TradeQuery{
|
|
OutTradeNo: req.TradeNo,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if alipayResp.TradeStatus != "TRADE_SUCCESS" {
|
|
return fiber.NewError(fiber.StatusBadRequest, "支付未完成，请确认后重试")
|
|
}
|
|
|
|
// 创建套餐
|
|
payment, err := strconv.ParseFloat(alipayResp.ReceiptAmount, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
paidAt, err := time.Parse("2006-01-02 15:04:05", alipayResp.SendPayDate)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = s.Resource.CreateResourcePrepared(c.Context(), req.TradeNo, alipayResp.TradeNo, payment, paidAt)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func CreateResourceByWechat(c *fiber.Ctx) error {
|
|
// 检查权限
|
|
_, err := auth.Protect(c, []s.PayloadType{s.PayloadUser}, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 解析请求参数
|
|
req := new(PaidCreateResourceReq)
|
|
if err := c.BodyParser(req); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 验证支付结果
|
|
wechatPayResp, _, err := g.WechatPay.Native.QueryOrderByOutTradeNo(c.Context(), native.QueryOrderByOutTradeNoRequest{
|
|
OutTradeNo: &req.TradeNo,
|
|
Mchid: &env.WechatPayMchId,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if *wechatPayResp.TradeState != "SUCCESS" {
|
|
return fiber.NewError(fiber.StatusBadRequest, "支付未完成，请确认后重试")
|
|
}
|
|
|
|
// 创建套餐
|
|
payment := float64(*wechatPayResp.Amount.PayerTotal) / 100
|
|
|
|
paidAt, err := time.Parse(time.RFC3339, *wechatPayResp.SuccessTime)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = s.Resource.CreateResourcePrepared(c.Context(), req.TradeNo, *wechatPayResp.OutTradeNo, payment, paidAt)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func CreateResourceByBalance(c *fiber.Ctx) error {
|
|
|
|
// 检查权限
|
|
authCtx, err := auth.Protect(c, []s.PayloadType{s.PayloadUser}, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 解析请求参数
|
|
req := new(CreateResourceReq)
|
|
if err := c.BodyParser(req); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 创建套餐
|
|
err = s.Resource.CreateResourceImmediately(&req.CreateResourceData, authCtx.Payload.Id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// endregion
|