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

feat: add qbit category

Browse Source
This commit is contained in:
Simon Ding
2024-10-11 13:20:03 +08:00
parent f2faefa837
commit 98fae05e16
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pkg/qbittorrent/qbittorrent.go
View File

@@ -11,6 +11,8 @@ import (
"github.com/pkg/errors"
)
const btCategory = "polaris"
type Info struct {
URL string
User string
@@ -18,7 +20,8 @@ type Info struct {
}
type Client struct {
c *qbt.Client
c *qbt.Client
category string
Info
}
@@ -36,11 +39,11 @@ func NewClient(url, user, pass string) (*Client, error) {
return nil, err
}
return &Client{c: qb, Info: Info{URL: url, User: user, Password: pass}}, nil
return &Client{c: qb, category: btCategory, Info: Info{URL: url, User: user, Password: pass}}, nil
}
func (c *Client) GetAll() ([]pkg.Torrent, error) {
tt, err := c.c.Torrents(qbt.TorrentsOptions{})
tt, err := c.c.Torrents(qbt.TorrentsOptions{Category: &c.category})
if err != nil {
return nil, errors.Wrap(err, "get torrents")
}
@@ -66,7 +69,7 @@ func (c *Client) Download(link, dir string) (pkg.Torrent, error) {
if err != nil {
return nil, errors.Wrap(err, "get hash")
}
err = c.c.DownloadLinks([]string{magnet}, qbt.DownloadOptions{Savepath: &dir})
err = c.c.DownloadLinks([]string{magnet}, qbt.DownloadOptions{Savepath: &dir, Category: &c.category})
if err != nil {
return nil, errors.Wrap(err, "qbt download")
}