Merge pull request '✨ 肯德基疯狂星期四文案指令新增支持群消息' (#4) from dev into main

Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/4

entity/plugindata.go

@@ -0,0 +1,13 @@
+package entity
+
+// PluginData
+// @description: 插件数据
+type PluginData struct {
+	UserId     string `json:"userId"`     // 用户Id
+	PluginCode string `json:"pluginCode"` // 插件编码
+	Data       string `json:"data"`       // 数据
+}
+
+func (PluginData) TableName() string {
+	return "t_plugin_data"
+}

initialization/plugin.go

@@ -5,6 +5,7 @@ import (
 	"go-wechat/model"
 	plugin "go-wechat/plugin"
 	"go-wechat/plugin/plugins"
+	"go-wechat/service"
 )
 
 // Plugin
@@ -22,6 +23,12 @@ func Plugin() {
 		return true
 	}, plugins.SaveToDb)
 
+	// 私聊指令消息
+	dispatcher.RegisterHandler(func(m *model.Message) bool {
+		// 私聊消息 或 群聊艾特机器人并且以/开头的消息
+		return (m.IsPrivateText() || (m.IsAt() && m.CleanContentStartWith("/"))) && service.CheckIsEnableCommand(m.FromUser)
+	}, plugins.Command)
+
 	// AI消息插件
 	dispatcher.RegisterHandler(func(m *model.Message) bool {
 		// 群内@或者私聊文字消息

model/leigod.go

@@ -0,0 +1,73 @@
+package model
+
+// LeiGodLoginResp
+// @description: 雷神登录返回
+type LeiGodLoginResp struct {
+	LoginInfo struct {
+		AccountToken string `json:"account_token"` // Token
+		ExpiryTime   string `json:"expiry_time"`   // 有效期
+		NnToken      string `json:"nn_token"`
+	} `json:"login_info"`
+	UserInfo struct {
+		Nickname   string `json:"nickname"`
+		Email      string `json:"email"`
+		Mobile     string `json:"mobile"`
+		Avatar     string `json:"avatar"`
+		RegionCode int    `json:"region_code"`
+	} `json:"user_info"`
+}
+
+// LeiGodUserInfoResp
+// @description: 雷神用户信息返回
+type LeiGodUserInfoResp struct {
+	UserPauseTime         int    `json:"user_pause_time"`
+	Nickname              string `json:"nickname"`
+	Email                 string `json:"email"`
+	CountryCode           string `json:"country_code"`
+	Mobile                string `json:"mobile"`
+	UserName              string `json:"user_name"`
+	MasterAccount         string `json:"master_account"`
+	Birthday              string `json:"birthday"`
+	PublicIp              string `json:"public_ip"`
+	Sex                   string `json:"sex"`
+	LastLoginTime         string `json:"last_login_time"`
+	LastLoginIp           string `json:"last_login_ip"`
+	PauseStatus           string `json:"pause_status"`    // 暂停状态
+	PauseStatusId         int    `json:"pause_status_id"` // 暂停状态 0未暂停1已暂停
+	LastPauseTime         string `json:"last_pause_time"` // 最后一次暂停时间
+	VipLevel              string `json:"vip_level"`
+	Avatar                string `json:"avatar"`
+	AvatarNew             string `json:"avatar_new"`
+	PackageId             string `json:"package_id"`
+	IsSwitchPackage       int    `json:"is_switch_package"`
+	PackageTitle          string `json:"package_title"`
+	PackageLevel          string `json:"package_level"`
+	BillingType           string `json:"billing_type"`
+	Lang                  string `json:"lang"`
+	StopedRemaining       string `json:"stoped_remaining"`
+	ExpiryTime            string `json:"expiry_time"`      // 剩余时长
+	ExpiryTimeSamp        int    `json:"expiry_time_samp"` // 剩余时长秒数
+	Address               string `json:"address"`
+	MobileContactType     string `json:"mobile_contact_type"`
+	MobileContactNumber   string `json:"mobile_contact_number"`
+	MobileContactTitle    string `json:"mobile_contact_title"`
+	RegionCode            int    `json:"region_code"`
+	IsPayUser             string `json:"is_pay_user"`
+	WallLogSwitch         string `json:"wall_log_switch"`
+	IsSetAdminPass        int    `json:"is_set_admin_pass"`
+	ExpiredExperienceTime string `json:"expired_experience_time"`
+	ExperienceExpiryTime  string `json:"experience_expiry_time"`
+	ExperienceTime        int    `json:"experience_time"`
+	FirstInvoiceDiscount  int    `json:"first_invoice_discount"`
+	NnNumber              string `json:"nn_number"`
+	UserSignature         string `json:"user_signature"`
+	MobileExpiryTime      string `json:"mobile_expiry_time"`
+	MobileExpiryTimeSamp  int    `json:"mobile_expiry_time_samp"`
+	MobilePauseStatus     int    `json:"mobile_pause_status"`
+	BlackExpiredTime      string `json:"black_expired_time"`
+	MobileExperienceTime  string `json:"mobile_experience_time"`
+	SuperTime             string `json:"super_time"`
+	NowDate               string `json:"now_date"`
+	NowTimeSamp           int    `json:"now_time_samp"`
+	UserEarnMinutes       string `json:"user_earn_minutes"`
+}

model/message.go

@@ -3,6 +3,7 @@ package model
 import (
 	"encoding/xml"
 	"go-wechat/types"
+	"regexp"
 	"strings"
 )
 
@@ -99,3 +100,28 @@ func (m Message) IsPrivateText() bool {
 	// 发信人不以@chatroom结尾且消息类型为文本
 	return !strings.HasSuffix(m.FromUser, "chatroom") && m.Type == types.MsgTypeText
 }
+
+// CleanContentStartWith
+// @description: 判断是否包含指定消息前缀
+// @receiver m
+// @param prefix
+// @return bool
+func (m Message) CleanContentStartWith(prefix string) bool {
+	content := m.Content
+
+	// 如果是@消息，过滤掉@的内容
+	if m.IsAt() {
+		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, " ")
+
+	return strings.HasPrefix(content, prefix)
+}

plugin/plugins/command.go

@@ -0,0 +1,51 @@
+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)
+	default:
+		utils.SendMessage(m.FromUser, m.GroupUser, "指令错误", 0)
+	}
+
+	// 中止后续消息处理
+	m.Abort()
+}

