jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-05-21 09:57:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: build in torrent client

Browse Source
This commit is contained in:
Simon Ding
2025-03-28 17:46:20 +08:00
parent 74e680c1ee
commit 050e6446a7
8 changed files with 85 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
server/core/client.go
View File

@@ -6,6 +6,7 @@ import (
"polaris/ent/downloadclients"
"polaris/log"
"polaris/pkg"
"polaris/pkg/buildin"
"polaris/pkg/qbittorrent"
"polaris/pkg/tmdb"
"polaris/pkg/transmission"
@@ -69,7 +70,7 @@ func (c *Client) reloadTasks() {
log.Warnf("get task error: %v", err)
continue
}
c.tasks[t.ID] = &Task{Torrent: to}
c.tasks[t.ID] = &Task{Torrent: to}
} else if t.Link != "" {
to, err := transmission.NewTorrent(transmission.Config{
URL: dl.URL,
@@ -80,7 +81,7 @@ func (c *Client) reloadTasks() {
log.Warnf("get task error: %v", err)
continue
}
c.tasks[t.ID] = &Task{Torrent: to}
c.tasks[t.ID] = &Task{Torrent: to}
}
} else if dl.Implementation == downloadclients.ImplementationQbittorrent {
if t.Hash != "" {
@@ -94,7 +95,7 @@ func (c *Client) reloadTasks() {
continue
}
c.tasks[t.ID] = &Task{Torrent: to}
} else if t.Link != "" {
to, err := qbittorrent.NewTorrent(qbittorrent.Info{
URL: dl.URL,
@@ -106,7 +107,7 @@ func (c *Client) reloadTasks() {
continue
}
c.tasks[t.ID] = &Task{Torrent: to}
}
}
@@ -114,6 +115,11 @@ func (c *Client) reloadTasks() {
log.Infof("------ task reloading done ------")
}
func (c *Client) buildInDownloader() (pkg.Downloader, error) {
dir := c.db.GetDownloadDir()
return buildin.NewDownloader(dir)
}
func (c *Client) GetDownloadClient() (pkg.Downloader, *ent.DownloadClients, error) {
downloaders := c.db.GetAllDonloadClients()
for _, d := range downloaders {
@@ -139,8 +145,16 @@ func (c *Client) GetDownloadClient() (pkg.Downloader, *ent.DownloadClients, erro
continue
}
return qbt, d, nil
} else if d.Implementation == downloadclients.ImplementationBuildin {
bin, err := c.buildInDownloader()
if err != nil {
log.Warnf("connect to download client error: %v", d.URL)
continue
}
return bin, d, nil
}
}
return nil, nil, errors.Errorf("no available download client")
}