fix: MessageID String() repr

message/message.go

@@ -1,6 +1,7 @@
 package message
 
 import (
+	"bytes"
 	"crypto/md5"
 	"encoding/base64"
 	"encoding/hex"
@@ -325,6 +326,19 @@ func NewMessageIDFromInteger(raw int64) (m MessageID) {
 	return
 }
 
+func (m MessageID) MarshalJSON() ([]byte, error) {
+	sb := bytes.NewBuffer(make([]byte, 0, len(m.s)+2))
+	_, err := strconv.ParseInt(m.s, 10, 64)
+	if err != nil {
+		sb.WriteByte('"')
+		json.HTMLEscape(sb, []byte(m.s))
+		sb.WriteByte('"')
+	} else {
+		sb.WriteString(m.s)
+	}
+	return sb.Bytes(), nil
+}
+
 func (m MessageID) String() string {
 	return m.s
 }

message/message_test.go

@@ -0,0 +1,43 @@
+package message
+
+import (
+	"encoding/json"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestMessageID(t *testing.T) {
+	id := NewMessageIDFromString("test case")
+	msg := Message([]MessageSegment{Reply(id)})
+	data, err := json.Marshal(&msg)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(string(data))
+	if !assert.Equal(t, `[{"type":"reply","data":{"id":"test case"}}]`, string(data)) {
+		t.Fail()
+	}
+	id = NewMessageIDFromInteger(-1008611)
+	paras := map[string]interface{}{}
+	paras["message_id"] = id
+	data, err = json.Marshal(&paras)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(string(data))
+	if !assert.Equal(t, `{"message_id":-1008611}`, string(data)) {
+		t.Fail()
+	}
+	id = NewMessageIDFromString("test case")
+	paras = map[string]interface{}{}
+	paras["message_id"] = id
+	data, err = json.Marshal(&paras)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(string(data))
+	if !assert.Equal(t, `{"message_id":"test case"}`, string(data)) {
+		t.Fail()
+	}
+}