|
|
package explorer
|
|
|
|
|
|
import (
|
|
|
"context"
|
|
|
"encoding/gob"
|
|
|
"fmt"
|
|
|
"math"
|
|
|
"path"
|
|
|
"strings"
|
|
|
"time"
|
|
|
|
|
|
model "github.com/cloudreve/Cloudreve/v3/models"
|
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/auth"
|
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/cache"
|
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/filesystem"
|
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/hashid"
|
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/serializer"
|
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/task"
|
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/util"
|
|
|
"github.com/gin-gonic/gin"
|
|
|
)
|
|
|
|
|
|
// ItemMoveService 处理多文件/目录移动
|
|
|
type ItemMoveService struct {
|
|
|
SrcDir string `json:"src_dir" binding:"required,min=1,max=65535"`
|
|
|
Src ItemIDService `json:"src"`
|
|
|
Dst string `json:"dst" binding:"required,min=1,max=65535"`
|
|
|
}
|
|
|
|
|
|
// ItemRenameService 处理多文件/目录重命名
|
|
|
type ItemRenameService struct {
|
|
|
Src ItemIDService `json:"src"`
|
|
|
NewName string `json:"new_name" binding:"required,min=1,max=255"`
|
|
|
}
|
|
|
|
|
|
// ItemService 处理多文件/目录相关服务
|
|
|
type ItemService struct {
|
|
|
Items []uint `json:"items"`
|
|
|
Dirs []uint `json:"dirs"`
|
|
|
}
|
|
|
|
|
|
// ItemIDService 处理多文件/目录相关服务，字段值为HashID，可通过Raw()方法获取原始ID
|
|
|
type ItemIDService struct {
|
|
|
Items []string `json:"items"`
|
|
|
Dirs []string `json:"dirs"`
|
|
|
Source *ItemService
|
|
|
Force bool `json:"force"`
|
|
|
UnlinkOnly bool `json:"unlink"`
|
|
|
}
|
|
|
|
|
|
// ItemCompressService 文件压缩任务服务
|
|
|
type ItemCompressService struct {
|
|
|
Src ItemIDService `json:"src"`
|
|
|
Dst string `json:"dst" binding:"required,min=1,max=65535"`
|
|
|
Name string `json:"name" binding:"required,min=1,max=255"`
|
|
|
}
|
|
|
|
|
|
// ItemDecompressService 文件解压缩任务服务
|
|
|
type ItemDecompressService struct {
|
|
|
Src string `json:"src"`
|
|
|
Dst string `json:"dst" binding:"required,min=1,max=65535"`
|
|
|
Encoding string `json:"encoding"`
|
|
|
}
|
|
|
|
|
|
// ItemPropertyService 获取对象属性服务
|
|
|
type ItemPropertyService struct {
|
|
|
ID string `binding:"required"`
|
|
|
TraceRoot bool `form:"trace_root"`
|
|
|
IsFolder bool `form:"is_folder"`
|
|
|
}
|
|
|
|
|
|
func init() {
|
|
|
gob.Register(ItemIDService{})
|
|
|
}
|
|
|
|
|
|
// Raw 批量解码HashID，获取原始ID
|
|
|
func (service *ItemIDService) Raw() *ItemService {
|
|
|
if service.Source != nil {
|
|
|
return service.Source
|
|
|
}
|
|
|
|
|
|
service.Source = &ItemService{
|
|
|
Dirs: make([]uint, 0, len(service.Dirs)),
|
|
|
Items: make([]uint, 0, len(service.Items)),
|
|
|
}
|
|
|
for _, folder := range service.Dirs {
|
|
|
id, err := hashid.DecodeHashID(folder, hashid.FolderID)
|
|
|
if err == nil {
|
|
|
service.Source.Dirs = append(service.Source.Dirs, id)
|
|
|
}
|
|
|
}
|
|
|
for _, file := range service.Items {
|
|
|
id, err := hashid.DecodeHashID(file, hashid.FileID)
|
|
|
if err == nil {
|
|
|
service.Source.Items = append(service.Source.Items, id)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
return service.Source
|
|
|
}
|
|
|
|
|
|
// CreateDecompressTask 创建文件解压缩任务
|
|
|
func (service *ItemDecompressService) CreateDecompressTask(c *gin.Context) serializer.Response {
|
|
|
// 创建文件系统
|
|
|
fs, err := filesystem.NewFileSystemFromContext(c)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeCreateFSError, "", err)
|
|
|
}
|
|
|
defer fs.Recycle()
|
|
|
|
|
|
// 检查用户组权限
|
|
|
if !fs.User.Group.OptionsSerialized.ArchiveTask {
|
|
|
return serializer.Err(serializer.CodeGroupNotAllowed, "", nil)
|
|
|
}
|
|
|
|
|
|
// 存放目录是否存在
|
|
|
if exist, _ := fs.IsPathExist(service.Dst); !exist {
|
|
|
return serializer.Err(serializer.CodeParentNotExist, "", nil)
|
|
|
}
|
|
|
|
|
|
// 压缩包是否存在
|
|
|
exist, file := fs.IsFileExist(service.Src)
|
|
|
if !exist {
|
|
|
return serializer.Err(serializer.CodeFileNotFound, "", nil)
|
|
|
}
|
|
|
|
|
|
// 文件尺寸限制
|
|
|
if fs.User.Group.OptionsSerialized.DecompressSize != 0 && file.Size > fs.User.Group.
|
|
|
OptionsSerialized.DecompressSize {
|
|
|
return serializer.Err(serializer.CodeFileTooLarge, "", nil)
|
|
|
}
|
|
|
|
|
|
// 支持的压缩格式后缀
|
|
|
var (
|
|
|
suffixes = []string{".zip", ".gz", ".xz", ".tar", ".rar"}
|
|
|
matched bool
|
|
|
)
|
|
|
for _, suffix := range suffixes {
|
|
|
if strings.HasSuffix(file.Name, suffix) {
|
|
|
matched = true
|
|
|
break
|
|
|
}
|
|
|
}
|
|
|
if !matched {
|
|
|
return serializer.Err(serializer.CodeUnsupportedArchiveType, "", nil)
|
|
|
}
|
|
|
|
|
|
// 创建任务
|
|
|
job, err := task.NewDecompressTask(fs.User, service.Src, service.Dst, service.Encoding)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeCreateTaskError, "", err)
|
|
|
}
|
|
|
task.TaskPoll.Submit(job)
|
|
|
|
|
|
return serializer.Response{}
|
|
|
|
|
|
}
|
|
|
|
|
|
// CreateCompressTask 创建文件压缩任务
|
|
|
func (service *ItemCompressService) CreateCompressTask(c *gin.Context) serializer.Response {
|
|
|
// 创建文件系统
|
|
|
fs, err := filesystem.NewFileSystemFromContext(c)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeCreateFSError, "", err)
|
|
|
}
|
|
|
defer fs.Recycle()
|
|
|
|
|
|
// 检查用户组权限
|
|
|
if !fs.User.Group.OptionsSerialized.ArchiveTask {
|
|
|
return serializer.Err(serializer.CodeGroupNotAllowed, "", nil)
|
|
|
}
|
|
|
|
|
|
// 补齐压缩文件扩展名（如果没有）
|
|
|
if !strings.HasSuffix(service.Name, ".zip") {
|
|
|
service.Name += ".zip"
|
|
|
}
|
|
|
|
|
|
// 存放目录是否存在，是否重名
|
|
|
if exist, _ := fs.IsPathExist(service.Dst); !exist {
|
|
|
return serializer.Err(serializer.CodeParentNotExist, "", nil)
|
|
|
}
|
|
|
if exist, _ := fs.IsFileExist(path.Join(service.Dst, service.Name)); exist {
|
|
|
return serializer.ParamErr("File "+service.Name+" already exist", nil)
|
|
|
}
|
|
|
|
|
|
// 检查文件名合法性
|
|
|
if !fs.ValidateLegalName(context.Background(), service.Name) {
|
|
|
return serializer.Err(serializer.CodeIllegalObjectName, "", nil)
|
|
|
}
|
|
|
if !fs.ValidateExtension(context.Background(), service.Name) {
|
|
|
return serializer.Err(serializer.CodeFileTypeNotAllowed, "", nil)
|
|
|
}
|
|
|
|
|
|
// 递归列出待压缩子目录
|
|
|
folders, err := model.GetRecursiveChildFolder(service.Src.Raw().Dirs, fs.User.ID, true)
|
|
|
if err != nil {
|
|
|
return serializer.DBErr("Failed to list folders", err)
|
|
|
}
|
|
|
|
|
|
// 列出所有待压缩文件
|
|
|
files, err := model.GetChildFilesOfFolders(&folders)
|
|
|
if err != nil {
|
|
|
return serializer.DBErr("Failed to list files", err)
|
|
|
}
|
|
|
|
|
|
// 计算待压缩文件大小
|
|
|
var totalSize uint64
|
|
|
for i := 0; i < len(files); i++ {
|
|
|
totalSize += files[i].Size
|
|
|
}
|
|
|
|
|
|
// 文件尺寸限制
|
|
|
if fs.User.Group.OptionsSerialized.CompressSize != 0 && totalSize > fs.User.Group.
|
|
|
OptionsSerialized.CompressSize {
|
|
|
return serializer.Err(serializer.CodeFileTooLarge, "", nil)
|
|
|
}
|
|
|
|
|
|
// 按照平均压缩率计算用户空间是否足够
|
|
|
compressRatio := 0.4
|
|
|
spaceNeeded := uint64(math.Round(float64(totalSize) * compressRatio))
|
|
|
if fs.User.GetRemainingCapacity() < spaceNeeded {
|
|
|
return serializer.Err(serializer.CodeInsufficientCapacity, "", err)
|
|
|
}
|
|
|
|
|
|
// 创建任务
|
|
|
job, err := task.NewCompressTask(fs.User, path.Join(service.Dst, service.Name), service.Src.Raw().Dirs,
|
|
|
service.Src.Raw().Items)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeCreateTaskError, "", err)
|
|
|
}
|
|
|
task.TaskPoll.Submit(job)
|
|
|
|
|
|
return serializer.Response{}
|
|
|
|
|
|
}
|
|
|
|
|
|
// Archive 创建归档
|
|
|
func (service *ItemIDService) Archive(ctx context.Context, c *gin.Context) serializer.Response {
|
|
|
// 创建文件系统
|
|
|
fs, err := filesystem.NewFileSystemFromContext(c)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeCreateFSError, "", err)
|
|
|
}
|
|
|
defer fs.Recycle()
|
|
|
|
|
|
// 检查用户组权限
|
|
|
if !fs.User.Group.OptionsSerialized.ArchiveDownload {
|
|
|
return serializer.Err(serializer.CodeGroupNotAllowed, "", nil)
|
|
|
}
|
|
|
|
|
|
// 创建打包下载会话
|
|
|
ttl := model.GetIntSetting("archive_timeout", 30)
|
|
|
downloadSessionID := util.RandStringRunes(16)
|
|
|
cache.Set("archive_"+downloadSessionID, *service, ttl)
|
|
|
cache.Set("archive_user_"+downloadSessionID, *fs.User, ttl)
|
|
|
signURL, err := auth.SignURI(
|
|
|
auth.General,
|
|
|
fmt.Sprintf("/api/v3/file/archive/%s/archive.zip", downloadSessionID),
|
|
|
int64(ttl),
|
|
|
)
|
|
|
|
|
|
return serializer.Response{
|
|
|
Code: 0,
|
|
|
Data: signURL.String(),
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// Delete 删除对象
|
|
|
func (service *ItemIDService) Delete(ctx context.Context, c *gin.Context) serializer.Response {
|
|
|
// 创建文件系统
|
|
|
fs, err := filesystem.NewFileSystemFromContext(c)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodePolicyNotAllowed, err.Error(), err)
|
|
|
}
|
|
|
defer fs.Recycle()
|
|
|
|
|
|
force, unlink := false, false
|
|
|
if fs.User.Group.OptionsSerialized.AdvanceDelete {
|
|
|
force = service.Force
|
|
|
unlink = service.UnlinkOnly
|
|
|
}
|
|
|
|
|
|
// 删除对象
|
|
|
items := service.Raw()
|
|
|
err = fs.Delete(ctx, items.Dirs, items.Items, force, unlink)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeNotSet, err.Error(), err)
|
|
|
}
|
|
|
|
|
|
return serializer.Response{
|
|
|
Code: 0,
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
// Move 移动对象
|
|
|
func (service *ItemMoveService) Move(ctx context.Context, c *gin.Context) serializer.Response {
|
|
|
// 创建文件系统
|
|
|
fs, err := filesystem.NewFileSystemFromContext(c)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeCreateFSError, "", err)
|
|
|
}
|
|
|
defer fs.Recycle()
|
|
|
|
|
|
// 移动对象
|
|
|
items := service.Src.Raw()
|
|
|
err = fs.Move(ctx, items.Dirs, items.Items, service.SrcDir, service.Dst)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeNotSet, err.Error(), err)
|
|
|
}
|
|
|
|
|
|
return serializer.Response{
|
|
|
Code: 0,
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
// Copy 复制对象
|
|
|
func (service *ItemMoveService) Copy(ctx context.Context, c *gin.Context) serializer.Response {
|
|
|
// 复制操作只能对一个目录或文件对象进行操作
|
|
|
if len(service.Src.Items)+len(service.Src.Dirs) > 1 {
|
|
|
return filesystem.ErrOneObjectOnly
|
|
|
}
|
|
|
|
|
|
// 创建文件系统
|
|
|
fs, err := filesystem.NewFileSystemFromContext(c)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeCreateFSError, "", err)
|
|
|
}
|
|
|
defer fs.Recycle()
|
|
|
|
|
|
// 复制对象
|
|
|
err = fs.Copy(ctx, service.Src.Raw().Dirs, service.Src.Raw().Items, service.SrcDir, service.Dst)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeNotSet, err.Error(), err)
|
|
|
}
|
|
|
|
|
|
return serializer.Response{
|
|
|
Code: 0,
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
// Rename 重命名对象
|
|
|
func (service *ItemRenameService) Rename(ctx context.Context, c *gin.Context) serializer.Response {
|
|
|
// 重命名作只能对一个目录或文件对象进行操作
|
|
|
if len(service.Src.Items)+len(service.Src.Dirs) > 1 {
|
|
|
return filesystem.ErrOneObjectOnly
|
|
|
}
|
|
|
|
|
|
// 创建文件系统
|
|
|
fs, err := filesystem.NewFileSystemFromContext(c)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeCreateFSError, "", err)
|
|
|
}
|
|
|
defer fs.Recycle()
|
|
|
|
|
|
// 重命名对象
|
|
|
err = fs.Rename(ctx, service.Src.Raw().Dirs, service.Src.Raw().Items, service.NewName)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeNotSet, err.Error(), err)
|
|
|
}
|
|
|
|
|
|
return serializer.Response{
|
|
|
Code: 0,
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// GetProperty 获取对象的属性
|
|
|
func (service *ItemPropertyService) GetProperty(ctx context.Context, c *gin.Context) serializer.Response {
|
|
|
userCtx, _ := c.Get("user")
|
|
|
user := userCtx.(*model.User)
|
|
|
|
|
|
var props serializer.ObjectProps
|
|
|
props.QueryDate = time.Now()
|
|
|
|
|
|
// 如果是文件对象
|
|
|
if !service.IsFolder {
|
|
|
res, err := hashid.DecodeHashID(service.ID, hashid.FileID)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeNotFound, "", err)
|
|
|
}
|
|
|
|
|
|
file, err := model.GetFilesByIDs([]uint{res}, user.ID)
|
|
|
if err != nil {
|
|
|
return serializer.DBErr("Failed to query file records", err)
|
|
|
}
|
|
|
|
|
|
props.CreatedAt = file[0].CreatedAt
|
|
|
props.UpdatedAt = file[0].UpdatedAt
|
|
|
props.Policy = file[0].GetPolicy().Name
|
|
|
props.Size = file[0].Size
|
|
|
|
|
|
// 查找父目录
|
|
|
if service.TraceRoot {
|
|
|
parent, err := model.GetFoldersByIDs([]uint{file[0].FolderID}, user.ID)
|
|
|
if err != nil {
|
|
|
return serializer.DBErr("Parent folder record not exist", err)
|
|
|
}
|
|
|
|
|
|
if err := parent[0].TraceRoot(); err != nil {
|
|
|
return serializer.DBErr("Failed to trace root folder", err)
|
|
|
}
|
|
|
|
|
|
props.Path = path.Join(parent[0].Position, parent[0].Name)
|
|
|
}
|
|
|
} else {
|
|
|
res, err := hashid.DecodeHashID(service.ID, hashid.FolderID)
|
|
|
if err != nil {
|
|
|
return serializer.Err(serializer.CodeNotFound, "", err)
|
|
|
}
|
|
|
|
|
|
folder, err := model.GetFoldersByIDs([]uint{res}, user.ID)
|
|
|
if err != nil {
|
|
|
return serializer.DBErr("Failed to query folder records", err)
|
|
|
}
|
|
|
|
|
|
props.CreatedAt = folder[0].CreatedAt
|
|
|
props.UpdatedAt = folder[0].UpdatedAt
|
|
|
|
|
|
// 如果对象是目录, 先尝试返回缓存结果
|
|
|
if cacheRes, ok := cache.Get(fmt.Sprintf("folder_props_%d", res)); ok {
|
|
|
res := cacheRes.(serializer.ObjectProps)
|
|
|
res.CreatedAt = props.CreatedAt
|
|
|
res.UpdatedAt = props.UpdatedAt
|
|
|
return serializer.Response{Data: res}
|
|
|
}
|
|
|
|
|
|
// 统计子目录
|
|
|
childFolders, err := model.GetRecursiveChildFolder([]uint{folder[0].ID},
|
|
|
user.ID, true)
|
|
|
if err != nil {
|
|
|
return serializer.DBErr("Failed to list child folders", err)
|
|
|
}
|
|
|
props.ChildFolderNum = len(childFolders) - 1
|
|
|
|
|
|
// 统计子文件
|
|
|
files, err := model.GetChildFilesOfFolders(&childFolders)
|
|
|
if err != nil {
|
|
|
return serializer.DBErr("Failed to list child files", err)
|
|
|
}
|
|
|
|
|
|
// 统计子文件个数和大小
|
|
|
props.ChildFileNum = len(files)
|
|
|
for i := 0; i < len(files); i++ {
|
|
|
props.Size += files[i].Size
|
|
|
}
|
|
|
|
|
|
// 查找父目录
|
|
|
if service.TraceRoot {
|
|
|
if err := folder[0].TraceRoot(); err != nil {
|
|
|
return serializer.DBErr("Failed to list child folders", err)
|
|
|
}
|
|
|
|
|
|
props.Path = folder[0].Position
|
|
|
}
|
|
|
|
|
|
// 如果列取对象是目录，则缓存结果
|
|
|
cache.Set(fmt.Sprintf("folder_props_%d", res), props,
|
|
|
model.GetIntSetting("folder_props_timeout", 300))
|
|
|
}
|
|
|
|
|
|
return serializer.Response{
|
|
|
Code: 0,
|
|
|
Data: props,
|
|
|
}
|
|
|
}
|
