package main

import (
	"fmt"
	"strconv"
	"strings"

	"github.com/animenotifier/arn"
	"github.com/fatih/color"
)

func main() {
	color.Yellow("Refreshing episode information for each anime.")

	if InvokeShellArgs() {
		return
	}

	highPriority := []*arn.Anime{}
	mediumPriority := []*arn.Anime{}
	lowPriority := []*arn.Anime{}

	for anime := range arn.MustStreamAnime() {
		if anime.GetMapping("shoboi/anime") == "" {
			continue
		}

		switch anime.Status {
		case "current":
			highPriority = append(highPriority, anime)
		case "upcoming":
			mediumPriority = append(mediumPriority, anime)
		default:
			lowPriority = append(lowPriority, anime)
		}
	}

	color.Cyan("High priority queue (%d):", len(highPriority))
	refreshQueue(highPriority)

	color.Cyan("Medium priority queue (%d):", len(mediumPriority))
	refreshQueue(mediumPriority)

	color.Cyan("Low priority queue (%d):", len(lowPriority))
	refreshQueue(lowPriority)

	color.Green("Finished.")
}

func refreshQueue(queue []*arn.Anime) {
	for _, anime := range queue {
		refresh(anime)
	}
}

func refresh(anime *arn.Anime) {
	fmt.Println(anime.ID, "|", anime.Title.Canonical, "|", anime.GetMapping("shoboi/anime"))

	episodeCount := len(anime.Episodes().Items)
	availableEpisodeCount := anime.Episodes().AvailableCount()

	err := anime.RefreshEpisodes()

	if err != nil {
		if strings.Contains(err.Error(), "missing a Shoboi ID") {
			return
		}

		color.Red(err.Error())
	} else {
		fmt.Println("+"+strconv.Itoa(len(anime.Episodes().Items)-episodeCount)+" airing", "|", "+"+strconv.Itoa(anime.Episodes().AvailableCount()-availableEpisodeCount)+" available")
	}
}