init

pkg/filesystem/fsctx/context.go

@@ -0,0 +1,44 @@
+package fsctx
+
+type key int
+
+const (
+	// GinCtx Gin的上下文
+	GinCtx key = iota
+	// PathCtx 文件或目录的虚拟路径
+	PathCtx
+	// FileModelCtx 文件数据库模型
+	FileModelCtx
+	// FolderModelCtx 目录数据库模型
+	FolderModelCtx
+	// HTTPCtx HTTP请求的上下文
+	HTTPCtx
+	// UploadPolicyCtx 上传策略，一般为slave模式下使用
+	UploadPolicyCtx
+	// UserCtx 用户
+	UserCtx
+	// ThumbSizeCtx 缩略图尺寸
+	ThumbSizeCtx
+	// FileSizeCtx 文件大小
+	FileSizeCtx
+	// ShareKeyCtx 分享文件的 HashID
+	ShareKeyCtx
+	// LimitParentCtx 限制父目录
+	LimitParentCtx
+	// IgnoreDirectoryConflictCtx 忽略目录重名冲突
+	IgnoreDirectoryConflictCtx
+	// RetryCtx 失败重试次数
+	RetryCtx
+	// ForceUsePublicEndpointCtx 强制使用公网 Endpoint
+	ForceUsePublicEndpointCtx
+	// CancelFuncCtx Context 取消函數
+	CancelFuncCtx
+	// 文件在从机节点中的路径
+	SlaveSrcPath
+	// Webdav目标名称
+	WebdavDstName
+	// WebDAVCtx WebDAV
+	WebDAVCtx
+	// WebDAV反代Url
+	WebDAVProxyUrlCtx
+)

pkg/filesystem/fsctx/stream.go

@@ -0,0 +1,123 @@
+package fsctx
+
+import (
+	"errors"
+	"github.com/HFO4/aliyun-oss-go-sdk/oss"
+	"io"
+	"time"
+)
+
+type WriteMode int
+
+const (
+	Overwrite WriteMode = 0x00001
+	// Append 只适用于本地策略
+	Append WriteMode = 0x00002
+	Nop    WriteMode = 0x00004
+)
+
+type UploadTaskInfo struct {
+	Size            uint64
+	MimeType        string
+	FileName        string
+	VirtualPath     string
+	Mode            WriteMode
+	Metadata        map[string]string
+	LastModified    *time.Time
+	SavePath        string
+	UploadSessionID *string
+	AppendStart     uint64
+	Model           interface{}
+	Src             string
+}
+
+// Get mimetype of uploaded file, if it's not defined, detect it from file name
+func (u *UploadTaskInfo) DetectMimeType() string {
+	if u.MimeType != "" {
+		return u.MimeType
+	}
+
+	return oss.TypeByExtension(u.FileName)
+}
+
+// FileHeader 上传来的文件数据处理器
+type FileHeader interface {
+	io.Reader
+	io.Closer
+	io.Seeker
+	Info() *UploadTaskInfo
+	SetSize(uint64)
+	SetModel(fileModel interface{})
+	Seekable() bool
+}
+
+// FileStream 用户传来的文件
+type FileStream struct {
+	Mode            WriteMode
+	LastModified    *time.Time
+	Metadata        map[string]string
+	File            io.ReadCloser
+	Seeker          io.Seeker
+	Size            uint64
+	VirtualPath     string
+	Name            string
+	MimeType        string
+	SavePath        string
+	UploadSessionID *string
+	AppendStart     uint64
+	Model           interface{}
+	Src             string
+}
+
+func (file *FileStream) Read(p []byte) (n int, err error) {
+	if file.File != nil {
+		return file.File.Read(p)
+	}
+
+	return 0, io.EOF
+}
+
+func (file *FileStream) Close() error {
+	if file.File != nil {
+		return file.File.Close()
+	}
+
+	return nil
+}
+
+func (file *FileStream) Seek(offset int64, whence int) (int64, error) {
+	if file.Seekable() {
+		return file.Seeker.Seek(offset, whence)
+	}
+
+	return 0, errors.New("no seeker")
+}
+
+func (file *FileStream) Seekable() bool {
+	return file.Seeker != nil
+}
+
+func (file *FileStream) Info() *UploadTaskInfo {
+	return &UploadTaskInfo{
+		Size:            file.Size,
+		MimeType:        file.MimeType,
+		FileName:        file.Name,
+		VirtualPath:     file.VirtualPath,
+		Mode:            file.Mode,
+		Metadata:        file.Metadata,
+		LastModified:    file.LastModified,
+		SavePath:        file.SavePath,
+		UploadSessionID: file.UploadSessionID,
+		AppendStart:     file.AppendStart,
+		Model:           file.Model,
+		Src:             file.Src,
+	}
+}
+
+func (file *FileStream) SetSize(size uint64) {
+	file.Size = size
+}
+
+func (file *FileStream) SetModel(fileModel interface{}) {
+	file.Model = fileModel
+}