Feat: vas for group / storage pack

service/vas/quota.go

@@ -0,0 +1,97 @@
+package vas
+
+import (
+	"encoding/json"
+	model "github.com/HFO4/cloudreve/models"
+	"github.com/HFO4/cloudreve/pkg/payment"
+	"github.com/HFO4/cloudreve/pkg/serializer"
+	"github.com/gin-gonic/gin"
+)
+
+// GeneralVASService 通用增值服务
+type GeneralVASService struct {
+}
+
+// CreateOrderService 创建订单服务
+type CreateOrderService struct {
+	Action string `json:"action" binding:"required,eq=group|eq=pack"`
+	Method string `json:"method" binding:"required,eq=alipay|eq=score|eq=payjs"`
+	ID     int64  `json:"id" binding:"required"`
+	Num    int    `json:"num" binding:"required,min=1,max=99"`
+}
+
+// Create 创建新订单
+func (service *CreateOrderService) Create(c *gin.Context, user *model.User) serializer.Response {
+	// 取得当前商品信息
+	packs, groups, err := decodeProductInfo()
+	if err != nil {
+		return serializer.Err(serializer.CodeInternalSetting, "无法解析商品设置", err)
+	}
+
+	// 查找要购买的商品
+	var (
+		pack  *serializer.PackProduct
+		group *serializer.GroupProducts
+	)
+	if service.Action == "group" {
+		for _, v := range groups {
+			if v.ID == service.ID {
+				group = &v
+				break
+			}
+		}
+	} else {
+		for _, v := range packs {
+			if v.ID == service.ID {
+				pack = &v
+				break
+			}
+		}
+	}
+	if pack == nil && group == nil {
+		return serializer.Err(serializer.CodeNotFound, "商品不存在", nil)
+	}
+
+	// 创建订单
+	res, err := payment.NewOrder(pack, group, service.Num, service.Method, user)
+	if err != nil {
+		return serializer.Err(serializer.CodeNotSet, err.Error(), err)
+	}
+
+	return serializer.Response{Data: res}
+
+}
+
+// Products 获取商品信息
+func (service *GeneralVASService) Products(c *gin.Context, user *model.User) serializer.Response {
+	options := model.GetSettingByNames("alipay_enabled", "payjs_enabled")
+	packs, groups, err := decodeProductInfo()
+	if err != nil {
+		return serializer.Err(serializer.CodeInternalSetting, "无法解析商品设置", err)
+	}
+
+	return serializer.BuildProductResponse(groups, packs, options["alipay_enabled"] == "1", options["payjs_enabled"] == "1")
+}
+
+func decodeProductInfo() ([]serializer.PackProduct, []serializer.GroupProducts, error) {
+	options := model.GetSettingByNames("pack_data", "group_sell_data", "alipay_enabled", "payjs_enabled")
+
+	var (
+		packs  []serializer.PackProduct
+		groups []serializer.GroupProducts
+	)
+	if err := json.Unmarshal([]byte(options["pack_data"]), &packs); err != nil {
+		return nil, nil, err
+	}
+	if err := json.Unmarshal([]byte(options["group_sell_data"]), &groups); err != nil {
+		return nil, nil, err
+	}
+
+	return packs, groups, nil
+}
+
+// Quota 获取容量配额信息
+func (service *GeneralVASService) Quota(c *gin.Context, user *model.User) serializer.Response {
+	packs := user.GetAvailableStoragePacks()
+	return serializer.BuildUserQuotaResponse(user, packs)
+}