Feat: support apply append mode and overwrite mode for FileStream

pkg/filesystem/archive.go

@@ -308,7 +308,7 @@ func (fs *FileSystem) Decompress(ctx context.Context, src, dst string) error {
 					Size:        uint64(size),
 					Name:        path.Base(dst),
 					VirtualPath: path.Dir(dst),
-					Mode:        fsctx.Create,
+					Mode:        0,
 				})
 				fileStream.Close()
 				if err != nil {

pkg/filesystem/driver/local/handler.go

@@ -93,7 +93,7 @@ func (handler Driver) Put(ctx context.Context, file fsctx.FileHeader) error {
 	dst := util.RelativePath(filepath.FromSlash(fileInfo.SavePath))
 
 	// 如果非 Overwrite，则检查是否有重名冲突
-	if fileInfo.Mode == fsctx.Create {
+	if fileInfo.Mode&fsctx.Overwrite != fsctx.Overwrite {
 		if util.Exists(dst) {
 			util.Log().Warning("物理同名文件已存在或不可用: %s", dst)
 			return errors.New("物理同名文件已存在或不可用")
@@ -115,21 +115,21 @@ func (handler Driver) Put(ctx context.Context, file fsctx.FileHeader) error {
 		err error
 	)
 
-	if fileInfo.Mode == fsctx.Append {
-		// 如果是追加模式，则直接打开文件
-		out, err = os.OpenFile(dst, os.O_APPEND|os.O_CREATE|os.O_WRONLY, Perm)
+	openMode := os.O_CREATE | os.O_RDWR
+	if fileInfo.Mode&fsctx.Append == fsctx.Append {
+		openMode |= os.O_APPEND
 	} else {
-		// 创建或覆盖目标文件
-		out, err = os.Create(dst)
+		openMode |= os.O_TRUNC
 	}
 
+	out, err = os.OpenFile(dst, openMode, Perm)
 	if err != nil {
 		util.Log().Warning("无法打开或创建文件，%s", err)
 		return err
 	}
 	defer out.Close()
 
-	if fileInfo.Mode == fsctx.Append {
+	if fileInfo.Mode&fsctx.Append == fsctx.Append {
 		stat, err := out.Stat()
 		if err != nil {
 			util.Log().Warning("无法读取文件信息，%s", err)
@@ -144,7 +144,7 @@ func (handler Driver) Put(ctx context.Context, file fsctx.FileHeader) error {
 				return fmt.Errorf("覆盖分片时发生错误: %w", err)
 			}
 
-			out, err = os.OpenFile(dst, os.O_APPEND|os.O_CREATE|os.O_WRONLY, Perm)
+			out, err = os.OpenFile(dst, openMode, Perm)
 			defer out.Close()
 			if err != nil {
 				util.Log().Warning("无法打开或创建文件，%s", err)

pkg/filesystem/driver/onedrive/api.go

@@ -260,9 +260,9 @@ func (client *Client) UploadChunk(ctx context.Context, uploadURL string, chunk *
 func (client *Client) Upload(ctx context.Context, file fsctx.FileHeader) error {
 	fileInfo := file.Info()
 	// 决定是否覆盖文件
-	overwrite := "replace"
-	if fileInfo.Mode != fsctx.Create {
-		overwrite = "fail"
+	overwrite := "fail"
+	if fileInfo.Mode&fsctx.Overwrite == fsctx.Overwrite {
+		overwrite = "replace"
 	}
 
 	size := int(fileInfo.Size)

pkg/filesystem/driver/oss/handler.go

@@ -237,11 +237,7 @@ func (handler Driver) Put(ctx context.Context, file fsctx.FileHeader) error {
 	credentialTTL := model.GetIntSetting("upload_credential_timeout", 3600)
 
 	// 是否允许覆盖
-	overwrite := true
-	if fileInfo.Mode == fsctx.Create {
-		overwrite = false
-	}
-
+	overwrite := fileInfo.Mode&fsctx.Overwrite == fsctx.Overwrite
 	options := []oss.Option{
 		oss.Expires(time.Now().Add(time.Duration(credentialTTL) * time.Second)),
 		oss.ForbidOverWrite(!overwrite),

pkg/filesystem/driver/remote/handler.go

@@ -173,9 +173,9 @@ func (handler Driver) Put(ctx context.Context, file fsctx.FileHeader) error {
 	fileName := url.QueryEscape(path.Base(fileInfo.SavePath))
 
 	// 决定是否要禁用文件覆盖
-	overwrite := "true"
-	if fileInfo.Mode != fsctx.Create {
-		overwrite = "false"
+	overwrite := "false"
+	if fileInfo.Mode&fsctx.Overwrite == fsctx.Overwrite {
+		overwrite = "true"
 	}
 
 	// 上传文件

pkg/filesystem/fsctx/stream.go

@@ -8,11 +8,10 @@ import (
 type WriteMode int
 
 const (
-	Overwrite WriteMode = iota
+	Overwrite WriteMode = 0x00001
 	// Append 只适用于本地策略
-	Append
-	Create
-	Nop
+	Append = 0x00002
+	Nop    = 0x00004
 )
 
 type UploadTaskInfo struct {

pkg/filesystem/hooks.go

@@ -151,6 +151,7 @@ func HookCleanFileContent(ctx context.Context, fs *FileSystem, file fsctx.FileHe
 		File:     ioutil.NopCloser(strings.NewReader("")),
 		SavePath: file.Info().SavePath,
 		Size:     0,
+		Model:    fsctx.Overwrite,
 	})
 }
 

pkg/filesystem/upload.go

@@ -51,7 +51,7 @@ func (fs *FileSystem) Upload(ctx context.Context, file *fsctx.FileStream) (err e
 	go fs.CancelUpload(ctx, savePath, file)
 
 	// 保存文件
-	if file.Mode != fsctx.Nop {
+	if file.Mode&fsctx.Nop != fsctx.Nop {
 		err = fs.Handler.Put(ctx, file)
 		if err != nil {
 			fs.Trigger(ctx, "AfterUploadFailed", file)
@@ -73,7 +73,7 @@ func (fs *FileSystem) Upload(ctx context.Context, file *fsctx.FileStream) (err e
 		return err
 	}
 
-	if file.Mode == fsctx.Create {
+	if file.Mode&fsctx.Overwrite == 0 {
 		fileInfo := file.Info()
 		util.Log().Info(
 			"新文件PUT:%s , 大小:%d, 上传者:%s",