package main import ( "fmt" "strings" "github.com/animenotifier/arn" "github.com/fatih/color" ) func main() { color.Yellow("Refreshing episode information for each anime.") defer arn.Node.Close() if InvokeShellArgs() { return } highPriority := []*arn.Anime{} mediumPriority := []*arn.Anime{} lowPriority := []*arn.Anime{} for anime := range arn.StreamAnime() { if anime.GetMapping("shoboi/anime") == "" { continue } // The rest gets sorted by airing status 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 { faint := color.New(color.Faint).SprintFunc() episodes := anime.Episodes() fmt.Println(faint(episodes)) fmt.Printf("+%d episodes | +%d available (%d total)\n", len(episodes.Items)-episodeCount, episodes.AvailableCount()-availableEpisodeCount, len(episodes.Items)) println() } }