You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
3.5 KiB
118 lines
3.5 KiB
package serializer
|
|
|
|
import (
|
|
"path"
|
|
"time"
|
|
|
|
model "github.com/cloudreve/Cloudreve/v3/models"
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/aria2/rpc"
|
|
)
|
|
|
|
// DownloadListResponse 下载列表响应条目
|
|
type DownloadListResponse struct {
|
|
UpdateTime time.Time `json:"update"`
|
|
UpdateInterval int `json:"interval"`
|
|
Name string `json:"name"`
|
|
Status int `json:"status"`
|
|
Dst string `json:"dst"`
|
|
Total uint64 `json:"total"`
|
|
Downloaded uint64 `json:"downloaded"`
|
|
Speed int `json:"speed"`
|
|
Info rpc.StatusInfo `json:"info"`
|
|
NodeName string `json:"node"`
|
|
}
|
|
|
|
// FinishedListResponse 已完成任务条目
|
|
type FinishedListResponse struct {
|
|
Name string `json:"name"`
|
|
GID string `json:"gid"`
|
|
Status int `json:"status"`
|
|
Dst string `json:"dst"`
|
|
Error string `json:"error"`
|
|
Total uint64 `json:"total"`
|
|
Files []rpc.FileInfo `json:"files"`
|
|
TaskStatus int `json:"task_status"`
|
|
TaskError string `json:"task_error"`
|
|
CreateTime time.Time `json:"create"`
|
|
UpdateTime time.Time `json:"update"`
|
|
NodeName string `json:"node"`
|
|
}
|
|
|
|
// BuildFinishedListResponse 构建已完成任务条目
|
|
func BuildFinishedListResponse(tasks []model.Download) Response {
|
|
resp := make([]FinishedListResponse, 0, len(tasks))
|
|
|
|
for i := 0; i < len(tasks); i++ {
|
|
fileName := tasks[i].StatusInfo.BitTorrent.Info.Name
|
|
if len(tasks[i].StatusInfo.Files) == 1 {
|
|
fileName = path.Base(tasks[i].StatusInfo.Files[0].Path)
|
|
}
|
|
|
|
// 过滤敏感信息
|
|
for i2 := 0; i2 < len(tasks[i].StatusInfo.Files); i2++ {
|
|
tasks[i].StatusInfo.Files[i2].Path = path.Base(tasks[i].StatusInfo.Files[i2].Path)
|
|
}
|
|
|
|
download := FinishedListResponse{
|
|
Name: fileName,
|
|
GID: tasks[i].GID,
|
|
Status: tasks[i].Status,
|
|
Error: tasks[i].Error,
|
|
Dst: tasks[i].Dst,
|
|
Total: tasks[i].TotalSize,
|
|
Files: tasks[i].StatusInfo.Files,
|
|
TaskStatus: -1,
|
|
UpdateTime: tasks[i].UpdatedAt,
|
|
CreateTime: tasks[i].CreatedAt,
|
|
NodeName: tasks[i].NodeName,
|
|
}
|
|
|
|
if tasks[i].Task != nil {
|
|
download.TaskError = tasks[i].Task.Error
|
|
download.TaskStatus = tasks[i].Task.Status
|
|
}
|
|
|
|
resp = append(resp, download)
|
|
}
|
|
|
|
return Response{Data: resp}
|
|
}
|
|
|
|
// BuildDownloadingResponse 构建正在下载的列表响应
|
|
func BuildDownloadingResponse(tasks []model.Download, intervals map[uint]int) Response {
|
|
resp := make([]DownloadListResponse, 0, len(tasks))
|
|
|
|
for i := 0; i < len(tasks); i++ {
|
|
fileName := ""
|
|
if len(tasks[i].StatusInfo.Files) > 0 {
|
|
fileName = path.Base(tasks[i].StatusInfo.Files[0].Path)
|
|
}
|
|
|
|
// 过滤敏感信息
|
|
tasks[i].StatusInfo.Dir = ""
|
|
for i2 := 0; i2 < len(tasks[i].StatusInfo.Files); i2++ {
|
|
tasks[i].StatusInfo.Files[i2].Path = path.Base(tasks[i].StatusInfo.Files[i2].Path)
|
|
}
|
|
|
|
interval := 10
|
|
if actualInterval, ok := intervals[tasks[i].ID]; ok {
|
|
interval = actualInterval
|
|
}
|
|
|
|
resp = append(resp, DownloadListResponse{
|
|
UpdateTime: tasks[i].UpdatedAt,
|
|
UpdateInterval: interval,
|
|
Name: fileName,
|
|
Status: tasks[i].Status,
|
|
Dst: tasks[i].Dst,
|
|
Total: tasks[i].TotalSize,
|
|
Downloaded: tasks[i].DownloadedSize,
|
|
Speed: tasks[i].Speed,
|
|
Info: tasks[i].StatusInfo,
|
|
NodeName: tasks[i].NodeName,
|
|
})
|
|
}
|
|
|
|
return Response{Data: resp}
|
|
}
|