package main

import (
	"bytes"
	_ "embed"
	"encoding/json"
	"go/format"
	"log"
	"os"
	"text/template"

	"github.com/Tnze/go-mc/internal/generateutils"
)

type registry struct {
	Default string `json:"default"`
	Entries map[string]struct {
		ProtocolID int `json:"protocol_id"`
	}
	ProtocolID int `json:"protocol_id"`
}

//go:embed registries.json
var registersJson []byte

//go:embed template.go.tmpl
var tempSource string

type tempData struct {
	PackageName string
	Default     string
	Entries     []string
	TypeName    string
}

var temp = template.Must(template.
	New("block_template").
	Funcs(template.FuncMap{
		"UpperTheFirst": generateutils.UpperTheFirst,
		"ToGoTypeName":  generateutils.ToGoTypeName,
		"Generator":     func() string { return "data/registry/generate.go" },
	}).
	Parse(tempSource),
)

func main() {
	var registries map[string]registry
	if err := json.Unmarshal(registersJson, &registries); err != nil {
		log.Fatal(err)
	}

	r := registries["minecraft:block_entity_type"]
	entries := make([]string, len(r.Entries))
	for name, v := range r.Entries {
		entries[v.ProtocolID] = name
	}

	var buff bytes.Buffer
	err := temp.Execute(&buff, tempData{
		PackageName: "blockentitytype",
		Default:     r.Default,
		Entries:     entries,
		TypeName:    "BlockEntityType",
	})
	if err != nil {
		log.Fatal(err)
	}

	formattedSource, err := format.Source(buff.Bytes())
	if err != nil {
		log.Fatal(err)
	}

	err = os.WriteFile("blockentitytype/blockentitytype.go", formattedSource, 0o666)
	if err != nil {
		log.Fatal(err)
	}
}