package filesystem import ( "context" "fmt" model "github.com/HFO4/cloudreve/models" "github.com/HFO4/cloudreve/pkg/serializer" "github.com/HFO4/cloudreve/pkg/util" "path" ) /* ================= 路径/目录相关 ================= */ // Object 文件或者目录 type Object struct { ID uint `json:"id"` Name string `json:"name"` Path string `json:"path"` Pic string `json:"pic"` Size uint64 `json:"size"` Type string `json:"type"` Date string `json:"date"` } // Delete 递归删除对象 func (fs *FileSystem) Delete(ctx context.Context, dirs, files []string) error { // 已删除的总容量,map用于去重 var deletedStorage = make(map[uint]uint64) // 已删除的文件ID var deletedFileIDs = make([]uint, 0, len(fs.FileTarget)) // 删除失败的文件的父目录ID // 所有文件的ID var allFileIDs = make([]uint, 0, len(fs.FileTarget)) // 列出要删除的目录 if len(dirs) > 0 { err := fs.ListDeleteDirs(ctx, dirs) if err != nil { return err } } // 列出要删除的文件 if len(files) > 0 { err := fs.ListDeleteFiles(ctx, files) if err != nil { return err } // 去除待删除文件中包含软连接的部分 filesToBeDelete, err := model.RemoveFilesWithSoftLinks(fs.FileTarget) if err != nil { return ErrDBListObjects.WithError(err) } // 根据存储策略将文件分组 policyGroup := fs.GroupFileByPolicy(ctx, filesToBeDelete) // 按照存储策略分组删除对象 failed := fs.deleteGroupedFile(ctx, policyGroup) for i := 0; i < len(fs.FileTarget); i++ { if util.ContainsString(failed[fs.FileTarget[i].PolicyID], fs.FileTarget[i].SourceName) { // TODO 删除失败时不删除文件记录及父目录 } else { deletedFileIDs = append(deletedFileIDs, fs.FileTarget[i].ID) deletedStorage[fs.FileTarget[i].ID] = fs.FileTarget[i].Size } allFileIDs = append(allFileIDs, fs.FileTarget[i].ID) } } // 删除文件记录 err := model.DeleteFileByIDs(allFileIDs) if err != nil { return ErrDBListObjects.WithError(err) } // 归还容量 var total uint64 for _, value := range deletedStorage { total += value } fs.User.IncreaseStorage(total) // 删除目录 var allFolderIDs = make([]uint, 0, len(fs.DirTarget)) for _, value := range fs.DirTarget { allFolderIDs = append(allFolderIDs, value.ID) } err = model.DeleteFolderByIDs(allFolderIDs) if err != nil { return ErrDBListObjects.WithError(err) } if notDeleted := len(fs.FileTarget) - len(deletedFileIDs); notDeleted > 0 { return serializer.NewError(serializer.CodeNotFullySuccess, fmt.Sprintf("有 %d 个文件未能成功删除,已删除它们的文件记录", notDeleted), nil) } return nil } // ListDeleteDirs 递归列出要删除目录,及目录下所有文件 func (fs *FileSystem) ListDeleteDirs(ctx context.Context, dirs []string) error { // 列出所有递归子目录 folders, err := model.GetRecursiveChildFolder(dirs, fs.User.ID) if err != nil { return ErrDBListObjects.WithError(err) } fs.SetTargetDir(&folders) // 检索目录下的子文件 files, err := model.GetChildFilesOfFolders(&folders) if err != nil { return ErrDBListObjects.WithError(err) } fs.SetTargetFile(&files) return nil } // ListDeleteFiles 根据给定的路径列出要删除的文件 func (fs *FileSystem) ListDeleteFiles(ctx context.Context, paths []string) error { files, err := model.GetFileByPaths(paths, fs.User.ID) if err != nil { return ErrDBListObjects.WithError(err) } fs.SetTargetFile(&files) return nil } // List 列出路径下的内容, // pathProcessor为最终对象路径的处理钩子。 // 有些情况下(如在分享页面列对象)时, // 路径需要截取掉被分享目录路径之前的部分。 func (fs *FileSystem) List(ctx context.Context, dirPath string, pathProcessor func(string) string) ([]Object, error) { // 获取父目录 isExist, folder := fs.IsPathExist(dirPath) // 不存在时返回空的结果 if !isExist { return nil, nil } // 获取子目录 childFolders, _ := folder.GetChildFolder() // 获取子文件 childFiles, _ := folder.GetChildFile() // 汇总处理结果 objects := make([]Object, 0, len(childFiles)+len(childFolders)) // 所有对象的父目录 var processedPath string for _, folder := range childFolders { // 路径处理钩子, // 所有对象父目录都是一样的,所以只处理一次 if processedPath == "" { if pathProcessor != nil { processedPath = pathProcessor(folder.Position) } else { processedPath = folder.Position } } objects = append(objects, Object{ ID: folder.ID, Name: folder.Name, Path: processedPath, Pic: "", Size: 0, Type: "dir", Date: folder.CreatedAt.Format("2006-01-02 15:04:05"), }) } for _, file := range childFiles { if processedPath == "" { if pathProcessor != nil { processedPath = pathProcessor(file.Dir) } else { processedPath = file.Dir } } objects = append(objects, Object{ ID: file.ID, Name: file.Name, Path: processedPath, Pic: file.PicInfo, Size: file.Size, Type: "file", Date: file.CreatedAt.Format("2006-01-02 15:04:05"), }) } return objects, nil } // CreateDirectory 根据给定的完整创建目录,不会递归创建 func (fs *FileSystem) CreateDirectory(ctx context.Context, fullPath string) error { // 获取要创建目录的父路径和目录名 fullPath = path.Clean(fullPath) base := path.Dir(fullPath) dir := path.Base(fullPath) // 检查目录名是否合法 if !fs.ValidateLegalName(ctx, dir) { return ErrIllegalObjectName } // 父目录是否存在 isExist, parent := fs.IsPathExist(base) if !isExist { return ErrPathNotExist } // 是否有同名目录 b, _ := fs.IsPathExist(fullPath) if b { return ErrFolderExisted } // 是否有同名文件 if ok, _ := fs.IsFileExist(path.Join(base, dir)); ok { return ErrFileExisted } // 创建目录 newFolder := model.Folder{ Name: dir, ParentID: parent.ID, Position: base, OwnerID: fs.User.ID, } _, err := newFolder.Create() return err } // IsPathExist 返回给定目录是否存在 // 如果存在就返回目录 func (fs *FileSystem) IsPathExist(path string) (bool, *model.Folder) { folder, err := model.GetFolderByPath(path, fs.User.ID) return err == nil, &folder } // IsFileExist 返回给定路径的文件是否存在 func (fs *FileSystem) IsFileExist(fullPath string) (bool, model.File) { basePath := path.Dir(fullPath) fileName := path.Base(fullPath) file, err := model.GetFileByPathAndName(basePath, fileName, fs.User.ID) return err == nil, file }