Feat: WebDAV mount folders

service/setting/webdav.go

@@ -2,6 +2,8 @@ package setting
 
 import (
 	model "github.com/HFO4/cloudreve/models"
+	"github.com/HFO4/cloudreve/pkg/filesystem"
+	"github.com/HFO4/cloudreve/pkg/hashid"
 	"github.com/HFO4/cloudreve/pkg/serializer"
 	"github.com/HFO4/cloudreve/pkg/util"
 	"github.com/gin-gonic/gin"
@@ -22,6 +24,65 @@ type WebDAVAccountCreateService struct {
 	Name string `json:"name" binding:"required,min=1,max=255"`
 }
 
+// WebDAVMountCreateService WebDAV 挂载创建服务
+type WebDAVMountCreateService struct {
+	Path   string `json:"path" binding:"required,min=1,max=65535"`
+	Policy string `json:"policy" binding:"required,min=1"`
+}
+
+// Create 创建目录挂载
+func (service *WebDAVMountCreateService) Create(c *gin.Context, user *model.User) serializer.Response {
+	// 创建文件系统
+	fs, err := filesystem.NewFileSystem(user)
+	if err != nil {
+		return serializer.Err(serializer.CodePolicyNotAllowed, err.Error(), err)
+	}
+	defer fs.Recycle()
+
+	// 检索要挂载的目录
+	exist, folder := fs.IsPathExist(service.Path)
+	if !exist {
+		return serializer.Err(serializer.CodeNotFound, "路径不存在", err)
+	}
+
+	// 检索要挂载的存储策略
+	policyID, err := hashid.DecodeHashID(service.Policy, hashid.PolicyID)
+	if err != nil {
+		return serializer.Err(serializer.CodeNotFound, "存储策略不存在", err)
+	}
+
+	// 检查存储策略是否可用
+	if policy, err := model.GetPolicyByID(policyID); err != nil || !util.ContainsUint(user.Group.PolicyList, policy.ID) {
+		return serializer.Err(serializer.CodeNotFound, "存储策略不可用", err)
+	}
+
+	// 挂载
+	if err := folder.Mount(policyID); err != nil {
+		return serializer.Err(serializer.CodeDBError, "挂载失败", err)
+	}
+
+	return serializer.Response{
+		Data: map[string]interface{}{
+			"id": hashid.HashID(folder.ID, hashid.FolderID),
+		},
+	}
+}
+
+// Unmount 取消目录挂载
+func (service *WebDAVListService) Unmount(c *gin.Context, user *model.User) serializer.Response {
+	folderID, _ := c.Get("object_id")
+	folder, err := model.GetFoldersByIDs([]uint{folderID.(uint)}, user.ID)
+	if err != nil || len(folder) == 0 {
+		return serializer.Err(serializer.CodeNotFound, "目录不存在", err)
+	}
+
+	if err := folder[0].Mount(0); err != nil {
+		return serializer.Err(serializer.CodeDBError, "取消挂载失败", err)
+	}
+
+	return serializer.Response{}
+}
+
 // Create 创建WebDAV账户
 func (service *WebDAVAccountCreateService) Create(c *gin.Context, user *model.User) serializer.Response {
 	account := model.Webdav{
@@ -53,7 +114,12 @@ func (service *WebDAVAccountService) Delete(c *gin.Context, user *model.User) se
 // Accounts 列出WebDAV账号
 func (service *WebDAVListService) Accounts(c *gin.Context, user *model.User) serializer.Response {
 	accounts := model.ListWebDAVAccounts(user.ID)
+
+	// 查找挂载了存储策略的目录
+	folders := model.GetMountedFolders(user.ID)
+
 	return serializer.Response{Data: map[string]interface{}{
 		"accounts": accounts,
+		"folders":  serializer.BuildMountedFolderRes(folders, user.Group.PolicyList),
 	}}
 }