84 lines
1.7 KiB
Go
84 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
_ "embed"
|
|
"encoding/json"
|
|
"go/format"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"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, ®istries); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
generateRegistry(registries["minecraft:block_entity_type"], "BlockEntityType", "blockentitytype")
|
|
generateRegistry(registries["minecraft:entity_type"], "EntityType", "entitytype")
|
|
}
|
|
|
|
func generateRegistry(r registry, typeName, packageName string) {
|
|
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: packageName,
|
|
Default: r.Default,
|
|
Entries: entries,
|
|
TypeName: typeName,
|
|
})
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
formattedSource, err := format.Source(buff.Bytes())
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = os.WriteFile(filepath.Join(packageName, packageName+".go"), formattedSource, 0o666)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|