李寻欢
3da8b327d0
Add AI command functionality to handle AI commands in the WeChat plugin. Includes options to enable or disable AI features.
66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package command
|
|
|
|
import (
|
|
"fmt"
|
|
"go-wechat/client"
|
|
"go-wechat/entity"
|
|
"go-wechat/utils"
|
|
"log"
|
|
"strings"
|
|
)
|
|
|
|
// AiCmd
|
|
// @description: AI指令
|
|
// @param userId
|
|
// @param groupUserId
|
|
// @param cmd
|
|
func AiCmd(userId, groupUserId, cmd string) {
|
|
// 判断发信人是不是群主
|
|
can := false
|
|
if strings.Contains(userId, "@chatroom") {
|
|
// 判断是不是群主
|
|
err := client.MySQL.Model(&entity.GroupUser{}).
|
|
Where("group_id = ?", userId).
|
|
Where("wxid = ?", groupUserId).
|
|
Pluck("is_admin", &can).Error
|
|
if err != nil {
|
|
log.Printf("查询群主失败: %v", err)
|
|
return
|
|
}
|
|
}
|
|
if !can {
|
|
utils.SendMessage(userId, groupUserId, "您不是群主，无法使用指令", 0)
|
|
return
|
|
}
|
|
|
|
var err error
|
|
replyMsg := "操作成功"
|
|
|
|
switch cmd {
|
|
case "enable", "启用", "打开":
|
|
err = setAiEnable(userId, true)
|
|
case "disable", "停用", "禁用", "关闭":
|
|
err = setAiEnable(userId, false)
|
|
default:
|
|
replyMsg = "指令错误"
|
|
}
|
|
if err != nil {
|
|
log.Printf("AI指令执行失败: %v", err)
|
|
replyMsg = fmt.Sprintf("指令执行错误: %v", err)
|
|
}
|
|
utils.SendMessage(userId, groupUserId, replyMsg, 0)
|
|
}
|
|
|
|
// setAiEnable
|
|
// @description: 设置AI启用状态
|
|
// @param userId
|
|
// @param enable
|
|
// @return err
|
|
func setAiEnable(userId string, enable bool) (err error) {
|
|
// 更新
|
|
err = client.MySQL.Model(&entity.Friend{}).
|
|
Where("wxid = ?", userId).
|
|
Update("enable_ai", enable).Error
|
|
return
|
|
}
|