plugin/plugins/command/help.go

@@ -0,0 +1,29 @@
+package command
+
+import (
+	"go-wechat/plugin"
+	"go-wechat/utils"
+)
+
+// HelpCmd
+// @description: 帮助指令
+func HelpCmd(m *plugin.MessageContext) {
+	str := `帮助菜单:
+指令消息必须以'/'开头，比如: '/帮助'。
+支持的指令:
+
+#1. 雷神加速器
+/ls option args
+option: 指令选项，可选值: 
+  绑定账户：'绑定'、'b'，参数: 账户名 密码 [-f]，-f表示强制绑定，非必传项
+  详情: '详情'、'i'
+  暂停: '暂停'、'p'
+示例: 绑定:
+/ls 绑定 123456 123456 或者 /ls b 123456 123456
+
+#2. 肯德基疯狂星期四文案
+/kfc、/肯德基
+`
+	utils.SendMessage(m.FromUser, m.GroupUser, str, 0)
+
+}

plugin/plugins/command/kfc.go

@@ -0,0 +1,95 @@
+package command
+
+import (
+	"github.com/go-resty/resty/v2"
+	"go-wechat/utils"
+	"log"
+)
+
+// KfcCrazyThursdayCmd
+// @description: 肯德基疯狂星期四文案
+// @param userId string 发信人
+func KfcCrazyThursdayCmd(userId string) {
+	// 随机选一个接口调用
+	str := kfcApi1()
+	if str == "" {
+		str = kfcApi2()
+	}
+	if str == "" {
+		str = kfcApi3()
+	}
+	if str == "" {
+		str = "文案获取失败"
+	}
+
+	// 发送消息
+	utils.SendMessage(userId, "", str, 0)
+}
+
+// kfcApi1
+// @description: 肯德基疯狂星期四文案接口1
+// @return string
+func kfcApi1() string {
+	res := resty.New()
+	resp, err := res.R().
+		Post("https://api.jixs.cc/api/wenan-fkxqs/index.php")
+	if err != nil {
+		log.Panicf("KFC接口1文案获取失败: %s", err.Error())
+	}
+	log.Printf("KFC接口1文案获取结果: %s", resp.String())
+	return resp.String()
+}
+
+// kfcApi2
+// @description: 肯德基疯狂星期四文案接口2
+// @return string
+func kfcApi2() string {
+	type result struct {
+		Code int    `json:"code"`
+		Text string `json:"text"`
+		Data struct {
+			Msg string `json:"msg"`
+		} `json:"data"`
+	}
+
+	var resData result
+
+	res := resty.New()
+	resp, err := res.R().
+		SetResult(&resData).
+		Post("https://api.jixs.cc/api/wenan-fkxqs/index.php")
+	if err != nil {
+		log.Panicf("KFC接口2文案获取失败: %s", err.Error())
+	}
+	log.Printf("KFC接口2文案获取结果: %s", resp.String())
+	if resData.Data.Msg != "" {
+		return resData.Data.Msg
+	}
+	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()
+}

