message.go

package message

import (
	"encoding/base64"
	"fmt"
	"strconv"
	"strings"

	jsoniter "github.com/json-iterator/go"
)

// Message impl the array form of message
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/array.md#%E6%95%B0%E7%BB%84%E6%A0%BC%E5%BC%8F
type Message []MessageSegment

// MessageSegment impl the single message
// MessageSegment 消息数组
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/array.md#%E6%95%B0%E7%BB%84%E6%A0%BC%E5%BC%8F
type MessageSegment struct {
	Type string            `json:"type"`
	Data map[string]string `json:"data"`
}

// EscapeCQText escapes special characters in a non-media plain message.\
//
// CQ码字符转换
func EscapeCQText(str string) string {
	str = strings.ReplaceAll(str, "&", "&")
	str = strings.ReplaceAll(str, "[", "[")
	str = strings.ReplaceAll(str, "]", "]")
	return str
}

// UnescapeCQText unescapes special characters in a non-media plain message.
//
// CQ码反解析
func UnescapeCQText(str string) string {
	str = strings.ReplaceAll(str, "]", "]")
	str = strings.ReplaceAll(str, "[", "[")
	str = strings.ReplaceAll(str, "&", "&")
	return str
}

// EscapeCQCodeText escapes special characters in a cqcode value.
//
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/string.md#%E8%BD%AC%E4%B9%89
//
// cq码字符转换
func EscapeCQCodeText(str string) string {
	str = strings.ReplaceAll(str, "&", "&")
	str = strings.ReplaceAll(str, "[", "[")
	str = strings.ReplaceAll(str, "]", "]")
	str = strings.ReplaceAll(str, ",", ",")
	return str
}

// UnescapeCQCodeText unescapes special characters in a cqcode value.
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/string.md#%E8%BD%AC%E4%B9%89
//
// cq码反解析
func UnescapeCQCodeText(str string) string {
	str = strings.ReplaceAll(str, ",", ",")
	str = strings.ReplaceAll(str, "]", "]")
	str = strings.ReplaceAll(str, "[", "[")
	str = strings.ReplaceAll(str, "&", "&")
	return str
}

// CQCode 将数组消息转换为CQ码
func (m MessageSegment) CQCode() string {
	cqcode := "[CQ:" + m.Type  // 消息类型
	for k, v := range m.Data { // 消息参数
		cqcode = cqcode + "," + k + "=" + EscapeCQCodeText(v)
	}
	return cqcode + "]"
}

// Text 纯文本
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#%E7%BA%AF%E6%96%87%E6%9C%AC
func Text(text ...interface{}) MessageSegment {
	return MessageSegment{
		Type: "text",
		Data: map[string]string{
			"text": fmt.Sprint(text...),
		},
	}
}

// Face QQ表情
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#qq-%E8%A1%A8%E6%83%85
func Face(id int) MessageSegment {
	return MessageSegment{
		Type: "face",
		Data: map[string]string{
			"id": strconv.Itoa(id),
		},
	}
}

// ImageC 普通图片
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#%E5%9B%BE%E7%89%87
func ImageC(file interface{}) MessageSegment {
	switch file.(type) {
	case string:
		if strings.HasPrefix(file.(string), "http://") || strings.HasPrefix(file.(string), "https://") {
			return MessageSegment{
				Type: "image",
				Data: map[string]string{
					"file": file.(string),
				},
			}
		} else {
			return MessageSegment{
				Type: "image",
				Data: map[string]string{
					"file": "file:///" + file.(string),
				},
			}
		}
	case []byte:
		return MessageSegment{
			Type: "image",
			Data: map[string]string{
				"file": base64.StdEncoding.EncodeToString(file.([]byte)),
			},
		}
	default:
		return MessageSegment{
			Type: "image",
			Data: map[string]string{
				"file": file.(string),
			},
		}
	}
}

// Image 普通图片
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#%E5%9B%BE%E7%89%87
func Image(file string) MessageSegment {
	return MessageSegment{
		Type: "image",
		Data: map[string]string{
			"file": file,
		},
	}
}

// Record 语音
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#%E8%AF%AD%E9%9F%B3
func Record(file string) MessageSegment {
	return MessageSegment{
		Type: "record",
		Data: map[string]string{
			"file": file,
		},
	}
}

// At @某人
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#%E6%9F%90%E4%BA%BA
func At(qq int64) MessageSegment {
	return MessageSegment{
		Type: "at",
		Data: map[string]string{
			"qq": strconv.FormatInt(qq, 10),
		},
	}
}

