Feat: get share info / unlock share

pull/247/head
HFO4 5 years ago
parent 94b13393a9
commit 0977b36f8b

@ -19,6 +19,11 @@ type Share struct {
RemainDownloads int // 剩余下载配额,负值标识无限制
Expires *time.Time // 过期时间,空值表示无过期时间
Score int // 每人次下载扣除积分
// 数据库忽略字段
User User `gorm:"PRELOAD:false,association_autoupdate:false"`
File File `gorm:"PRELOAD:false,association_autoupdate:false"`
Folder Folder `gorm:"PRELOAD:false,association_autoupdate:false"`
}
// Create 创建分享
@ -32,6 +37,7 @@ func (share *Share) Create() (uint, error) {
}
// GetShareByHashID 根据HashID查找分享
// TODO 测试
func GetShareByHashID(hashID string) *Share {
id, err := hashid.DecodeHashID(hashID, hashid.ShareID)
if err != nil {
@ -45,3 +51,59 @@ func GetShareByHashID(hashID string) *Share {
return &share
}
// IsAvailable 返回此分享是否可用(是否过期)
// TODO 测试
func (share *Share) IsAvailable() bool {
if share.RemainDownloads == 0 {
return false
}
if share.Expires != nil && time.Now().After(*share.Expires) {
return false
}
// 检查源对象是否存在
var sourceID uint
if share.IsDir {
folder := share.GetSourceFolder()
sourceID = folder.ID
} else {
file := share.GetSourceFile()
sourceID = file.ID
}
if sourceID == 0 {
return false
}
return true
}
// GetCreator 获取分享的创建者
func (share *Share) GetCreator() *User {
if share.User.ID == 0 {
share.User, _ = GetUserByID(share.UserID)
}
return &share.User
}
// GetSourceFolder 获取源目录
func (share *Share) GetSourceFolder() *Folder {
if share.Folder.ID == 0 {
folders, _ := GetFoldersByIDs([]uint{share.SourceID}, share.UserID)
if len(folders) > 0 {
share.Folder = folders[0]
}
}
return &share.Folder
}
// GetSourceFile 获取源文件
func (share *Share) GetSourceFile() *File {
if share.File.ID == 0 {
files, _ := GetFilesByIDs([]uint{share.SourceID}, share.UserID)
if len(files) > 0 {
share.File = files[0]
}
}
return &share.File
}

@ -0,0 +1,70 @@
package serializer
import (
model "github.com/HFO4/cloudreve/models"
"github.com/HFO4/cloudreve/pkg/hashid"
"time"
)
// Share 分享序列化
type Share struct {
Locked bool `json:"locked"`
IsDir bool `json:"is_dir"`
Score int `json:"score"`
CreateDate string `json:"create_date,omitempty"`
Downloads int `json:"downloads"`
Views int `json:"views"`
Expire int64 `json:"expire"`
Creator *shareCreator `json:"creator,omitempty"`
Source *shareSource `json:"source,omitempty"`
}
type shareCreator struct {
Key string `json:"key"`
Nick string `json:"nick"`
GroupName string `json:"group_name"`
}
type shareSource struct {
Name string `json:"name"`
Size uint64 `json:"size"`
}
// BuildShareResponse 构建获取分享信息响应
func BuildShareResponse(share *model.Share, unlocked bool) Share {
creator := share.GetCreator()
resp := Share{
Locked: !unlocked,
Creator: &shareCreator{
Key: hashid.HashID(creator.ID, hashid.UserID),
Nick: creator.Nick,
GroupName: creator.Group.Name,
},
Score: share.Score,
CreateDate: share.CreatedAt.Format("2006-01-02 15:04:05"),
}
if !unlocked {
return resp
}
resp.IsDir = share.IsDir
resp.Downloads = share.Downloads
resp.Views = share.Views
if share.Expires != nil {
resp.Expire = share.Expires.Unix() - time.Now().Unix()
}
if share.IsDir {
} else {
source := share.GetSourceFile()
resp.Source = &shareSource{
Name: source.Name,
Size: source.Size,
}
}
return resp
}

@ -19,11 +19,6 @@ type ShareCreateService struct {
Score int `json:"score" binding:"gte=0"`
}
// ShareGetService 获取分享服务
type ShareGetService struct {
Password string `form:"password" binding:"max=255"`
}
// Create 创建新分享
func (service *ShareCreateService) Create(c *gin.Context) serializer.Response {
user := currentUser(c)
@ -51,11 +46,12 @@ func (service *ShareCreateService) Create(c *gin.Context) serializer.Response {
}
newShare := model.Share{
Password: service.Password,
IsDir: service.IsDir,
UserID: user.ID,
SourceID: service.SourceID,
Score: service.Score,
Password: service.Password,
IsDir: service.IsDir,
UserID: user.ID,
SourceID: service.SourceID,
Score: service.Score,
RemainDownloads: -1,
}
// 如果开启了自动过期
@ -85,20 +81,6 @@ func (service *ShareCreateService) Create(c *gin.Context) serializer.Response {
}
// Get 获取分享内容
func (service *ShareGetService) Get(c *gin.Context) serializer.Response {
user := currentUser(c)
share := model.GetShareByHashID(c.Param("id"))
if share == nil {
return serializer.Err(serializer.CodeNotFound, "分享不存在或已被取消", nil)
}
return serializer.Response{
Code: 0,
Data: user,
}
}
func currentUser(c *gin.Context) *model.User {
var user *model.User
if userCtx, ok := c.Get("user"); ok {

@ -0,0 +1,41 @@
package share
import (
"fmt"
model "github.com/HFO4/cloudreve/models"
"github.com/HFO4/cloudreve/pkg/serializer"
"github.com/HFO4/cloudreve/pkg/util"
"github.com/gin-gonic/gin"
)
// ShareGetService 获取分享服务
type ShareGetService struct {
Password string `form:"password" binding:"max=255"`
}
// Get 获取分享内容
func (service *ShareGetService) Get(c *gin.Context) serializer.Response {
share := model.GetShareByHashID(c.Param("id"))
if share == nil || !share.IsAvailable() {
return serializer.Err(serializer.CodeNotFound, "分享不存在或已被取消", nil)
}
// 是否已解锁
unlocked := true
if share.Password != "" {
sessionKey := fmt.Sprintf("share_unlock_%d", share.ID)
unlocked = util.GetSession(c, sessionKey) != nil
if !unlocked && service.Password != "" {
// 如果未解锁,且指定了密码,则尝试解锁
if service.Password == share.Password {
unlocked = true
util.SetSession(c, map[string]interface{}{sessionKey: true})
}
}
}
return serializer.Response{
Code: 0,
Data: serializer.BuildShareResponse(share, unlocked),
}
}
Loading…
Cancel
Save