feat: downloader task stats (#339)

pkg/download/downloader.go

@@ -35,7 +35,8 @@ const (
 )
 
 var (
-	ErrTaskNotFound = errors.New("task not found")
+	ErrTaskNotFound        = errors.New("task not found")
+	ErrUnSupportedProtocol = errors.New("unsupported protocol")
 )
 
 type Listener func(event *Event)
@@ -211,7 +212,7 @@ func (d *Downloader) parseFb(url string) (fetcher.FetcherBuilder, error) {
 	if ok {
 		return fetchBuilder, nil
 	}
-	return nil, errors.New("unsupported protocol")
+	return nil, ErrUnSupportedProtocol
 }
 
 func (d *Downloader) setupFetcher(fetcher fetcher.Fetcher) {
@@ -508,6 +509,15 @@ func (d *Downloader) DeleteByStatues(statues []base.Status, force bool) (err err
 	return
 }
 
+func (d *Downloader) Stats(id string) (sr any, err error) {
+	task := d.GetTask(id)
+	if task == nil {
+		return sr, ErrTaskNotFound
+	}
+	sr = task.fetcher.Stats()
+	return
+}
+
 func (d *Downloader) doDelete(task *Task, force bool) (err error) {
 	err = func() error {
 		if task.fetcher != nil {
@@ -754,6 +764,7 @@ func (d *Downloader) doCreate(fetcher fetcher.Fetcher, opts *base.Options) (task
 	task := NewTask()
 	task.fetcherBuilder = fb
 	task.fetcher = fetcher
+	task.Protocol = fetcher.Name()
 	task.Meta = fetcher.Meta()
 	task.Progress = &Progress{}
 	initTask(task)

pkg/download/model.go

@@ -7,6 +7,7 @@ import (
 	"github.com/GopeedLab/gopeed/internal/protocol/http"
 	"github.com/GopeedLab/gopeed/pkg/base"
 	"github.com/GopeedLab/gopeed/pkg/util"
+
 	gonanoid "github.com/matoous/go-nanoid/v2"
 	"net/url"
 	"sync"
@@ -22,6 +23,7 @@ type Task struct {
 	ID        string               `json:"id"`
 	Meta      *fetcher.FetcherMeta `json:"meta"`
 	Status    base.Status          `json:"status"`
+	Protocol  string               `json:"protocol"`
 	Progress  *Progress            `json:"progress"`
 	CreatedAt time.Time            `json:"createdAt"`
 	UpdatedAt time.Time            `json:"updatedAt"`

pkg/rest/api.go

@@ -281,6 +281,21 @@ func DoProxy(w http.ResponseWriter, r *http.Request) {
 	w.Write(buf)
 }
 
+func GetStats(w http.ResponseWriter, r *http.Request) {
+	vars := mux.Vars(r)
+	taskId := vars["id"]
+	if taskId == "" {
+		WriteJson(w, model.NewErrorResult("param invalid: id", model.CodeInvalidParam))
+		return
+	}
+	statsResult, err := Downloader.Stats(taskId)
+	if err != nil {
+		writeError(w, err.Error())
+		return
+	}
+	WriteJson(w, model.NewOkResult(statsResult))
+}
+
 func convertStatues(statues []string) []base.Status {
 	result := make([]base.Status, 0)
 	for _, status := range statues {

pkg/rest/server.go

@@ -98,6 +98,7 @@ func BuildServer(startCfg *model.StartConfig) (*http.Server, net.Listener, error
 	r.Methods(http.MethodDelete).Path("/api/v1/tasks").HandlerFunc(DeleteTasks)
 	r.Methods(http.MethodGet).Path("/api/v1/tasks/{id}").HandlerFunc(GetTask)
 	r.Methods(http.MethodGet).Path("/api/v1/tasks").HandlerFunc(GetTasks)
+	r.Methods(http.MethodGet).Path("/api/v1/tasks/{id}/stats").HandlerFunc(GetStats)
 	r.Methods(http.MethodGet).Path("/api/v1/config").HandlerFunc(GetConfig)
 	r.Methods(http.MethodPut).Path("/api/v1/config").HandlerFunc(PutConfig)
 	r.Methods(http.MethodPost).Path("/api/v1/extensions").HandlerFunc(InstallExtension)