fix: put as task from web

internal/aria2/monitor.go

@@ -147,7 +147,7 @@ func (m *Monitor) Complete() error {
 				if err != nil {
 					return errors.Wrapf(err, "failed to open file %s", file.Path)
 				}
-				stream := model.FileStream{
+				stream := &model.FileStream{
 					Obj: model.Object{
 						Name:     path.Base(file.Path),
 						Size:     size,

internal/fs/put.go

@@ -7,6 +7,7 @@ import (
 	"github.com/alist-org/alist/v3/internal/model"
 	"github.com/alist-org/alist/v3/internal/operations"
 	"github.com/alist-org/alist/v3/pkg/task"
+	"github.com/alist-org/alist/v3/pkg/utils"
 	"github.com/pkg/errors"
 	"sync/atomic"
 )
@@ -24,6 +25,13 @@ func putAsTask(dstDirPath string, file model.FileStreamer) error {
 	if err != nil {
 		return errors.WithMessage(err, "failed get account")
 	}
+	if file.NeedStore() {
+		tempFile, err := utils.CreateTempFile(file)
+		if err != nil {
+			return errors.Wrapf(err, "failed to create temp file")
+		}
+		file.SetReadCloser(tempFile)
+	}
 	UploadTaskManager.Submit(task.WithCancelCtx(&task.Task[uint64]{
 		Name: fmt.Sprintf("upload %s to [%s](%s)", file.GetName(), account.GetAccount().VirtualPath, dstDirActualPath),
 		Func: func(task *task.Task[uint64]) error {

internal/fs/util.go

@@ -65,7 +65,7 @@ func getFileStreamFromLink(file model.Obj, link *model.Link) (model.FileStreamer
 	if mimetype == "" {
 		mimetype = "application/octet-stream"
 	}
-	stream := model.FileStream{
+	stream := &model.FileStream{
 		Obj:        file,
 		ReadCloser: rc,
 		Mimetype:   mimetype,

internal/model/obj.go

@@ -19,6 +19,9 @@ type FileStreamer interface {
 	io.ReadCloser
 	Obj
 	GetMimetype() string
+	SetReadCloser(io.ReadCloser)
+	NeedStore() bool
+	GetReadCloser() io.ReadCloser
 }
 
 type URL interface {

internal/model/stream.go

@@ -7,9 +7,22 @@ import (
 type FileStream struct {
 	Obj
 	io.ReadCloser
-	Mimetype string
+	Mimetype     string
+	WebPutAsTask bool
 }
 
 func (f FileStream) GetMimetype() string {
 	return f.Mimetype
 }
+
+func (f FileStream) NeedStore() bool {
+	return f.WebPutAsTask
+}
+
+func (f *FileStream) GetReadCloser() io.ReadCloser {
+	return f.ReadCloser
+}
+
+func (f *FileStream) SetReadCloser(rc io.ReadCloser) {
+	f.ReadCloser = rc
+}

internal/operations/fs.go

@@ -5,6 +5,7 @@ import (
 	"github.com/alist-org/alist/v3/internal/conf"
 	"github.com/alist-org/alist/v3/internal/errs"
 	log "github.com/sirupsen/logrus"
+	"os"
 	stdpath "path"
 	"strings"
 	"time"
@@ -222,6 +223,14 @@ func Remove(ctx context.Context, account driver.Driver, path string) error {
 }
 
 func Put(ctx context.Context, account driver.Driver, dstDirPath string, file model.FileStreamer, up driver.UpdateProgress) error {
+	defer func() {
+		if f, ok := file.GetReadCloser().(*os.File); ok {
+			err := os.RemoveAll(f.Name())
+			if err != nil {
+				log.Errorf("failed to remove file [%s]", f.Name())
+			}
+		}
+	}()
 	defer func() {
 		if err := file.Close(); err != nil {
 			log.Errorf("failed to close file streamer, %v", err)
@@ -241,6 +250,7 @@ func Put(ctx context.Context, account driver.Driver, dstDirPath string, file mod
 		up = func(p int) {}
 	}
 	err = account.Put(ctx, parentDir, file, up)
+	log.Debugf("put file [%s] done", file.GetName())
 	if err == nil {
 		// clear cache
 		key := stdpath.Join(account.GetAccount().VirtualPath, dstDirPath)