feat(task): add speed monitor (#7655)

internal/offline_download/115/client.go

@@ -107,6 +107,7 @@ func (p *Cloud115) Status(task *tool.DownloadTask) (*tool.Status, error) {
 			s.Progress = t.Percent
 			s.Status = t.GetStatus()
 			s.Completed = t.IsDone()
+			s.TotalBytes = t.Size
 			if t.IsFailed() {
 				s.Err = fmt.Errorf(t.GetStatus())
 			}

internal/offline_download/aria2/aria2.go

@@ -82,7 +82,7 @@ func (a *Aria2) Status(task *tool.DownloadTask) (*tool.Status, error) {
 	if err != nil {
 		return nil, err
 	}
-	total, err := strconv.ParseUint(info.TotalLength, 10, 64)
+	total, err := strconv.ParseInt(info.TotalLength, 10, 64)
 	if err != nil {
 		total = 0
 	}
@@ -91,8 +91,9 @@ func (a *Aria2) Status(task *tool.DownloadTask) (*tool.Status, error) {
 		downloaded = 0
 	}
 	s := &tool.Status{
-		Completed: info.Status == "complete",
-		Err:       err,
+		Completed:  info.Status == "complete",
+		Err:        err,
+		TotalBytes: total,
 	}
 	s.Progress = float64(downloaded) / float64(total) * 100
 	if len(info.FollowedBy) != 0 {

internal/offline_download/http/client.go

@@ -83,6 +83,7 @@ func (s SimpleHttp) Run(task *tool.DownloadTask) error {
 	}
 	defer file.Close()
 	fileSize := resp.ContentLength
+	task.SetTotalBytes(fileSize)
 	err = utils.CopyWithCtx(task.Ctx(), file, resp.Body, fileSize, task.SetProgress)
 	return err
 }

internal/offline_download/pikpak/pikpak.go

@@ -3,6 +3,7 @@ package pikpak
 import (
 	"context"
 	"fmt"
+	"strconv"
 
 	"github.com/alist-org/alist/v3/drivers/pikpak"
 	"github.com/alist-org/alist/v3/internal/errs"
@@ -105,6 +106,10 @@ func (p *PikPak) Status(task *tool.DownloadTask) (*tool.Status, error) {
 			s.Progress = float64(t.Progress)
 			s.Status = t.Message
 			s.Completed = (t.Phase == "PHASE_TYPE_COMPLETE")
+			s.TotalBytes, err = strconv.ParseInt(t.FileSize, 10, 64)
+			if err != nil {
+				s.TotalBytes = 0
+			}
 			if t.Phase == "PHASE_TYPE_ERROR" {
 				s.Err = fmt.Errorf(t.Message)
 			}

internal/offline_download/qbit/qbit.go

@@ -64,6 +64,7 @@ func (a *QBittorrent) Status(task *tool.DownloadTask) (*tool.Status, error) {
 		return nil, err
 	}
 	s := &tool.Status{}
+	s.TotalBytes = info.Size
 	s.Progress = float64(info.Completed) / float64(info.Size) * 100
 	switch info.State {
 	case qbittorrent.UPLOADING, qbittorrent.PAUSEDUP, qbittorrent.QUEUEDUP, qbittorrent.STALLEDUP, qbittorrent.FORCEDUP, qbittorrent.CHECKINGUP:

internal/offline_download/tool/add.go

@@ -29,7 +29,7 @@ type AddURLArgs struct {
 	DeletePolicy DeletePolicy
 }
 
-func AddURL(ctx context.Context, args *AddURLArgs) (task.TaskInfoWithCreator, error) {
+func AddURL(ctx context.Context, args *AddURLArgs) (task.TaskExtensionInfo, error) {
 	// get tool
 	tool, err := Tools.Get(args.Tool)
 	if err != nil {
@@ -81,7 +81,7 @@ func AddURL(ctx context.Context, args *AddURLArgs) (task.TaskInfoWithCreator, er
 
 	taskCreator, _ := ctx.Value("user").(*model.User) // taskCreator is nil when convert failed
 	t := &DownloadTask{
-		TaskWithCreator: task.TaskWithCreator{
+		TaskExtension: task.TaskExtension{
 			Creator: taskCreator,
 		},
 		Url:          args.URL,

internal/offline_download/tool/base.go

@@ -16,11 +16,12 @@ type AddUrlArgs struct {
 }
 
 type Status struct {
-	Progress  float64
-	NewGID    string
-	Completed bool
-	Status    string
-	Err       error
+	TotalBytes int64
+	Progress   float64
+	NewGID     string
+	Completed  bool
+	Status     string
+	Err        error
 }
 
 type Tool interface {

internal/offline_download/tool/download.go

@@ -14,7 +14,7 @@ import (
 )
 
 type DownloadTask struct {
-	task.TaskWithCreator
+	task.TaskExtension
 	Url               string       `json:"url"`
 	DstDirPath        string       `json:"dst_dir_path"`
 	TempDir           string       `json:"temp_dir"`
@@ -28,6 +28,9 @@ type DownloadTask struct {
 }
 
 func (t *DownloadTask) Run() error {
+	t.ClearEndTime()
+	t.SetStartTime(time.Now())
+	defer func() { t.SetEndTime(time.Now()) }()
 	if t.tool == nil {
 		tool, err := Tools.Get(t.Toolname)
 		if err != nil {
@@ -131,6 +134,7 @@ func (t *DownloadTask) Update() (bool, error) {
 	}
 	t.callStatusRetried = 0
 	t.SetProgress(info.Progress)
+	t.SetTotalBytes(info.TotalBytes)
 	t.Status = fmt.Sprintf("[%s]: %s", t.tool.Name(), info.Status)
 	if info.NewGID != "" {
 		log.Debugf("followen by: %+v", info.NewGID)
@@ -171,16 +175,18 @@ func (t *DownloadTask) Complete() error {
 	// upload files
 	for i := range files {
 		file := files[i]
-		TransferTaskManager.Add(&TransferTask{
-			TaskWithCreator: task.TaskWithCreator{
-				Creator: t.Creator,
+		tsk := &TransferTask{
+			TaskExtension: task.TaskExtension{
+				Creator: t.GetCreator(),
 			},
 			file:         file,
 			DstDirPath:   t.DstDirPath,
 			TempDir:      t.TempDir,
 			DeletePolicy: t.DeletePolicy,
 			FileDir:      file.Path,
-		})
+		}
+		tsk.SetTotalBytes(file.Size)
+		TransferTaskManager.Add(tsk)
 	}
 	return nil
 }

internal/offline_download/tool/transfer.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
+	"time"
 
 	"github.com/alist-org/alist/v3/internal/model"
 	"github.com/alist-org/alist/v3/internal/op"
@@ -16,7 +17,7 @@ import (
 )
 
 type TransferTask struct {
-	task.TaskWithCreator
+	task.TaskExtension
 	FileDir      string       `json:"file_dir"`
 	DstDirPath   string       `json:"dst_dir_path"`
 	TempDir      string       `json:"temp_dir"`
@@ -25,6 +26,9 @@ type TransferTask struct {
 }
 
 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{}) {

internal/offline_download/transmission/client.go

@@ -150,6 +150,7 @@ func (t *Transmission) Status(task *tool.DownloadTask) (*tool.Status, error) {
 		Err:       err,
 	}
 	s.Progress = *info.PercentDone * 100
+	s.TotalBytes = int64(*info.SizeWhenDone / 8)
 
 	switch *info.Status {
 	case transmissionrpc.TorrentStatusCheckWait,