142 lines
3.3 KiB
Go
142 lines
3.3 KiB
Go
package sonarr
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/tls"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
)
|
|
|
|
type Client struct {
|
|
baseURL string
|
|
apiKey string
|
|
httpClient *http.Client
|
|
}
|
|
|
|
func NewClient(baseURL, apiKey string, tlsSkipVerify bool) *Client {
|
|
client := &http.Client{}
|
|
if strings.HasPrefix(baseURL, "https") && tlsSkipVerify {
|
|
client.Transport = &http.Transport{
|
|
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
|
}
|
|
}
|
|
return &Client{
|
|
baseURL: strings.TrimRight(baseURL, "/"),
|
|
apiKey: apiKey,
|
|
httpClient: client,
|
|
}
|
|
}
|
|
|
|
func (c *Client) do(method, path string, body interface{}) ([]byte, error) {
|
|
var reqBody io.Reader
|
|
if body != nil {
|
|
data, err := json.Marshal(body)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("marshal request: %w", err)
|
|
}
|
|
reqBody = bytes.NewReader(data)
|
|
}
|
|
|
|
req, err := http.NewRequest(method, c.baseURL+path, reqBody)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("create request: %w", err)
|
|
}
|
|
req.Header.Set("X-Api-Key", c.apiKey)
|
|
if body != nil {
|
|
req.Header.Set("Content-Type", "application/json")
|
|
}
|
|
|
|
resp, err := c.httpClient.Do(req)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("request failed: %w", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
respBody, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("read response: %w", err)
|
|
}
|
|
|
|
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
|
return nil, fmt.Errorf("API error %d: %s", resp.StatusCode, string(respBody))
|
|
}
|
|
|
|
return respBody, nil
|
|
}
|
|
|
|
func (c *Client) List() ([]Series, error) {
|
|
data, err := c.do("GET", "/api/v3/series", nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var series []Series
|
|
if err := json.Unmarshal(data, &series); err != nil {
|
|
return nil, fmt.Errorf("parse response: %w", err)
|
|
}
|
|
return series, nil
|
|
}
|
|
|
|
func (c *Client) Lookup(term string) ([]Series, error) {
|
|
data, err := c.do("GET", "/api/v3/series/lookup?term="+url.QueryEscape(term), nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var series []Series
|
|
if err := json.Unmarshal(data, &series); err != nil {
|
|
return nil, fmt.Errorf("parse response: %w", err)
|
|
}
|
|
return series, nil
|
|
}
|
|
|
|
func (c *Client) Get(id int) (*Series, error) {
|
|
data, err := c.do("GET", fmt.Sprintf("/api/v3/series/%d", id), nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var s Series
|
|
if err := json.Unmarshal(data, &s); err != nil {
|
|
return nil, fmt.Errorf("parse response: %w", err)
|
|
}
|
|
return &s, nil
|
|
}
|
|
|
|
func (c *Client) QualityProfiles() ([]QualityProfile, error) {
|
|
data, err := c.do("GET", "/api/v3/qualityprofile", nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var profiles []QualityProfile
|
|
if err := json.Unmarshal(data, &profiles); err != nil {
|
|
return nil, fmt.Errorf("parse response: %w", err)
|
|
}
|
|
return profiles, nil
|
|
}
|
|
|
|
func (c *Client) RootFolders() ([]RootFolder, error) {
|
|
data, err := c.do("GET", "/api/v3/rootfolder", nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var folders []RootFolder
|
|
if err := json.Unmarshal(data, &folders); err != nil {
|
|
return nil, fmt.Errorf("parse response: %w", err)
|
|
}
|
|
return folders, nil
|
|
}
|
|
|
|
func (c *Client) Add(s Series) (*Series, error) {
|
|
data, err := c.do("POST", "/api/v3/series", s)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var result Series
|
|
if err := json.Unmarshal(data, &result); err != nil {
|
|
return nil, fmt.Errorf("parse response: %w", err)
|
|
}
|
|
return &result, nil
|
|
}
|