examples/receive-and-send: 增加私信的 example

Author: vissong

examples/receive-and-send/README.md

@@ -14,3 +14,12 @@
 
 返回一个 ark 消息
 
+`dm xxx`
+
+进入私信流程，机器人会给你发一条私信
+
+## 启动方法
+
+- 进入到 receive-and-send 目录
+- 修改 config.yaml.demo 为 config.yaml，并在其中配置你的机器人 appid 和 token，注意要符合 yaml 格式要求
+- 执行 `go build` 之后执行 `receive-and-send` 或者直接运行 `go run *.go`

examples/receive-and-send/config.yaml.demo

@@ -0,0 +1,3 @@
+# 在这个配置文件中补充你的 appid 和 bot token，并修改文件名为 config.yaml
+appid:
+token:

examples/receive-and-send/main.go

@@ -41,7 +41,7 @@ func main() {
 	// 根据不同的回调，生成 intents
 	intent := websocket.RegisterHandlers(
 		ATMessageEventHandler(), ReadyHandler(), ErrorNotifyHandler(), GuildEventHandler(),
-		MemberEventHandler(), ChannelEventHandler(),
+		MemberEventHandler(), ChannelEventHandler(), DirectMessageHandler(),
 	)
 	if err = botgo.NewSessionManager().Start(wsInfo, botToken, &intent); err != nil {
 		log.Fatalln(err)
@@ -90,6 +90,13 @@ func MemberEventHandler() websocket.GuildMemberEventHandler {
 	}
 }
 
+func DirectMessageHandler() websocket.DirectMessageEventHandler {
+	return func(event *dto.WSPayload, data *dto.WSDirectMessageData) error {
+		fmt.Println(data)
+		return nil
+	}
+}
+
 func getConfigPath(name string) string {
 	_, filename, _, ok := runtime.Caller(1)
 	if ok {

examples/receive-and-send/process.go

@@ -31,6 +31,11 @@ func (p Processor) ProcessMessage(input string, data *dto.WSATMessageData) error
 			IgnoreGetMessageError: true,
 		},
 	}
+	// 进入到私信逻辑
+	if cmd.Cmd == "dm" {
+		p.dmHandler(data)
+		return nil
+	}
 	if cmd.Cmd == "time" {
 		toCreate.Content = genReplyContent(data)
 	}
@@ -57,6 +62,30 @@ func parseCommand(input string) *CMD {
 	}
 }
 
+func (p Processor) dmHandler(data *dto.WSATMessageData) {
+	dm, err := p.api.CreateDirectMessage(
+		context.Background(), &dto.DirectMessageToCreate{
+			SourceGuildID: data.GuildID,
+			RecipientID:   data.Author.ID,
+		},
+	)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+
+	toCreate := &dto.MessageToCreate{
+		Content: "默认私信回复",
+	}
+	_, err = p.api.PostDirectMessage(
+		context.Background(), dm, toCreate,
+	)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+}
+
 func genReplyContent(data *dto.WSATMessageData) string {
 	var tpl = `你好：%s
 在子频道 %s 收到消息。