elastictv

type Credits struct {
	Actor    []string `json:"actor,omitempty"`
	Director []string `json:"director,omitempty"`
	Other    []string `json:"other,omitempty"`
}

type Description struct {
	Source string `json:"source,omitempty"`
	Text   string `json:"text,omitempty"`
}

type ElasticTV struct {
	Client    *elasticsearch.Client
	Providers []SearchableProvider
	Index     index
}

type Episode struct {
	AirDate     string      `json:"air_date,omitempty"`
	Description Description `json:"description"`
	IDs         IDs         `json:"ids"`
	Image       string      `json:"image,omitempty"`
	Rating      *Rating     `json:"rating,omitempty"`
	Timestamp   string      `json:"@timestamp"`
	Title       string      `json:"title"`
	TVShowID    int         `json:"tvshow_id,omitempty"`
	Episode     uint16      `json:"episode"`
	Season      uint16      `json:"season"`
}

type IDs struct {
	IMDb string `json:"imdb,omitempty"`
	TMDb int    `json:"tmdb,omitempty"`
}

type LookupCommonParams struct {
	Title    []string
	Director []string
	Actor    []string
	Other    []string
	Country  []string
	Genre    []string
	IMDBID   string
}

type LookupEpisodeParams struct {
	LookupCommonParams
	EpisodeTitle string
	Season       uint16
	Episode      uint16
}

type LookupMovieParams struct {
	LookupCommonParams
	Year uint16
}

type Query struct {
	Query struct {
		Bool struct {
			Should []interface{} `json:"should,omitempty"`
			Must   []interface{} `json:"must,omitempty"`
		} `json:"bool"`
	} `json:"query"`
}

type Rating struct {
	Source string  `json:"source,omitempty"`
	Value  float32 `json:"value,omitempty"`
}

type SearchEpisodesParams struct {
	IDs     IDs
	Episode uint16
	Season  uint16
}

type SearchTitlesParams struct {
	Query     string `json:"query,omitempty"`
	Attribute string `json:"attribute,omitempty"`
	Year      uint16 `json:"year,omitempty"`
	Type      string `json:"type,omitempty"`
	Timestamp string `json:"@timestamp,omitempty"`
}

type SearchableProvider interface {
	Name() string
	Init(estv *ElasticTV) (SearchableProvider, error)
	SearchTitles(SearchTitlesParams) error
	SearchEpisodes(*Title, uint16, uint16) error
}

type Title struct {
	Alias       []string    `json:"alias,omitempty"`
	Country     []string    `json:"country,omitempty"`
	Credits     Credits     `json:"credits"`
	Description Description `json:"description"`
	Genre       []string    `json:"genre,omitempty"`
	IDs         IDs         `json:"ids"`
	Image       string      `json:"image,omitempty"`
	Language    string      `json:"language,omitempty"`
	Rating      *Rating     `json:"rating,omitempty"`
	Timestamp   string      `json:"@timestamp"`
	Title       string      `json:"title"`
	Type        string      `json:"type"`
	Year        uint16      `json:"year,omitempty"`
	Tagline     string      `json:"tagline,omitempty"`
}