334 lines
8.0 KiB
Go
334 lines
8.0 KiB
Go
package push
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/tidwall/gjson"
|
|
|
|
"github.com/sjkhsl/study_xxqg/conf"
|
|
"github.com/sjkhsl/study_xxqg/lib"
|
|
"github.com/sjkhsl/study_xxqg/lib/state"
|
|
"github.com/sjkhsl/study_xxqg/model"
|
|
"github.com/sjkhsl/study_xxqg/utils"
|
|
)
|
|
|
|
var (
|
|
qq *QQ
|
|
)
|
|
|
|
type qqPlugin func(event *Event, args []string)
|
|
|
|
type QQ struct {
|
|
postAdd string
|
|
plugins map[string]qqPlugin
|
|
}
|
|
|
|
func InitQQ() *QQ {
|
|
config := conf.GetConfig()
|
|
q := new(QQ)
|
|
qq = q
|
|
q.postAdd = config.QQ.PostAddr
|
|
q.plugins = make(map[string]qqPlugin, 1)
|
|
q.newPlugin("user", qqGetUser)
|
|
q.newPlugin("score", qqGetScore)
|
|
q.newPlugin("fail", qqGetFailUser)
|
|
q.newPlugin("study", qqStudy)
|
|
q.newPlugin("login", qqLogin)
|
|
q.newPlugin("help", qqHelp)
|
|
return q
|
|
}
|
|
|
|
func (q *QQ) newPlugin(text string, plugin qqPlugin) {
|
|
q.plugins[text] = plugin
|
|
}
|
|
|
|
func (q *QQ) ServeHTTP(writer http.ResponseWriter, request *http.Request) {
|
|
data, _ := io.ReadAll(request.Body)
|
|
go q.handle(data)
|
|
writer.WriteHeader(204)
|
|
}
|
|
|
|
func (q *QQ) handle(data []byte) {
|
|
defer func() {
|
|
err := recover()
|
|
if err != nil {
|
|
log.Errorln("qq消息处理错误")
|
|
}
|
|
}()
|
|
config := conf.GetConfig()
|
|
e := new(Event)
|
|
err := json.Unmarshal(data, e)
|
|
if err != nil {
|
|
log.Errorln(err.Error())
|
|
return
|
|
}
|
|
e.qq = q
|
|
if e.PostType == "message" {
|
|
log.Infoln("收到qq消息 ==》 " + e.Message)
|
|
// 遍历白名单列表
|
|
for _, id := range config.QQ.WhiteList {
|
|
if e.GroupId == id || e.UserId == id {
|
|
for text, plugin := range q.plugins {
|
|
messages := strings.Split(e.Message, " ")
|
|
if messages[0] == "."+text {
|
|
plugin(e, messages[1:])
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
log.Infoln("消息来源不在白名单中！")
|
|
|
|
}
|
|
}
|
|
|
|
func qqHelp(event *Event, _ []string) {
|
|
help := ".user 查询用户\n.fail 查询过期用户\n.study 对一个用户进行学习\n.score 查询用户分数\n.login 登录一个用户"
|
|
event.Send(help)
|
|
}
|
|
|
|
func qqLogin(event *Event, _ []string) {
|
|
core := &lib.Core{ShowBrowser: conf.GetConfig().ShowBrowser, Push: func(id string, kind string, message string) {
|
|
if kind == "flush" {
|
|
event.Send(message)
|
|
} else {
|
|
if conf.GetConfig().LogLevel == "debug" {
|
|
event.Send(message)
|
|
}
|
|
}
|
|
}}
|
|
_, err := core.L(conf.GetConfig().Retry.Times, "")
|
|
if err != nil {
|
|
event.Send(err.Error())
|
|
}
|
|
}
|
|
|
|
func qqStudy(event *Event, args []string) {
|
|
users, err := model.Query()
|
|
if err != nil {
|
|
event.Send(err.Error())
|
|
return
|
|
}
|
|
var user *model.User
|
|
if len(users) == 1 {
|
|
user = users[0]
|
|
} else {
|
|
if len(args) <= 0 {
|
|
event.Send("缺少序号参数，请输入 .study 序号")
|
|
return
|
|
} else {
|
|
index, err := strconv.Atoi(args[0])
|
|
if err != nil {
|
|
event.Send(err.Error())
|
|
return
|
|
}
|
|
user = users[index]
|
|
}
|
|
}
|
|
core := &lib.Core{ShowBrowser: conf.GetConfig().ShowBrowser, Push: func(id string, kind string, message string) {
|
|
if kind == "flush" {
|
|
event.Send(message)
|
|
} else {
|
|
if conf.GetConfig().LogLevel == "debug" {
|
|
event.Send(message)
|
|
}
|
|
}
|
|
}}
|
|
core.Init()
|
|
state.Add(user.Uid, core)
|
|
defer state.Delete(user.Uid)
|
|
defer core.Quit()
|
|
lib.Study(core, user)
|
|
|
|
}
|
|
|
|
func qqGetScore(event *Event, _ []string) {
|
|
users, err := model.Query()
|
|
if err != nil {
|
|
event.Send(err.Error())
|
|
return
|
|
}
|
|
for _, user := range users {
|
|
score, err := lib.GetUserScore(user.ToCookies())
|
|
if err != nil {
|
|
event.Send(err.Error())
|
|
continue
|
|
}
|
|
event.Send(user.Nick + "\n" + lib.FormatScore(score))
|
|
}
|
|
}
|
|
|
|
func qqGetFailUser(event *Event, _ []string) {
|
|
user, err := model.QueryFailUser()
|
|
if err != nil {
|
|
event.Send(err.Error())
|
|
return
|
|
}
|
|
result := ""
|
|
for i, user := range user {
|
|
result += fmt.Sprintf("%d %v %v\n", i, user.Nick, utils.Stamp2Str(user.LoginTime))
|
|
}
|
|
event.Send(result)
|
|
}
|
|
|
|
func qqGetUser(event *Event, _ []string) {
|
|
users, err := model.Query()
|
|
if err != nil {
|
|
event.Send(err.Error())
|
|
return
|
|
}
|
|
result := ""
|
|
for i, user := range users {
|
|
result += fmt.Sprintf("%d %v %v\n", i, user.Nick, utils.Stamp2Str(user.LoginTime))
|
|
}
|
|
event.Send(result)
|
|
}
|
|
|
|
type (
|
|
anonymous struct {
|
|
Id int `json:"id"`
|
|
Name string `json:"name"`
|
|
Flag string `json:"flag"`
|
|
}
|
|
|
|
Files struct {
|
|
Id string `json:"id"`
|
|
Name string `json:"name"`
|
|
Size int64 `json:"size"`
|
|
Busid int64 `json:"busid"`
|
|
FileUrl string `json:"url"`
|
|
}
|
|
|
|
Status struct {
|
|
AppEnabled bool `json:"app_enabled"`
|
|
AppGood bool `json:"app_good"`
|
|
AppInitialized bool `json:"app_initialized"`
|
|
Good bool `json:"good"`
|
|
Online bool `json:"online"`
|
|
PluginsGood interface{} `json:"plugins_good"`
|
|
Stat struct {
|
|
PacketReceived int `json:"packet_received"`
|
|
PacketSent int `json:"packet_sent"`
|
|
PacketLost int `json:"packet_lost"`
|
|
MessageReceived int `json:"message_received"`
|
|
MessageSent int `json:"message_sent"`
|
|
DisconnectTimes int `json:"disconnect_times"`
|
|
LostTimes int `json:"lost_times"`
|
|
LastMessageTime int `json:"last_message_time"`
|
|
} `json:"stat"`
|
|
}
|
|
|
|
MessageIds struct {
|
|
MessageID int32 `json:"message_id"`
|
|
}
|
|
|
|
Senders struct {
|
|
Age int `json:"age"`
|
|
Area string `json:"area"`
|
|
Card string `json:"card"`
|
|
Level string `json:"level"`
|
|
NickName string `json:"nickname"`
|
|
Role string `json:"role"`
|
|
Sex string `json:"sex"`
|
|
Title string `json:"title"`
|
|
UserId int `json:"user_id"`
|
|
}
|
|
|
|
// Event
|
|
/*
|
|
* 事件
|
|
*
|
|
*/
|
|
Event struct {
|
|
qq *QQ
|
|
Anonymous anonymous `json:"anonymous"`
|
|
Font int `json:"font"`
|
|
GroupId int64 `json:"group_id"`
|
|
Message string `json:"message"`
|
|
MessageType string `json:"message_type"`
|
|
PostType string `json:"post_type"`
|
|
RawMessage string `json:"raw_message"`
|
|
SelfId int64 `json:"self_id"`
|
|
Sender Senders `json:"sender"`
|
|
SubType string `json:"sub_type"`
|
|
UserId int64 `json:"user_id"`
|
|
Time int `json:"time"`
|
|
NoticeType string `json:"notice_type"`
|
|
RequestType string `json:"request_type"`
|
|
Comment string `json:"comment"`
|
|
Flag string `json:"flag"`
|
|
OperatorID int `json:"operator_id"`
|
|
File Files `json:"file"`
|
|
Duration int64 `json:"duration"`
|
|
TargetId int64 `json:"target_id"` // 运气王id
|
|
HonorType string `json:"honor_type"`
|
|
MetaEventType string `json:"meta_event_type"`
|
|
Status Status `json:"status"`
|
|
Interval int `json:"interval"`
|
|
CardNew string `json:"card_new"` // 新名片
|
|
CardOld string `json:"card_old"` // 旧名片
|
|
MessageIds
|
|
|
|
GuildID int64 `json:"guild_id"`
|
|
ChannelID int64 `json:"channel_id"`
|
|
}
|
|
)
|
|
|
|
func (e *Event) sendGroupMsg(groupId int64, message any) int {
|
|
if _, ok := message.(string); ok {
|
|
message = map[string]any{
|
|
"type": "text",
|
|
"data": map[string]any{
|
|
"text": message,
|
|
},
|
|
}
|
|
}
|
|
response, err := utils.GetClient().R().SetHeader("Authorization", conf.GetConfig().QQ.AccessToken).SetBodyJsonMarshal(map[string]any{
|
|
"action": "send_group_msg",
|
|
"params": map[string]any{
|
|
"group_id": groupId,
|
|
"message": message,
|
|
},
|
|
}).Post(e.qq.postAdd)
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
return int(gjson.GetBytes(response.Bytes(), "data").Int())
|
|
}
|
|
|
|
func (e *Event) Send(message any) {
|
|
if e.MessageType == "group" {
|
|
e.sendGroupMsg(e.GroupId, message)
|
|
} else {
|
|
e.sendPrivateMsg(e.UserId, message)
|
|
}
|
|
}
|
|
|
|
func (e *Event) sendPrivateMsg(userId int64, message any) int {
|
|
if _, ok := message.(string); ok {
|
|
message = map[string]any{
|
|
"type": "text",
|
|
"data": map[string]any{
|
|
"text": message,
|
|
},
|
|
}
|
|
}
|
|
response, err := utils.GetClient().R().SetHeader("Authorization", conf.GetConfig().QQ.AccessToken).SetBodyJsonMarshal(map[string]any{
|
|
"action": "send_private_msg",
|
|
"params": map[string]any{
|
|
"user_id": userId,
|
|
"message": message,
|
|
},
|
|
}).Post(e.qq.postAdd)
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
return int(gjson.GetBytes(response.Bytes(), "data").Int())
|
|
}
|