✨ 肯德基疯狂星期四文案指令新增支持群消息

plugin/plugins/command.go

@@ -4,6 +4,7 @@ import (
 	"go-wechat/plugin"
 	"go-wechat/plugin/plugins/command"
 	"go-wechat/utils"
+	"regexp"
 	"strings"
 )
 
@@ -11,13 +12,27 @@ import (
 // @description: 自定义指令
 // @param m
 func Command(m *plugin.MessageContext) {
+	// 如果是群聊，提取出消息
+	content := m.Content
+
+	if m.IsGroup() {
+		re := regexp.MustCompile(`@([^ | ]+)`)
+		matches := re.FindStringSubmatch(content)
+		if len(matches) > 0 {
+			// 过滤掉第一个匹配到的
+			content = strings.Replace(content, matches[0], "", 1)
+		}
+		// 去掉最前面的空格
+		content = strings.TrimLeft(content, " ")
+		content = strings.TrimLeft(content, " ")
+	}
 	// 判断是不是指令
-	if !strings.HasPrefix(m.Content, "/") {
+	if !strings.HasPrefix(content, "/") {
 		return
 	}
 
 	// 用空格分割消息，下标0表示指令
-	msgArray := strings.Split(m.Content, " ")
+	msgArray := strings.Split(content, " ")
 	cmd := msgArray[0]
 
 	switch cmd {

plugin/plugins/command/kfc.go

@@ -10,11 +10,14 @@ import (
 // @description: 肯德基疯狂星期四文案
 // @param userId string 发信人
 func KfcCrazyThursdayCmd(userId string) {
-	// 接口调用
+	// 随机选一个接口调用
 	str := kfcApi1()
 	if str == "" {
 		str = kfcApi2()
 	}
+	if str == "" {
+		str = kfcApi3()
+	}
 	if str == "" {
 		str = "文案获取失败"
 	}
@@ -64,3 +67,29 @@ func kfcApi2() string {
 	}
 	return resp.String()
 }
+
+// kfcApi3
+// @description: 肯德基疯狂星期四文案接口3
+// @return string
+func kfcApi3() string {
+	type result struct {
+		Code int    `json:"code"`
+		Msg  string `json:"msg"`
+		Text string `json:"text"`
+	}
+
+	var resData result
+
+	res := resty.New()
+	resp, err := res.R().
+		SetResult(&resData).
+		Post("https://api.pearktrue.cn/api/kfc")
+	if err != nil {
+		log.Panicf("KFC接口3文案获取失败: %s", err.Error())
+	}
+	log.Printf("KFC接口3文案获取结果: %s", resp.String())
+	if resData.Text != "" {
+		return resData.Text
+	}
+	return resp.String()
+}