examples/receive-and-send: demo 增加精华消息和表情表态的指令

Author: vissong

examples/receive-and-send/cmd_action.go

@@ -0,0 +1,42 @@
+package main
+
+import (
+	"context"
+	"log"
+
+	"github.com/tencent-connect/botgo/dto"
+)
+
+func (p Processor) setEmoji(ctx context.Context, channelID string, messageID string) {
+	err := p.api.CreateMessageReaction(
+		ctx, channelID, messageID, dto.Emoji{
+			ID:   "307",
+			Type: 1,
+		},
+	)
+	if err != nil {
+		log.Println(err)
+	}
+}
+
+func (p Processor) setPins(ctx context.Context, channelID, msgID string) {
+	_, err := p.api.AddPins(ctx, channelID, msgID)
+	if err != nil {
+		log.Println(err)
+	}
+}
+
+func (p Processor) setAnnounces(ctx context.Context, data *dto.WSATMessageData) {
+	if _, err := p.api.CreateChannelAnnounces(
+		ctx, data.ChannelID,
+		&dto.ChannelAnnouncesToCreate{MessageID: data.ID},
+	); err != nil {
+		log.Println(err)
+	}
+}
+
+func (p Processor) sendReply(ctx context.Context, channelID string, toCreate *dto.MessageToCreate) {
+	if _, err := p.api.PostMessage(ctx, channelID, toCreate); err != nil {
+		log.Println(err)
+	}
+}

examples/receive-and-send/process.go

@@ -37,34 +37,26 @@ func (p Processor) ProcessMessage(input string, data *dto.WSATMessageData) error
 		return nil
 	}
 
-	var reply bool
-	var announcement bool
 	switch cmd.Cmd {
+	case "hi":
+		p.sendReply(ctx, data.ChannelID, toCreate)
 	case "time":
 		toCreate.Content = genReplyContent(data)
-		reply = true
+		p.sendReply(ctx, data.ChannelID, toCreate)
 	case "ark":
 		toCreate.Ark = genReplyArk(data)
-		reply = true
-	case "hi":
-		reply = true
+		p.sendReply(ctx, data.ChannelID, toCreate)
 	case "公告":
-		announcement = true
-	default:
-	}
-	// 是否命中上面的指令，不回复多余的内容
-	if reply {
-		if _, err := p.api.PostMessage(ctx, data.ChannelID, toCreate); err != nil {
-			log.Println(err)
+		p.setAnnounces(ctx, data)
+	case "pin":
+		if data.MessageReference != nil {
+			p.setPins(ctx, data.ChannelID, data.MessageReference.MessageID)
 		}
-	}
-	if announcement {
-		if _, err := p.api.CreateChannelAnnounces(
-			ctx, data.ChannelID,
-			&dto.ChannelAnnouncesToCreate{MessageID: data.ID},
-		); err != nil {
-			log.Println(err)
+	case "emoji":
+		if data.MessageReference != nil {
+			p.setEmoji(ctx, data.ChannelID, data.MessageReference.MessageID)
 		}
+	default:
 	}
 
 	return nil