mirror of
https://github.com/idanoo/autobrr
synced 2025-07-23 16:59:12 +00:00
feat(lists): integrate Omegabrr (#1885)
* feat(lists): integrate Omegabrr * feat(lists): add missing lists index * feat(lists): add db repo * feat(lists): add db migrations * feat(lists): labels * feat(lists): url lists and more arrs * fix(lists): db migrations client_id wrong type * fix(lists): db fields * feat(lists): create list form wip * feat(lists): show in list and create * feat(lists): update and delete * feat(lists): trigger via webhook * feat(lists): add webhook handler * fix(arr): encode json to pointer * feat(lists): rename endpoint to lists * feat(lists): fetch tags from arr * feat(lists): process plaintext lists * feat(lists): add background refresh job * run every 6th hour with a random start delay between 1-35 seconds * feat(lists): refresh on save and improve logging * feat(lists): cast arr client to pointer * feat(lists): improve error handling * feat(lists): reset shows field with match release * feat(lists): filter opts all lists * feat(lists): trigger on update if enabled * feat(lists): update option for lists * feat(lists): show connected filters in list * feat(lists): missing listSvc dep * feat(lists): cleanup * feat(lists): typo arr list * feat(lists): radarr include original * feat(lists): rename ExcludeAlternateTitle to IncludeAlternateTitle * fix(lists): arr client type conversion to pointer * fix(actions): only log panic recover if err not nil * feat(lists): show spinner on save * feat(lists): show icon in filters list * feat(lists): change icon color in filters list * feat(lists): delete relations on filter delete
This commit is contained in:
parent
b68ae334ca
commit
221bc35371
77 changed files with 5025 additions and 254 deletions
154
pkg/arr/sonarr/sonarr.go
Normal file
154
pkg/arr/sonarr/sonarr.go
Normal file
|
@ -0,0 +1,154 @@
|
|||
// Copyright (c) 2021 - 2024, Ludvig Lundgren and the autobrr contributors.
|
||||
// SPDX-License-Identifier: GPL-2.0-or-later
|
||||
|
||||
package sonarr
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"log"
|
||||
|
||||
"github.com/autobrr/autobrr/pkg/arr"
|
||||
"github.com/autobrr/autobrr/pkg/errors"
|
||||
"github.com/autobrr/autobrr/pkg/sharedhttp"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
Hostname string
|
||||
APIKey string
|
||||
|
||||
// basic auth username and password
|
||||
BasicAuth bool
|
||||
Username string
|
||||
Password string
|
||||
|
||||
Log *log.Logger
|
||||
}
|
||||
|
||||
type ClientInterface interface {
|
||||
Test(ctx context.Context) (*SystemStatusResponse, error)
|
||||
Push(ctx context.Context, release Release) ([]string, error)
|
||||
}
|
||||
|
||||
type Client struct {
|
||||
config Config
|
||||
http *http.Client
|
||||
|
||||
Log *log.Logger
|
||||
}
|
||||
|
||||
// New create new sonarr Client
|
||||
func New(config Config) *Client {
|
||||
httpClient := &http.Client{
|
||||
Timeout: time.Second * 120,
|
||||
Transport: sharedhttp.Transport,
|
||||
}
|
||||
|
||||
c := &Client{
|
||||
config: config,
|
||||
http: httpClient,
|
||||
Log: log.New(io.Discard, "", log.LstdFlags),
|
||||
}
|
||||
|
||||
if config.Log != nil {
|
||||
c.Log = config.Log
|
||||
}
|
||||
|
||||
return c
|
||||
}
|
||||
|
||||
func (c *Client) Test(ctx context.Context) (*SystemStatusResponse, error) {
|
||||
status, res, err := c.get(ctx, "system/status")
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "could not make Test")
|
||||
}
|
||||
|
||||
if status == http.StatusUnauthorized {
|
||||
return nil, errors.New("unauthorized: bad credentials")
|
||||
}
|
||||
|
||||
c.Log.Printf("sonarr system/status status: (%v) response: %v\n", status, string(res))
|
||||
|
||||
response := SystemStatusResponse{}
|
||||
if err = json.Unmarshal(res, &response); err != nil {
|
||||
return nil, errors.Wrap(err, "could not unmarshal data")
|
||||
}
|
||||
|
||||
return &response, nil
|
||||
}
|
||||
|
||||
func (c *Client) Push(ctx context.Context, release Release) ([]string, error) {
|
||||
status, res, err := c.postBody(ctx, "release/push", release)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "could not push release to sonarr")
|
||||
}
|
||||
|
||||
c.Log.Printf("sonarr release/push status: (%v) response: %v\n", status, string(res))
|
||||
|
||||
if status == http.StatusBadRequest {
|
||||
badRequestResponses := make([]*BadRequestResponse, 0)
|
||||
|
||||
if err = json.Unmarshal(res, &badRequestResponses); err != nil {
|
||||
return nil, errors.Wrap(err, "could not unmarshal data")
|
||||
}
|
||||
|
||||
rejections := []string{}
|
||||
for _, response := range badRequestResponses {
|
||||
rejections = append(rejections, response.String())
|
||||
}
|
||||
|
||||
return rejections, nil
|
||||
}
|
||||
|
||||
pushResponse := make([]PushResponse, 0)
|
||||
if err = json.Unmarshal(res, &pushResponse); err != nil {
|
||||
return nil, errors.Wrap(err, "could not unmarshal data")
|
||||
}
|
||||
|
||||
// log and return if rejected
|
||||
if pushResponse[0].Rejected {
|
||||
rejections := strings.Join(pushResponse[0].Rejections, ", ")
|
||||
|
||||
c.Log.Printf("sonarr release/push rejected %v reasons: %q\n", release.Title, rejections)
|
||||
return pushResponse[0].Rejections, nil
|
||||
}
|
||||
|
||||
// successful push
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
func (c *Client) GetAllSeries(ctx context.Context) ([]Series, error) {
|
||||
return c.GetSeries(ctx, 0)
|
||||
}
|
||||
|
||||
func (c *Client) GetSeries(ctx context.Context, tvdbID int64) ([]Series, error) {
|
||||
params := make(url.Values)
|
||||
if tvdbID != 0 {
|
||||
params.Set("tvdbId", strconv.FormatInt(tvdbID, 10))
|
||||
}
|
||||
|
||||
data := make([]Series, 0)
|
||||
err := c.getJSON(ctx, "series", params, &data)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "could not get tags")
|
||||
}
|
||||
|
||||
return data, nil
|
||||
}
|
||||
|
||||
func (c *Client) GetTags(ctx context.Context) ([]*arr.Tag, error) {
|
||||
data := make([]*arr.Tag, 0)
|
||||
err := c.getJSON(ctx, "tag", nil, &data)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "could not get tags")
|
||||
}
|
||||
|
||||
return data, nil
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue