add world module for in-memory block storage and integrate getblock command in messenger

cmd/strom/messenger/main.go

@@ -2,11 +2,16 @@ package messenger
 
 import (
 	"flag"
+	"fmt"
+	"strconv"
+	"strings"
 
 	"github.com/admin-else/strom/cmd/strom/cmd_util"
 	"github.com/admin-else/strom/mc/bot/chat"
 	"github.com/admin-else/strom/mc/bot/keepalive"
+	"github.com/admin-else/strom/mc/bot/world"
 	"github.com/admin-else/strom/mc/client"
+	"github.com/admin-else/strom/mc/data"
 	"github.com/admin-else/strom/mc/event"
 	"github.com/admin-else/strom/mc/proto"
 	"github.com/admin-else/strom/mc/proto_generated/v1_21_11"
@@ -20,7 +25,8 @@ var (
 
 type Messenger struct {
 	*proto.Conn
-	chat *chat.Module
+	chat  *chat.Module
+	world *world.Module
 }
 
 func (m *Messenger) OnChat(e *v1_21_11.PlayToClientPacketPlayerChat) (err error) {
@@ -39,7 +45,46 @@ func (m *Messenger) OnChatSystem(e *v1_21_11.PlayToClientPacketSystemChat) (err
 }
 
 func (m *Messenger) OnStdin(e event.Stdin) (err error) {
-	return m.chat.SendMessage(e.Val)
+	input := strings.TrimSpace(e.Val)
+	if strings.HasPrefix(input, ">say ") {
+		return m.chat.SendMessage(strings.TrimSpace(strings.TrimPrefix(input, ">say")))
+	}
+	if strings.HasPrefix(input, ">getblock ") {
+		return m.handleGetBlock(strings.TrimSpace(strings.TrimPrefix(input, ">getblock")))
+	}
+	return m.chat.SendMessage(input)
+}
+
+func (m *Messenger) handleGetBlock(args string) (err error) {
+	fields := strings.Fields(args)
+	if len(fields) != 3 {
+		m.Log.Info("getblock", "error", "expected >getblock <x> <y> <z>")
+		return nil
+	}
+
+	coords := make([]int64, 3)
+	for i, f := range fields {
+		coords[i], err = strconv.ParseInt(f, 10, 32)
+		if err != nil {
+			m.Log.Info("getblock", "error", fmt.Sprintf("bad coordinate %q", f))
+			return nil
+		}
+	}
+
+	stateId, err := m.world.GetBlock(int32(coords[0]), int32(coords[1]), int32(coords[2]))
+	if err != nil {
+		m.Log.Info("getblock", "error", err.Error())
+		return nil
+	}
+
+	block, ok := data.LookupBlockByStateId(m.Version, stateId)
+	if !ok {
+		m.Log.Info("getblock", "stateId", stateId, "name", "unknown")
+		return nil
+	}
+
+	m.Log.Info("getblock", "x", coords[0], "y", coords[1], "z", coords[2], "name", block.Name, "stateId", stateId)
+	return nil
 }
 
 func Run(args []string) (err error) {
@@ -62,9 +107,14 @@ func Run(args []string) (err error) {
 	if err != nil {
 		return
 	}
+
+	// Shareable world storage. Other bots can subscribe to the same World.
+	w := world.NewWorld(c.Version, -64, 384)
+
 	m := &Messenger{
-		Conn: c,
-		chat: chatMod,
+		Conn:  c,
+		chat:  chatMod,
+		world: world.Start(c, w),
 	}
 
 	event.StartListingStdin(m.Loop)