implementing storing chest data into Manager.Screens in bot/screen (#279)

bot/msg/chat.go

@@ -175,6 +175,30 @@ func (m *Manager) SendMessage(msg string) error {
 	return err
 }
 
+// SendMessage send chat message to server.
+// Doesn't support sending message with signature currently.
+func (m *Manager) SendCommand(command string) error {
+	if len(command) > 256 {
+		return errors.New("message length greater than 256")
+	}
+	var salt int64
+	if err := binary.Read(rand.Reader, binary.BigEndian, &salt); err != nil {
+		return err
+	}
+
+	err := m.c.Conn.WritePacket(pk.Marshal(
+		packetid.ServerboundChatCommand,
+		pk.String(command),
+		pk.Long(time.Now().UnixMilli()),
+		pk.Long(salt),
+		pk.Ary[pk.VarInt]{Ary: []pk.Tuple{}},
+		sign.HistoryUpdate{
+			Acknowledged: pk.NewFixedBitSet(20),
+		},
+	))
+	return err
+}
+
 var (
 	InvalidChatPacket       = errors.New("invalid chat packet")
 	ValidationFailed  error = bot.DisconnectErr(chat.TranslateMsg("multiplayer.disconnect.chat_validation_failed"))

bot/screen/chest.go

@@ -0,0 +1,39 @@
+package screen
+
+import (
+	"errors"
+
+	"github.com/Tnze/go-mc/chat"
+	"github.com/Tnze/go-mc/data/inventory"
+)
+
+type Chest struct {
+	Type  inventory.InventoryID
+	Title chat.Message
+	Slots []Slot
+	Rows  int
+}
+
+func (c *Chest) onSetSlot(i int, slot Slot) error {
+	if i < 0 || i >= len(c.Slots) {
+		return errors.New("slot index out of bounds")
+	}
+	c.Slots[i] = slot
+	return nil
+}
+
+func (c *Chest) onClose() error {
+	return nil
+}
+
+func (c *Chest) Container() []Slot {
+	return c.Slots[0 : c.Rows*9]
+}
+
+func (c *Chest) Main() []Slot {
+	return c.Slots[c.Rows*9 : c.Rows*9+27]
+}
+
+func (c *Chest) Hotbar() []Slot {
+	return c.Slots[c.Rows*9+27 : (c.Rows+4)*9]
+}

bot/screen/events.go

@@ -1,7 +1,9 @@
 package screen
 
+import "github.com/Tnze/go-mc/chat"
+
 type EventsListener struct {
-	Open    func(id int) error
+	Open    func(id int, container_type int32, title chat.Message) error
 	SetSlot func(id, index int) error
 	Close   func(id int) error
 }

bot/screen/screen.go

@@ -81,11 +81,23 @@ func (m *Manager) onOpenScreen(p pk.Packet) error {
 	if err := p.Scan(&ContainerID, &Type, &Title); err != nil {
 		return Error{err}
 	}
-	//if c, ok := m.Screens[byte(ContainerID)]; ok {
-	// TODO: Create the specified container
-	//}
+	if _, ok := m.Screens[int(ContainerID)]; !ok {
+		TypeInt32 := int32(Type)
+		if TypeInt32 < 6 {
+			Rows := TypeInt32 + 1
+			chest := Chest{
+				Type:  TypeInt32,
+				Slots: make([]Slot, 9*Rows),
+				Rows:  int(Rows),
+				Title: Title,
+			}
+			m.Screens[int(ContainerID)] = &chest
+		}
+	} else {
+		return errors.New("container id already exists in screens")
+	}
 	if m.events.Open != nil {
-		if err := m.events.Open(int(ContainerID)); err != nil {
+		if err := m.events.Open(int(ContainerID), int32(Type), Title); err != nil {
 			return Error{err}
 		}
 	}