package arr

import (
	"context"
	"fmt"

	"bellhop/internal/config"
)

// Sonarr drives a Sonarr v3/v4 instance.
type Sonarr struct {
	h                *httpClient
	qualityProfileID int
	rootFolder       string
}

func NewSonarr(cfg *config.ArrConfig) *Sonarr {
	return &Sonarr{
		h:                newHTTP(cfg.URL, cfg.APIKey),
		qualityProfileID: cfg.QualityProfileID,
		rootFolder:       cfg.RootFolder,
	}
}

func (s *Sonarr) Search(ctx context.Context, term string) ([]Result, error) {
	body, err := s.h.get(ctx, "/api/v3/series/lookup", map[string]string{"term": term})
	if err != nil {
		return nil, err
	}
	return parseLookup(body, "title")
}

func (s *Sonarr) Add(ctx context.Context, res Result) error {
	if len(res.raw) == 0 {
		return fmt.Errorf("sonarr: empty result")
	}
	body, err := buildAddBody(res.raw, map[string]any{
		"qualityProfileId": s.qualityProfileID,
		"rootFolderPath":   s.rootFolder,
		"monitored":        true,
		"seasonFolder":     true,
		"addOptions": map[string]any{
			"monitor":                      "all",
			"searchForMissingEpisodes":     true,
			"searchForCutoffUnmetEpisodes": false,
		},
	})
	if err != nil {
		return err
	}
	_, err = s.h.post(ctx, "/api/v3/series", body)
	return err
}

var _ Client = (*Sonarr)(nil)