package oplcli

import (
	"fmt"
	"io"
	"log/slog"
	"net/http"
	"os"
	"path"
)

type artUrls struct {
	cover       string
	cover2      string
	background  string
	disc        string
	spine       string
	logo        string
	screenshot  string
	screenshot2 string
}

func getArtUrlForGameFromGithub(gameID string) artUrls {
	const baseURL = "https://raw.githubusercontent.com/Luden02/psx-ps2-opl-art-database/refs/heads/main/PS2"

	genURL := func(kind string) string {
		return fmt.Sprintf("%s/%s/%s_%s.png", baseURL, gameID, gameID, kind)
	}

	return artUrls{
		cover:       genURL("COV"),
		cover2:      genURL("COV2"),
		background:  genURL("BG_00"),
		disc:        genURL("ICO"),
		spine:       genURL("LAB"),
		logo:        genURL("LGO"),
		screenshot:  genURL("SCR_00"),
		screenshot2: genURL("SCR_01"),
	}
}

func downloadArtForGame(gameID string, targetDir string) {
	urls := getArtUrlForGameFromGithub(gameID)

	totalDownloads := 0

	dl := func(url, kind string) {
		file := path.Join(targetDir, fmt.Sprintf("%s_%s.png", gameID, kind))
		found, err := downloadFile(url, file)
		if err != nil {
			slog.Warn("Failed to download art file", "gameID", gameID, "url", url)
		}
		if found {
			slog.Debug("Downloaded art file", "gameID", gameID, "kind", kind, "target", file)
			totalDownloads++
		}
	}

	dl(urls.cover, "COV")
	dl(urls.cover2, "COV2")
	dl(urls.disc, "ICO")
	dl(urls.background, "BG")
	dl(urls.screenshot, "SCR")
	dl(urls.screenshot2, "SRC2")
	dl(urls.spine, "LAB")

	if totalDownloads == 0 {
		slog.Error("No art files found", "gameID", gameID)
	} else {
		slog.Info("Finished downloading art files", "gameID", gameID, "total", totalDownloads)
	}
}

func downloadFile(url, target string) (bool, error) {
	out, err := os.Create(target)
	if err != nil {
		return false, err
	}
	defer out.Close()

	resp, err := http.Get(url)
	if err != nil {
		return false, err
	}
	defer resp.Body.Close()

	if resp.StatusCode == 404 {
		return false, nil
	}

	if resp.StatusCode >= 400 {
		return false, fmt.Errorf("status code: %s", resp.Status)
	}

	n, err := io.Copy(out, resp.Body)
	if err != nil {
		return false, err
	}

	if n == 0 {
		return false, fmt.Errorf("Got empty file")
	}

	return true, nil
}