|
|
@ -42,12 +42,63 @@ func (folder *Folder) GetChildFolder() ([]Folder, error) {
|
|
|
|
return folders, result.Error
|
|
|
|
return folders, result.Error
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// GetRecursiveChildFolder 查找所有递归子目录
|
|
|
|
// GetRecursiveChildFolder 查找所有递归子目录,包括自身
|
|
|
|
func GetRecursiveChildFolder(dirs []string, uid uint) ([]Folder, error) {
|
|
|
|
func GetRecursiveChildFolder(dirs []string, uid uint, includeSelf bool) ([]Folder, error) {
|
|
|
|
folders := make([]Folder, 0, len(dirs))
|
|
|
|
folders := make([]Folder, 0, len(dirs))
|
|
|
|
search := util.BuildRegexp(dirs, "^", "/", "|")
|
|
|
|
var err error
|
|
|
|
result := DB.Where("(owner_id = ? and position_absolute REGEXP ?) or position_absolute in (?)", uid, search, dirs).Find(&folders)
|
|
|
|
if conf.DatabaseConfig.Type == "mysql" {
|
|
|
|
return folders, result.Error
|
|
|
|
|
|
|
|
|
|
|
|
// MySQL 下使用正则查询
|
|
|
|
|
|
|
|
search := util.BuildRegexp(dirs, "^", "/", "|")
|
|
|
|
|
|
|
|
result := DB.Where("(owner_id = ? and position_absolute REGEXP ?) or (owner_id = ? and position_absolute in (?))", uid, search, uid, dirs).Find(&folders)
|
|
|
|
|
|
|
|
err = result.Error
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// SQLite 下使用递归查询
|
|
|
|
|
|
|
|
var parFolders []Folder
|
|
|
|
|
|
|
|
result := DB.Where("owner_id = ? and position_absolute in (?)", uid, dirs).Find(&parFolders)
|
|
|
|
|
|
|
|
if result.Error != nil {
|
|
|
|
|
|
|
|
return folders, err
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 整理父目录的ID
|
|
|
|
|
|
|
|
var parentIDs = make([]uint, 0, len(parFolders))
|
|
|
|
|
|
|
|
for _, folder := range parFolders {
|
|
|
|
|
|
|
|
parentIDs = append(parentIDs, folder.ID)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if includeSelf {
|
|
|
|
|
|
|
|
// 合并至最终结果
|
|
|
|
|
|
|
|
folders = append(folders, parFolders...)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
parFolders = []Folder{}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 递归查询子目录,最大递归65535次
|
|
|
|
|
|
|
|
for i := 0; i < 65535; i++ {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
result = DB.Where("owner_id = ? and parent_id in (?)", uid, parentIDs).Find(&parFolders)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 查询结束条件
|
|
|
|
|
|
|
|
if len(parFolders) == 0 {
|
|
|
|
|
|
|
|
break
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 整理父目录的ID
|
|
|
|
|
|
|
|
parentIDs = make([]uint, 0, len(parFolders))
|
|
|
|
|
|
|
|
for _, folder := range parFolders {
|
|
|
|
|
|
|
|
parentIDs = append(parentIDs, folder.ID)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 合并至最终结果
|
|
|
|
|
|
|
|
folders = append(folders, parFolders...)
|
|
|
|
|
|
|
|
parFolders = []Folder{}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return folders, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// DeleteFolderByIDs 根据给定ID批量删除目录记录
|
|
|
|
// DeleteFolderByIDs 根据给定ID批量删除目录记录
|
|
|
@ -60,10 +111,10 @@ func DeleteFolderByIDs(ids []uint) error {
|
|
|
|
// return path.Join(folder.Position,folder.Name)
|
|
|
|
// return path.Join(folder.Position,folder.Name)
|
|
|
|
//}
|
|
|
|
//}
|
|
|
|
|
|
|
|
|
|
|
|
// MoveFileTo 将此目录下的文件递归移动至dstFolder
|
|
|
|
// MoveFileTo 将此目录下的文件移动至dstFolder
|
|
|
|
func (folder *Folder) MoveFileTo(files []string, dstFolder *Folder) error {
|
|
|
|
func (folder *Folder) MoveFileTo(files []string, dstFolder *Folder) error {
|
|
|
|
// 更改顶级要移动文件的父目录指向
|
|
|
|
// 更改顶级要移动文件的父目录指向
|
|
|
|
err := DB.Model(File{}).Where("dir in (?) and user_id = ?", folder.PositionAbsolute, folder.OwnerID).
|
|
|
|
err := DB.Model(File{}).Where("name in (?) and user_id = ? and dir = ?", files, folder.OwnerID, folder.PositionAbsolute).
|
|
|
|
Update(map[string]interface{}{
|
|
|
|
Update(map[string]interface{}{
|
|
|
|
"folder_id": dstFolder.ID,
|
|
|
|
"folder_id": dstFolder.ID,
|
|
|
|
"dir": dstFolder.PositionAbsolute,
|
|
|
|
"dir": dstFolder.PositionAbsolute,
|
|
|
@ -78,10 +129,25 @@ func (folder *Folder) MoveFileTo(files []string, dstFolder *Folder) error {
|
|
|
|
|
|
|
|
|
|
|
|
// MoveFolderTo 将此目录下的目录移动至dstFolder
|
|
|
|
// MoveFolderTo 将此目录下的目录移动至dstFolder
|
|
|
|
func (folder *Folder) MoveFolderTo(dirs []string, dstFolder *Folder) error {
|
|
|
|
func (folder *Folder) MoveFolderTo(dirs []string, dstFolder *Folder) error {
|
|
|
|
// 生成全局路径
|
|
|
|
// 生成绝对路径
|
|
|
|
fullDirs := make([]string, len(dirs))
|
|
|
|
fullDirs := make([]string, len(dirs))
|
|
|
|
for i := 0; i < len(dirs); i++ {
|
|
|
|
for i := 0; i < len(dirs); i++ {
|
|
|
|
fullDirs[i] = path.Join(folder.PositionAbsolute, dirs[i])
|
|
|
|
fullDirs[i] = path.Join(
|
|
|
|
|
|
|
|
folder.PositionAbsolute,
|
|
|
|
|
|
|
|
path.Base(dirs[i]),
|
|
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var subFolders = make([][]Folder, len(fullDirs))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 更新被移动的目录递归的子目录和文件
|
|
|
|
|
|
|
|
for key, parentDir := range fullDirs {
|
|
|
|
|
|
|
|
// 检索被移动的目录的所有子目录
|
|
|
|
|
|
|
|
toBeMoved, err := GetRecursiveChildFolder([]string{parentDir}, folder.OwnerID, true)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
|
|
return err
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
subFolders[key] = toBeMoved
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// 更改顶级要移动目录的父目录指向
|
|
|
|
// 更改顶级要移动目录的父目录指向
|
|
|
@ -90,41 +156,45 @@ func (folder *Folder) MoveFolderTo(dirs []string, dstFolder *Folder) error {
|
|
|
|
"parent_id": dstFolder.ID,
|
|
|
|
"parent_id": dstFolder.ID,
|
|
|
|
"position": dstFolder.PositionAbsolute,
|
|
|
|
"position": dstFolder.PositionAbsolute,
|
|
|
|
"position_absolute": gorm.Expr(util.BuildConcat("?", "name", conf.DatabaseConfig.Type), util.FillSlash(dstFolder.PositionAbsolute)),
|
|
|
|
"position_absolute": gorm.Expr(util.BuildConcat("?", "name", conf.DatabaseConfig.Type), util.FillSlash(dstFolder.PositionAbsolute)),
|
|
|
|
}).
|
|
|
|
}).Error
|
|
|
|
Error
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// 更新被移动的目录递归的子目录
|
|
|
|
// 更新被移动的目录递归的子目录和文件
|
|
|
|
for _, parentDir := range fullDirs {
|
|
|
|
for _, toBeMoved := range subFolders {
|
|
|
|
toBeMoved, err := GetRecursiveChildFolder([]string{parentDir}, folder.OwnerID)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
|
|
return err
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO 找到更好的修改办法
|
|
|
|
// TODO 找到更好的修改办法
|
|
|
|
for _, subFolder := range toBeMoved {
|
|
|
|
|
|
|
|
newPosition := path.Join(dstFolder.PositionAbsolute, strings.Replace(subFolder.Position, folder.Position, "", 1))
|
|
|
|
for key, subFolder := range toBeMoved {
|
|
|
|
DB.Model(&subFolder).Updates(map[string]interface{}{
|
|
|
|
// 每个分组的第一个目录已经变更指向,直接跳过
|
|
|
|
"position": newPosition,
|
|
|
|
if key > 0 {
|
|
|
|
"position_absolute": path.Join(newPosition, subFolder.Name),
|
|
|
|
newPosition := path.Join(dstFolder.PositionAbsolute, strings.Replace(subFolder.Position, folder.Position, "", 1))
|
|
|
|
})
|
|
|
|
DB.Model(&subFolder).Updates(map[string]interface{}{
|
|
|
|
|
|
|
|
"position": newPosition,
|
|
|
|
|
|
|
|
"position_absolute": path.Join(newPosition, subFolder.Name),
|
|
|
|
|
|
|
|
})
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 更新被移动的目录递归的子文件
|
|
|
|
// 抽离所有子目录的ID
|
|
|
|
for _, parentDir := range fullDirs {
|
|
|
|
var subFolderIDs = make([]uint, len(toBeMoved))
|
|
|
|
toBeMoved, err := GetRecursiveByPaths([]string{parentDir}, folder.OwnerID)
|
|
|
|
for key, subFolder := range toBeMoved {
|
|
|
|
|
|
|
|
subFolderIDs[key] = subFolder.ID
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 获取子目录下的所有子文件
|
|
|
|
|
|
|
|
toBeMovedFile, err := GetFilesByParentIDs(subFolderIDs, folder.OwnerID)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// TODO 找到更好的修改办法
|
|
|
|
for _, subFile := range toBeMovedFile {
|
|
|
|
for _, subFile := range toBeMoved {
|
|
|
|
newPosition := path.Join(dstFolder.PositionAbsolute, strings.Replace(subFile.Dir, folder.PositionAbsolute, "", 1))
|
|
|
|
newPosition := path.Join(dstFolder.PositionAbsolute, strings.Replace(subFile.Dir, folder.Position, "", 1))
|
|
|
|
|
|
|
|
DB.Model(&subFile).Updates(map[string]interface{}{
|
|
|
|
DB.Model(&subFile).Updates(map[string]interface{}{
|
|
|
|
"dir": newPosition,
|
|
|
|
"dir": newPosition,
|
|
|
|
})
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
return nil
|
|
|
|
return nil
|
|
|
|