chore: add is it support upload config for driver

AlistGo · Jun 20, 2022 · a61bb6a · a61bb6a
1 parent 31ff31d
commit a61bb6a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/internal/driver/config.go b/internal/driver/config.go
@@ -7,4 +7,5 @@ type Config struct {
 	OnlyProxy     bool
 	NoNeedSetLink bool
 	NoCache       bool
+	NoUpload      bool
 }
diff --git a/internal/fs/put.go b/internal/fs/put.go
@@ -15,6 +15,9 @@ var UploadTaskManager = task.NewTaskManager()
 // Put add as a put task
 func Put(ctx context.Context, account driver.Driver, parentPath string, file model.FileStreamer) error {
 	account, actualParentPath, err := operations.GetAccountAndActualPath(parentPath)
+	if account.Config().NoUpload {
+		return errors.New("upload is not supported")
+	}
 	if err != nil {
 		return errors.WithMessage(err, "failed get account")
 	}