|
|
|
package serializer
|
|
|
|
|
|
|
|
import (
|
|
|
|
model "github.com/HFO4/cloudreve/models"
|
|
|
|
"github.com/zyxar/argo/rpc"
|
|
|
|
"path"
|
|
|
|
)
|
|
|
|
|
|
|
|
// DownloadListResponse 下载列表响应条目
|
|
|
|
type DownloadListResponse struct {
|
|
|
|
UpdateTime int64 `json:"update"`
|
|
|
|
UpdateInterval int `json:"interval"`
|
|
|
|
Name string `json:"name"`
|
|
|
|
Status int `json:"status"`
|
|
|
|
UserID uint `json:"uid"`
|
|
|
|
Error string `json:"error"`
|
|
|
|
Dst string `json:"dst"`
|
|
|
|
Total uint64 `json:"total"`
|
|
|
|
Downloaded uint64 `json:"downloaded"`
|
|
|
|
Speed int `json:"speed"`
|
|
|
|
Info rpc.StatusInfo `json:"info"`
|
|
|
|
}
|
|
|
|
|
|
|
|
// BuildDownloadingResponse 构建正在下载的列表响应
|
|
|
|
func BuildDownloadingResponse(tasks []model.Download) Response {
|
|
|
|
resp := make([]DownloadListResponse, 0, len(tasks))
|
|
|
|
interval := model.GetIntSetting("aria2_interval", 10)
|
|
|
|
|
|
|
|
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)
|
|
|
|
}
|
|
|
|
|
|
|
|
resp = append(resp, DownloadListResponse{
|
|
|
|
UpdateTime: tasks[i].UpdatedAt.Unix(),
|
|
|
|
UpdateInterval: interval,
|
|
|
|
Name: fileName,
|
|
|
|
Status: tasks[i].Status,
|
|
|
|
UserID: tasks[i].UserID,
|
|
|
|
Error: tasks[i].Error,
|
|
|
|
Dst: tasks[i].Dst,
|
|
|
|
Total: tasks[i].TotalSize,
|
|
|
|
Downloaded: tasks[i].DownloadedSize,
|
|
|
|
Speed: tasks[i].Speed,
|
|
|
|
Info: tasks[i].StatusInfo,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
return Response{Data: resp}
|
|
|
|
}
|