159 lines
4.0 KiB
Go
159 lines
4.0 KiB
Go
package serializer
|
||
|
||
import (
|
||
model "github.com/cloudreve/Cloudreve/v3/models"
|
||
"github.com/cloudreve/Cloudreve/v3/pkg/hashid"
|
||
"github.com/cloudreve/Cloudreve/v3/pkg/util"
|
||
"time"
|
||
)
|
||
|
||
type quota struct {
|
||
Base uint64 `json:"base"`
|
||
Pack uint64 `json:"pack"`
|
||
Used uint64 `json:"used"`
|
||
Total uint64 `json:"total"`
|
||
Packs []storagePacks `json:"packs"`
|
||
}
|
||
|
||
type storagePacks struct {
|
||
Name string `json:"name"`
|
||
Size uint64 `json:"size"`
|
||
ActivateDate time.Time `json:"activate_date"`
|
||
Expiration int `json:"expiration"`
|
||
ExpirationDate time.Time `json:"expiration_date"`
|
||
}
|
||
|
||
// MountedFolders 已挂载的目录
|
||
type MountedFolders struct {
|
||
ID string `json:"id"`
|
||
Name string `json:"name"`
|
||
PolicyName string `json:"policy_name"`
|
||
}
|
||
|
||
type policyOptions struct {
|
||
Name string `json:"name"`
|
||
ID string `json:"id"`
|
||
}
|
||
|
||
type nodeOptions struct {
|
||
Name string `json:"name"`
|
||
ID uint `json:"id"`
|
||
}
|
||
|
||
// BuildPolicySettingRes 构建存储策略选项选择
|
||
func BuildPolicySettingRes(policies []model.Policy) Response {
|
||
options := make([]policyOptions, 0, len(policies))
|
||
for _, policy := range policies {
|
||
options = append(options, policyOptions{
|
||
Name: policy.Name,
|
||
ID: hashid.HashID(policy.ID, hashid.PolicyID),
|
||
})
|
||
}
|
||
|
||
return Response{
|
||
Data: options,
|
||
}
|
||
}
|
||
|
||
// BuildMountedFolderRes 构建已挂载目录响应，list为当前用户可用存储策略ID
|
||
func BuildMountedFolderRes(folders []model.Folder, list []uint) []MountedFolders {
|
||
res := make([]MountedFolders, 0, len(folders))
|
||
for _, folder := range folders {
|
||
single := MountedFolders{
|
||
ID: hashid.HashID(folder.ID, hashid.FolderID),
|
||
Name: folder.Name,
|
||
PolicyName: "[Invalid Policy]",
|
||
}
|
||
if policy, err := model.GetPolicyByID(folder.PolicyID); err == nil && util.ContainsUint(list, policy.ID) {
|
||
single.PolicyName = policy.Name
|
||
}
|
||
|
||
res = append(res, single)
|
||
}
|
||
|
||
return res
|
||
}
|
||
|
||
// BuildUserQuotaResponse 序列化用户存储配额概况响应
|
||
func BuildUserQuotaResponse(user *model.User, packs []model.StoragePack) Response {
|
||
packSize := user.GetAvailablePackSize()
|
||
res := quota{
|
||
Base: user.Group.MaxStorage,
|
||
Pack: packSize,
|
||
Used: user.Storage,
|
||
Total: packSize + user.Group.MaxStorage,
|
||
Packs: make([]storagePacks, 0, len(packs)),
|
||
}
|
||
for _, pack := range packs {
|
||
res.Packs = append(res.Packs, storagePacks{
|
||
Name: pack.Name,
|
||
Size: pack.Size,
|
||
ActivateDate: *pack.ActiveTime,
|
||
Expiration: int(pack.ExpiredTime.Sub(*pack.ActiveTime).Seconds()),
|
||
ExpirationDate: *pack.ExpiredTime,
|
||
})
|
||
}
|
||
|
||
return Response{
|
||
Data: res,
|
||
}
|
||
}
|
||
|
||
// PackProduct 容量包商品
|
||
type PackProduct struct {
|
||
ID int64 `json:"id"`
|
||
Name string `json:"name"`
|
||
Size uint64 `json:"size"`
|
||
Time int64 `json:"time"`
|
||
Price int `json:"price"`
|
||
Score int `json:"score"`
|
||
}
|
||
|
||
// GroupProducts 用户组商品
|
||
type GroupProducts struct {
|
||
ID int64 `json:"id"`
|
||
Name string `json:"name"`
|
||
GroupID uint `json:"group_id"`
|
||
Time int64 `json:"time"`
|
||
Price int `json:"price"`
|
||
Score int `json:"score"`
|
||
Des []string `json:"des"`
|
||
Highlight bool `json:"highlight"`
|
||
}
|
||
|
||
// BuildProductResponse 构建增值服务商品响应
|
||
func BuildProductResponse(groups []GroupProducts, packs []PackProduct,
|
||
wechat, alipay, payjs, custom bool, customName string, scorePrice int) Response {
|
||
// 隐藏响应中的用户组ID
|
||
for i := 0; i < len(groups); i++ {
|
||
groups[i].GroupID = 0
|
||
}
|
||
return Response{
|
||
Data: map[string]interface{}{
|
||
"packs": packs,
|
||
"groups": groups,
|
||
"alipay": alipay,
|
||
"wechat": wechat,
|
||
"payjs": payjs,
|
||
"custom": custom,
|
||
"custom_name": customName,
|
||
"score_price": scorePrice,
|
||
},
|
||
}
|
||
}
|
||
|
||
// BuildNodeOptionRes 构建可用节点列表响应
|
||
func BuildNodeOptionRes(nodes []*model.Node) Response {
|
||
options := make([]nodeOptions, 0, len(nodes))
|
||
for _, node := range nodes {
|
||
options = append(options, nodeOptions{
|
||
Name: node.Name,
|
||
ID: node.ID,
|
||
})
|
||
}
|
||
|
||
return Response{
|
||
Data: options,
|
||
}
|
||
}
|