forked from lxh/go-wxhelper
李寻欢
3da8b327d0
Add AI command functionality to handle AI commands in the WeChat plugin. Includes options to enable or disable AI features.
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
package plugins
|
||
|
||
import (
|
||
"go-wechat/plugin"
|
||
"go-wechat/plugin/plugins/command"
|
||
"go-wechat/utils"
|
||
"regexp"
|
||
"strings"
|
||
)
|
||
|
||
// Command
|
||
// @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(content, "/") {
|
||
return
|
||
}
|
||
|
||
// 用空格分割消息，下标0表示指令
|
||
msgArray := strings.Split(content, " ")
|
||
cmd := msgArray[0]
|
||
|
||
switch cmd {
|
||
case "/帮助", "/h", "/help", "/?", "/？":
|
||
command.HelpCmd(m)
|
||
case "/雷神", "/ls":
|
||
command.LeiGodCmd(m.FromUser, msgArray[1], msgArray[2:]...)
|
||
case "/肯德基", "/kfc":
|
||
command.KfcCrazyThursdayCmd(m.FromUser)
|
||
case "/ai":
|
||
command.AiCmd(m.FromUser, m.GroupUser, msgArray[1])
|
||
default:
|
||
utils.SendMessage(m.FromUser, m.GroupUser, "指令错误", 0)
|
||
}
|
||
|
||
// 中止后续消息处理
|
||
m.Abort()
|
||
}
|