55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
package chat
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"strconv"
|
|
|
|
"github.com/Tnze/go-mc/nbt"
|
|
pk "github.com/Tnze/go-mc/net/packet"
|
|
)
|
|
|
|
// ReadFrom decode Message in a Text component
|
|
func (m *Message) ReadFrom(r io.Reader) (int64, error) {
|
|
var code pk.String
|
|
n, err := code.ReadFrom(r)
|
|
if err != nil {
|
|
return n, err
|
|
}
|
|
err = nbt.Unmarshal([]byte(code), (*Message)(m))
|
|
return n, err
|
|
}
|
|
|
|
// WriteTo encode Message into a Text component
|
|
func (m Message) WriteTo(w io.Writer) (int64, error) {
|
|
code, err := nbt.Marshal(Message(m))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return pk.String(code).WriteTo(w)
|
|
}
|
|
|
|
func (m Message) MarshalNBT() ([]byte, error) {
|
|
if m.Translate != "" {
|
|
return nbt.Marshal(translateMsg(m))
|
|
} else {
|
|
return nbt.Marshal(rawMsgStruct(m))
|
|
}
|
|
}
|
|
|
|
func (m *Message) UnmarshalNBT(raw []byte) (err error) {
|
|
if len(raw) == 0 {
|
|
return io.EOF
|
|
}
|
|
switch raw[0] {
|
|
case nbt.TagString:
|
|
return nbt.Unmarshal(raw, &m.Text) // Unmarshal as jsonString
|
|
case nbt.TagCompound:
|
|
return nbt.Unmarshal(raw, (*rawMsgStruct)(m)) // Unmarshal as jsonMsg
|
|
case nbt.TagList:
|
|
return nbt.Unmarshal(raw, &m.Extra) // Unmarshal as []Message
|
|
default:
|
|
return errors.New("unknown chat message type: '" + strconv.FormatUint(uint64(raw[0]), 16) + "'")
|
|
}
|
|
}
|