initial code (#6)

* initial code commit

sonarr/api.go

@@ -0,0 +1,149 @@
+package sonarr
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"github.com/l3uddz/nabarr/media"
+	"github.com/l3uddz/nabarr/util"
+	"github.com/lucperkins/rek"
+	"net/url"
+	"strconv"
+	"strings"
+)
+
+var (
+	ErrItemNotFound = errors.New("not found")
+)
+
+func (c *Client) getSystemStatus() (*systemStatus, error) {
+	// send request
+	resp, err := rek.Get(util.JoinURL(c.apiURL, "/system/status"), rek.Headers(c.apiHeaders),
+		rek.Timeout(c.apiTimeout))
+	if err != nil {
+		return nil, fmt.Errorf("request system status: %w", err)
+	}
+	defer resp.Body().Close()
+
+	// validate response
+	if resp.StatusCode() != 200 {
+		return nil, fmt.Errorf("validate system status response: %s", resp.Status())
+	}
+
+	// decode response
+	b := new(systemStatus)
+	if err := json.NewDecoder(resp.Body()).Decode(b); err != nil {
+		return nil, fmt.Errorf("decode system status response: %w", err)
+	}
+
+	return b, nil
+}
+
+func (c *Client) getQualityProfileId(profileName string) (int, error) {
+	// send request
+	resp, err := rek.Get(util.JoinURL(c.apiURL, "/profile"), rek.Headers(c.apiHeaders),
+		rek.Timeout(c.apiTimeout))
+	if err != nil {
+		return 0, fmt.Errorf("request quality profiles: %w", err)
+	}
+	defer resp.Body().Close()
+
+	// validate response
+	if resp.StatusCode() != 200 {
+		return 0, fmt.Errorf("validate quality profiles response: %s", resp.Status())
+	}
+
+	// decode response
+	b := new([]qualityProfile)
+	if err := json.NewDecoder(resp.Body()).Decode(b); err != nil {
+		return 0, fmt.Errorf("decode quality profiles response: %w", err)
+	}
+
+	// find quality profile
+	for _, profile := range *b {
+		if strings.EqualFold(profile.Name, profileName) {
+			return profile.Id, nil
+		}
+	}
+
+	return 0, errors.New("quality profile not found")
+}
+
+func (c *Client) lookupMediaItem(item *media.Item) (*lookupRequest, error) {
+	// prepare request
+	reqUrl, err := util.URLWithQuery(util.JoinURL(c.apiURL, "/series/lookup"),
+		url.Values{"term": []string{fmt.Sprintf("tvdb:%s", item.TvdbId)}})
+	if err != nil {
+		return nil, fmt.Errorf("generate series lookup request url: %w", err)
+	}
+
+	// send request
+	resp, err := rek.Get(reqUrl, rek.Headers(c.apiHeaders), rek.Timeout(c.apiTimeout))
+	if err != nil {
+		return nil, fmt.Errorf("request series lookup: %w", err)
+	}
+	defer resp.Body().Close()
+
+	// validate response
+	if resp.StatusCode() != 200 {
+		return nil, fmt.Errorf("validate series lookup response: %s", resp.Status())
+	}
+
+	// decode response
+	b := new([]lookupRequest)
+	if err := json.NewDecoder(resp.Body()).Decode(b); err != nil {
+		return nil, fmt.Errorf("decode series lookup response: %w", err)
+	}
+
+	// find series
+	for _, s := range *b {
+		if strconv.Itoa(s.TvdbId) == item.TvdbId {
+			return &s, nil
+		}
+	}
+
+	return nil, fmt.Errorf("series lookup tvdbId: %v: %w", item.TvdbId, ErrItemNotFound)
+}
+
+func (c *Client) AddMediaItem(item *media.Item) error {
+	// prepare request
+	tvdbId, err := strconv.Atoi(item.TvdbId)
+	if err != nil {
+		return fmt.Errorf("converting tvdb id to int: %q", item.TvdbId)
+	}
+
+	req := addRequest{
+		Title:            item.Title,
+		TitleSlug:        item.Slug,
+		Year:             item.Year,
+		QualityProfileId: c.qualityProfileId,
+		Images:           []string{},
+		Tags:             []string{},
+		Monitored:        true,
+		RootFolderPath:   c.rootFolder,
+		AddOptions: addOptions{
+			SearchForMissingEpisodes:   true,
+			IgnoreEpisodesWithFiles:    false,
+			IgnoreEpisodesWithoutFiles: false,
+		},
+		Seasons:      []string{},
+		SeriesType:   "standard",
+		SeasonFolder: true,
+		TvdbId:       tvdbId,
+	}
+
+	// send request
+	resp, err := rek.Post(util.JoinURL(c.apiURL, "/series"), rek.Headers(c.apiHeaders), rek.Json(req),
+		rek.Timeout(c.apiTimeout))
+	if err != nil {
+		return fmt.Errorf("request add series: %w", err)
+	}
+	defer resp.Body().Close()
+
+	// validate response
+	if resp.StatusCode() != 200 && resp.StatusCode() != 201 {
+		return fmt.Errorf("validate add series response: %s", resp.Status())
+	}
+
+	return nil
+}