manualops

type ActivityTracker interface {
	Start(action, detail string, source activity.ActivitySource) string
	End(id string)
	Fail(id string)
	Progress(id string, current, total int, detail string)
}

type BGTracker interface {
	Add(delta int)
	Done()
}

type DownloadAccepted struct {
	ActivityID string `json:"activity_id"`
	Status     string `json:"status"`
}

type DownloadRequest struct {
	Provider    api.ProviderID `json:"provider"`
	SubtitleID  string         `json:"subtitle_id"`
	Language    string         `json:"language"`
	ReleaseName string         `json:"release_name,omitempty"`
	MediaType   api.MediaType  `json:"media_type,omitempty"`

	Score      int  `json:"score,omitempty"`
	Season     int  `json:"season,omitempty"`
	Episode    int  `json:"episode,omitempty"`
	ArrID      int  `json:"media_id,omitempty"`
	TopPick    bool `json:"top_pick,omitempty"`
	HearingImp bool `json:"hearing_impaired,omitempty"`
	Forced     bool `json:"forced,omitempty"`
	// contains filtered or unexported fields
}

type DownloadStore interface {
	SearchStore
	NextManualNumber(ctx context.Context, mediaType api.MediaType, mediaID, language string, variant api.Variant) int
	UpsertSubtitleFile(ctx context.Context, mediaType api.MediaType, mediaID string, sf *api.SubtitleFile) error
	SetSyncOffset(ctx context.Context, path string, offsetMs int64) error
	SaveDownload(ctx context.Context, rec *api.DownloadRecord) error
}

type EventPublisher interface {
	PublishNotify(level events.NotifyLevel, text string)
	PublishCoverageUpdate(mediaType api.MediaType, mediaID, language, source string)
}

type Handler struct {
	// contains filtered or unexported fields
}

type HandlerDeps struct {
	DBFunc     func() DownloadStore
	Activity   ActivityTracker
	Alerts     activity.WarnRecorder
	Events     EventPublisher
	StateFunc  func() *LiveState
	BGTracker  BGTracker
	ServerCtx  func() context.Context
	Resolve    *resolve.Resolver
	DecodeJSON func(w http.ResponseWriter, r *http.Request, v any, maxSize int64) bool
}

type LiveState struct {
	Cfg       api.ConfigProvider
	Engine    api.SearchEngine
	Scorer    api.Scorer
	Sonarr    ManualSonarrClient
	Radarr    ManualRadarrClient
	SonarrLib ResolveSonarrClient
	RadarrLib ResolveRadarrClient
	Providers []api.Provider
}

type ManualRadarrClient interface {
	GetMovieByID(ctx context.Context, movieID int) (arrapi.Movie, error)
	RescanMovie(ctx context.Context, movieID int) error
}

type ManualSearchResponse struct {
	Results []SearchResult `json:"results"`
}

type ManualSonarrClient interface {
	GetSeriesByID(ctx context.Context, seriesID int) (arrapi.Series, error)
	RescanSeries(ctx context.Context, seriesID int) error
}

type ResolveCandidate struct {
	MediaType api.MediaType `json:"media_type"`
	Title     string        `json:"title"`
	MediaID   int           `json:"media_id"`
	Year      int           `json:"year,omitempty"`
}

type ResolveQueryParams struct {
	Season  *int
	Episode *int
	Title   string
	Imdb    string
	Type    string // "", "series", or "movie"
	Tmdb    int
}

type ResolveRadarrClient interface {
	GetMovies(ctx context.Context) ([]arrapi.Movie, error)
}

type ResolveResponse struct {
	Items      []ResolvedItem     `json:"items,omitempty"`
	Candidates []ResolveCandidate `json:"candidates,omitempty"`
	Resolved   bool               `json:"resolved"`
}

type ResolveSearchIDs struct {
	Imdb string `json:"imdb,omitempty"`
	Tvdb int    `json:"tvdb,omitempty"`
	Tmdb int    `json:"tmdb,omitempty"`
}

type ResolveSonarrClient interface {
	GetSeries(ctx context.Context) ([]arrapi.Series, error)
	GetEpisodes(ctx context.Context, seriesID int) ([]arrapi.Episode, error)
}

type ResolvedItem struct {
	MediaType api.MediaType    `json:"media_type"`
	Title     string           `json:"title"`
	SearchIDs ResolveSearchIDs `json:"search_ids"`
	MediaID   int              `json:"media_id"`
	Year      int              `json:"year,omitempty"`
	Season    int              `json:"season,omitempty"`
	Episode   int              `json:"episode,omitempty"`
}

type SearchDeps struct {
	DB       SearchStore
	Activity ActivityTracker
	Alerts   activity.WarnRecorder
	Events   EventPublisher
}

type SearchResult struct {
	Matches     map[string]int `json:"matches,omitempty"`
	Provider    api.ProviderID `json:"provider"`
	Language    string         `json:"language"`
	ReleaseName string         `json:"release_name"`
	MatchedBy   string         `json:"matched_by"`
	SubtitleID  string         `json:"subtitle_id"`
	// Tier is the score's quality-tier label, computed server-side by the
	// scorer (same table as /api/score): the remote CLI renders it and has
	// no scorer of its own.
	Tier       api.ScoreTier `json:"tier"`
	Score      int           `json:"score"`
	HearingImp bool          `json:"hearing_impaired"`
	Forced     bool          `json:"forced"`
	OnDisk     bool          `json:"on_disk"`
}

type SearchStore interface {
	DownloadedRefs(ctx context.Context, mediaType api.MediaType, mediaID, language string) ([]api.DownloadedRef, error)
	ClearManualLock(ctx context.Context, mediaType api.MediaType, mediaID, language string, variant api.Variant) error
}