zhamao-robot/zhamao-framework
1
0
Fork 0
mirror of https://github.com/zhamao-robot/zhamao-framework.git synced 2026-07-23 00:25:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Finished single websocket framework. It works.

Browse Source 
完成了全反向websocket框架，全面支持多机器人
This commit is contained in:
jerry
2018-06-15 09:18:53 +08:00
parent ab6408e0eb
commit 2de4337600
17 changed files with 601 additions and 441 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
src/cqbot/utils/APIHandler.php
View File

@@ -9,14 +9,14 @@
class APIHandler
{
static function execute($cmd, $res = null) {
if (!isset($cmd[0])) return;
switch ($cmd[0]) {
if (!isset($cmd["type"])) return false;
switch ($cmd["type"]) {
case "set_friend_add_request":
$id = $cmd[1];
$id = $cmd["params"][0];
$msg = "Hi～你好！";
$msg .= "\n第一次见面请多关照！";
CQUtil::sendPrivateMsg($id, $msg);
break;
CQUtil::sendPrivateMsg($id, $msg, $cmd["self_id"]);
return true;
case "get_friend_list":
$friend = $res["data"][0]["friends"];
$list = [];
@@ -25,9 +25,9 @@ class APIHandler
}
Buffer::set("friend_list", $list);
Console::put(Console::setColor("已读取" . count(Buffer::get("friend_list")) . "个好友", "blue"));
break;
return true;
case "update_group_member_list":
$group_id = $cmd[1];
$group_id = $cmd["params"][0];
$info_data = $res["data"];
Console::info(Console::setColor("Updating group $group_id members, it will take several minutes.", "yellow"));
foreach ($info_data as $k => $v) {
@@ -35,11 +35,11 @@ class APIHandler
CQUtil::getGroup($group_id)->setMember($v["user_id"], $s);
$s->updateData();
}
break;
return true;
case "update_group_member_info":
$info_data = $res["data"];
$group = $cmd[1];
$user = $cmd[2];
$group = $cmd["params"][0];
$user = $cmd["params"][1];
$g = CQUtil::getGroup($group);
$member = $g->getMember($user);
$member->setAttribute($info_data);
@@ -48,43 +48,44 @@ class APIHandler
$member->setLastSentTime($info_data["last_sent_time"]);
$member->setRole($info_data["role"]);
Console::info("Updated group member information: " . $group . ":" . $user);
break;
return true;
case "update_group_info":
$group = $res["data"];
$current = $cmd[1];
$current = $cmd["params"][0];
$list = [];
foreach ($group as $k => $v) {
$list[$v["group_id"]] = $group[$k];
}
if (!isset($list[$current]) && Buffer::array_key_exists("groups", $current)) {
Buffer::unset("groups", $current);
break;
return true;
}
$g = CQUtil::getGroup($current);
$g->setGroupName($list[$current]["group_name"]);
$g->setPrefix($list[$current]["prefix"]);
break;
return true;
case "get_group_member_list":
$group_data = $res["data"];
$ls = Buffer::get("group_list");
$group_id = $cmd[1];
$group_id = $cmd["params"][0];
$ls[$group_id]["member"] = $group_data;
$group = new Group($group_id, $ls[$group_id]);
$group = new Group($group_id, $ls[$group_id], $cmd["self_id"]);
//TODO: 添加获取API时多账号对群的实例化的支持
Buffer::appendKey("groups", $group_id, $group);
break;
return true;
case "get_group_list":
$group = $res["data"];
$list = [];
foreach ($group as $k => $v) {
$list[$v["group_id"]] = $group[$k];
CQUtil::sendAPI(["action" => "get_group_member_list", "params" => ["group_id" => $v["group_id"]]], ["get_group_member_list", $v["group_id"]]);
CQUtil::sendAPI(CQUtil::getApiConnectionByQQ($cmd["self_id"])->fd, ["action" => "get_group_member_list", "params" => ["group_id" => $v["group_id"]]], ["get_group_member_list", $v["group_id"]]);
}
Buffer::set("group_list", $list);
Console::put(Console::setColor("已读取" . count(Buffer::get("group_list")) . "个群", "blue"));
break;
return true;
case "get_version_info":
Buffer::set("version_info", $res["data"]);
break;
return true;
}
return false;
}
}