plugin/plugins/command/leigod.go

@@ -0,0 +1,205 @@
+package command
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"go-wechat/client"
+	"go-wechat/entity"
+	"go-wechat/model"
+	"go-wechat/utils"
+	"go-wechat/vo"
+	"gorm.io/gorm"
+	"log"
+	"strings"
+)
+
+// leiGod
+// @description: 雷神加速器相关接口
+type leiGodI interface {
+	binding(string, string, bool) string // 绑定雷神加速器账号
+	info() string                        // 账户详情
+	pause() string                       // 暂停加速
+}
+
+type leiGod struct {
+	userId string // 用户Id
+}
+
+// newLeiGod
+// @description: 创建一个雷神加速器实例
+// @param userId
+// @return leiGodI
+func newLeiGod(userId string) leiGodI {
+	return &leiGod{userId: userId}
+}
+
+// LeiGodCmd
+// @description: 雷神加速器指令
+// @param userId
+// @param cmd
+// @param args
+// @return string
+func LeiGodCmd(userId, cmd string, args ...string) {
+	lg := newLeiGod(userId)
+
+	var replyMsg string
+	switch cmd {
+	case "绑定", "b":
+		var force bool
+		if len(args) == 3 && args[2] == "-f" {
+			force = true
+		}
+		replyMsg = lg.binding(args[0], args[1], force)
+	case "详情", "i":
+		replyMsg = lg.info()
+	case "暂停", "p":
+		replyMsg = lg.pause()
+	default:
+		replyMsg = "指令错误"
+	}
+
+	// 返回消息
+	if strings.TrimSpace(replyMsg) != "" {
+		utils.SendMessage(userId, "", replyMsg, 0)
+	}
+}
+
+// binding
+// @description: 绑定雷神加速器账号
+// @receiver l
+// @param account
+// @param password
+// @param force
+// @return flag
+func (l leiGod) binding(account, password string, force bool) (replyMsg string) {
+	log.Printf("用户[%s]绑定雷神加速器账号[%s] -> %s", l.userId, account, password)
+
+	// 取出已绑定的账号
+	var data entity.PluginData
+	client.MySQL.Where("user_id = ?", l.userId).Where("plugin_code = 'leigod'").First(&data)
+
+	var ac vo.LeiGodAccount
+	if data.UserId != "" {
+		if err := json.Unmarshal([]byte(data.Data), &ac); err != nil {
+			log.Printf("用户[%s]已绑定雷神账号解析失败: %v", l.userId, err)
+			return
+		}
+		log.Printf("用户[%s]已绑定账号[%s]", l.userId, ac.Account)
+	}
+
+	// 如果已经绑定账号，且不是强制绑定，则返回
+	if ac.Account != "" && !force {
+		replyMsg = "您已绑定账号[" + ac.Account + "]，如需更换请使用 -f 参数: \n/雷神 绑定 账号 密码 -f"
+		return
+	}
+
+	accountStr := fmt.Sprintf("{\"account\": \"%s\", \"password\":\"%s\"}", account, password)
+
+	// 绑定账号
+	var err error
+	if data.UserId != "" {
+		// 修改
+		err = client.MySQL.Model(&data).
+			Where("user_id = ?", l.userId).
+			Where("plugin_code = 'leigod'").
+			Update("data", accountStr).Error
+	} else {
+		// 新增
+		data = entity.PluginData{
+			UserId:     l.userId,
+			PluginCode: "leigod",
+			Data:       accountStr,
+		}
+		err = client.MySQL.Create(&data).Error
+	}
+
+	if err != nil {
+		log.Printf("用户[%s]绑定雷神账号失败: %v", l.userId, err)
+		replyMsg = "绑定失败: " + err.Error()
+	} else {
+		replyMsg = "绑定成功"
+	}
+
+	return
+}
+
+// info
+// @description: 账户详情
+// @receiver l
+// @return replyMsg
+func (l leiGod) info() (replyMsg string) {
+	log.Printf("用户[%s]获取雷神账户详情", l.userId)
+
+	// 取出已绑定的账号
+	var data entity.PluginData
+	err := client.MySQL.Where("user_id = ?", l.userId).Where("plugin_code = 'leigod'").First(&data).Error
+	if err != nil {
+		log.Printf("用户[%s]获取雷神账户详情失败: %v", l.userId, err)
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			replyMsg = "您还未绑定账号，请先绑定后再使用，绑定指定:\n/雷神 绑定 你的账号 你的密码"
+		} else {
+			replyMsg = "系统错误: " + err.Error()
+		}
+		return
+	}
+
+	// 解析为结构体
+	var ac vo.LeiGodAccount
+	if err = json.Unmarshal([]byte(data.Data), &ac); err != nil {
+		log.Printf("用户[%s]已绑定雷神账号解析失败: %v", l.userId, err)
+		replyMsg = "系统炸了，请耐心等待修复"
+		return
+	}
+
+	lgu := utils.LeiGodUtil(ac.Account, ac.Password)
+	if err = lgu.Login(); err != nil {
+		return "登录失败: " + err.Error()
+	}
+	var ui model.LeiGodUserInfoResp
+	if ui, err = lgu.Info(); err != nil {
+		return "获取详情失败: " + err.Error()
+	}
+	replyMsg = fmt.Sprintf("#账户 %s\n#剩余时长 %s\n#暂停状态 %s\n#最后暂停时间 %s",
+		ui.Mobile, ui.ExpiryTime, ui.PauseStatus, ui.LastPauseTime)
+	return
+}
+
+// pause
+// @description: 暂停加速
+// @receiver l
+// @return flag
+func (l leiGod) pause() (replyMsg string) {
+	log.Printf("用户[%s]暂停加速", l.userId)
+
+	// 取出已绑定的账号
+	var data entity.PluginData
+	err := client.MySQL.Where("user_id = ?", l.userId).Where("plugin_code = 'leigod'").First(&data).Error
+	if err != nil {
+		log.Printf("用户[%s]获取雷神账户详情失败: %v", l.userId, err)
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			replyMsg = "您还未绑定账号，请先绑定后再使用，绑定指定:\n/雷神 绑定 你的账号 你的密码"
+		} else {
+			replyMsg = "系统错误: " + err.Error()
+		}
+		return
+	}
+
+	// 解析为结构体
+	var ac vo.LeiGodAccount
+	if err = json.Unmarshal([]byte(data.Data), &ac); err != nil {
+		log.Printf("用户[%s]已绑定雷神账号解析失败: %v", l.userId, err)
+		replyMsg = "系统炸了，请耐心等待修复"
+		return
+	}
+
+	lgu := utils.LeiGodUtil(ac.Account, ac.Password)
+	if err = lgu.Login(); err != nil {
+		return "登录失败: " + err.Error()
+	}
+	if err = lgu.Pause(); err != nil {
+		return "暂停失败: " + err.Error()
+	}
+
+	return "暂停成功"
+}

