300 lines
7.2 KiB
Go
300 lines
7.2 KiB
Go
// Package chat implements Minecraft's chat message encoding system.
|
|
//
|
|
// The type Message is the Minecraft chat message. Can be encoded as JSON
|
|
// or net/packet.Field .
|
|
//
|
|
// It's very recommended that use SetLanguage before using Message.String or Message.ClearString,
|
|
// or the `github.com/Tnze/go-mc/data/en-us` will be used.
|
|
// Note: The package of data/lang/... will SetLanguage on theirs init() so you don't need to call by your self.
|
|
//
|
|
// Some of these docs is copied from https://wiki.vg/Chat.
|
|
package chat
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"regexp"
|
|
"strings"
|
|
|
|
en_us "github.com/Tnze/go-mc/data/lang/en-us"
|
|
pk "github.com/Tnze/go-mc/net/packet"
|
|
)
|
|
|
|
// Colors
|
|
const (
|
|
Black = "black"
|
|
DarkBlue = "dark_blue"
|
|
DarkGreen = "dark_green"
|
|
DarkAqua = "dark_aqua"
|
|
DarkRed = "dark_red"
|
|
DarkPurple = "dark_purple"
|
|
Gold = "gold"
|
|
Gray = "gray"
|
|
DarkGray = "dark_gray"
|
|
Blue = "blue"
|
|
Green = "green"
|
|
Aqua = "aqua"
|
|
Red = "red"
|
|
LightPurple = "light_purple"
|
|
Yellow = "yellow"
|
|
White = "white"
|
|
)
|
|
|
|
// Message is a message sent by other
|
|
type Message struct {
|
|
Text string `json:"text,omitempty"`
|
|
|
|
Bold bool `json:"bold,omitempty"` //粗体
|
|
Italic bool `json:"italic,omitempty"` //斜体
|
|
UnderLined bool `json:"underlined,omitempty"` //下划线
|
|
StrikeThrough bool `json:"strikethrough,omitempty"` //删除线
|
|
Obfuscated bool `json:"obfuscated,omitempty"` //随机
|
|
// Font of the message, could be one of minecraft:uniform, minecraft:alt or minecraft:default
|
|
// This option is only valid on 1.16+, otherwise the property is ignored.
|
|
Font string `json:"font,omitempty"` //字体
|
|
Color string `json:"color,omitempty"` //颜色
|
|
|
|
// Insertion contains text to insert. Only used for messages in chat.
|
|
// When shift is held, clicking the component inserts the given text
|
|
// into the chat box at the cursor (potentially replacing selected text).
|
|
Insertion string `json:"insertion,omitempty"`
|
|
ClickEvent *ClickEvent `json:"clickEvent,omitempty"`
|
|
HoverEvent *HoverEvent `json:"hoverEvent,omitempty"`
|
|
|
|
Translate string `json:"translate,omitempty"`
|
|
With []json.RawMessage `json:"with,omitempty"`
|
|
Extra []Message `json:"extra,omitempty"`
|
|
}
|
|
|
|
type jsonMsg Message
|
|
|
|
//UnmarshalJSON decode json to Message
|
|
func (m *Message) UnmarshalJSON(raw []byte) (err error) {
|
|
if len(raw) == 0 {
|
|
return io.EOF
|
|
}
|
|
// The right way to distinguish JSON String and Object
|
|
// is to look up the first character.
|
|
if raw[0] == '"' {
|
|
err = json.Unmarshal(raw, &m.Text) // Unmarshal as jsonString
|
|
} else {
|
|
err = json.Unmarshal(raw, (*jsonMsg)(m)) // Unmarshal as jsonMsg
|
|
}
|
|
return
|
|
}
|
|
|
|
// ReadFrom decode Message in a ChatMsg packet
|
|
func (m *Message) ReadFrom(r io.Reader) (int64, error) {
|
|
var Len pk.VarInt
|
|
if n, err := Len.ReadFrom(r); err != nil {
|
|
return n, err
|
|
}
|
|
lr := &io.LimitedReader{R: r, N: int64(Len)}
|
|
err := json.NewDecoder(lr).Decode(m)
|
|
return int64(Len) - lr.N, err
|
|
}
|
|
|
|
// WriteTo encode Message into a ChatMsg packet
|
|
func (m Message) WriteTo(w io.Writer) (int64, error) {
|
|
code, err := json.Marshal(m)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return pk.String(code).WriteTo(w)
|
|
}
|
|
|
|
func (m Message) Append(extraMsg ...Message) Message {
|
|
origLen := len(m.Extra)
|
|
finalLen := origLen + len(extraMsg)
|
|
if cap(m.Extra) < len(m.Extra)+len(extraMsg) {
|
|
// pre expansion
|
|
extra := make([]Message, finalLen)
|
|
copy(extra, m.Extra)
|
|
m.Extra = extra
|
|
}
|
|
for _, v := range extraMsg {
|
|
m.Extra = append(m.Extra, v)
|
|
}
|
|
return m
|
|
}
|
|
|
|
func (m Message) SetColor(color string) Message {
|
|
m.Color = color
|
|
return m
|
|
}
|
|
|
|
func Text(str string) Message {
|
|
return Message{Text: str}
|
|
}
|
|
|
|
func TranslateMsg(key string, with ...Message) (m Message) {
|
|
m.Translate = key
|
|
m.With = make([]json.RawMessage, len(with))
|
|
for i, v := range with {
|
|
m.With[i], _ = json.Marshal(v)
|
|
}
|
|
return
|
|
}
|
|
|
|
var fmtCode = map[byte]string{
|
|
'0': "30",
|
|
'1': "34",
|
|
'2': "32",
|
|
'3': "36",
|
|
'4': "31",
|
|
'5': "35",
|
|
'6': "33",
|
|
'7': "37",
|
|
'8': "90",
|
|
'9': "94",
|
|
'a': "92",
|
|
'b': "96",
|
|
'c': "91",
|
|
'd': "95",
|
|
'e': "93",
|
|
'f': "97",
|
|
|
|
// 'k':"", //random
|
|
'l': "1",
|
|
'm': "9",
|
|
'n': "4",
|
|
'o': "3",
|
|
'r': "0",
|
|
}
|
|
var colors = map[string]string{
|
|
"black": "30",
|
|
"dark_blue": "34",
|
|
"dark_green": "32",
|
|
"dark_aqua": "36",
|
|
"dark_red": "31",
|
|
"dark_purple": "35",
|
|
"gold": "33",
|
|
"gray": "37",
|
|
"dark_gray": "90",
|
|
"blue": "94",
|
|
"green": "92",
|
|
"aqua": "96",
|
|
"red": "91",
|
|
"light_purple": "95",
|
|
"yellow": "93",
|
|
"white": "97",
|
|
}
|
|
|
|
// translateMap is the translation table.
|
|
// By default, it's en-us.
|
|
var translateMap = en_us.Map
|
|
|
|
// SetLanguage set the translate map to this map.
|
|
func SetLanguage(trans map[string]string) {
|
|
translateMap = trans
|
|
}
|
|
|
|
// ClearString return the message String without escape sequence for ansi color.
|
|
func (m Message) ClearString() string {
|
|
var msg strings.Builder
|
|
text, _ := TransCtrlSeq(m.Text, false)
|
|
msg.WriteString(text)
|
|
|
|
//handle translate
|
|
if m.Translate != "" {
|
|
args := make([]interface{}, len(m.With))
|
|
for i, v := range m.With {
|
|
var arg Message
|
|
_ = arg.UnmarshalJSON(v) //ignore error
|
|
args[i] = arg.ClearString()
|
|
}
|
|
|
|
if translateMap != nil {
|
|
_, _ = fmt.Fprintf(&msg, translateMap[m.Translate], args...)
|
|
} else {
|
|
_, _ = fmt.Fprint(&msg, m.Translate, m.With)
|
|
}
|
|
}
|
|
|
|
if m.Extra != nil {
|
|
for i := range m.Extra {
|
|
msg.WriteString(m.Extra[i].ClearString())
|
|
}
|
|
}
|
|
return msg.String()
|
|
}
|
|
|
|
// String return the message string with escape sequence for ansi color.
|
|
// To convert Translated Message to string, you must set
|
|
// On Windows, you may want print this string using github.com/matte/go-colorable.
|
|
func (m Message) String() string {
|
|
var msg, format strings.Builder
|
|
if m.Bold {
|
|
format.WriteString("1;")
|
|
}
|
|
if m.Italic {
|
|
format.WriteString("3;")
|
|
}
|
|
if m.UnderLined {
|
|
format.WriteString("4;")
|
|
}
|
|
if m.StrikeThrough {
|
|
format.WriteString("9;")
|
|
}
|
|
if m.Color != "" {
|
|
format.WriteString(colors[m.Color] + ";")
|
|
}
|
|
if format.Len() > 0 {
|
|
msg.WriteString("\033[" + format.String()[:format.Len()-1] + "m")
|
|
}
|
|
|
|
text, ok := TransCtrlSeq(m.Text, true)
|
|
msg.WriteString(text)
|
|
|
|
//handle translate
|
|
if m.Translate != "" {
|
|
args := make([]interface{}, len(m.With))
|
|
for i, v := range m.With {
|
|
var arg Message
|
|
_ = arg.UnmarshalJSON(v) //ignore error
|
|
args[i] = arg
|
|
}
|
|
|
|
if translateMap != nil {
|
|
_, _ = fmt.Fprintf(&msg, translateMap[m.Translate], args...)
|
|
} else {
|
|
_, _ = fmt.Fprint(&msg, m.Translate, m.With)
|
|
}
|
|
}
|
|
|
|
if m.Extra != nil {
|
|
for i := range m.Extra {
|
|
msg.WriteString(m.Extra[i].String())
|
|
}
|
|
}
|
|
|
|
if format.Len() > 0 || ok {
|
|
msg.WriteString("\033[0m")
|
|
}
|
|
return msg.String()
|
|
}
|
|
|
|
var fmtPat = regexp.MustCompile("(?i)§[0-9A-FK-OR]")
|
|
|
|
// TransCtrlSeq will transform control sequences into ANSI code
|
|
// or simply filter them. Depends on the second argument.
|
|
// if the str contains control sequences, returned change=true.
|
|
func TransCtrlSeq(str string, ansi bool) (dst string, change bool) {
|
|
dst = fmtPat.ReplaceAllStringFunc(
|
|
str,
|
|
func(str string) string {
|
|
f, ok := fmtCode[str[2]]
|
|
if ok {
|
|
if ansi {
|
|
change = true
|
|
return "\033[" + f + "m" // enable, add ANSI code
|
|
}
|
|
return "" //disable, remove the § code
|
|
}
|
|
return str //not a § code
|
|
},
|
|
)
|
|
return
|
|
}
|