cloudreve/routers/controllers/vas.go

package controllers

import (
	"github.com/HFO4/cloudreve/pkg/payment"
	"github.com/HFO4/cloudreve/pkg/util"
	"github.com/HFO4/cloudreve/service/vas"
	"github.com/gin-gonic/gin"
	"github.com/qingwg/payjs/notify"
	"github.com/smartwalle/alipay/v3"
)

// GetQuota 获取容量配额信息
func GetQuota(c *gin.Context) {
	var service vas.GeneralVASService
	if err := c.ShouldBindUri(&service); err == nil {
		res := service.Quota(c, CurrentUser(c))
		c.JSON(200, res)
	} else {
		c.JSON(200, ErrorResponse(err))
	}
}

// GetProduct 获取商品信息
func GetProduct(c *gin.Context) {
	var service vas.GeneralVASService
	if err := c.ShouldBindUri(&service); err == nil {
		res := service.Products(c, CurrentUser(c))
		c.JSON(200, res)
	} else {
		c.JSON(200, ErrorResponse(err))
	}
}

// NewOrder 新建支付订单
func NewOrder(c *gin.Context) {
	var service vas.CreateOrderService
	if err := c.ShouldBindJSON(&service); err == nil {
		res := service.Create(c, CurrentUser(c))
		c.JSON(200, res)
	} else {
		c.JSON(200, ErrorResponse(err))
	}
}

// OrderStatus 查询订单状态
func OrderStatus(c *gin.Context) {
	var service vas.OrderService
	if err := c.ShouldBindUri(&service); err == nil {
		res := service.Status(c, CurrentUser(c))
		c.JSON(200, res)
	} else {
		c.JSON(200, ErrorResponse(err))
	}
}

// GetRedeemInfo 获取兑换码信息
func GetRedeemInfo(c *gin.Context) {
	var service vas.RedeemService
	if err := c.ShouldBindUri(&service); err == nil {
		res := service.Query(c)
		c.JSON(200, res)
	} else {
		c.JSON(200, ErrorResponse(err))
	}
}

// DoRedeem 获取兑换码信息
func DoRedeem(c *gin.Context) {
	var service vas.RedeemService
	if err := c.ShouldBindUri(&service); err == nil {
		res := service.Redeem(c, CurrentUser(c))
		c.JSON(200, res)
	} else {
		c.JSON(200, ErrorResponse(err))
	}
}

// AlipayCallback 支付宝回调
func AlipayCallback(c *gin.Context) {
	pay, err := payment.NewPaymentInstance("alipay")
	if err != nil {
		util.Log().Debug("[支付宝回调] 无法创建支付宝客户端, %s", err)
		c.Status(400)
		return
	}

	res, err := pay.(*payment.Alipay).Client.GetTradeNotification(c.Request)
	if err != nil {
		util.Log().Debug("[支付宝回调] 回调验证失败, %s", err)
		c.Status(403)
		return
	}

	if res != nil && res.TradeStatus == "TRADE_SUCCESS" {
		// 支付成功
		if err := payment.OrderPaid(res.OutTradeNo); err != nil {
			util.Log().Debug("[支付宝回调] 支付处理失败, %s", err)
		}
	}

	// 确认收到通知消息
	alipay.AckNotification(c.Writer)
}

// PayJSCallback PayJS回调
func PayJSCallback(c *gin.Context) {
	pay, err := payment.NewPaymentInstance("payjs")
	if err != nil {
		util.Log().Debug("[PayJS回调] 无法创建支付宝客户端, %s", err)
		c.Status(400)
		return
	}

	payNotify := pay.(*payment.PayJSClient).Client.GetNotify(c.Request, c.Writer)

	//设置接收消息的处理方法
	payNotify.SetMessageHandler(func(msg notify.Message) {
		if err := payment.OrderPaid(msg.OutTradeNo); err != nil {
			util.Log().Debug("[PayJS回调] 支付处理失败, %s", err)
		}
	})

	//处理消息接收以及回复
	err = payNotify.Serve()
	if err != nil {
		util.Log().Debug("[PayJS回调] 回调处理失败, %s", err)
		return
	}

	//发送回复的消息
	payNotify.SendResponseMsg()

}

// QQCallback QQ互联回调
func QQCallback(c *gin.Context) {
	var service vas.QQCallbackService
	if err := c.ShouldBindJSON(&service); err == nil {
		res := service.Callback(c, CurrentUser(c))
		c.JSON(200, res)
	} else {
		c.JSON(200, ErrorResponse(err))
	}
}