feat(clients): add Readarr support (#490)

* Add initial Readarr support

* Readarr working with MaM

* feat(clients): readarr add tests
This commit is contained in:
voltron4lyfe 2022-10-14 10:56:42 -07:00 committed by GitHub
parent b7d2161fdb
commit 71d0424b61
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
16 changed files with 626 additions and 6 deletions

View file

@ -0,0 +1,68 @@
package action
import (
"context"
"time"
"github.com/autobrr/autobrr/internal/domain"
"github.com/autobrr/autobrr/pkg/errors"
"github.com/autobrr/autobrr/pkg/readarr"
)
func (s *service) readarr(action domain.Action, release domain.Release) ([]string, error) {
s.log.Trace().Msg("action READARR")
// TODO validate data
// get client for action
client, err := s.clientSvc.FindByID(context.TODO(), action.ClientID)
if err != nil {
return nil, errors.Wrap(err, "readarr could not find client: %v", action.ClientID)
}
// return early if no client found
if client == nil {
return nil, errors.New("no client found")
}
// initial config
cfg := readarr.Config{
Hostname: client.Host,
APIKey: client.Settings.APIKey,
Log: s.subLogger,
}
// only set basic auth if enabled
if client.Settings.Basic.Auth {
cfg.BasicAuth = client.Settings.Basic.Auth
cfg.Username = client.Settings.Basic.Username
cfg.Password = client.Settings.Basic.Password
}
arr := readarr.New(cfg)
r := readarr.Release{
Title: release.TorrentName,
DownloadUrl: release.TorrentURL,
Size: int64(release.Size),
Indexer: release.Indexer,
DownloadProtocol: "torrent",
Protocol: "torrent",
PublishDate: time.Now().Format(time.RFC3339),
}
rejections, err := arr.Push(r)
if err != nil {
return nil, errors.Wrap(err, "readarr: failed to push release: %v", r)
}
if rejections != nil {
s.log.Debug().Msgf("readarr: release push rejected: %v, indexer %v to %v reasons: '%v'", r.Title, r.Indexer, client.Host, rejections)
return rejections, nil
}
s.log.Debug().Msgf("readarr: successfully pushed release: %v, indexer %v to %v", r.Title, r.Indexer, client.Host)
return nil, nil
}

View file

@ -66,6 +66,9 @@ func (s *service) RunAction(action *domain.Action, release domain.Release) ([]st
case domain.ActionTypeWhisparr:
rejections, err = s.whisparr(*action, release)
case domain.ActionTypeReadarr:
rejections, err = s.readarr(*action, release)
default:
s.log.Warn().Msgf("unsupported action type: %v", action.Type)
return rejections, err