service/friend.go

@@ -50,3 +50,13 @@ func GetAllEnableChatRank() (records []entity.Friend, err error) {
 	err = client.MySQL.Where("enable_chat_rank = ?", 1).Where("wxid LIKE '%@chatroom'").Find(&records).Error
 	return
 }
+
+// CheckIsEnableCommand
+// @description: 检查用户是否启用了指令
+// @param userId
+// @return flag
+func CheckIsEnableCommand(userId string) (flag bool) {
+	var coo int64
+	client.MySQL.Model(&entity.Friend{}).Where("enable_command = 1").Where("wxid = ?", userId).Count(&coo)
+	return coo > 0
+}

utils/leigod.go

@@ -0,0 +1,169 @@
+package utils
+
+import (
+	"crypto/md5"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"github.com/go-resty/resty/v2"
+	"go-wechat/model"
+	"log"
+)
+
+// LeiGod
+// @description: 雷神加速器相关接口
+type LeiGod interface {
+	Login() error                            // 登录
+	Info() (model.LeiGodUserInfoResp, error) // 获取用户信息
+	Pause() error                            // 暂停加速
+}
+
+type leiGod struct {
+	account, password string // 账号、密码
+	token             string
+}
+
+// LeiGodUtil
+// @description: 创建一个雷神加速器工具类
+// @param userId
+// @return leiGodI
+func LeiGodUtil(account, password string) LeiGod {
+	// 把密码md5一下
+	hash := md5.New()
+	hash.Write([]byte(password))
+	password = fmt.Sprintf("%x", hash.Sum(nil))
+
+	return &leiGod{account: account, password: password}
+}
+
+// Login
+// @description: 登录
+// @receiver l
+// @return string
+func (l *leiGod) Login() (err error) {
+	// 组装参数
+	param := map[string]any{
+		"account_token": nil,
+		"country_code":  86,
+		"lang":          "zh_CN",
+		"os_type":       4,
+		"mobile_num":    l.account,
+		"username":      l.account,
+		"password":      l.password,
+		"region_code":   1,
+		"src_channel":   "guanwang",
+		"sem_ad_img_url": map[string]any{
+			"btn_yrl": "",
+			"url":     "",
+		},
+	}
+	pbs, _ := json.Marshal(param)
+
+	var loginResp model.Response[any]
+	var resp *resty.Response
+
+	res := resty.New()
+	resp, err = res.R().
+		SetHeader("Content-Type", "application/json;chartset=utf-8").
+		SetBody(string(pbs)).
+		SetResult(&loginResp).
+		Post("https://webapi.leigod.com/api/auth/login")
+	if err != nil {
+		log.Panicf("雷神加速器登录失败: %s", err.Error())
+	}
+	log.Printf("雷神加速器登录结果: %s", unicodeToText(resp.String()))
+
+	// 返回状态码不是0表示有错
+	if loginResp.Code != 0 {
+		return errors.New(loginResp.Msg)
+	}
+
+	// 将Data字段转为结构体
+	var bs []byte
+	if bs, err = json.Marshal(loginResp.Data); err != nil {
+		return
+	}
+
+	var loginInfo model.LeiGodLoginResp
+	if err = json.Unmarshal(bs, &loginInfo); err != nil {
+		return
+	}
+
+	if loginInfo.LoginInfo.AccountToken != "" {
+		l.token = loginInfo.LoginInfo.AccountToken
+	}
+
+	return
+}
+
+// Info
+// @description: 获取用户信息
+// @receiver l
+// @return string
+func (l *leiGod) Info() (ui model.LeiGodUserInfoResp, err error) {
+	// 组装参数
+	param := map[string]any{
+		"account_token": l.token,
+		"lang":          "zh_CN",
+		"os_type":       4,
+	}
+	pbs, _ := json.Marshal(param)
+
+	var userInfoResp model.Response[model.LeiGodUserInfoResp]
+	var resp *resty.Response
+
+	res := resty.New()
+	resp, err = res.R().
+		SetHeader("Content-Type", "application/json;chartset=utf-8").
+		SetBody(string(pbs)).
+		SetResult(&userInfoResp).
+		Post("https://webapi.leigod.com/api/user/info")
+	if err != nil {
+		log.Panicf("雷神加速器用户信息获取失败: %s", err.Error())
+	}
+	log.Printf("雷神加速器用户信息获取结果: %s", unicodeToText(resp.String()))
+
+	// 返回状态码不是0表示有错
+	if userInfoResp.Code != 0 {
+		err = errors.New(userInfoResp.Msg)
+		return
+	}
+
+	return userInfoResp.Data, err
+}
+
+// Pause
+// @description: 暂停加速
+// @receiver l
+// @return string
+func (l *leiGod) Pause() (err error) {
+	// 组装参数
+	param := map[string]any{
+		"account_token": l.token,
+		"lang":          "zh_CN",
+		"os_type":       4,
+	}
+	pbs, _ := json.Marshal(param)
+
+	var pauseResp model.Response[any]
+	var resp *resty.Response
+
+	res := resty.New()
+	resp, err = res.R().
+		SetHeader("Content-Type", "application/json;chartset=utf-8").
+		SetBody(string(pbs)).
+		SetResult(&pauseResp).
+		Post("https://webapi.leigod.com/api/user/pause")
+	if err != nil {
+		log.Panicf("雷神加速器暂停失败: %s", err.Error())
+	}
+	log.Printf("雷神加速器暂停结果: %s", unicodeToText(resp.String()))
+
+	// 返回状态码不是0表示有错
+	if pauseResp.Code != 0 {
+		err = errors.New(pauseResp.Msg)
+		return
+	}
+
+	return
+}

utils/string.go

@@ -0,0 +1,15 @@
+package utils
+
+import (
+	"strconv"
+	"strings"
+)
+
+// unicodeToText
+// @description: unicode转文本
+// @param str
+// @return dst
+func unicodeToText(str string) (dst string) {
+	dst, _ = strconv.Unquote(strings.Replace(strconv.Quote(str), `\\u`, `\u`, -1))
+	return
+}

vo/leigod.go

@@ -0,0 +1,8 @@
+package vo
+
+// LeiGodAccount
+// @description: 雷神账号
+type LeiGodAccount struct {
+	Account  string `json:"account"`  // 账号
+	Password string `json:"password"` // 密码
+}