Initial release
This commit is contained in:
@@ -0,0 +1,153 @@
|
||||
package radarr
|
||||
|
||||
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() ([]Movie, error) {
|
||||
data, err := c.do("GET", "/api/v3/movie", nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var movies []Movie
|
||||
if err := json.Unmarshal(data, &movies); err != nil {
|
||||
return nil, fmt.Errorf("parse response: %w", err)
|
||||
}
|
||||
return movies, nil
|
||||
}
|
||||
|
||||
func (c *Client) Lookup(term string) ([]Movie, error) {
|
||||
data, err := c.do("GET", "/api/v3/movie/lookup?term="+url.QueryEscape(term), nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var movies []Movie
|
||||
if err := json.Unmarshal(data, &movies); err != nil {
|
||||
return nil, fmt.Errorf("parse response: %w", err)
|
||||
}
|
||||
return movies, nil
|
||||
}
|
||||
|
||||
func (c *Client) LookupIMDB(imdbID string) (*Movie, error) {
|
||||
data, err := c.do("GET", "/api/v3/movie/lookup/imdb?imdbId="+url.QueryEscape(imdbID), nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var movie Movie
|
||||
if err := json.Unmarshal(data, &movie); err != nil {
|
||||
return nil, fmt.Errorf("parse response: %w", err)
|
||||
}
|
||||
return &movie, nil
|
||||
}
|
||||
|
||||
func (c *Client) Get(id int) (*Movie, error) {
|
||||
data, err := c.do("GET", fmt.Sprintf("/api/v3/movie/%d", id), nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var m Movie
|
||||
if err := json.Unmarshal(data, &m); err != nil {
|
||||
return nil, fmt.Errorf("parse response: %w", err)
|
||||
}
|
||||
return &m, 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(m Movie) (*Movie, error) {
|
||||
data, err := c.do("POST", "/api/v3/movie", m)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var result Movie
|
||||
if err := json.Unmarshal(data, &result); err != nil {
|
||||
return nil, fmt.Errorf("parse response: %w", err)
|
||||
}
|
||||
return &result, nil
|
||||
}
|
||||
Reference in New Issue
Block a user