Feat: get share info / unlock share

models/share.go

@@ -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
+}

pkg/serializer/share.go

@@ -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
+
+}

service/share/manage.go

@@ -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)
@@ -56,6 +51,7 @@ func (service *ShareCreateService) Create(c *gin.Context) serializer.Response {
 		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 {

service/share/visit.go

@@ -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),
+	}
+}