From 085ef0e369457e53d98fbdc4cf9f7d9759bed9c7 Mon Sep 17 00:00:00 2001
From: Tnze <cjd001113@outlook.com>
Date: Sun, 28 Jul 2024 23:32:53 +0800
Subject: [PATCH] Provide .Block() method for block.State

---
 level/block/block.go | 36 +++++++++++++++++++++++++++++-------
 1 file changed, 29 insertions(+), 7 deletions(-)

diff --git a/level/block/block.go b/level/block/block.go
index 4aace420..7f59aca9 100644
--- a/level/block/block.go
+++ b/level/block/block.go
@@ -37,6 +37,31 @@ type (
 	}
 )
 
+func (s *State) Block() (Block, error) {
+	// Get an empty Block object
+	block, ok := FromID[s.Name]
+	if !ok {
+		return nil, UnknownBlockErr{s.Name}
+	}
+
+	// Set property values
+	if s.Properties.Type != nbt.TagEnd {
+		err := s.Properties.Unmarshal(&block)
+		if err != nil {
+			return nil, err
+		}
+	}
+	return block, nil
+}
+
+type UnknownBlockErr struct {
+	Name string
+}
+
+func (u UnknownBlockErr) Error() string {
+	return "unknown block: " + u.Name
+}
+
 func init() {
 	var states []State
 	// decompress
@@ -51,15 +76,12 @@ func init() {
 	ToStateID = make(map[Block]StateID, len(states))
 	StateList = make([]Block, 0, len(states))
 	for _, state := range states {
-		block := FromID[state.Name]
-		if state.Properties.Type != nbt.TagEnd {
-			err := state.Properties.Unmarshal(&block)
-			if err != nil {
-				panic(err)
-			}
+		block, err := state.Block()
+		if err != nil {
+			panic(err)
 		}
 		if _, ok := ToStateID[block]; ok {
-			panic(fmt.Errorf("state %#v already exist", block))
+			panic(fmt.Errorf("state %#v already exists", block))
 		}
 		ToStateID[block] = StateID(len(StateList))
 		StateList = append(StateList, block)