Add tags implement

registry/README.md

@@ -0,0 +1,33 @@
+# Registry System
+
+instance {
+    ResourceLocation {
+        namespace: string
+        path: string
+    }
+
+    ResourceKey<T> {
+        registryName: ResourceLocation
+        location: ResourceLocation
+    }
+
+    registry<T> {
+        getId(T): int
+        byId(int): T
+
+        getKey(T): ResourceLocation
+        getResourceKey(T): ResourceKey<T>
+        get(ResourceKey<T>): T
+        get(ResourceLocation): T
+
+        getTags(TagKey<T>): (TagKey<T>, *T[])[]
+        getTagNames(): TagKey<T>[]
+        resetTags()
+        bindTags((TagKey<T>, *T[])[])
+    }[]
+
+    TagKey<T> {
+        *Registry<T>
+        ResourceLocation
+    }
+}

registry/codec.go

@@ -1,10 +1,12 @@
 package registry
 
 import (
+	"io"
 	"reflect"
 
 	"github.com/Tnze/go-mc/chat"
 	"github.com/Tnze/go-mc/nbt"
+	pk "github.com/Tnze/go-mc/net/packet"
 )
 
 type NetworkCodec struct {
@@ -72,7 +74,12 @@ type Dimension struct {
 	MonsterSpawnBlockLightLimit int32          `nbt:"monster_spawn_block_light_limit"`
 }
 
-func (c *NetworkCodec) Registry(id string) any {
+type RegistryCodec interface {
+	pk.FieldDecoder
+	ReadTagsFrom(r io.Reader) (int64, error)
+}
+
+func (c *NetworkCodec) Registry(id string) RegistryCodec {
 	codecVal := reflect.ValueOf(c).Elem()
 	codecTyp := codecVal.Type()
 	numField := codecVal.NumField()
@@ -82,7 +89,7 @@ func (c *NetworkCodec) Registry(id string) any {
 			continue
 		}
 		if registryID == id {
-			return codecVal.Field(i).Addr().Interface()
+			return codecVal.Field(i).Addr().Interface().(RegistryCodec)
 		}
 	}
 	return nil

registry/network.go

@@ -1,14 +1,15 @@
 package registry
 
 import (
+	"errors"
 	"io"
+	"strconv"
 
 	pk "github.com/Tnze/go-mc/net/packet"
 )
 
 func (reg *Registry[E]) ReadFrom(r io.Reader) (int64, error) {
 	var length pk.VarInt
-
 	n, err := length.ReadFrom(r)
 	if err != nil {
 		return n, err
@@ -33,7 +34,7 @@ func (reg *Registry[E]) ReadFrom(r io.Reader) (int64, error) {
 		}
 
 		if hasData {
-			n3, err = pk.NBTField{V:&data, AllowUnknownFields: true}.ReadFrom(r)
+			n3, err = pk.NBTField{V: &data, AllowUnknownFields: true}.ReadFrom(r)
 			if err != nil {
 				return n + n1 + n2 + n3, err
 			}
@@ -44,3 +45,49 @@ func (reg *Registry[E]) ReadFrom(r io.Reader) (int64, error) {
 	}
 	return n, nil
 }
+
+func (reg *Registry[E]) ReadTagsFrom(r io.Reader) (int64, error) {
+	var count pk.VarInt
+	n, err := count.ReadFrom(r)
+	if err != nil {
+		return n, err
+	}
+
+	var tag pk.Identifier
+	var length pk.VarInt
+	for i := 0; i < int(count); i++ {
+		var n1, n2, n3 int64
+
+		n1, err = tag.ReadFrom(r)
+		if err != nil {
+			return n + n1, err
+		}
+
+		n2, err = length.ReadFrom(r)
+		if err != nil {
+			return n + n1 + n2, err
+		}
+
+		n += n1 + n2
+		values := make([]*E, length)
+
+		var id pk.VarInt
+		for i := 0; i < int(length); i++ {
+			n3, err = id.ReadFrom(r)
+			if err != nil {
+				return n + n3, err
+			}
+
+			if id < 0 || int(id) >= len(reg.values) {
+				err = errors.New("invalid id: " + strconv.Itoa(int(id)))
+				return n + n3, err
+			}
+
+			values[i] = &reg.values[id]
+			n += n3
+		}
+
+		reg.tags[string(tag)] = values
+	}
+	return n, nil
+}

registry/registry.go

@@ -4,6 +4,7 @@ type Registry[E any] struct {
 	keys    map[string]int32
 	values  []E
 	indices map[*E]int32
+	tags    map[string][]*E
 }
 
 func NewRegistry[E any]() Registry[E] {
@@ -11,9 +12,17 @@ func NewRegistry[E any]() Registry[E] {
 		keys:    make(map[string]int32),
 		values:  make([]E, 0, 256),
 		indices: make(map[*E]int32),
+		tags:    make(map[string][]*E),
 	}
 }
 
+func (r *Registry[E]) Clear() {
+	r.keys = make(map[string]int32)
+	r.values = r.values[:0]
+	r.indices = make(map[*E]int32)
+	r.tags = make(map[string][]*E)
+}
+
 func (r *Registry[E]) Get(key string) (int32, *E) {
 	id, ok := r.keys[key]
 	if !ok {
@@ -38,8 +47,14 @@ func (r *Registry[E]) Put(name string, data E) (id int32, val *E) {
 	return
 }
 
-func (r *Registry[E]) Clear() {
-	r.keys = make(map[string]int32)
-	r.values = r.values[:0]
-	r.indices = make(map[*E]int32)
-}
+// func (r *Registry[E]) BindTags(tag string, ids []int32) error {
+// 	values := make([]*E, len(ids))
+// 	for i, id := range ids {
+// 		if id < 0 || id >= int32(len(r.values)) {
+// 			return errors.New("invalid id: " + strconv.Itoa(int(id)))
+// 		}
+// 		values[i] = &r.values[id]
+// 	}
+// 	r.tags[tag] = values
+// 	return nil
+// }