mirror of
https://github.com/idanoo/autobrr
synced 2025-07-23 16:59:12 +00:00
feat(qbittorrent): Add more methods (#443)
* (feat): qbittorrent: Add funcs * sync * Delete go.mod * more more more * a cat for your trouble * no pointers for you * maybe they weren't so bad afterall.
This commit is contained in:
Kyle Sanderson
GitHub
parent
9813f5718d
commit
5a7614f954
2 changed files with 230 additions and 2 deletions
|
|
@ -59,11 +59,27 @@ type TorrentTracker struct {
|
|||
Status TrackerStatus `json:"status"`
|
||||
NumPeers int `json:"num_peers"`
|
||||
NumSeeds int `json:"num_seeds"`
|
||||
NumLeechers int `json:"num_leeches"`
|
||||
NumLeechers int `json:"num_leechers"`
|
||||
NumDownloaded int `json:"num_downloaded"`
|
||||
Message string `json:"msg"`
|
||||
}
|
||||
|
||||
type TorrentFiles []struct {
|
||||
Availability int `json:"availability"`
|
||||
Index int `json:"index"`
|
||||
IsSeed bool `json:"is_seed,omitempty"`
|
||||
Name string `json:"name"`
|
||||
PieceRange []int `json:"piece_range"`
|
||||
Priority int `json:"priority"`
|
||||
Progress int `json:"progress"`
|
||||
Size int `json:"size"`
|
||||
}
|
||||
|
||||
type Category struct {
|
||||
Name string `json:"name"`
|
||||
SavePath string `json:"savePath"`
|
||||
}
|
||||
|
||||
type TorrentState string
|
||||
|
||||
const (
|
||||
|
|
@ -113,7 +129,7 @@ const (
|
|||
TorrentStateCheckingDl TorrentState = "checkingDL"
|
||||
|
||||
// Torrent is forced to downloading to ignore queue limit
|
||||
TorrentStateForceDl TorrentState = "forceDL"
|
||||
TorrentStateForcedDl TorrentState = "forcedDL"
|
||||
|
||||
// Checking resume data on qBt startup
|
||||
TorrentStateCheckingResumeData TorrentState = "checkingResumeData"
|
||||
|
|
|
|||
|
|
@ -275,3 +275,215 @@ func (c *Client) GetTransferInfo() (*TransferInfo, error) {
|
|||
|
||||
return &info, nil
|
||||
}
|
||||
|
||||
func (c *Client) Resume(hashes []string) error {
|
||||
// Add hashes together with | separator
|
||||
hv := strings.Join(hashes, "|")
|
||||
opts := map[string]string{
|
||||
"hashes": hv,
|
||||
}
|
||||
|
||||
resp, err := c.get("torrents/resume", opts)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "could not resume torrents: %v", hashes)
|
||||
} else if resp.StatusCode != http.StatusOK {
|
||||
return errors.Wrap(err, "could not resume torrents: %v unexpected status: %v", hashes, resp.StatusCode)
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Client) SetForceStart(hashes []string, value bool) error {
|
||||
// Add hashes together with | separator
|
||||
hv := strings.Join(hashes, "|")
|
||||
opts := map[string]string{
|
||||
"hashes": hv,
|
||||
"value": strconv.FormatBool(value),
|
||||
}
|
||||
|
||||
resp, err := c.get("torrents/setForceStart", opts)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "could not setForceStart torrents: %v", hashes)
|
||||
} else if resp.StatusCode != http.StatusOK {
|
||||
return errors.Wrap(err, "could not setForceStart torrents: %v unexpected status: %v", hashes, resp.StatusCode)
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Client) Recheck(hashes []string) error {
|
||||
// Add hashes together with | separator
|
||||
hv := strings.Join(hashes, "|")
|
||||
opts := map[string]string{
|
||||
"hashes": hv,
|
||||
}
|
||||
|
||||
resp, err := c.get("torrents/recheck", opts)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "could not recheck torrents: %v", hashes)
|
||||
} else if resp.StatusCode != http.StatusOK {
|
||||
return errors.Wrap(err, "could not recheck torrents: %v unexpected status: %v", hashes, resp.StatusCode)
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Client) Pause(hashes []string) error {
|
||||
// Add hashes together with | separator
|
||||
hv := strings.Join(hashes, "|")
|
||||
opts := map[string]string{
|
||||
"hashes": hv,
|
||||
}
|
||||
|
||||
resp, err := c.get("torrents/pause", opts)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "could not pause torrents: %v", hashes)
|
||||
} else if resp.StatusCode != http.StatusOK {
|
||||
return errors.Wrap(err, "could not pause torrents: %v unexpected status: %v", hashes, resp.StatusCode)
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Client) SetAutoManagement(hashes []string, enable bool) error {
|
||||
// Add hashes together with | separator
|
||||
hv := strings.Join(hashes, "|")
|
||||
opts := map[string]string{
|
||||
"hashes": hv,
|
||||
"enable": strconv.FormatBool(enable),
|
||||
}
|
||||
|
||||
resp, err := c.get("torrents/setAutoManagement", opts)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "could not setAutoManagement torrents: %v", hashes)
|
||||
} else if resp.StatusCode != http.StatusOK {
|
||||
return errors.Wrap(err, "could not setAutoManagement torrents: %v unexpected status: %v", hashes, resp.StatusCode)
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Client) CreateCategory(category string, path string) error {
|
||||
opts := map[string]string{
|
||||
"category": category,
|
||||
"savePath": path,
|
||||
}
|
||||
|
||||
resp, err := c.get("torrents/createCategory", opts)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "could not createCategory torrents: %v", category)
|
||||
} else if resp.StatusCode != http.StatusOK {
|
||||
return errors.Wrap(err, "could not createCategory torrents: %v unexpected status: %v", category, resp.StatusCode)
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Client) EditCategory(category string, path string) error {
|
||||
opts := map[string]string{
|
||||
"category": category,
|
||||
"savePath": path,
|
||||
}
|
||||
|
||||
resp, err := c.get("torrents/editCategory", opts)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "could not editCategory torrents: %v", category)
|
||||
} else if resp.StatusCode != http.StatusOK {
|
||||
return errors.Wrap(err, "could not editCategory torrents: %v unexpected status: %v", category, resp.StatusCode)
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Client) RemoveCategories(categories[] string) error {
|
||||
opts := map[string]string{
|
||||
"categories": strings.Join(categories, "\n"),
|
||||
}
|
||||
|
||||
resp, err := c.get("torrents/removeCategories", opts)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "could not removeCategories torrents: %v", opts["categories"])
|
||||
} else if resp.StatusCode != http.StatusOK {
|
||||
return errors.Wrap(err, "could not removeCategories torrents: %v unexpected status: %v", opts["categories"], resp.StatusCode)
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Client) SetCategory(hashes []string, category string) error {
|
||||
// Add hashes together with | separator
|
||||
hv := strings.Join(hashes, "|")
|
||||
opts := map[string]string{
|
||||
"hashes": hv,
|
||||
"category": category,
|
||||
}
|
||||
|
||||
resp, err := c.get("torrents/setCategory", opts)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "could not setCategory torrents: %v", hashes)
|
||||
} else if resp.StatusCode != http.StatusOK {
|
||||
return errors.Wrap(err, "could not setCategory torrents: %v unexpected status: %v", hashes, resp.StatusCode)
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Client) GetFilesInformation(hash string) (*TorrentFiles, error) {
|
||||
opts := map[string]string{
|
||||
"hash": hash,
|
||||
}
|
||||
|
||||
resp, err := c.get("torrents/files", opts)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "could not get files info")
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
|
||||
body, readErr := ioutil.ReadAll(resp.Body)
|
||||
if readErr != nil {
|
||||
return nil, errors.Wrap(readErr, "could not read body")
|
||||
}
|
||||
|
||||
var info TorrentFiles
|
||||
err = json.Unmarshal(body, &info)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(readErr, "could not unmarshal body")
|
||||
}
|
||||
|
||||
return &info, nil
|
||||
}
|
||||
|
||||
func (c *Client) GetCategories() (map[string]Category, error) {
|
||||
resp, err := c.get("torrents/categories", nil)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "could not get files info")
|
||||
}
|
||||
|
||||
defer resp.Body.Close()
|
||||
|
||||
body, readErr := ioutil.ReadAll(resp.Body)
|
||||
if readErr != nil {
|
||||
return nil, errors.Wrap(readErr, "could not read body")
|
||||
}
|
||||
|
||||
m := make(map[string]Category)
|
||||
err = json.Unmarshal(body, &m)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(readErr, "could not unmarshal body")
|
||||
}
|
||||
|
||||
return m, nil
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue