platform/web/handlers/resource.go

package handlers

import (
	"errors"
	"platform/pkg/u"
	"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 {
	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, []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.
		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)))
	}

	var resource []*m.Resource
	err = do.Debug().
		Order(q.Resource.CreatedAt.Desc()).
		Offset(req.GetOffset()).
		Limit(req.GetLimit()).
		Scan(&resource)
	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, []services.PayloadType{services.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 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,
			ResourceNo: services.ID.GenReadable("res"),
		}
		err = q.Resource.Create(&resource)
		if err != nil {
			return err
		}
		resourcePss := m.ResourcePss{
			ResourceID: resource.ID,
			Type:       req.Type,
			Live:       req.Live,
			Quota:      req.Quota,
			Expire:     common.LocalDateTime(time.Now().Add(time.Duration(req.Expire) * time.Second)),
			DailyLimit: req.DailyLimit,
		}
		err = q.ResourcePss.Create(&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