jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-06-25 18:25:05 +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

9
pkg/qbittorrent/qbittorrent.go
View File

@@ -11,6 +11,8 @@ import (
"github.com/pkg/errors" "github.com/pkg/errors"
) )
const btCategory = "polaris"
type Info struct { type Info struct {
URL string URL string
User string User string
@@ -19,6 +21,7 @@ type Info struct {
type Client struct { type Client struct {
c *qbt.Client c *qbt.Client
category string
Info Info
} }
@@ -36,11 +39,11 @@ func NewClient(url, user, pass string) (*Client, error) {
return nil, err 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) { 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 { if err != nil {
return nil, errors.Wrap(err, "get torrents") return nil, errors.Wrap(err, "get torrents")
} }
@@ -66,7 +69,7 @@ func (c *Client) Download(link, dir string) (pkg.Torrent, error) {
if err != nil { if err != nil {
return nil, errors.Wrap(err, "get hash") 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 { if err != nil {
return nil, errors.Wrap(err, "qbt download") return nil, errors.Wrap(err, "qbt download")
} }

2
server/server.go
View File

@@ -76,7 +76,7 @@ func (s *Server) Serve() error {
activity.GET("/", HttpHandler(s.GetAllActivities)) activity.GET("/", HttpHandler(s.GetAllActivities))
activity.POST("/delete", HttpHandler(s.RemoveActivity)) activity.POST("/delete", HttpHandler(s.RemoveActivity))
activity.GET("/media/:id", HttpHandler(s.GetMediaDownloadHistory)) activity.GET("/media/:id", HttpHandler(s.GetMediaDownloadHistory))
activity.GET("/torrents", HttpHandler(s.GetAllTorrents)) //activity.GET("/torrents", HttpHandler(s.GetAllTorrents))
} }
tv := api.Group("/media") tv := api.Group("/media")