92 lines
2.1 KiB
Go
92 lines
2.1 KiB
Go
// sniffRegistryCodec is an example that acts as a client,
|
|
// connects to the server and saves its RegistryCodec to a .nbt file.
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"os"
|
|
|
|
//"github.com/mattn/go-colorable"
|
|
|
|
"github.com/Tnze/go-mc/bot"
|
|
"github.com/Tnze/go-mc/bot/basic"
|
|
_ "github.com/Tnze/go-mc/data/lang/zh-cn"
|
|
"github.com/Tnze/go-mc/data/packetid"
|
|
"github.com/Tnze/go-mc/nbt"
|
|
pk "github.com/Tnze/go-mc/net/packet"
|
|
)
|
|
|
|
var (
|
|
address = flag.String("address", "127.0.0.1", "The server address")
|
|
client *bot.Client
|
|
player *basic.Player
|
|
)
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
// log.SetOutput(colorable.NewColorableStdout())
|
|
client = bot.NewClient()
|
|
client.Auth.Name = "Daze"
|
|
player = basic.NewPlayer(client, basic.DefaultSettings, basic.EventsListener{})
|
|
|
|
// To receive the raw NBT data, create a new packet handler
|
|
// instead of just reading player.RegistryCodec in GameStart event.
|
|
client.Events.AddListener(bot.PacketHandler{
|
|
ID: packetid.ClientboundLogin,
|
|
Priority: 50,
|
|
F: onLogin,
|
|
})
|
|
|
|
// Login
|
|
err := client.JoinServer(*address)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
log.Println("Login success")
|
|
|
|
// JoinGame
|
|
for {
|
|
if err = client.HandleGame(); err == nil {
|
|
panic("HandleGame never return nil")
|
|
}
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func onLogin(p pk.Packet) error {
|
|
var DimensionNames []pk.Identifier
|
|
var RegistryCodec nbt.RawMessage
|
|
err := p.Scan(
|
|
new(pk.Int), // Entity ID
|
|
new(pk.Boolean), // Is hardcore
|
|
new(pk.Byte), // Gamemode
|
|
new(pk.Byte), // Previous Gamemode
|
|
pk.Array(&DimensionNames), // Dimension Names
|
|
pk.NBT(&RegistryCodec), // Registry Codec (Only care about this)
|
|
// ...Ignored
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = saveToFile("RegistryCodec.nbt", RegistryCodec)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
log.Print("Successfully written RegistryCodec.nbt")
|
|
return nil
|
|
}
|
|
|
|
func saveToFile(filename string, value any) (err error) {
|
|
f, err := os.Create(filename)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
defer func(f *os.File) {
|
|
if err2 := f.Close(); err == nil {
|
|
err = err2
|
|
}
|
|
}(f)
|
|
return nbt.NewEncoder(f).Encode(value, "")
|
|
}
|