Anime.go

package shoboi

import (
	"errors"
	"net/http"
	"regexp"
	"sort"
	"strconv"
	"strings"
)

var episodeNameRegex = regexp.MustCompile(`\*(\d+)\*(.*)`)

// Anime ...
type Anime struct {
	ID            string `xml:"TID" json:"TID"`
	CategoryID    string `xml:"Cat" json:"Cat"`
	Comment       string `xml:"Comment" json:"Comment"`
	FirstChannel  string `xml:"FirstCh" json:"FirstCh"`
	FirstEndMonth string `xml:"FirstEndMonth" json:"FirstEndMonth"`
	FirstEndYear  string `xml:"FirstEndYear" json:"FirstEndYear"`
	FirstMonth    string `xml:"FirstMonth" json:"FirstMonth"`
	FirstYear     string `xml:"FirstYear" json:"FirstYear"`
	Keywords      string `xml:"Keywords" json:"Keywords"`
	ShortTitle    string `xml:"ShortTitle" json:"ShortTitle"`
	SubTitles     string `xml:"SubTitles" json:"SubTitles"`
	TitleJapanese string `xml:"Title" json:"Title"`
	TitleEnglish  string `xml:"TitleEN" json:"TitleEN"`
	TitleHiragana string `xml:"TitleYomi" json:"TitleYomi"`
	TitleFlag     string `xml:"TitleFlag" json:"TitleFlag"`
	UpdatedAt     string `xml:"LastUpdate" json:"LastUpdate"`
	UserPoint     string `xml:"UserPoint" json:"UserPoint"`
	UserPointRank string `xml:"UserPointRank" json:"UserPointRank"`
}

// GetAnime ...
func GetAnime(tid string) (*Anime, error) {
	titleFull := &TitleFull{}
	resp, err := get("http://cal.syoboi.jp/json.php?Req=TitleFull&TID=" + tid)

	if resp.StatusCode() != http.StatusOK {
		return nil, errors.New("Invalid status code: " + strconv.Itoa(resp.StatusCode()))
	}

	if err != nil {
		return nil, err
	}

	err = resp.Unmarshal(titleFull)

	if err != nil {
		return nil, err
	}

	if titleFull.Titles == nil || titleFull.Titles[tid] == nil {
		return nil, errors.New("Invalid data: Titles is nil")
	}

	return titleFull.Titles[tid], nil
}

// Episodes ...
func (anime *Anime) Episodes() []*Episode {
	episodes := anime.NamedEpisodes()
	episodeNumber := 1
	index := 0

	if len(episodes) > 0 {
		episodeNumber = episodes[0].Number
	}

	for {
		airingDate := GetAiringDate(anime.ID, episodeNumber)

		if index >= len(episodes) {
			if airingDate == nil {
				break
			}

			episodes = append(episodes, &Episode{
				anime:  anime,
				Number: episodeNumber,
			})
		}

		episodes[index].AiringDate = airingDate

		episodeNumber++
		index++
	}

	sort.Slice(episodes, func(i, j int) bool {
		return episodes[i].Number < episodes[j].Number
	})

	return episodes
}

// NamedEpisodes ...
func (anime *Anime) NamedEpisodes() []*Episode {
	if anime.SubTitles == "" {
		return nil
	}

	episodes := []*Episode{}

	episodeNames := strings.Split(anime.SubTitles, "\r\n")
	for _, episode := range episodeNames {
		episode = strings.TrimSpace(episode)
		matches := episodeNameRegex.FindStringSubmatch(episode)

		if len(matches) > 2 {
			episodeNumber, _ := strconv.Atoi(matches[1])

			episodes = append(episodes, &Episode{
				Number:        episodeNumber,
				TitleJapanese: matches[2],
				anime:         anime,
			})
		}
	}

	return episodes
}