// Music 音乐分享
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#%E9%9F%B3%E4%B9%90%E5%88%86%E4%BA%AB-
func Music(mType string, id int64) MessageSegment {
	return MessageSegment{
		Type: "music",
		Data: map[string]string{
			"type": mType,
			"id":   strconv.FormatInt(id, 10),
		},
	}
}

// CustomMusic 音乐自定义分享
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#%E9%9F%B3%E4%B9%90%E8%87%AA%E5%AE%9A%E4%B9%89%E5%88%86%E4%BA%AB-
func CustomMusic(url, audio, title string) MessageSegment {
	return MessageSegment{
		Type: "music",
		Data: map[string]string{
			"type":  "custom",
			"url":   url,
			"audio": audio,
			"title": title,
		},
	}
}

// Reply 回复
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#%E5%9B%9E%E5%A4%8D
func Reply(id int64) MessageSegment {
	return MessageSegment{
		Type: "reply",
		Data: map[string]string{
			"id": strconv.FormatInt(id, 10),
		},
	}
}

// Forward 合并转发
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#%E5%90%88%E5%B9%B6%E8%BD%AC%E5%8F%91-
func Forward(id string) MessageSegment {
	return MessageSegment{
		Type: "forward",
		Data: map[string]string{
			"id": id,
		},
	}
}

// Node 合并转发节点
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#%E5%90%88%E5%B9%B6%E8%BD%AC%E5%8F%91%E8%8A%82%E7%82%B9-
func Node(id int64) MessageSegment {
	return MessageSegment{
		Type: "node",
		Data: map[string]string{
			"id": strconv.FormatInt(id, 10),
		},
	}
}

// CustomNode 自定义合并转发节点
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#%E5%90%88%E5%B9%B6%E8%BD%AC%E5%8F%91%E8%87%AA%E5%AE%9A%E4%B9%89%E8%8A%82%E7%82%B9
func CustomNode(nickname string, userID int64, content interface{}) MessageSegment {
	var str string
	if s, ok := content.(string); ok {
		str = s
	} else {
		str, _ = jsoniter.MarshalToString(content)
	}
	return MessageSegment{
		Type: "node",
		Data: map[string]string{
			"uin":     strconv.FormatInt(userID, 10),
			"name":    nickname,
			"content": str,
		},
	}
}

// XML 消息
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#xml-%E6%B6%88%E6%81%AF
func XML(data string) MessageSegment {
	return MessageSegment{
		Type: "xml",
		Data: map[string]string{
			"data": data,
		},
	}
}

// JSON 消息
// https://github.com/howmanybots/onebot/blob/master/v11/specs/message/segment.md#xml-%E6%B6%88%E6%81%AF
func JSON(data string) MessageSegment {
	return MessageSegment{
		Type: "json",
		Data: map[string]string{
			"data": data,
		},
	}
}

// Expand CQCode

// Gift 群礼物
// https://github.com/Mrs4s/go-cqhttp/blob/master/docs/cqhttp.md#%E7%A4%BC%E7%89%A9
//
// Deprecated: 群礼物改版
func Gift(userID string, giftID string) MessageSegment {
	return MessageSegment{
		Type: "gift",
		Data: map[string]string{
			"qq": userID,
			"id": giftID,
		},
	}
}

// Poke 戳一戳
// https://github.com/Mrs4s/go-cqhttp/blob/master/docs/cqhttp.md#%E6%88%B3%E4%B8%80%E6%88%B3
func Poke(userID int64) MessageSegment {
	return MessageSegment{
		Type: "poke",
		Data: map[string]string{
			"qq": strconv.FormatInt(userID, 10),
		},
	}
}

// TTS 文本转语音
// https://github.com/Mrs4s/go-cqhttp/blob/master/docs/cqhttp.md#%E6%96%87%E6%9C%AC%E8%BD%AC%E8%AF%AD%E9%9F%B3
func TTS(text string) MessageSegment {
	return MessageSegment{
		Type: "tts",
		Data: map[string]string{
			"text": text,
		},
	}
}

// Add 为 MessageSegment 的 Data 增加一个字段
func (m MessageSegment) Add(key string, val interface{}) MessageSegment {
	m.Data[key] = fmt.Sprint(val)
	return m
}

func (m MessageSegment) Delete(key string) MessageSegment {
	delete(m.Data, key)
	return m
}

// ReplyWithMessage returns a reply message
func ReplyWithMessage(messageID int64, m ...MessageSegment) Message {
	return append(Message{Reply(messageID)}, m...)
}