platform/web/handlers/resource.go

package handlers

import (
	"errors"
	"platform/web/auth"
	"platform/web/common"
	m "platform/web/models"
	q "platform/web/queries"
	"platform/web/services"
	"time"

	"github.com/gofiber/fiber/v2"
)

// region ListResourcePss

type ListResourcePssReq struct {
	Page         int        `json:"page" validate:"required"`
	Size         int        `json:"size" validate:"required"`
	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, []services.PayloadType{services.PayloadUser}, []string{})
	if err != nil {
		return err
	}

	// 解析请求参数
	req := new(ListResourcePssReq)
	if err := c.BodyParser(req); err != nil {
		return err
	}

	// 查询资源列表
	do := q.Resource.
		Join(q.ResourcePss, q.ResourcePss.ResourceID.EqCol(q.Resource.ID)).
		Where(q.Resource.UserID.Eq(authContext.Payload.Id))
	if req.Active != nil {
		do = do.Where(q.Resource.Active.Is(*req.Active))
	}
	if req.Type != nil {
		do = do.Where(q.ResourcePss.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.Expire.Gte(*req.ExpireAfter))
	}
	if req.ExpireBefore != nil {
		do = do.Where(q.ResourcePss.Expire.Lte(*req.ExpireBefore))
	}

	total, err := do.Debug().
		Count()
	if err != nil {
		return err
	}

	var resourcePss []*m.ResourcePss
	err = do.Debug().
		Select(q.ResourcePss.ALL).
		Order(q.ResourcePss.CreatedAt.Desc()).
		Offset((req.Page - 1) * req.Size).
		Limit(req.Size).
		Scan(&resourcePss)
	if err != nil {
		return err
	}

	return c.JSON(common.PageResp{
		Total: int(total),
		Page:  req.Page,
		Size:  req.Size,
		List:  resourcePss,
	})
}

// endregion

// region CreateResourceByBalance

type CreateResourceByBalanceReq struct {
	Type       int32 `json:"type" validate:"required"`
	Live       int32 `json:"live" validate:"required"`
	Expire     int32 `json:"expire" validate:"required"`
	Quota      int32 `json:"quota" validate:"required"`
	DailyLimit int32 `json:"daily_limit" validate:"required"`
}

// CreateResourceByBalance 通过余额创建资源
func CreateResourceByBalance(c *fiber.Ctx) error {

	// 检查权限
	authContext, err := auth.Protect(c, []services.PayloadType{services.PayloadUser}, []string{})
	if err != nil {
		return err
	}

	// 解析请求参数
	req := new(CreateResourceByBalanceReq)
	if err := c.BodyParser(req); err != nil {
		return err
	}

	err = q.Q.Transaction(func(q *q.Query) error {
		// 检查用户
		user, err := q.User.Where(q.User.ID.Eq(authContext.Payload.Id)).Take()
		if err != nil {
			return err
		}

		// 计算价格
		var amount = 100
		var payment = 100

		// 检查余额
		if user.Balance < float64(amount)/100 {
			return fiber.NewError(fiber.StatusBadRequest, "余额不足")
		}

		// 更新用户余额
		user.Balance -= float64(payment)
		_, err = q.User.
			Where(q.User.ID.Eq(authContext.Payload.Id)).
			Update(q.User.Balance, user.Balance)
		if err != nil {
			return err
		}

		// 创建资源
		resource := m.Resource{
			UserID: authContext.Payload.Id,
		}
		err = q.Resource.Save(&resource)
		if err != nil {
			return err
		}
		resourcePss := m.ResourcePss{
			ResourceID: resource.ID,
			Type:       req.Type,
			Live:       req.Live,
			Quota:      req.Quota,
			Expire:     time.Now().Add(time.Duration(req.Expire) * time.Second),
			DailyLimit: req.DailyLimit,
		}
		err = q.ResourcePss.Save(&resourcePss)
		if err != nil {
			return err
		}

		// 生成账单
		bill := m.Bill{
			UserID:     authContext.Payload.Id,
			ResourceID: resource.ID,
			BillNo:     services.ID.GenReadable("bil"),
			Info:       "购买套餐",
			Type:       1,
			Status:     1,
		}
		err = q.Bill.
			Omit(q.Bill.TradeID, q.Bill.RefundID).
			Create(&bill)
		if err != nil {
			return err
		}

		return nil
	})
	if err != nil {
		return err
	}

	return nil
}

// endregion

// region CreateResourceByAlipayCallback

type CreateResourceByAlipayCallbackReq struct {
}

// CreateResourceByAlipayCallback 支付宝支付回调
func CreateResourceByAlipayCallback(c *fiber.Ctx) error {

	// 根据支付类型执行不同流程：
	// 1. 支付宝或微信（即时支付）
	// - 更新订单状态
	// - 生成账单
	// - 生成资源

	return errors.New("not implemented")
}

// endregion

// region CreateResourceByWechatCallback

type CreateResourceByWechatCallbackReq struct {
}

// CreateResourceByWechatCallback 微信支付回调
func CreateResourceByWechatCallback(c *fiber.Ctx) error {
	return errors.New("not implemented")
}

// endregion