mirror of https://github.com/Xhofe/alist
87 lines
2.6 KiB
Go
87 lines
2.6 KiB
Go
package qbit
|
|
|
|
import (
|
|
"github.com/alist-org/alist/v3/internal/conf"
|
|
"github.com/alist-org/alist/v3/internal/errs"
|
|
"github.com/alist-org/alist/v3/internal/model"
|
|
"github.com/alist-org/alist/v3/internal/offline_download/tool"
|
|
"github.com/alist-org/alist/v3/internal/setting"
|
|
"github.com/alist-org/alist/v3/pkg/qbittorrent"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type QBittorrent struct {
|
|
client qbittorrent.Client
|
|
}
|
|
|
|
func (a *QBittorrent) Run(task *tool.DownloadTask) error {
|
|
return errs.NotSupport
|
|
}
|
|
|
|
func (a *QBittorrent) Name() string {
|
|
return "qBittorrent"
|
|
}
|
|
|
|
func (a *QBittorrent) Items() []model.SettingItem {
|
|
// qBittorrent settings
|
|
return []model.SettingItem{
|
|
{Key: conf.QbittorrentUrl, Value: "http://admin:adminadmin@localhost:8080/", Type: conf.TypeString, Group: model.OFFLINE_DOWNLOAD, Flag: model.PRIVATE},
|
|
{Key: conf.QbittorrentSeedtime, Value: "0", Type: conf.TypeNumber, Group: model.OFFLINE_DOWNLOAD, Flag: model.PRIVATE},
|
|
}
|
|
}
|
|
|
|
func (a *QBittorrent) Init() (string, error) {
|
|
a.client = nil
|
|
url := setting.GetStr(conf.QbittorrentUrl)
|
|
qbClient, err := qbittorrent.New(url)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
a.client = qbClient
|
|
return "ok", nil
|
|
}
|
|
|
|
func (a *QBittorrent) IsReady() bool {
|
|
return a.client != nil
|
|
}
|
|
|
|
func (a *QBittorrent) AddURL(args *tool.AddUrlArgs) (string, error) {
|
|
err := a.client.AddFromLink(args.Url, args.TempDir, args.UID)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return args.UID, nil
|
|
}
|
|
|
|
func (a *QBittorrent) Remove(task *tool.DownloadTask) error {
|
|
err := a.client.Delete(task.GID, false)
|
|
return err
|
|
}
|
|
|
|
func (a *QBittorrent) Status(task *tool.DownloadTask) (*tool.Status, error) {
|
|
info, err := a.client.GetInfo(task.GID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
s := &tool.Status{}
|
|
s.TotalBytes = info.Size
|
|
s.Progress = float64(info.Completed) / float64(info.Size) * 100
|
|
switch info.State {
|
|
case qbittorrent.UPLOADING, qbittorrent.PAUSEDUP, qbittorrent.QUEUEDUP, qbittorrent.STALLEDUP, qbittorrent.FORCEDUP, qbittorrent.CHECKINGUP:
|
|
s.Completed = true
|
|
case qbittorrent.ALLOCATING, qbittorrent.DOWNLOADING, qbittorrent.METADL, qbittorrent.PAUSEDDL, qbittorrent.QUEUEDDL, qbittorrent.STALLEDDL, qbittorrent.CHECKINGDL, qbittorrent.FORCEDDL, qbittorrent.CHECKINGRESUMEDATA, qbittorrent.MOVING:
|
|
s.Status = "[qBittorrent] downloading"
|
|
case qbittorrent.ERROR, qbittorrent.MISSINGFILES, qbittorrent.UNKNOWN:
|
|
s.Err = errors.Errorf("[qBittorrent] failed to download %s, error: %s", task.GID, info.State)
|
|
default:
|
|
s.Err = errors.Errorf("[qBittorrent] unknown error occurred downloading %s", task.GID)
|
|
}
|
|
return s, nil
|
|
}
|
|
|
|
var _ tool.Tool = (*QBittorrent)(nil)
|
|
|
|
func init() {
|
|
tool.Tools.Add(&QBittorrent{})
|
|
}
|