Feat: mount storage policy for folders in WebDAV

pkg/webdav/file.go

@@ -48,8 +48,8 @@ func moveFiles(ctx context.Context, fs *filesystem.FileSystem, src FileInfo, dst
 	} else {
 		err = fs.Move(
 			ctx,
-			fileIDs,
 			folderIDs,
+			fileIDs,
 			src.GetPosition(),
 			path.Dir(dst),
 		)
@@ -81,7 +81,7 @@ func copyFiles(ctx context.Context, fs *filesystem.FileSystem, src FileInfo, dst
 			return http.StatusInternalServerError, err
 		}
 	} else {
-		err := fs.Copy(ctx, []uint{src.(*model.File).ID}, []uint{}, src.(*model.File).Position, dst)
+		err := fs.Copy(ctx, []uint{}, []uint{src.(*model.File).ID}, src.(*model.File).Position, path.Dir(dst))
 		if err != nil {
 			return http.StatusInternalServerError, err
 		}

pkg/webdav/webdav.go

@@ -356,6 +356,17 @@ func (h *Handler) handlePut(w http.ResponseWriter, r *http.Request, fs *filesyst
 		fs.Use("AfterValidateFailed", filesystem.HookGiveBackCapacity)
 		ctx = context.WithValue(ctx, fsctx.FileModelCtx, *originFile)
 	} else {
+		// 检查父目录指定存储策略
+		if exist, folder := fs.IsPathExist(filePath); exist {
+			if folder.PolicyID != 0 {
+				// 尝试获取并重设存储策略
+				if policy, err := model.GetPolicyByID(fs.User.GetPolicyID(folder.PolicyID)); err == nil {
+					fs.User.Policy = policy
+					fs.DispatchHandler()
+				}
+			}
+		}
+
 		// 给文件系统分配钩子
 		fs.Use("BeforeUpload", filesystem.HookValidateFile)
 		fs.Use("BeforeUpload", filesystem.HookValidateCapacity)