102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package chat
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"io"
|
|
"strconv"
|
|
|
|
"git.konjactw.dev/falloutBot/go-mc/nbt"
|
|
pk "git.konjactw.dev/falloutBot/go-mc/net/packet"
|
|
)
|
|
|
|
// ReadFrom decode Message in a Text component
|
|
func (m *Message) ReadFrom(r io.Reader) (int64, error) {
|
|
return pk.NBT(m).ReadFrom(r)
|
|
}
|
|
|
|
// WriteTo encode Message into a Text component
|
|
func (m Message) WriteTo(w io.Writer) (int64, error) {
|
|
return pk.NBT(&m).WriteTo(w)
|
|
}
|
|
|
|
func (m Message) TagType() byte {
|
|
return nbt.TagCompound
|
|
}
|
|
|
|
func (m Message) MarshalNBT(w io.Writer) error {
|
|
if m.Translate != "" {
|
|
return nbt.NewEncoder(w).Encode(translateMsg(m), "")
|
|
} else {
|
|
return nbt.NewEncoder(w).Encode(rawMsgStruct(m), "")
|
|
}
|
|
}
|
|
|
|
func (m *Message) UnmarshalNBT(tagType byte, r nbt.DecoderReader) error {
|
|
// Re-combine the tagType into the reader, and create a nbt decoder
|
|
tagReader := bytes.NewReader([]byte{tagType})
|
|
decoder := nbt.NewDecoder(io.MultiReader(tagReader, r))
|
|
decoder.NetworkFormat(true) // TagType directlly followed the body
|
|
|
|
switch tagType {
|
|
case nbt.TagString:
|
|
_, err := decoder.Decode(&m.Text)
|
|
return err
|
|
case nbt.TagCompound:
|
|
_, err := decoder.Decode((*rawMsgStruct)(m))
|
|
return err
|
|
case nbt.TagList:
|
|
_, err := decoder.Decode(&m.Extra)
|
|
return err
|
|
default:
|
|
return errors.New("unknown chat message type: '" + strconv.FormatUint(uint64(tagType), 16) + "'")
|
|
}
|
|
}
|
|
|
|
func (t *TranslateArgs) UnmarshalNBT(tagType byte, r nbt.DecoderReader) error {
|
|
tagReader := bytes.NewReader([]byte{tagType})
|
|
decoder := nbt.NewDecoder(io.MultiReader(tagReader, r))
|
|
decoder.NetworkFormat(true) // TagType directlly followed the body
|
|
|
|
switch tagType {
|
|
case nbt.TagList:
|
|
var value []Message
|
|
if _, err := decoder.Decode(&value); err != nil {
|
|
return err
|
|
}
|
|
for _, v := range value {
|
|
*t = append(*t, v)
|
|
}
|
|
return nil
|
|
case nbt.TagByteArray:
|
|
var value []int8
|
|
if _, err := decoder.Decode(&value); err != nil {
|
|
return err
|
|
}
|
|
for _, v := range value {
|
|
*t = append(*t, strconv.FormatInt(int64(v), 10))
|
|
}
|
|
return nil
|
|
case nbt.TagIntArray:
|
|
var value []int32
|
|
if _, err := decoder.Decode(&value); err != nil {
|
|
return err
|
|
}
|
|
for _, v := range value {
|
|
*t = append(*t, strconv.FormatInt(int64(v), 10))
|
|
}
|
|
return nil
|
|
case nbt.TagLongArray:
|
|
var value []int64
|
|
if _, err := decoder.Decode(&value); err != nil {
|
|
return err
|
|
}
|
|
for _, v := range value {
|
|
*t = append(*t, strconv.FormatInt(int64(v), 10))
|
|
}
|
|
return nil
|
|
default:
|
|
return errors.New("unknown translation args type: '" + strconv.FormatUint(uint64(tagType), 16) + "'")
|
|
}
|
|
}
|