Feat: delete objects

models/file.go

@@ -3,6 +3,7 @@ package model
 import (
 	"github.com/HFO4/cloudreve/pkg/util"
 	"github.com/jinzhu/gorm"
+	"path"
 )
 
 // File 文件
@@ -31,7 +32,7 @@ func (file *File) Create() (uint, error) {
 	return file.ID, nil
 }
 
-// GetFileByPathAndName 给定路径、文件名、用户ID，查找文件
+// GetFileByPathAndName 给定路径(s)、文件名、用户ID，查找文件
 func GetFileByPathAndName(path string, name string, uid uint) (File, error) {
 	var file File
 	result := DB.Where("user_id = ? AND dir = ? AND name=?", uid, path, name).First(&file)
@@ -45,6 +46,20 @@ func (folder *Folder) GetChildFile() ([]File, error) {
 	return files, result.Error
 }
 
+// GetChildFilesOfFolders 批量检索目录子文件
+func GetChildFilesOfFolders(folders *[]Folder) ([]File, error) {
+	// 将所有待删除目录ID抽离，以便检索文件
+	folderIDs := make([]uint, 0, len(*folders))
+	for _, value := range *folders {
+		folderIDs = append(folderIDs, value.ID)
+	}
+
+	// 检索文件
+	var files []File
+	result := DB.Where("folder_id in (?)", folderIDs).Find(&files)
+	return files, result.Error
+}
+
 // GetPolicy 获取文件所属策略
 // TODO:test
 func (file *File) GetPolicy() *Policy {
@@ -53,3 +68,52 @@ func (file *File) GetPolicy() *Policy {
 	}
 	return &file.Policy
 }
+
+// GetFileByPaths 根据给定的文件路径(s)查找文件
+func GetFileByPaths(paths []string, uid uint) ([]File, error) {
+	var files []File
+	tx := DB
+	for _, value := range paths {
+		base := path.Base(value)
+		dir := path.Dir(value)
+		tx = tx.Or("dir = ? and name = ? and user_id = ?", dir, base, uid)
+	}
+	result := tx.Find(&files)
+	return files, result.Error
+}
+
+// RemoveFilesWithSoftLinks 去除给定的文件列表中有软链接的文件
+func RemoveFilesWithSoftLinks(files []File) ([]File, error) {
+	// 结果值
+	filteredFiles := make([]File, 0)
+
+	// 查询软链接的文件
+	var filesWithSoftLinks []File
+	tx := DB
+	for _, value := range files {
+		tx = tx.Or("source_name = ? and policy_id = ? and id != ?", value.SourceName, value.GetPolicy().ID, value.ID)
+	}
+	result := tx.Find(&filesWithSoftLinks)
+	if result.Error != nil {
+		return nil, result.Error
+	}
+
+	// 过滤具有软连接的文件
+	for i := 0; i < len(files); i++ {
+		for _, value := range filesWithSoftLinks {
+			if value.PolicyID != files[i].PolicyID || value.SourceName != files[i].SourceName {
+				filteredFiles = append(filteredFiles, files[i])
+				break
+			}
+		}
+	}
+
+	return filteredFiles, nil
+
+}
+
+// DeleteFileByIDs 根据给定ID批量删除文件记录
+func DeleteFileByIDs(ids []uint) error {
+	result := DB.Where("id in (?)", ids).Delete(&File{})
+	return result.Error
+}