feat(chatops-lark): support bot mention in group chat commands

- Add support for parsing commands in group chats when the bot is mentioned
- Introduce new message type detection for group and private messages
- Update README to include bot name configuration
- Implement bot mention detection using regex and mentions parsing
- Add new command parsing logic for group chat scenarios

Author: purelind

chatops-lark/README.md

@@ -8,6 +8,7 @@ You can run it by following steps:
   ```yaml
   cherry-pick-invite.audit_webhook: <your_audit_lark_webhook>
   cherry-pick-invite.github_token: <your_github_token>
+  bot_name: <your_bot_name> # for @bot mention in group chat
   ```
 2. Run the lark bot app:
   ```bash

chatops-lark/pkg/events/handler/root.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"regexp"
 	"slices"
 	"strings"
 	"time"
@@ -20,6 +21,10 @@ import (
 const (
 	ctxKeyGithubToken     = "github_token"
 	ctxKeyLarkSenderEmail = "lark.sender.email"
+
+	// Message types
+	msgTypePrivate = "private"
+	msgTypeGroup   = "group"
 )
 
 var (
@@ -59,9 +64,11 @@ var (
 )
 
 type Command struct {
-	Name   string
-	Args   []string
-	Sender *CommandSender
+	Name    string
+	Args    []string
+	Sender  *CommandSender
+	MsgType string // Message type: private or group
+	ChatID  string // Group ID, only valid when MsgType is group
 }
 
 type CommandSender struct {
@@ -124,7 +131,14 @@ func (r *rootHandler) Handle(ctx context.Context, event *larkim.P2MessageReceive
 	}
 
 	hl := log.With().Str("eventId", eventID).Logger()
-	command := shouldHandle(event)
+
+	// Get bot name from config with type assertion
+	botName := ""
+	if name, ok := r.Config["bot_name"].(string); ok {
+		botName = name
+	}
+
+	command := shouldHandle(event, botName)
 	if command == nil {
 		hl.Debug().Msg("none command received")
 		return nil
@@ -271,7 +285,76 @@ func (r *rootHandler) deleteReaction(reqMsgID, reactionID string) error {
 	return nil
 }
 
-func shouldHandle(event *larkim.P2MessageReceiveV1) *Command {
+// determineMessageType determines the message type and checks if the bot was mentioned
+func determineMessageType(event *larkim.P2MessageReceiveV1, botName string) (msgType string, chatID string, isMentionBot bool) {
+	// Check if the message is from a group chat
+	if event.Event.Message.ChatId != nil && event.Event.Message.ChatType != nil && *event.Event.Message.ChatType == "group" {
+		msgType = msgTypeGroup
+		chatID = *event.Event.Message.ChatId
+		isMentionBot = checkIfBotMentioned(event, botName)
+	} else {
+		msgType = msgTypePrivate
+	}
+
+	return msgType, chatID, isMentionBot
+}
+
+// checkIfBotMentioned checks if the bot was mentioned in the message
+func checkIfBotMentioned(event *larkim.P2MessageReceiveV1, botName string) bool {
+	if event.Event.Message.Mentions == nil || len(event.Event.Message.Mentions) == 0 {
+		return false
+	}
+
+	for _, mention := range event.Event.Message.Mentions {
+		if mention == nil {
+			continue
+		}
+		if mention.Name != nil {
+			// check if the bot was mentioned
+			if *mention.Name == botName {
+				return true
+			}
+		}
+	}
+	return false
+}
+
+// parseGroupCommand parses commands from group messages
+// Supported format: @bot /command arg1 arg2...
+func parseGroupCommand(event *larkim.P2MessageReceiveV1) *Command {
+	messageContent := strings.TrimSpace(*event.Event.Message.Content)
+
+	var content commandLarkMsgContent
+	if err := json.Unmarshal([]byte(messageContent), &content); err != nil {
+		return nil
+	}
+
+	// In Lark messages, mentions are converted to @_user_X format
+	// Use regex to match commands after @_user_X: /command arg1 arg2...
+	re := regexp.MustCompile(`@_user_\d+\s+(/\S+)(.*)`)
+	matches := re.FindStringSubmatch(content.Text)
+
+	if len(matches) < 3 {
+		return nil
+	}
+
+	commandName := matches[1]
+	argsStr := strings.TrimSpace(matches[2])
+	var args []string
+	if argsStr != "" {
+		args = strings.Fields(argsStr)
+	}
+
+	// Check if it's a privileged command
+	if slices.Contains(privilegedCommandList, commandName) {
+		return nil
+	}
+
+	return &Command{Name: commandName, Args: args}
+}
+
+// parsePrivateCommand parses commands from private messages
+func parsePrivateCommand(event *larkim.P2MessageReceiveV1) *Command {
 	messageContent := strings.TrimSpace(*event.Event.Message.Content)
 
 	var content commandLarkMsgContent
@@ -286,3 +369,15 @@ func shouldHandle(event *larkim.P2MessageReceiveV1) *Command {
 
 	return &Command{Name: messageParts[0], Args: messageParts[1:]}
 }
+
+func shouldHandle(event *larkim.P2MessageReceiveV1, botName string) *Command {
+	// Determine message type and whether the bot was mentioned
+	msgType, _, isMentionBot := determineMessageType(event, botName)
+
+	if msgType == msgTypeGroup && isMentionBot {
+		return parseGroupCommand(event)
+	} else if msgType == msgTypePrivate {
+		return parsePrivateCommand(event)
+	}
+	return nil
+}