mirror of
https://github.com/idanoo/autobrr
synced 2025-07-23 16:59:12 +00:00
85 lines
2.3 KiB
Go
85 lines
2.3 KiB
Go
// Copyright (c) 2021 - 2024, Ludvig Lundgren and the autobrr contributors.
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
package action
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/autobrr/autobrr/internal/domain"
|
|
"github.com/autobrr/autobrr/pkg/errors"
|
|
"github.com/autobrr/autobrr/pkg/sonarr"
|
|
)
|
|
|
|
func (s *service) sonarr(ctx context.Context, action *domain.Action, release domain.Release) ([]string, error) {
|
|
s.log.Trace().Msg("action SONARR")
|
|
|
|
// TODO validate data
|
|
|
|
// get client for action
|
|
client, err := s.clientSvc.FindByID(ctx, action.ClientID)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "sonarr 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 := sonarr.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
|
|
}
|
|
|
|
externalClientId := client.Settings.ExternalDownloadClientId
|
|
if action.ExternalDownloadClientID > 0 {
|
|
externalClientId = int(action.ExternalDownloadClientID)
|
|
}
|
|
|
|
externalClient := client.Settings.ExternalDownloadClient
|
|
if action.ExternalDownloadClient != "" {
|
|
externalClient = action.ExternalDownloadClient
|
|
}
|
|
|
|
r := sonarr.Release{
|
|
Title: release.TorrentName,
|
|
InfoUrl: release.InfoURL,
|
|
DownloadUrl: release.DownloadURL,
|
|
MagnetUrl: release.MagnetURI,
|
|
Size: int64(release.Size),
|
|
Indexer: release.Indexer.Identifier,
|
|
DownloadClientId: externalClientId,
|
|
DownloadClient: externalClient,
|
|
DownloadProtocol: string(release.Protocol),
|
|
Protocol: string(release.Protocol),
|
|
PublishDate: time.Now().Format(time.RFC3339),
|
|
}
|
|
|
|
arr := sonarr.New(cfg)
|
|
|
|
rejections, err := arr.Push(ctx, r)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "sonarr: failed to push release: %v", r)
|
|
}
|
|
|
|
if rejections != nil {
|
|
s.log.Debug().Msgf("sonarr: release push rejected: %v, indexer %v to %v reasons: '%v'", r.Title, r.Indexer, client.Host, rejections)
|
|
|
|
return rejections, nil
|
|
}
|
|
|
|
s.log.Debug().Msgf("sonarr: successfully pushed release: %v, indexer %v to %v", r.Title, r.Indexer, client.Host)
|
|
|
|
return nil, nil
|
|
}
|