mirror of
https://github.com/Mrs4s/go-cqhttp.git
synced 2025-05-05 03:23:49 +08:00
497 lines
16 KiB
Go
497 lines
16 KiB
Go
package server
|
||
|
||
import (
|
||
"fmt"
|
||
"net/http"
|
||
"strconv"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
|
||
"github.com/Mrs4s/go-cqhttp/coolq"
|
||
"github.com/Mrs4s/go-cqhttp/global"
|
||
"github.com/gorilla/websocket"
|
||
log "github.com/sirupsen/logrus"
|
||
"github.com/tidwall/gjson"
|
||
)
|
||
|
||
type websocketServer struct {
|
||
bot *coolq.CQBot
|
||
token string
|
||
eventConn []*websocketConn
|
||
eventConnMutex sync.Mutex
|
||
handshake string
|
||
}
|
||
|
||
type websocketClient struct {
|
||
conf *global.GoCQReverseWebsocketConfig
|
||
token string
|
||
bot *coolq.CQBot
|
||
|
||
universalConn *websocketConn
|
||
eventConn *websocketConn
|
||
}
|
||
|
||
type websocketConn struct {
|
||
*websocket.Conn
|
||
sync.Mutex
|
||
}
|
||
|
||
var WebsocketServer = &websocketServer{}
|
||
var upgrader = websocket.Upgrader{
|
||
CheckOrigin: func(r *http.Request) bool {
|
||
return true
|
||
},
|
||
}
|
||
|
||
func (s *websocketServer) Run(addr, authToken string, b *coolq.CQBot) {
|
||
s.token = authToken
|
||
s.bot = b
|
||
s.handshake = fmt.Sprintf(`{"_post_method":2,"meta_event_type":"lifecycle","post_type":"meta_event","self_id":%d,"sub_type":"connect","time":%d}`,
|
||
s.bot.Client.Uin, time.Now().Unix())
|
||
b.OnEventPush(s.onBotPushEvent)
|
||
http.HandleFunc("/event", s.event)
|
||
http.HandleFunc("/api", s.api)
|
||
http.HandleFunc("/", s.any)
|
||
go func() {
|
||
log.Infof("CQ Websocket 服务器已启动: %v", addr)
|
||
log.Fatal(http.ListenAndServe(addr, nil))
|
||
}()
|
||
}
|
||
|
||
func NewWebsocketClient(conf *global.GoCQReverseWebsocketConfig, authToken string, b *coolq.CQBot) *websocketClient {
|
||
return &websocketClient{conf: conf, token: authToken, bot: b}
|
||
}
|
||
|
||
func (c *websocketClient) Run() {
|
||
if !c.conf.Enabled {
|
||
return
|
||
}
|
||
if c.conf.ReverseUrl != "" {
|
||
c.connectUniversal()
|
||
} else {
|
||
if c.conf.ReverseApiUrl != "" {
|
||
c.connectApi()
|
||
}
|
||
if c.conf.ReverseEventUrl != "" {
|
||
c.connectEvent()
|
||
}
|
||
}
|
||
c.bot.OnEventPush(c.onBotPushEvent)
|
||
}
|
||
|
||
func (c *websocketClient) connectApi() {
|
||
log.Infof("开始尝试连接到反向Websocket API服务器: %v", c.conf.ReverseApiUrl)
|
||
header := http.Header{
|
||
"X-Client-Role": []string{"API"},
|
||
"X-Self-ID": []string{strconv.FormatInt(c.bot.Client.Uin, 10)},
|
||
"User-Agent": []string{"CQHttp/4.15.0"},
|
||
}
|
||
if c.token != "" {
|
||
header["Authorization"] = []string{"Token " + c.token}
|
||
}
|
||
conn, _, err := websocket.DefaultDialer.Dial(c.conf.ReverseApiUrl, header)
|
||
if err != nil {
|
||
log.Warnf("连接到反向Websocket API服务器 %v 时出现错误: %v", c.conf.ReverseApiUrl, err)
|
||
if c.conf.ReverseReconnectInterval != 0 {
|
||
time.Sleep(time.Millisecond * time.Duration(c.conf.ReverseReconnectInterval))
|
||
c.connectApi()
|
||
}
|
||
return
|
||
}
|
||
log.Infof("已连接到反向Websocket API服务器 %v", c.conf.ReverseApiUrl)
|
||
wrappedConn := &websocketConn{Conn: conn}
|
||
go c.listenApi(wrappedConn, false)
|
||
}
|
||
|
||
func (c *websocketClient) connectEvent() {
|
||
log.Infof("开始尝试连接到反向Websocket Event服务器: %v", c.conf.ReverseEventUrl)
|
||
header := http.Header{
|
||
"X-Client-Role": []string{"Event"},
|
||
"X-Self-ID": []string{strconv.FormatInt(c.bot.Client.Uin, 10)},
|
||
"User-Agent": []string{"CQHttp/4.15.0"},
|
||
}
|
||
if c.token != "" {
|
||
header["Authorization"] = []string{"Token " + c.token}
|
||
}
|
||
conn, _, err := websocket.DefaultDialer.Dial(c.conf.ReverseEventUrl, header)
|
||
if err != nil {
|
||
log.Warnf("连接到反向Websocket Event服务器 %v 时出现错误: %v", c.conf.ReverseEventUrl, err)
|
||
if c.conf.ReverseReconnectInterval != 0 {
|
||
time.Sleep(time.Millisecond * time.Duration(c.conf.ReverseReconnectInterval))
|
||
c.connectEvent()
|
||
}
|
||
return
|
||
}
|
||
log.Infof("已连接到反向Websocket Event服务器 %v", c.conf.ReverseEventUrl)
|
||
c.eventConn = &websocketConn{Conn: conn}
|
||
}
|
||
|
||
func (c *websocketClient) connectUniversal() {
|
||
log.Infof("开始尝试连接到反向Websocket Universal服务器: %v", c.conf.ReverseUrl)
|
||
header := http.Header{
|
||
"X-Client-Role": []string{"Universal"},
|
||
"X-Self-ID": []string{strconv.FormatInt(c.bot.Client.Uin, 10)},
|
||
"User-Agent": []string{"CQHttp/4.15.0"},
|
||
}
|
||
if c.token != "" {
|
||
header["Authorization"] = []string{"Token " + c.token}
|
||
}
|
||
conn, _, err := websocket.DefaultDialer.Dial(c.conf.ReverseUrl, header)
|
||
if err != nil {
|
||
log.Warnf("连接到反向Websocket Universal服务器 %v 时出现错误: %v", c.conf.ReverseUrl, err)
|
||
if c.conf.ReverseReconnectInterval != 0 {
|
||
time.Sleep(time.Millisecond * time.Duration(c.conf.ReverseReconnectInterval))
|
||
c.connectUniversal()
|
||
}
|
||
return
|
||
}
|
||
wrappedConn := &websocketConn{Conn: conn}
|
||
go c.listenApi(wrappedConn, true)
|
||
c.universalConn = wrappedConn
|
||
}
|
||
|
||
func (c *websocketClient) listenApi(conn *websocketConn, u bool) {
|
||
defer conn.Close()
|
||
for {
|
||
_, buf, err := conn.ReadMessage()
|
||
if err != nil {
|
||
log.Warnf("监听反向WS API时出现错误: %v", err)
|
||
break
|
||
}
|
||
|
||
go conn.handleRequest(c.bot, buf)
|
||
|
||
}
|
||
if c.conf.ReverseReconnectInterval != 0 {
|
||
time.Sleep(time.Millisecond * time.Duration(c.conf.ReverseReconnectInterval))
|
||
if !u {
|
||
go c.connectApi()
|
||
}
|
||
}
|
||
}
|
||
|
||
func (c *websocketClient) onBotPushEvent(m coolq.MSG) {
|
||
if c.eventConn != nil {
|
||
log.Debugf("向WS服务器 %v 推送Event: %v", c.eventConn.RemoteAddr().String(), m.ToJson())
|
||
conn := c.eventConn
|
||
conn.Lock()
|
||
defer conn.Unlock()
|
||
_ = c.eventConn.SetWriteDeadline(time.Now().Add(time.Second * 15))
|
||
if err := c.eventConn.WriteJSON(m); err != nil {
|
||
log.Warnf("向WS服务器 %v 推送Event时出现错误: %v", c.eventConn.RemoteAddr().String(), err)
|
||
_ = c.eventConn.Close()
|
||
if c.conf.ReverseReconnectInterval != 0 {
|
||
time.Sleep(time.Millisecond * time.Duration(c.conf.ReverseReconnectInterval))
|
||
c.connectEvent()
|
||
}
|
||
}
|
||
}
|
||
if c.universalConn != nil {
|
||
log.Debugf("向WS服务器 %v 推送Event: %v", c.universalConn.RemoteAddr().String(), m.ToJson())
|
||
conn := c.universalConn
|
||
conn.Lock()
|
||
defer conn.Unlock()
|
||
_ = c.universalConn.SetWriteDeadline(time.Now().Add(time.Second * 15))
|
||
if err := c.universalConn.WriteJSON(m); err != nil {
|
||
log.Warnf("向WS服务器 %v 推送Event时出现错误: %v", c.universalConn.RemoteAddr().String(), err)
|
||
_ = c.universalConn.Close()
|
||
if c.conf.ReverseReconnectInterval != 0 {
|
||
time.Sleep(time.Millisecond * time.Duration(c.conf.ReverseReconnectInterval))
|
||
c.connectUniversal()
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
func (s *websocketServer) event(w http.ResponseWriter, r *http.Request) {
|
||
if s.token != "" {
|
||
if auth := r.URL.Query().Get("access_token"); auth != s.token && auth != "" {
|
||
log.Warnf("已拒绝 %v 的 Websocket 请求: Token错误", r.RemoteAddr)
|
||
w.WriteHeader(401)
|
||
return
|
||
} else if auth := r.Header.Get("Authorization"); auth != "" {
|
||
if strings.SplitN(auth, " ", 2)[1] != s.token {
|
||
log.Warnf("已拒绝 %v 的 Websocket 请求: Token错误", r.RemoteAddr)
|
||
w.WriteHeader(401)
|
||
return
|
||
}
|
||
} else {
|
||
log.Warnf("已拒绝 %v 的 Websocket 请求: 空Token", r.RemoteAddr)
|
||
w.WriteHeader(401)
|
||
return
|
||
}
|
||
}
|
||
c, err := upgrader.Upgrade(w, r, nil)
|
||
if err != nil {
|
||
log.Warnf("处理 Websocket 请求时出现错误: %v", err)
|
||
return
|
||
}
|
||
err = c.WriteMessage(websocket.TextMessage, []byte(s.handshake))
|
||
if err != nil {
|
||
log.Warnf("Websocket 握手时出现错误: %v", err)
|
||
c.Close()
|
||
return
|
||
}
|
||
|
||
log.Infof("接受 Websocket 连接: %v (/event)", r.RemoteAddr)
|
||
|
||
conn := &websocketConn{Conn: c}
|
||
|
||
s.eventConnMutex.Lock()
|
||
s.eventConn = append(s.eventConn, conn)
|
||
s.eventConnMutex.Unlock()
|
||
}
|
||
|
||
func (s *websocketServer) api(w http.ResponseWriter, r *http.Request) {
|
||
if s.token != "" {
|
||
if auth := r.URL.Query().Get("access_token"); auth != s.token && auth != "" {
|
||
log.Warnf("已拒绝 %v 的 Websocket 请求: Token错误", r.RemoteAddr)
|
||
w.WriteHeader(401)
|
||
return
|
||
} else if auth := r.Header.Get("Authorization"); auth != "" {
|
||
if strings.SplitN(auth, " ", 2)[1] != s.token {
|
||
log.Warnf("已拒绝 %v 的 Websocket 请求: Token错误", r.RemoteAddr)
|
||
w.WriteHeader(401)
|
||
return
|
||
}
|
||
} else {
|
||
log.Warnf("已拒绝 %v 的 Websocket 请求: 空Token", r.RemoteAddr)
|
||
w.WriteHeader(401)
|
||
return
|
||
}
|
||
}
|
||
c, err := upgrader.Upgrade(w, r, nil)
|
||
if err != nil {
|
||
log.Warnf("处理 Websocket 请求时出现错误: %v", err)
|
||
return
|
||
}
|
||
log.Infof("接受 Websocket 连接: %v (/api)", r.RemoteAddr)
|
||
conn := &websocketConn{Conn: c}
|
||
go s.listenApi(conn)
|
||
}
|
||
|
||
func (s *websocketServer) any(w http.ResponseWriter, r *http.Request) {
|
||
if s.token != "" {
|
||
if auth := r.URL.Query().Get("access_token"); auth != s.token && auth != "" {
|
||
log.Warnf("已拒绝 %v 的 Websocket 请求: Token错误", r.RemoteAddr)
|
||
w.WriteHeader(401)
|
||
return
|
||
} else if auth := r.Header.Get("Authorization"); auth != "" {
|
||
if strings.SplitN(auth, " ", 2)[1] != s.token {
|
||
log.Warnf("已拒绝 %v 的 Websocket 请求: Token错误", r.RemoteAddr)
|
||
w.WriteHeader(401)
|
||
return
|
||
}
|
||
} else {
|
||
log.Warnf("已拒绝 %v 的 Websocket 请求: 空Token", r.RemoteAddr)
|
||
w.WriteHeader(401)
|
||
return
|
||
}
|
||
}
|
||
c, err := upgrader.Upgrade(w, r, nil)
|
||
if err != nil {
|
||
log.Warnf("处理 Websocket 请求时出现错误: %v", err)
|
||
return
|
||
}
|
||
err = c.WriteMessage(websocket.TextMessage, []byte(s.handshake))
|
||
if err != nil {
|
||
log.Warnf("Websocket 握手时出现错误: %v", err)
|
||
c.Close()
|
||
return
|
||
}
|
||
|
||
log.Infof("接受 Websocket 连接: %v (/)", r.RemoteAddr)
|
||
conn := &websocketConn{Conn: c}
|
||
s.eventConn = append(s.eventConn, conn)
|
||
s.listenApi(conn)
|
||
}
|
||
|
||
func (s *websocketServer) listenApi(c *websocketConn) {
|
||
defer c.Close()
|
||
for {
|
||
t, payload, err := c.ReadMessage()
|
||
if err != nil {
|
||
break
|
||
}
|
||
|
||
if t == websocket.TextMessage {
|
||
go c.handleRequest(s.bot, payload)
|
||
}
|
||
}
|
||
}
|
||
|
||
func (c *websocketConn) handleRequest(bot *coolq.CQBot, payload []byte) {
|
||
defer func() {
|
||
if err := recover(); err != nil {
|
||
log.Printf("处置WS命令时发生无法恢复的异常：%v", err)
|
||
c.Close()
|
||
}
|
||
}()
|
||
|
||
j := gjson.ParseBytes(payload)
|
||
t := strings.ReplaceAll(j.Get("action").Str, "_async", "")
|
||
log.Debugf("WS接收到API调用: %v 参数: %v", t, j.Get("params").Raw)
|
||
if f, ok := wsApi[t]; ok {
|
||
ret := f(bot, j.Get("params"))
|
||
if j.Get("echo").Exists() {
|
||
ret["echo"] = j.Get("echo").Value()
|
||
}
|
||
c.Lock()
|
||
defer c.Unlock()
|
||
_ = c.WriteJSON(ret)
|
||
}
|
||
}
|
||
|
||
func (s *websocketServer) onBotPushEvent(m coolq.MSG) {
|
||
s.eventConnMutex.Lock()
|
||
defer s.eventConnMutex.Unlock()
|
||
for i, l := 0, len(s.eventConn); i < l; i++ {
|
||
conn := s.eventConn[i]
|
||
log.Debugf("向WS客户端 %v 推送Event: %v", conn.RemoteAddr().String(), m.ToJson())
|
||
conn.Lock()
|
||
if err := conn.WriteMessage(websocket.TextMessage, []byte(m.ToJson())); err != nil {
|
||
_ = conn.Close()
|
||
next := i + 1
|
||
if next >= l {
|
||
next = l - 1
|
||
}
|
||
s.eventConn[i], s.eventConn[next] = s.eventConn[next], s.eventConn[i]
|
||
s.eventConn = append(s.eventConn[:next], s.eventConn[next+1:]...)
|
||
i--
|
||
l--
|
||
conn = nil
|
||
continue
|
||
}
|
||
conn.Unlock()
|
||
}
|
||
}
|
||
|
||
var wsApi = map[string]func(*coolq.CQBot, gjson.Result) coolq.MSG{
|
||
"get_login_info": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQGetLoginInfo()
|
||
},
|
||
"get_friend_list": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQGetFriendList()
|
||
},
|
||
"get_group_list": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQGetGroupList(p.Get("no_cache").Bool())
|
||
},
|
||
"get_group_info": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQGetGroupInfo(p.Get("group_id").Int())
|
||
},
|
||
"get_group_member_list": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQGetGroupMemberList(p.Get("group_id").Int())
|
||
},
|
||
"get_group_member_info": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQGetGroupMemberInfo(
|
||
p.Get("group_id").Int(), p.Get("user_id").Int(),
|
||
p.Get("no_cache").Bool(),
|
||
)
|
||
},
|
||
"send_msg": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
autoEscape := global.EnsureBool(p.Get("auto_escape"), false)
|
||
if p.Get("message_type").Str == "private" {
|
||
return bot.CQSendPrivateMessage(p.Get("user_id").Int(), p.Get("message"), autoEscape)
|
||
}
|
||
if p.Get("message_type").Str == "group" {
|
||
return bot.CQSendGroupMessage(p.Get("group_id").Int(), p.Get("message"), autoEscape)
|
||
}
|
||
if p.Get("group_id").Int() != 0 {
|
||
return bot.CQSendGroupMessage(p.Get("group_id").Int(), p.Get("message"), autoEscape)
|
||
}
|
||
if p.Get("user_id").Int() != 0 {
|
||
return bot.CQSendPrivateMessage(p.Get("user_id").Int(), p.Get("message"), autoEscape)
|
||
}
|
||
return coolq.MSG{}
|
||
},
|
||
"send_group_msg": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQSendGroupMessage(p.Get("group_id").Int(), p.Get("message"), global.EnsureBool(p.Get("auto_escape"), false))
|
||
},
|
||
"send_group_forward_msg": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQSendGroupForwardMessage(p.Get("group_id").Int(), p.Get("messages"))
|
||
},
|
||
"send_private_msg": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQSendPrivateMessage(p.Get("user_id").Int(), p.Get("message"), global.EnsureBool(p.Get("auto_escape"), false))
|
||
},
|
||
"delete_msg": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQDeleteMessage(int32(p.Get("message_id").Int()))
|
||
},
|
||
"set_friend_add_request": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
apr := true
|
||
if p.Get("approve").Exists() {
|
||
apr = p.Get("approve").Bool()
|
||
}
|
||
return bot.CQProcessFriendRequest(p.Get("flag").Str, apr)
|
||
},
|
||
"set_group_add_request": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
subType := p.Get("sub_type").Str
|
||
apr := true
|
||
if subType == "" {
|
||
subType = p.Get("type").Str
|
||
}
|
||
if p.Get("approve").Exists() {
|
||
apr = p.Get("approve").Bool()
|
||
}
|
||
return bot.CQProcessGroupRequest(p.Get("flag").Str, subType, p.Get("reason").Str, apr)
|
||
},
|
||
"set_group_card": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQSetGroupCard(p.Get("group_id").Int(), p.Get("user_id").Int(), p.Get("card").Str)
|
||
},
|
||
"set_group_special_title": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQSetGroupSpecialTitle(p.Get("group_id").Int(), p.Get("user_id").Int(), p.Get("special_title").Str)
|
||
},
|
||
"set_group_kick": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQSetGroupKick(p.Get("group_id").Int(), p.Get("user_id").Int(), p.Get("message").Str)
|
||
},
|
||
"set_group_ban": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQSetGroupBan(p.Get("group_id").Int(), p.Get("user_id").Int(), func() uint32 {
|
||
if p.Get("duration").Exists() {
|
||
return uint32(p.Get("duration").Int())
|
||
}
|
||
return 1800
|
||
}())
|
||
},
|
||
"set_group_whole_ban": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQSetGroupWholeBan(p.Get("group_id").Int(), func() bool {
|
||
if p.Get("enable").Exists() {
|
||
return p.Get("enable").Bool()
|
||
}
|
||
return true
|
||
}())
|
||
},
|
||
"set_group_name": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQSetGroupName(p.Get("group_id").Int(), p.Get("name").Str)
|
||
},
|
||
"set_group_leave": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQSetGroupLeave(p.Get("group_id").Int())
|
||
},
|
||
"get_image": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQGetImage(p.Get("file").Str)
|
||
},
|
||
"get_forward_msg": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQGetForwardMessage(p.Get("message_id").Str)
|
||
},
|
||
"get_group_msg": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQGetGroupMessage(int32(p.Get("message_id").Int()))
|
||
},
|
||
"get_group_honor_info": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQGetGroupHonorInfo(p.Get("group_id").Int(), p.Get("type").Str)
|
||
},
|
||
"can_send_image": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQCanSendImage()
|
||
},
|
||
"can_send_record": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQCanSendRecord()
|
||
},
|
||
"get_status": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQGetStatus()
|
||
},
|
||
"get_version_info": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQGetVersionInfo()
|
||
},
|
||
".handle_quick_operation": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
|
||
return bot.CQHandleQuickOperation(p.Get("context"), p.Get("operation"))
|
||
},
|
||
}
|