package serializer import ( "fmt" "github.com/HFO4/cloudreve/models" ) // CheckLogin 检查登录 func CheckLogin() Response { return Response{ Code: CodeCheckLogin, Msg: "未登录", } } // User 用户序列化器 type User struct { ID uint `json:"id"` Email string `json:"user_name"` Nickname string `json:"nickname"` Status int `json:"status"` Avatar string `json:"avatar"` CreatedAt int64 `json:"created_at"` PreferredTheme string `json:"preferred_theme"` Policy policy `json:"policy"` Group group `json:"group"` } type policy struct { SaveType string `json:"saveType"` MaxSize string `json:"maxSize"` AllowedType []string `json:"allowedType"` UploadURL string `json:"upUrl"` AllowGetSource bool `json:"allowSource"` } type group struct { AllowShare bool `json:"allowShare"` AllowRemoteDownload bool `json:"allowRemoteDownload"` AllowTorrentDownload bool `json:"allowTorrentDownload"` } type storage struct { Used uint64 `json:"used"` Free uint64 `json:"free"` Total uint64 `json:"total"` } // BuildUser 序列化用户 func BuildUser(user model.User) User { aria2Option := user.Group.GetAria2Option() return User{ ID: user.ID, Email: user.Email, Nickname: user.Nick, Status: user.Status, Avatar: user.Avatar, CreatedAt: user.CreatedAt.Unix(), PreferredTheme: user.OptionsSerialized.PreferredTheme, Policy: policy{ SaveType: user.Policy.Type, MaxSize: fmt.Sprintf("%.2fmb", float64(user.Policy.MaxSize)/(1024*1024)), AllowedType: user.Policy.OptionsSerialized.FileType, UploadURL: user.Policy.Server, AllowGetSource: user.Policy.IsOriginLinkEnable, }, Group: group{ AllowShare: user.Group.ShareEnabled, AllowRemoteDownload: aria2Option[0], AllowTorrentDownload: aria2Option[2], }, } } // BuildUserResponse 序列化用户响应 func BuildUserResponse(user model.User) Response { return Response{ Data: BuildUser(user), } } // BuildUserStorageResponse 序列化用户存储概况响应 func BuildUserStorageResponse(user model.User) Response { storageResp := storage{ Used: user.Storage, Free: user.Group.MaxStorage - user.Storage, Total: user.Group.MaxStorage, } if user.Group.MaxStorage < user.Storage { storageResp.Free = 0 } return Response{ Data: storageResp, } }