package main

import (
	"context"
	"database/sql"
	_ "embed"
	"flag"
	"fmt"
	"git.infra.nkode.tech/dkelly/nkode-core/repository"
	"git.infra.nkode.tech/dkelly/nkode-core/sqlite"
	_ "github.com/mattn/go-sqlite3"
	"log"
	"os"
	"path/filepath"
)

func main() {
	sqliteSchema, err := sqlite.FS.ReadFile("schema.sql")
	if err != nil {
		log.Fatal(err)
	}
	commandLine := flag.NewFlagSet(os.Args[0], flag.ExitOnError)
	dbPath := commandLine.String("db-path", "", "Path to the database")
	svgPath := commandLine.String("svg-path", "", "Path to the SVG directory")
	if err = commandLine.Parse(os.Args[1:]); err != nil {
		log.Fatalf("Failed to parse flags: %v", err)
	}
	if err = MakeTables(*dbPath, string(sqliteSchema)); err != nil {
		log.Fatal(err)
	}
	ctx := context.Background()
	sqliteRepo, err := repository.NewSqliteRepository(ctx, *dbPath)
	sqliteRepo.Start()
	defer func(sqliteRepo *repository.SqliteRepository) {
		if err := sqliteRepo.Stop(); err != nil {
			log.Fatal(err)
		}
	}(sqliteRepo)

	FlaticonToSqlite(sqliteRepo, *svgPath)
	log.Println(fmt.Sprintf("Successfully added all SVGs in %s to the database at %s\n", *svgPath, *dbPath))
}

func FlaticonToSqlite(repo *repository.SqliteRepository, svgDir string) {
	files, err := os.ReadDir(svgDir)
	if err != nil {
		log.Fatal(err)
	}

	for _, file := range files {
		if file.IsDir() || filepath.Ext(file.Name()) != ".svg" {
			continue
		}
		filePath := filepath.Join(svgDir, file.Name())

		content, err := os.ReadFile(filePath)
		if err != nil {
			log.Println("Error reading file:", filePath, err)
			continue
		}

		if err = repo.AddSvg(string(content)); err != nil {
			log.Fatal(err)
		}
	}

}

func MakeTables(dbPath string, schema string) error {
	if _, err := os.Stat(dbPath); os.IsNotExist(err) {
		if err = os.MkdirAll(filepath.Dir(dbPath), 0755); err != nil {
			return err
		}
		if _, err = os.Create(dbPath); err != nil {
			return err
		}
	}
	db, err := sql.Open("sqlite3", dbPath)
	if err != nil {
		return err
	}
	if _, err = db.Exec(schema); err != nil {
		return err
	}
	return db.Close()
}