package main

import (
	"compress/gzip"
	"errors"
	"os"
	"path"
	"path/filepath"
	"strings"

	"github.com/animenotifier/mal"

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

func main() {
	color.Yellow("Parsing MAL files")

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

	// Invoke via parameters
	if InvokeShellArgs() {
		return
	}

	if objectType == "all" || objectType == "anime" {
		readFiles(path.Join(arn.Root, "jobs", "mal-download", "anime"), readAnimeFile)
	}

	if objectType == "all" || objectType == "character" {
		readFiles(path.Join(arn.Root, "jobs", "mal-download", "character"), readCharacterFile)
	}
}

func readFiles(root string, onFile func(string) error) {
	count := 0

	filepath.Walk(root, func(name string, info os.FileInfo, err error) error {
		if err != nil {
			color.Red(err.Error())
			return err
		}

		if info.IsDir() {
			return nil
		}

		if !strings.HasSuffix(name, ".html.gz") {
			return nil
		}

		count++
		err = onFile(name)

		if err != nil {
			color.Red(err.Error())
		}

		// Always continue traversing the directory
		return nil
	})

	color.Cyan("%d files found", count)
}

func readAnimeFile(name string) error {
	file, err := os.Open(name)

	if err != nil {
		color.Red(err.Error())
		return err
	}

	defer file.Close()

	reader, err := gzip.NewReader(file)

	if err != nil {
		color.Red(err.Error())
		return err
	}

	anime, characters, err := malparser.ParseAnime(reader)

	if err != nil {
		color.Red(err.Error())
		return err
	}

	if anime.ID == "" {
		return errors.New("Empty ID")
	}

	for _, character := range characters {
		obj, err := arn.MAL.Get("Character", character.ID)

		if err != nil {
			arn.MAL.Set("Character", character.ID, character)
			continue
		}

		existing := obj.(*mal.Character)
		modified := false

		if existing.Name != character.Name {
			existing.Name = character.Name
			modified = true
		}

		if existing.Image != character.Image {
			existing.Image = character.Image
			modified = true
		}

		if modified {
			arn.MAL.Set("Character", existing.ID, existing)
		}
	}

	// fmt.Println(anime.ID, anime.Title)
	arn.MAL.Set("Anime", anime.ID, anime)
	return nil
}

func readCharacterFile(name string) error {
	file, err := os.Open(name)

	if err != nil {
		color.Red(err.Error())
		return err
	}

	defer file.Close()

	reader, err := gzip.NewReader(file)

	if err != nil {
		color.Red(err.Error())
		return err
	}

	character, err := malparser.ParseCharacter(reader)

	if err != nil {
		color.Red(err.Error())
		return err
	}

	if character.ID == "" {
		return errors.New("Empty ID")
	}

	// fmt.Println(character.ID, character.Name)
	arn.MAL.Set("Character", character.ID, character)
	return nil
}