package main

import (
	"flag"
	"fmt"
	"io/ioutil"
	"os"
	"path"

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

func main() {
	color.Yellow("Updating anime thumbnails")

	defer arn.Node.Close()
	defer color.Green("Finished.")

	// Parse flags
	var animeID string
	flag.StringVar(&animeID, "id", "", "ID of the anime that you want to refresh")
	flag.Parse()

	// Refresh 1 anime in case ID was specified
	if animeID != "" {
		anime, _ := arn.GetAnime(animeID)

		if anime != nil {
			sync(anime)
		}

		return
	}

	// Otherwise refresh all anime
	for anime := range arn.StreamAnime() {
		sync(anime)
	}
}

// sync refreshes the image of the given anime.
func sync(anime *arn.Anime) {
	base := path.Join(arn.Root, "/images/anime/original/", anime.ID)

	if _, err := os.Stat(base + ".png"); err == nil {
		update(anime, base+".png")
		return
	}

	if _, err := os.Stat(base + ".jpg"); err == nil {
		update(anime, base+".jpg")
		return
	}

	if _, err := os.Stat(base + ".jpeg"); err == nil {
		update(anime, base+".jpg")
		return
	}

	if _, err := os.Stat(base + ".gif"); err == nil {
		update(anime, base+".gif")
		return
	}

	if _, err := os.Stat(base + ".webp"); err == nil {
		update(anime, base+".webp")
		return
	}
}

// update expects a file to load as image for the anime and updates it.
func update(anime *arn.Anime, filePath string) {
	fmt.Println(anime.ID, anime)

	data, err := ioutil.ReadFile(filePath)

	if err != nil {
		return
	}

	anime.SetImageBytes(data)
	anime.Save()
}