show_library_trakt.go

package minutes

import (
	"encoding/json"
	"fmt"
	"strconv"

	trakt "github.com/42minutes/go-trakt"
)

// TraktLibrary is a read-only Trakt.tv library
type TraktLibrary struct {
	client *trakt.Client
}

// NewTraktLibrary returns a TraktLibrary
func NewTraktLibrary(client *trakt.Client) *TraktLibrary {
	return &TraktLibrary{
		client: client,
	}
}

// GetShow returns a Show
// or errors with ErrNotFound, or ErrInternalServer
func (l *TraktLibrary) GetShow(showID string) (*Show, error) {
	// log.Infof("> Get show 'trakt:%s'", showID)
	id, err := strconv.Atoi(showID)
	if err != nil {
		return nil, ErrInternalServer
	}
	show, result := l.client.Shows().One(id)
	if result.Err != nil {
		return nil, ErrInternalServer
	}
	// TODO: proper way to convert between structs
	sh := Show{}
	sh.ID = fmt.Sprintf("%d", show.IDs.Trakt)
	data, err := json.Marshal(show)
	if err != nil {
		return nil, ErrInternalServer
	}
	json.Unmarshal(data, &sh)

	// log.Infof(">> Got show 'trakt:%s' as '%s'", showID, sh.Title)

	return &sh, nil
}

// GetSeasons returns all Seasons for a show
// or errors with ErrNotFound, or ErrInternalServer
func (l *TraktLibrary) GetSeasons(showID string) ([]*Season, error) {
	id, err := strconv.Atoi(showID)
	if err != nil {
		return nil, ErrInternalServer
	}

	seasons, result := l.client.Seasons().All(id)
	if result.Err != nil {
		return nil, ErrInternalServer
	}

	seasRs := []*Season{}
	for _, season := range seasons {
		se := Season{}
		se.ID = fmt.Sprintf("%d", season.IDs.Trakt)
		se.ShowID = showID
		data, err := json.Marshal(season)
		if err != nil {
			return nil, ErrInternalServer
		}
		json.Unmarshal(data, &se)
		seasRs = append(seasRs, &se)
	}
	return seasRs, nil
}

// GetSeason returns a Season given a Show's ID and a Season number
// or errors with ErrNotFound, ErrMissingShow, or ErrInternalServer
// TODO not working and returs ErrNotImplemented
func (l *TraktLibrary) GetSeason(showID string, seasonNumber int) (*Season, error) {
	ses, err := l.GetSeasons(showID)
	if err != nil {
		return nil, ErrInternalServer
	}

	for _, se := range ses {
		if se.Number == seasonNumber {
			return se, nil
		}
	}

	return nil, ErrNotFound
}

// GetEpisode returns an Episode  given a Show's ID a Season number
// and Episode's number
// or errors with ErrNotFound, ErrMissingShow, or ErrInternalServer
func (l *TraktLibrary) GetEpisode(showID string, seasonNumber, episodeNumber int) (*Episode, error) {
	id, err := strconv.Atoi(showID)
	if err != nil {
		return nil, ErrInternalServer
	}

	episode, result := l.client.Episodes().OneBySeasonByNumber(
		id, seasonNumber, episodeNumber,
	)
	if result.Err != nil {
		return nil, ErrInternalServer
	}

	ep := Episode{}
	ep.ID = fmt.Sprintf("%d", episode.IDs.Trakt)
	ep.ShowID = showID
	data, err := json.Marshal(episode)
	if err != nil {
		return nil, ErrInternalServer
	}
	json.Unmarshal(data, &ep)
	return &ep, nil
}

// GetEpisodes returns all Shows for a show and season number
// or errors with ErrNotFound, or ErrInternalServer
func (l *TraktLibrary) GetEpisodes(showID string, seasonNumber int) ([]*Episode, error) {
	id, err := strconv.Atoi(showID)
	if err != nil {
		return nil, ErrInternalServer
	}

	episodes, result := l.client.Episodes().AllBySeason(id, seasonNumber)
	if result.Err != nil {
		return nil, ErrInternalServer
	}
	epsRs := []*Episode{}
	for _, episode := range episodes {
		ep := Episode{}
		ep.ShowID = showID
		ep.ID = fmt.Sprintf("%d", episode.IDs.Trakt)
		data, err := json.Marshal(episode)
		if err != nil {
			return nil, ErrInternalServer
		}
		json.Unmarshal(data, &ep)
		epsRs = append(epsRs, &ep)
	}
	return epsRs, nil
}

// QueryShowsByTitle returns all Shows that match a partial title ordered
// by their probability
// or errors with ErrInternalServer
func (l *TraktLibrary) QueryShowsByTitle(title string) ([]*Show, error) {
	shows, result := l.client.Shows().Search(title)
	if result.Err != nil {
		return nil, ErrInternalServer
	}

	shRs := []*Show{}
	for _, show := range shows {
		sh := &Show{}
		sh.ID = fmt.Sprintf("%d", show.Show.IDs.Trakt)
		data, err := json.Marshal(show.Show)
		if err != nil {
			return nil, ErrInternalServer
		}
		err = json.Unmarshal(data, &sh)
		shRs = append(shRs, sh)
	}
	return shRs, nil
}