feat: add upload progress and fix panic

This commit is contained in:
Simon Ding
2024-11-17 21:57:14 +08:00
parent 7d5ce8ba97
commit ba3f6de852
5 changed files with 23 additions and 5 deletions

View File

@@ -374,6 +374,15 @@ func (s *StorageInfo) ToWebDavSetting() WebdavSetting {
}
}
func (s *StorageInfo) ToAlistSetting() WebdavSetting {
return WebdavSetting{
URL: s.Settings["url"],
User: s.Settings["user"],
Password: s.Settings["password"],
ChangeFileHash: s.Settings["change_file_hash"],
}
}
type WebdavSetting struct {
URL string `json:"url"`
User string `json:"user"`

View File

@@ -17,8 +17,9 @@ import (
type Activity struct {
*ent.History
Progress int `json:"progress"`
SeedRatio float64 `json:"seed_ratio"`
Progress int `json:"progress"`
SeedRatio float64 `json:"seed_ratio"`
UploadProgress float64 `json:"upload_progress"`
}
func (s *Server) GetAllActivities(c *gin.Context) (interface{}, error) {
@@ -44,6 +45,9 @@ func (s *Server) GetAllActivities(c *gin.Context) (interface{}, error) {
} else {
a.SeedRatio = r
}
if task.UploadProgresser != nil {
a.UploadProgress = task.UploadProgresser()
}
}
}
activities = append(activities, a)

View File

@@ -207,6 +207,7 @@ func (c *Client) moveCompletedTask(id int) (err1 error) {
if err := stImpl.Copy(filepath.Join(c.db.GetDownloadDir(), torrentName), r.TargetDir); err != nil {
return errors.Wrap(err, "move file")
}
torrent.UploadProgresser = stImpl.UploadProgress
c.db.SetHistoryStatus(r.ID, history.StatusSeeding)
if len(episodeIds) > 0 {
@@ -285,6 +286,7 @@ func (c *Client) CheckDownloadedSeriesFiles(m *ent.Media) error {
type Task struct {
//Processing bool
pkg.Torrent
UploadProgresser func() float64
}
func (c *Client) DownloadSeriesAllEpisodes(id int) []string {

View File

@@ -42,7 +42,7 @@ func (s *Server) AddStorage(c *gin.Context) (interface{}, error) {
log.Infof("file name: %v", f.Name())
}
} else if in.Implementation == "alist" {
cfg := in.ToWebDavSetting()
cfg := in.ToAlistSetting()
_, err := storage.NewAlist(&alist.Config{URL: cfg.URL, Username: cfg.User, Password: cfg.Password}, in.TvPath)
if err != nil {
return nil, errors.Wrap(err, "alist")

View File

@@ -73,7 +73,7 @@ class Activity {
required this.saved,
required this.progress,
required this.size,
required this.seedRatio});
required this.seedRatio, required this.uploadProgress});
final int? id;
final int? mediaId;
@@ -86,6 +86,7 @@ class Activity {
final int? progress;
final int? size;
final double seedRatio;
final double uploadProgress;
factory Activity.fromJson(Map<String, dynamic> json) {
return Activity(
@@ -99,6 +100,8 @@ class Activity {
saved: json["saved"],
progress: json["progress"],
seedRatio: json["seed_ratio"],
size: json["size"]);
size: json["size"],
uploadProgress: json["upload_progress"]
);
}
}