feat(autobrr): support basic auth (#17)

autobrr · Jan 9, 2023 · c26f6be · c26f6be
1 parent ddff1d0
commit c26f6be
Show file tree

Hide file tree

Showing 5 changed files with 43 additions and 8 deletions.
diff --git a/README.md b/README.md
@@ -54,6 +54,17 @@ If you're trying to reach radarr or sonarr hosted on swizzin from some other loc
         - 15
 ```
 
+Same goes for autobrr if it's behind basic auth.
+
+```yaml
+  autobrr:
+    host: http://localhost:7474
+    apikey: YOUR_API_KEY
+    basicAuth:
+      user: username
+      pass: password
+```
+
 ### Tags
 
 This works for both sonarr and radarr.

diff --git a/internal/processor/radarr.go b/internal/processor/radarr.go
@@ -7,6 +7,7 @@ import (
 
 	"github.com/autobrr/omegabrr/internal/domain"
 	"github.com/autobrr/omegabrr/pkg/autobrr"
+
 	"github.com/rs/zerolog"
 	"github.com/rs/zerolog/log"
 	"golift.io/starr"
@@ -18,14 +19,14 @@ func (s Service) radarr(ctx context.Context, cfg *domain.ArrConfig, dryRun bool,
 
 	l.Debug().Msgf("gathering titles...")
 
-	movieTitles, err := s.processRadarr(ctx, cfg, &l)
+	titles, err := s.processRadarr(ctx, cfg, &l)
 	if err != nil {
 		return err
 	}
 
-	l.Debug().Msgf("got %v filter titles", len(movieTitles))
+	l.Debug().Msgf("got %v filter titles", len(titles))
 
-	joinedTitles := strings.Join(movieTitles, ",")
+	joinedTitles := strings.Join(titles, ",")
 
 	l.Trace().Msgf("%v", joinedTitles)
 

diff --git a/internal/processor/service.go b/internal/processor/service.go
@@ -29,6 +29,9 @@ func (s Service) Process(dryRun bool) error {
 	}
 
 	a := autobrr.NewClient(s.cfg.Clients.Autobrr.Host, s.cfg.Clients.Autobrr.Apikey)
+	if s.cfg.Clients.Autobrr.BasicAuth != nil {
+		a.SetBasicAuth(s.cfg.Clients.Autobrr.BasicAuth.User, s.cfg.Clients.Autobrr.BasicAuth.Pass)
+	}
 
 	log.Debug().Msgf("starting filter processing...")
 
@@ -72,6 +75,10 @@ func (s Service) GetFilters(ctx context.Context) ([]autobrr.Filter, error) {
 	}
 
 	a := autobrr.NewClient(s.cfg.Clients.Autobrr.Host, s.cfg.Clients.Autobrr.Apikey)
+	if s.cfg.Clients.Autobrr.BasicAuth != nil {
+		a.SetBasicAuth(s.cfg.Clients.Autobrr.BasicAuth.User, s.cfg.Clients.Autobrr.BasicAuth.Pass)
+	}
+
 	filters, err := a.GetFilters(ctx)
 	if err != nil {
 		return nil, err

diff --git a/internal/processor/sonarr.go b/internal/processor/sonarr.go
@@ -7,6 +7,7 @@ import (
 
 	"github.com/autobrr/omegabrr/internal/domain"
 	"github.com/autobrr/omegabrr/pkg/autobrr"
+
 	"github.com/rs/zerolog"
 	"github.com/rs/zerolog/log"
 	"golift.io/starr"
@@ -18,14 +19,14 @@ func (s Service) sonarr(ctx context.Context, cfg *domain.ArrConfig, dryRun bool,
 
 	l.Debug().Msgf("gathering titles...")
 
-	movieTitles, err := s.processSonarr(ctx, cfg, &l)
+	titles, err := s.processSonarr(ctx, cfg, &l)
 	if err != nil {
 		return err
 	}
 
-	l.Debug().Msgf("got %v filter titles", len(movieTitles))
+	l.Debug().Msgf("got %v filter titles", len(titles))
 
-	joinedTitles := strings.Join(movieTitles, ",")
+	joinedTitles := strings.Join(titles, ",")
 
 	l.Trace().Msgf("%v", joinedTitles)
 
@@ -41,7 +42,7 @@ func (s Service) sonarr(ctx context.Context, cfg *domain.ArrConfig, dryRun bool,
 			f := autobrr.UpdateFilter{MatchReleases: joinedTitles}
 
 			if err := brr.UpdateFilterByID(ctx, filterID, f); err != nil {
-				l.Error().Err(err).Msgf("something went wrong updating movie filter: %v", filterID)
+				l.Error().Err(err).Msgf("something went wrong updating tv filter: %v", filterID)
 				continue
 			}
 		}

diff --git a/pkg/autobrr/client.go b/pkg/autobrr/client.go
@@ -5,20 +5,24 @@ import (
 	"context"
 	"crypto/tls"
 	"encoding/json"
-	"github.com/rs/zerolog/log"
 	"io"
 	"net/http"
 	"net/http/httputil"
 	"strconv"
 	"time"
 
+	"github.com/rs/zerolog/log"
+
 	"github.com/pkg/errors"
 )
 
 type Client struct {
 	Host   string
 	APIKey string
 
+	BasicUser string
+	BasicPass string
+
 	client *http.Client
 }
 
@@ -44,6 +48,15 @@ func NewClient(host string, apikey string) *Client {
 	return c
 }
 
+func (c *Client) SetBasicAuth(user, pass string) {
+	if user != "" {
+		c.BasicUser = user
+	}
+	if pass != "" {
+		c.BasicPass = pass
+	}
+}
+
 func (c *Client) baseClient() {
 
 }
@@ -61,6 +74,7 @@ func (c *Client) GetFilters(ctx context.Context) ([]Filter, error) {
 		return nil, err
 	}
 
+	req.SetBasicAuth(c.BasicUser, c.BasicPass)
 	req.Header.Add("X-API-Token", c.APIKey)
 
 	res, err := c.client.Do(req)
@@ -99,6 +113,7 @@ func (c *Client) UpdateFilterByID(ctx context.Context, filterID int, filter Upda
 		return err
 	}
 
+	req.SetBasicAuth(c.BasicUser, c.BasicPass)
 	req.Header.Add("X-API-Token", c.APIKey)
 
 	res, err := c.client.Do(req)