mirror of https://github.com/Xhofe/alist
98 lines
2.5 KiB
Go
98 lines
2.5 KiB
Go
package tool
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/alist-org/alist/v3/internal/model"
|
|
"github.com/alist-org/alist/v3/internal/op"
|
|
"github.com/alist-org/alist/v3/internal/stream"
|
|
"github.com/alist-org/alist/v3/internal/task"
|
|
"github.com/alist-org/alist/v3/pkg/utils"
|
|
"github.com/pkg/errors"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/xhofe/tache"
|
|
)
|
|
|
|
type TransferTask struct {
|
|
task.TaskExtension
|
|
FileDir string `json:"file_dir"`
|
|
DstDirPath string `json:"dst_dir_path"`
|
|
TempDir string `json:"temp_dir"`
|
|
DeletePolicy DeletePolicy `json:"delete_policy"`
|
|
file File
|
|
}
|
|
|
|
func (t *TransferTask) Run() error {
|
|
t.ClearEndTime()
|
|
t.SetStartTime(time.Now())
|
|
defer func() { t.SetEndTime(time.Now()) }()
|
|
// check dstDir again
|
|
var err error
|
|
if (t.file == File{}) {
|
|
t.file, err = GetFile(t.FileDir)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to get file %s", t.FileDir)
|
|
}
|
|
}
|
|
storage, dstDirActualPath, err := op.GetStorageAndActualPath(t.DstDirPath)
|
|
if err != nil {
|
|
return errors.WithMessage(err, "failed get storage")
|
|
}
|
|
mimetype := utils.GetMimeType(t.file.Path)
|
|
rc, err := t.file.GetReadCloser()
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to open file %s", t.file.Path)
|
|
}
|
|
s := &stream.FileStream{
|
|
Ctx: nil,
|
|
Obj: &model.Object{
|
|
Name: filepath.Base(t.file.Path),
|
|
Size: t.file.Size,
|
|
Modified: t.file.Modified,
|
|
IsFolder: false,
|
|
},
|
|
Reader: rc,
|
|
Mimetype: mimetype,
|
|
Closers: utils.NewClosers(rc),
|
|
}
|
|
relDir, err := filepath.Rel(t.TempDir, filepath.Dir(t.file.Path))
|
|
if err != nil {
|
|
log.Errorf("find relation directory error: %v", err)
|
|
}
|
|
newDistDir := filepath.Join(dstDirActualPath, relDir)
|
|
return op.Put(t.Ctx(), storage, newDistDir, s, t.SetProgress)
|
|
}
|
|
|
|
func (t *TransferTask) GetName() string {
|
|
return fmt.Sprintf("transfer %s to [%s]", t.file.Path, t.DstDirPath)
|
|
}
|
|
|
|
func (t *TransferTask) GetStatus() string {
|
|
return "transferring"
|
|
}
|
|
|
|
func (t *TransferTask) OnSucceeded() {
|
|
if t.DeletePolicy == DeleteOnUploadSucceed || t.DeletePolicy == DeleteAlways {
|
|
err := os.Remove(t.file.Path)
|
|
if err != nil {
|
|
log.Errorf("failed to delete file %s, error: %s", t.file.Path, err.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
func (t *TransferTask) OnFailed() {
|
|
if t.DeletePolicy == DeleteOnUploadFailed || t.DeletePolicy == DeleteAlways {
|
|
err := os.Remove(t.file.Path)
|
|
if err != nil {
|
|
log.Errorf("failed to delete file %s, error: %s", t.file.Path, err.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
var (
|
|
TransferTaskManager *tache.Manager[*TransferTask]
|
|
)
|