shkong/go-cqhttp
1
0
Fork 0
mirror of https://github.com/Mrs4s/go-cqhttp.git synced 2025-05-05 03:23:49 +08:00
Code Issues Projects Releases Wiki Activity

websocket echo field supported.

Browse Source
This commit is contained in:
Mrs4s 2020-07-25 20:35:15 +08:00
parent f19c76d4e1
commit 549525236d
1 changed files with 58 additions and 54 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

112
server/websocket.go
View File

@ -116,7 +116,11 @@ func (s *websocketServer) listenApi(c *websocket.Conn) {
t := strings.ReplaceAll(j.Get("action").Str, "_async", "") //TODO: async support t := strings.ReplaceAll(j.Get("action").Str, "_async", "") //TODO: async support
log.Infof("API调用: %v", j.Get("action").Str) log.Infof("API调用: %v", j.Get("action").Str)
if f, ok := wsApi[t]; ok { if f, ok := wsApi[t]; ok {
_ = c.WriteMessage(websocket.TextMessage, []byte(f(s.bot, j.Get("params")))) ret := f(s.bot, j.Get("params"))
if j.Get("echo").Exists() {
ret["echo"] = j.Get("echo").Value()
}
_ = c.WriteJSON(ret)
} }
} }
} }
@ -139,54 +143,54 @@ func (s *websocketServer) onBotPushEvent(m coolq.MSG) {
} }
} }
var wsApi = map[string]func(*coolq.CQBot, gjson.Result) string{ var wsApi = map[string]func(*coolq.CQBot, gjson.Result) coolq.MSG{
"get_login_info": func(bot *coolq.CQBot, p gjson.Result) string { "get_login_info": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQGetLoginInfo().ToJson() return bot.CQGetLoginInfo()
}, },
"get_friend_list": func(bot *coolq.CQBot, p gjson.Result) string { "get_friend_list": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQGetFriendList().ToJson() return bot.CQGetFriendList()
}, },
"get_group_list": func(bot *coolq.CQBot, p gjson.Result) string { "get_group_list": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQGetGroupList().ToJson() return bot.CQGetGroupList()
}, },
"get_group_info": func(bot *coolq.CQBot, p gjson.Result) string { "get_group_info": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQGetGroupInfo(p.Get("group_id").Int()).ToJson() return bot.CQGetGroupInfo(p.Get("group_id").Int())
}, },
"get_group_member_list": func(bot *coolq.CQBot, p gjson.Result) string { "get_group_member_list": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQGetGroupMemberList(p.Get("group_id").Int()).ToJson() return bot.CQGetGroupMemberList(p.Get("group_id").Int())
}, },
"get_group_member_info": func(bot *coolq.CQBot, p gjson.Result) string { "get_group_member_info": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQGetGroupMemberInfo( return bot.CQGetGroupMemberInfo(
p.Get("group_id").Int(), p.Get("user_id").Int(), p.Get("group_id").Int(), p.Get("user_id").Int(),
p.Get("no_cache").Bool(), p.Get("no_cache").Bool(),
).ToJson() )
}, },
"send_msg": func(bot *coolq.CQBot, p gjson.Result) string { "send_msg": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
if p.Get("group_id").Int() != 0 { if p.Get("group_id").Int() != 0 {
return bot.CQSendGroupMessage(p.Get("group_id").Int(), p.Get("message").Str).ToJson() return bot.CQSendGroupMessage(p.Get("group_id").Int(), p.Get("message").Str)
} }
if p.Get("user_id").Int() != 0 { if p.Get("user_id").Int() != 0 {
return bot.CQSendPrivateMessage(p.Get("user_id").Int(), p.Get("message").Str).ToJson() return bot.CQSendPrivateMessage(p.Get("user_id").Int(), p.Get("message").Str)
} }
return "" return coolq.MSG{}
}, },
"send_group_msg": func(bot *coolq.CQBot, p gjson.Result) string { "send_group_msg": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQSendGroupMessage(p.Get("group_id").Int(), p.Get("message").Str).ToJson() return bot.CQSendGroupMessage(p.Get("group_id").Int(), p.Get("message").Str)
}, },
"send_private_msg": func(bot *coolq.CQBot, p gjson.Result) string { "send_private_msg": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQSendPrivateMessage(p.Get("user_id").Int(), p.Get("message").Str).ToJson() return bot.CQSendPrivateMessage(p.Get("user_id").Int(), p.Get("message").Str)
}, },
"delete_msg": func(bot *coolq.CQBot, p gjson.Result) string { "delete_msg": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQDeleteMessage(int32(p.Get("message_id").Int())).ToJson() return bot.CQDeleteMessage(int32(p.Get("message_id").Int()))
}, },
"set_friend_add_request": func(bot *coolq.CQBot, p gjson.Result) string { "set_friend_add_request": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
apr := true apr := true
if p.Get("approve").Exists() { if p.Get("approve").Exists() {
apr = p.Get("approve").Bool() apr = p.Get("approve").Bool()
} }
return bot.CQProcessFriendRequest(p.Get("flag").Str, apr).ToJson() return bot.CQProcessFriendRequest(p.Get("flag").Str, apr)
}, },
"set_group_add_request": func(bot *coolq.CQBot, p gjson.Result) string { "set_group_add_request": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
subType := p.Get("sub_type").Str subType := p.Get("sub_type").Str
apr := true apr := true
if subType == "" { if subType == "" {
@ -195,50 +199,50 @@ var wsApi = map[string]func(*coolq.CQBot, gjson.Result) string{
if p.Get("approve").Exists() { if p.Get("approve").Exists() {
apr = p.Get("approve").Bool() apr = p.Get("approve").Bool()
} }
return bot.CQProcessGroupRequest(p.Get("flag").Str, subType, apr).ToJson() return bot.CQProcessGroupRequest(p.Get("flag").Str, subType, apr)
}, },
"set_group_card": func(bot *coolq.CQBot, p gjson.Result) string { "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).ToJson() 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) string { "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).ToJson() 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) string { "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).ToJson() 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) string { "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(), uint32(p.Get("duration").Int())).ToJson() return bot.CQSetGroupBan(p.Get("group_id").Int(), p.Get("user_id").Int(), uint32(p.Get("duration").Int()))
}, },
"set_group_whole_ban": func(bot *coolq.CQBot, p gjson.Result) string { "set_group_whole_ban": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQSetGroupWholeBan(p.Get("group_id").Int(), func() bool { return bot.CQSetGroupWholeBan(p.Get("group_id").Int(), func() bool {
if p.Get("enable").Exists() { if p.Get("enable").Exists() {
return p.Get("enable").Bool() return p.Get("enable").Bool()
} }
return true return true
}()).ToJson() }())
}, },
"set_group_name": func(bot *coolq.CQBot, p gjson.Result) string { "set_group_name": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQSetGroupName(p.Get("group_id").Int(), p.Get("name").Str).ToJson() return bot.CQSetGroupName(p.Get("group_id").Int(), p.Get("name").Str)
}, },
"get_image": func(bot *coolq.CQBot, p gjson.Result) string { "get_image": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQGetImage(p.Get("file").Str).ToJson() return bot.CQGetImage(p.Get("file").Str)
}, },
"get_group_msg": func(bot *coolq.CQBot, p gjson.Result) string { "get_group_msg": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQGetGroupMessage(int32(p.Get("message_id").Int())).ToJson() return bot.CQGetGroupMessage(int32(p.Get("message_id").Int()))
}, },
"can_send_image": func(bot *coolq.CQBot, p gjson.Result) string { "can_send_image": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQCanSendImage().ToJson() return bot.CQCanSendImage()
}, },
"can_send_record": func(bot *coolq.CQBot, p gjson.Result) string { "can_send_record": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQCanSendRecord().ToJson() return bot.CQCanSendRecord()
}, },
"get_status": func(bot *coolq.CQBot, p gjson.Result) string { "get_status": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQGetStatus().ToJson() return bot.CQGetStatus()
}, },
"get_version_info": func(bot *coolq.CQBot, p gjson.Result) string { "get_version_info": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQGetVersionInfo().ToJson() return bot.CQGetVersionInfo()
}, },
".handle_quick_operation": func(bot *coolq.CQBot, p gjson.Result) string { ".handle_quick_operation": func(bot *coolq.CQBot, p gjson.Result) coolq.MSG {
return bot.CQHandleQuickOperation(p.Get("context"), p.Get("operation")).ToJson() return bot.CQHandleQuickOperation(p.Get("context"), p.Get("operation"))
}, },
} }