173 lines
4 KiB
Go
173 lines
4 KiB
Go
package filesystem
|
||
|
||
import (
|
||
"context"
|
||
"github.com/HFO4/cloudreve/models"
|
||
"github.com/HFO4/cloudreve/pkg/filesystem/local"
|
||
"github.com/HFO4/cloudreve/pkg/filesystem/response"
|
||
"github.com/gin-gonic/gin"
|
||
"io"
|
||
"net/url"
|
||
)
|
||
|
||
// FileHeader 上传来的文件数据处理器
|
||
type FileHeader interface {
|
||
io.Reader
|
||
io.Closer
|
||
GetSize() uint64
|
||
GetMIMEType() string
|
||
GetFileName() string
|
||
GetVirtualPath() string
|
||
}
|
||
|
||
// Handler 存储策略适配器
|
||
type Handler interface {
|
||
// 上传文件
|
||
Put(ctx context.Context, file io.ReadCloser, dst string, size uint64) error
|
||
|
||
// 删除一个或多个文件
|
||
Delete(ctx context.Context, files []string) ([]string, error)
|
||
|
||
// 获取文件
|
||
Get(ctx context.Context, path string) (response.RSCloser, error)
|
||
|
||
// 获取缩略图
|
||
Thumb(ctx context.Context, path string) (*response.ContentResponse, error)
|
||
|
||
// 获取外链/下载地址，
|
||
// url - 站点本身地址,
|
||
// isDownload - 是否直接下载
|
||
Source(ctx context.Context, path string, url url.URL, ttl int64, isDownload bool) (string, error)
|
||
}
|
||
|
||
// FileSystem 管理文件的文件系统
|
||
type FileSystem struct {
|
||
// 文件系统所有者
|
||
User *model.User
|
||
// 操作文件使用的存储策略
|
||
Policy *model.Policy
|
||
// 当前正在处理的文件对象
|
||
FileTarget []model.File
|
||
// 当前正在处理的目录对象
|
||
DirTarget []model.Folder
|
||
|
||
/*
|
||
钩子函数
|
||
*/
|
||
// 上传文件前
|
||
BeforeUpload []Hook
|
||
// 上传文件后
|
||
AfterUpload []Hook
|
||
// 文件保存成功，插入数据库验证失败后
|
||
AfterValidateFailed []Hook
|
||
// 用户取消上传后
|
||
AfterUploadCanceled []Hook
|
||
// 文件下载前
|
||
BeforeFileDownload []Hook
|
||
|
||
/*
|
||
文件系统处理适配器
|
||
*/
|
||
Handler Handler
|
||
}
|
||
|
||
// NewFileSystem 初始化一个文件系统
|
||
func NewFileSystem(user *model.User) (*FileSystem, error) {
|
||
fs := &FileSystem{
|
||
User: user,
|
||
}
|
||
|
||
// 分配存储策略适配器
|
||
err := fs.dispatchHandler()
|
||
|
||
// TODO 分配默认钩子
|
||
return fs, err
|
||
}
|
||
|
||
// NewAnonymousFileSystem 初始化匿名文件系统
|
||
func NewAnonymousFileSystem() (*FileSystem, error) {
|
||
fs := &FileSystem{
|
||
User: &model.User{},
|
||
}
|
||
|
||
anonymousGroup, err := model.GetGroupByID(3)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
fs.User.Group = anonymousGroup
|
||
return fs, nil
|
||
}
|
||
|
||
// dispatchHandler 根据存储策略分配文件适配器
|
||
func (fs *FileSystem) dispatchHandler() error {
|
||
var policyType string
|
||
var currentPolicy *model.Policy
|
||
|
||
if fs.Policy == nil {
|
||
// 如果没有具体指定，就是用用户当前存储策略
|
||
policyType = fs.User.Policy.Type
|
||
currentPolicy = &fs.User.Policy
|
||
} else {
|
||
policyType = fs.Policy.Type
|
||
currentPolicy = fs.Policy
|
||
}
|
||
|
||
// 根据存储策略类型分配适配器
|
||
switch policyType {
|
||
case "local":
|
||
fs.Handler = local.Handler{
|
||
Policy: currentPolicy,
|
||
}
|
||
return nil
|
||
default:
|
||
return ErrUnknownPolicyType
|
||
}
|
||
}
|
||
|
||
// NewFileSystemFromContext 从gin.Context创建文件系统
|
||
// TODO 用户不存在时使用匿名文件系统
|
||
func NewFileSystemFromContext(c *gin.Context) (*FileSystem, error) {
|
||
user, exist := c.Get("user")
|
||
if !exist {
|
||
return NewAnonymousFileSystem()
|
||
}
|
||
fs, err := NewFileSystem(user.(*model.User))
|
||
return fs, err
|
||
}
|
||
|
||
// SetTargetFile 设置当前处理的目标文件
|
||
func (fs *FileSystem) SetTargetFile(files *[]model.File) {
|
||
if len(fs.FileTarget) == 0 {
|
||
fs.FileTarget = *files
|
||
} else {
|
||
fs.FileTarget = append(fs.FileTarget, *files...)
|
||
}
|
||
|
||
}
|
||
|
||
// SetTargetDir 设置当前处理的目标目录
|
||
func (fs *FileSystem) SetTargetDir(dirs *[]model.Folder) {
|
||
if len(fs.DirTarget) == 0 {
|
||
fs.DirTarget = *dirs
|
||
} else {
|
||
fs.DirTarget = append(fs.DirTarget, *dirs...)
|
||
}
|
||
|
||
}
|
||
|
||
// SetTargetFileByIDs 根据文件ID设置目标文件，忽略用户ID
|
||
func (fs *FileSystem) SetTargetFileByIDs(ids []uint) error {
|
||
files, err := model.GetFilesByIDs(ids, 0)
|
||
if err != nil || len(files) == 0 {
|
||
return ErrFileExisted.WithError(err)
|
||
}
|
||
fs.SetTargetFile(&files)
|
||
return nil
|
||
}
|
||
|
||
// CleanTargets 清空目标
|
||
func (fs *FileSystem) CleanTargets() {
|
||
fs.FileTarget = []model.File{}
|
||
fs.DirTarget = []model.Folder{}
|
||
}
|