idanoo/autobrr
1
0
Fork 0
mirror of https://github.com/idanoo/autobrr synced 2025-07-23 08:49:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(downloadclient): qBittorrent url parse err handling (#1832)

Browse source 
* fix(downloadclient): qBittorrent url parse err handling

* fix(downloadclient): qBittorrent url parse err handling test
This commit is contained in:
ze0s 2024-11-24 00:53:59 +01:00 • committed by GitHub
parent a18284ecc6
commit f54c51fa06
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 22 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

13
internal/domain/client.go
View file

@ -158,17 +158,20 @@ func (c DownloadClient) Validate() error {
return nil
}
func (c DownloadClient) BuildLegacyHost() string {
func (c DownloadClient) BuildLegacyHost() (string, error) {
if c.Type == DownloadClientTypeQbittorrent {
return c.qbitBuildLegacyHost()
}
return ""
return c.Host, nil
}
// qbitBuildLegacyHost exists to support older configs
func (c DownloadClient) qbitBuildLegacyHost() string {
func (c DownloadClient) qbitBuildLegacyHost() (string, error) {
// parse url
u, _ := url.Parse(c.Host)
u, err := url.Parse(c.Host)
if err != nil {
return "", err
}
// reset Opaque
u.Opaque = ""
@ -200,5 +203,5 @@ func (c DownloadClient) qbitBuildLegacyHost() string {
}
// make into new string and return
return u.String()
return u.String(), nil
}

3
internal/domain/client_test.go
View file

@ -153,7 +153,8 @@ func TestDownloadClient_qbitBuildLegacyHost(t *testing.T) {
Password: tt.fields.Password,
Settings: tt.fields.Settings,
}
assert.Equalf(t, tt.want, c.qbitBuildLegacyHost(), "qbitBuildLegacyHost()")
got, _ := c.qbitBuildLegacyHost()
assert.Equalf(t, tt.want, got, "qbitBuildLegacyHost()")
})
}
}

7
internal/download_client/connection.go
View file

@ -71,8 +71,13 @@ func (s *service) testConnection(ctx context.Context, client domain.DownloadClie
}
func (s *service) testQbittorrentConnection(ctx context.Context, client domain.DownloadClient) error {
clientHost, err := client.BuildLegacyHost()
if err != nil {
return errors.Wrap(err, "error building qBittorrent host url: %s", client.Host)
}
qbtSettings := qbittorrent.Config{
Host: client.BuildLegacyHost(),
Host: clientHost,
TLSSkipVerify: client.TLSSkipVerify,
Username: client.Username,
Password: client.Password,

7
internal/download_client/service.go
View file

@ -181,8 +181,13 @@ func (s *service) GetClient(ctx context.Context, clientId int32) (*domain.Downlo
switch client.Type {
case domain.DownloadClientTypeQbittorrent:
clientHost, err := client.BuildLegacyHost()
if err != nil {
return nil, errors.Wrap(err, "error building qBittorrent host url: %v", client.Host)
}
client.Client = qbittorrent.NewClient(qbittorrent.Config{
Host: client.BuildLegacyHost(),
Host: clientHost,
Username: client.Username,
Password: client.Password,
TLSSkipVerify: client.TLSSkipVerify,