2
src/cqbot/utils/Buffer.php
View File

@@ -25,6 +25,8 @@ class Buffer
static $out_count;//发送消息数量
/** @var \swoole_atomic $out_count */
static $api_id;//API调用ID
/** @var WSConnection[] */
static $connect = [];
static function get($name){ return self::$data[$name] ?? null; }

196
src/cqbot/utils/CQUtil.php
View File

@@ -18,7 +18,7 @@ class CQUtil
Buffer::set("user", []);//清空用户列表
Buffer::set("time_send", false);//发送Timing数据到管理群
Buffer::set("cmd_prefix", DP::getJsonData("config.json")["cmd_prefix"] ?? "");//设置指令的前缀符号
Buffer::set("res_code", file_get_contents(WORKING_DIR."src/cqbot/Framework.php"));
Buffer::set("res_code", file_get_contents(WORKING_DIR . "src/cqbot/Framework.php"));
}
public static function saveAllFiles() {
@@ -41,65 +41,48 @@ class CQUtil
/**
* 生成报错日志
* @param $log
* @param $self_id
* @param string $head
* @param int $send_debug_message
*/
public static function errorLog($log, $head = "ERROR", $send_debug_message = 1) {
public static function errorLog($log, $self_id, $head = "ERROR", $send_debug_message = 1) {
Console::error($log, ($head === "ERROR") ? null : "[" . $head . "] ");
$time = date("Y-m-d H:i:s");
$msg = "[$head @ $time]: $log\n";
file_put_contents(DP::getDataFolder() . "log_error.txt", $msg, FILE_APPEND);
if (self::checkAPIConnection() === -1) {
file_put_contents(DP::getDataFolder() . "last_error.log", $msg, FILE_APPEND);
} else {
if ($send_debug_message)
self::sendDebugMsg($msg, 0);
}
if ($send_debug_message)
self::sendDebugMsg($msg, $self_id);
}
/**
* 发送调试信息到管理群（需先设置管理群号）
* @param $msg
* @param $self_id
* @param int $need_head
* @return null
*/
static function sendDebugMsg($msg, $need_head = 1) {
if (Framework::$admin_group == "") return null;
static function sendDebugMsg($msg, $self_id, $need_head = 1) {
if (Framework::$admin_group[$self_id] == "") return null;
if ($need_head)
$data = CQMsg("[DEBUG] " . date("H:i:s") . ": " . $msg, "group", Framework::$admin_group);
else
$data = CQMsg($msg, "group", Framework::$admin_group);
return self::APIPush($data);
}
/**
* 检查API端口连接情况
* @return int
*/
static function checkAPIConnection() {
if (Buffer::$api === null) return -1;//在framework链接API之前
if (Buffer::$api->isConnected() === false) {
//链接被断开
Buffer::$api->upgrade('/api/', function ($cli) {
self::sendDebugMsg("API重新链接成功");
self::APIPushDelayMsg();
});
return 0;
}
return 1;
$connect = CQUtil::getApiConnectionByQQ($self_id);
return self::sendAPI($connect->fd, $data, ["send_debug_msg"]);
}
/**
* 推送API，给API端口
* @param $fd
* @param $data
* @return bool
*/
static function APIPush($data) {
static function APIPush($fd, $data) {
if ($data == null || $data == "") {
Console::error("EMPTY DATA PUSH");
return false;
}
if (self::checkAPIConnection() === -1) {
/*if (self::checkAPIConnection() === -1) {
//忽略掉framework链接API之前的消息
self::errorlog("API推送失败，未发送的消息: \n" . $data, "API ERROR", 0);
return false;
@@ -107,20 +90,21 @@ class CQUtil
if (self::checkAPIConnection() === 0) {
self::APIPushAfterConnected($data);
return true;
}
if (Buffer::$api->push($data) === false) {
}*/
if (Buffer::$event->push($fd, $data) === false) {
$data = self::unicodeDecode($data);
self::errorlog("API推送失败，未发送的消息: \n" . $data, "API ERROR", 0);
self::sendErrorEmail("API推送失败", "未成功推送的消息：<br>$data<br>请检查酷q是否开启及网络链接情况<br>在此期间，机器人会中断所有消息处理<br>请及时处理");
$connect = self::getConnection($fd);
self::errorlog("API推送失败，未发送的消息: \n" . $data, $connect->getQQ(), "API ERROR", 0);
self::sendErrorEmail("API推送失败", "未成功推送的消息：<br>$data<br>请检查酷q是否开启及网络链接情况<br>在此期间，机器人会中断所有消息处理<br>请及时处理", $connect->getQQ());
return false;
}
return true;
}
/**
* 延迟推送在API连接断开后收到的消息函数
* 延迟推送在API连接断开后收到的消息函数//待定
*/
static function APIPushDelayMsg() {
/*static function APIPushDelayMsg() {
$delay_push_list = Buffer::get("delay_push");
$cur_time = time();
foreach ($delay_push_list as $item) {
@@ -131,10 +115,10 @@ class CQUtil
}
}
Buffer::set("delay_push", []);
}
}*/
/**
* 推迟推送API，用于酷Q重启后的重新连接API
* 推迟推送API，用于酷Q重启后的重新连接API//待定
* @param $data
*/
static function APIPushAfterConnected($data) {
@@ -195,11 +179,12 @@ class CQUtil
* @param $address
* @param $title
* @param $content
* @param $self_id
* @param string $name
* @param int $send_debug_message
* @return bool|string
*/
static function sendEmail($address, $title, $content, $name = "CQ开发团队", $send_debug_message = 1) {
static function sendEmail($address, $title, $content, $self_id, $name = "CQ开发团队", $send_debug_message = 1) {
$mail = new \PHPMailer(true);
try {
$mail->isSMTP();
@@ -228,12 +213,64 @@ class CQUtil
unset($mail);
return true;
} catch (\Exception $e) {
self::errorLog("发送邮件错误！错误信息：" . $info = $mail->ErrorInfo, "ERROR", $send_debug_message);
self::errorLog("发送邮件错误！错误信息：" . $info = $mail->ErrorInfo, $self_id, "ERROR", $send_debug_message);
unset($mail);
return $info;
}
}
/**
* 返回所有api、event连接
* @param string $type
* @return WSConnection[]
*/
static function getConnections($type = "all") {
switch ($type) {
case "all":
return Buffer::$connect;
case "event":
$ls = [];
foreach (Buffer::$connect as $fd => $connection) {
if ($connection->getType() === 0) {
$ls[$fd] = $connection;
}
}
return $ls;
case "api":
$ls = [];
foreach (Buffer::$connect as $fd => $connection) {
if ($connection->getType() === 1) {
$ls[$fd] = $connection;
}
}
return $ls;
default:
Console::error("寻找连接时链接类型传入错误！");
return [];
}
}
/**
* @param $fd
* @return WSConnection
*/
static function getConnection($fd) {
if (!isset(Buffer::$connect[$fd])) {
$s = new WSConnection(Buffer::$event, $fd);
Buffer::$connect[$fd] = $s;
}
return Buffer::$connect[$fd];
}
static function getApiConnectionByQQ($qq) {
foreach (self::getConnections() as $fd => $c) {
if ($c->getType() === 1 && $c->getQQ() == $qq) {
return $c;
}
}
return null;
}
/**
* 获取运行时间
* @param $time
@@ -313,10 +350,11 @@ class CQUtil
* 此功能基于sendMail，请看上方sendMail函数的介绍
* @param $title
* @param $content
* @param $self_id
* @param string $name
*/
static function sendErrorEmail($title, $content, $name = "机器人错误提示") {
self::sendEmail(["here your receive email address"], $title, $content, $name, 0);
static function sendErrorEmail($title, $content, $self_id, $name = "机器人错误提示") {
self::sendEmail(["here your receive email address"], $title, $content, $self_id, $name, 0);
}
/**
@@ -375,14 +413,22 @@ class CQUtil
* 发送群组消息，含控制台推出
* @param $groupId
* @param $msg
* @param string $self_id
* @return bool
*/
static function sendGroupMsg($groupId, $msg) {
static function sendGroupMsg($groupId, $msg, $self_id) {
$reply = ["action" => "send_group_msg", "params" => ["group_id" => $groupId, "message" => $msg]];
$reply["echo"] = $reply;
$reply["echo"]["time"] = time();
$reply = json_encode($reply);
if (self::APIPush($reply)) {
$connections = CQUtil::getApiConnectionByQQ($self_id);
if ($connections === null) {
Console::error("未找到qq号：" . $self_id . "的API连接");
return false;
} else {
$api_fd = $connections->fd;
}
if (self::sendAPI($api_fd, $reply, ["send_group_msg"])) {
if (Buffer::$data["info_level"] == 1) {
$out_count = Buffer::$out_count->get();
Console::put(Console::setColor(date("H:i:s "), "lightpurple") . Console::setColor("[{$out_count}]GROUP", "blue") . Console::setColor(" " . $groupId, "yellow") . Console::setColor(" > ", "gray") . $msg);
@@ -397,14 +443,22 @@ class CQUtil
* 发送私聊消息
* @param $userId
* @param $msg
* @param $self_id
* @return bool
*/
static function sendPrivateMsg($userId, $msg) {
static function sendPrivateMsg($userId, $msg, $self_id) {
$reply = ["action" => "send_private_msg", "params" => ["user_id" => $userId, "message" => $msg]];
$reply["echo"] = $reply;
$reply["echo"]["time"] = time();
$reply = json_encode($reply);
if (self::APIPush($reply)) {
$connections = CQUtil::getApiConnectionByQQ($self_id);
if ($connections === null) {
Console::error("未找到qq号：" . $self_id . "的API连接");
return false;
} else {
$api_fd = $connections->fd;
}
if (self::sendAPI($api_fd, $reply, ["send_private_msg"])) {
if (Buffer::$data["info_level"] == 1) {
$out_count = Buffer::$out_count->get();
Console::put(Console::setColor(date("H:i:s "), "lightpurple") . Console::setColor("[{$out_count}]PRIVATE", "blue") . Console::setColor(" " . $userId, "yellow") . Console::setColor(" > ", "gray") . $msg);
@@ -423,50 +477,26 @@ class CQUtil
/**
* 发送其他API，HTTP插件支持的其他API都可以发送。
* echo是返回内容，可以在APIHandler.php里面解析
* @param $fd
* @param $data
* @param $echo
* @return bool
*/
static function sendAPI($data, $echo) {
static function sendAPI($fd, $data, $echo) {
if (!is_array($data)) {
$api = [];
$api["action"] = $data;
} else {
$api = $data;
}
$rw = $echo;
$echo = [
"self_id" => self::getConnection($fd)->getQQ(),
"type" => array_shift($rw),
"params" => $rw
];
$api["echo"] = $echo;
self::APIPush(json_encode($api));
}
/**
* 删除一个和模块相关联的指令
* @param $name
* @return bool
*/
static function removeCommand($name) {
$list = Buffer::get("commands");
if (!isset($list[$name])) return false;
unset($list[$name]);
Buffer::set("commands", $list);
DP::setJsonData("commands.json", $list);
return true;
}
/**
* 添加一个指令给非callTask方式激活的模块。
* 注意：如果给callTask方式激活的模块添加指令，则在使用对应功能时会回复多次同样的内容
* @param $name
* @param $class
* @return bool
*/
static function addCommand($name, $class) {
if (!is_file(WORKING_DIR . 'src/cqbot/mods/' . $class . '.php')) {
return false;
}
$list = Buffer::get("commands");
$list[$name] = $class;
DP::setJsonData("commands.json", $list);
Buffer::set("commands", $list);
return true;
return self::APIPush($fd, json_encode($api));
}
/**
@@ -501,7 +531,7 @@ class CQUtil
/**
* 重启框架，此服务重启为全自动的
*/
static function reload(){
static function reload() {
Console::info("Reloading server");
self::saveAllFiles();
Buffer::$event->reload();
@@ -510,7 +540,7 @@ class CQUtil
/**
* 停止运行框架，需要用shell再次开启才能启动
*/
static function stop(){
static function stop() {
Console::info("Stopping server...");
self::saveAllFiles();
Buffer::$api->close();
@@ -610,7 +640,7 @@ class CQUtil
* @param $group_id
* @return Group|null
*/
static function getGroup($group_id){
static function getGroup($group_id) {
$d = Buffer::get("groups");
return $d[$group_id] ?? null;
}