arrman/radarr/client.go

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
}