Files
go-mc/data/registry/generate.go
2022-11-26 20:37:57 +08:00

79 lines
1.5 KiB
Go

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)
}
}