mirror of
https://github.com/zhamao-robot/zhamao-framework.git
synced 2026-03-19 05:34:53 +08:00
274 lines
8.0 KiB
PHP
274 lines
8.0 KiB
PHP
<?php
|
||
|
||
use Swoole\Coroutine;
|
||
use ZM\Config\ZMConfig;
|
||
use ZM\Console\Console;
|
||
use ZM\Context\Context;
|
||
use ZM\Event\EventManager;
|
||
use ZM\Framework;
|
||
use ZM\Store\ZMBuf;
|
||
use ZM\Utils\DataProvider;
|
||
use Swoole\Coroutine\System;
|
||
use ZM\Context\ContextInterface;
|
||
use ZM\Utils\ZMUtil;
|
||
|
||
|
||
function phar_classloader($p) {
|
||
$filepath = getClassPath($p);
|
||
if ($filepath === null) {
|
||
Console::debug("F:Warning: get class path wrongs.$p");
|
||
return;
|
||
}
|
||
try {
|
||
require_once $filepath;
|
||
} catch (Exception $e) {
|
||
echo "Error when finding class: " . $p . PHP_EOL;
|
||
die;
|
||
}
|
||
}
|
||
|
||
function getClassPath($class_name) {
|
||
$dir = str_replace("\\", "/", $class_name);
|
||
$dir2 = WORKING_DIR . "/src/" . $dir . ".php";
|
||
//echo "@@@".$dir2.PHP_EOL;
|
||
$dir2 = str_replace("\\", "/", $dir2);
|
||
if (file_exists($dir2)) return $dir2;
|
||
else {
|
||
$dir = DataProvider::getWorkingDir() . "/src/" . $dir . ".php";
|
||
//echo "###".$dir.PHP_EOL;
|
||
if (file_exists($dir)) return $dir;
|
||
else return null;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 使用自己定义的万（san）能分割函数
|
||
* @param $msg
|
||
* @param bool $ban_comma
|
||
* @return array
|
||
*/
|
||
function explodeMsg($msg, $ban_comma = false) {
|
||
$msg = str_replace(" ", "\n", trim($msg));
|
||
if (!$ban_comma) {
|
||
//$msg = str_replace("，", "\n", $msg);
|
||
$msg = str_replace("\t", "\n", $msg);
|
||
}
|
||
$msgs = explode("\n", $msg);
|
||
$ls = [];
|
||
foreach ($msgs as $k => $v) {
|
||
if (trim($v) == "") continue;
|
||
$ls[] = trim($v);
|
||
}
|
||
return $ls;
|
||
}
|
||
|
||
function unicode_decode($str) {
|
||
return preg_replace_callback('/\\\\u([0-9a-f]{4})/i', function ($matches) {
|
||
return mb_convert_encoding(pack("H*", $matches[1]), "UTF-8", "UCS-2BE");
|
||
}, $str);
|
||
}
|
||
|
||
/**
|
||
* 获取模块文件夹下的每个类文件的类名称
|
||
* @param $dir
|
||
* @param string $indoor_name
|
||
* @return array
|
||
*/
|
||
function getAllClasses($dir, $indoor_name) {
|
||
if (!is_dir($dir)) return [];
|
||
$list = scandir($dir);
|
||
$classes = [];
|
||
if ($list[0] == '.') unset($list[0], $list[1]);
|
||
foreach ($list as $v) {
|
||
//echo "Finding " . $dir . $v . PHP_EOL;
|
||
//echo "At " . $indoor_name . PHP_EOL;
|
||
if (is_dir($dir . $v)) $classes = array_merge($classes, getAllClasses($dir . $v . "/", $indoor_name . "\\" . $v));
|
||
elseif (mb_substr($v, -4) == ".php") {
|
||
$class_name = $indoor_name . "\\" . mb_substr($v, 0, -4);
|
||
$classes [] = $class_name;
|
||
}
|
||
}
|
||
return $classes;
|
||
}
|
||
|
||
function matchPattern($pattern, $context) {
|
||
if (mb_substr($pattern, 0, 1) == "" && mb_substr($context, 0, 1) == "")
|
||
return true;
|
||
if ('*' == mb_substr($pattern, 0, 1) && "" != mb_substr($pattern, 1, 1) && "" == mb_substr($context, 0, 1))
|
||
return false;
|
||
if (mb_substr($pattern, 0, 1) == mb_substr($context, 0, 1))
|
||
return matchPattern(mb_substr($pattern, 1), mb_substr($context, 1));
|
||
if (mb_substr($pattern, 0, 1) == "*")
|
||
return matchPattern(mb_substr($pattern, 1), $context) || matchPattern($pattern, mb_substr($context, 1));
|
||
return false;
|
||
}
|
||
|
||
function split_explode($del, $str, $divide_en = false) {
|
||
$str = explode($del, $str);
|
||
for ($i = 0; $i < mb_strlen($str[0]); $i++) {
|
||
if (
|
||
is_numeric(mb_substr($str[0], $i, 1)) &&
|
||
(
|
||
!is_numeric(mb_substr($str[0], $i - 1, 1)) &&
|
||
mb_substr($str[0], $i - 1, 1) != ' ' &&
|
||
ctype_alpha(mb_substr($str[0], $i - 1, 1)) === false
|
||
)
|
||
) {
|
||
$str[0] = mb_substr($str[0], 0, $i) . " " . mb_substr($str[0], $i);
|
||
} elseif (
|
||
$divide_en &&
|
||
ctype_alnum(mb_substr($str[0], $i, 1)) &&
|
||
!ctype_alnum(mb_substr($str[0], $i - 1, 1)) &&
|
||
mb_substr($str[0], $i - 1, 1) != ' '
|
||
) {
|
||
$str[0] = mb_substr($str[0], 0, $i) . ' ' . mb_substr($str[0], $i);
|
||
}
|
||
}
|
||
$str = implode($del, $str);
|
||
//echo $str."\n";
|
||
$ls = [];
|
||
foreach (explode($del, $str) as $k => $v) {
|
||
if (trim($v) == "") continue;
|
||
$ls[] = $v;
|
||
}
|
||
//var_dump($ls);
|
||
return $ls == [] ? [""] : $ls;
|
||
}
|
||
|
||
function matchArgs($pattern, $context) {
|
||
$result = [];
|
||
if (matchPattern($pattern, $context)) {
|
||
if (mb_strpos($pattern, "*") === false) return [];
|
||
$exp = explode("*", $pattern);
|
||
$i = 0;
|
||
foreach ($exp as $k => $v) {
|
||
//echo "[MATCH$k] " . $v . PHP_EOL;
|
||
if ($v == "" && $k == 0) continue;
|
||
elseif ($v == "" && $k == count($exp) - 1) {
|
||
$context = $context . "^EOL";
|
||
$v = "^EOL";
|
||
}
|
||
$cur_var = "";
|
||
//echo mb_substr($context, $i) . "|" . $v . PHP_EOL;
|
||
$ori = $i;
|
||
while (($a = mb_substr($context, $i, mb_strlen($v))) != $v && $a != "") {
|
||
$cur_var .= mb_substr($context, $i, 1);
|
||
++$i;
|
||
}
|
||
if ($i != $ori || $k == 1 || $k == count($exp) - 1) {
|
||
//echo $cur_var . PHP_EOL;
|
||
$result[] = $cur_var;
|
||
}
|
||
$i += mb_strlen($v);
|
||
}
|
||
return $result;
|
||
} else return false;
|
||
}
|
||
|
||
function connectIsQQ() {
|
||
return ctx()->getConnection()->getName() == 'qq';
|
||
}
|
||
|
||
function connectIsDefault() {
|
||
return ctx()->getConnection()->getName() == 'default';
|
||
}
|
||
|
||
function connectIs($type) {
|
||
return ctx()->getConnection()->getName() == $type;
|
||
}
|
||
|
||
function getAnnotations() {
|
||
$s = debug_backtrace()[1];
|
||
//echo json_encode($s, 128|256);
|
||
$list = [];
|
||
foreach(EventManager::$events as $event => $v) {
|
||
foreach($v as $ks => $vs) {
|
||
//echo get_class($vs).": ".$vs->class." => ".$vs->method.PHP_EOL;
|
||
if($vs->class == $s["class"] && $vs->method == $s["function"]) {
|
||
$list[get_class($vs)][]=$vs;
|
||
}
|
||
}
|
||
}
|
||
return $list;
|
||
}
|
||
|
||
function set_coroutine_params($array) {
|
||
$cid = Co::getCid();
|
||
if ($cid == -1) die("Cannot set coroutine params at none coroutine mode.");
|
||
if (isset(Context::$context[$cid])) Context::$context[$cid] = array_merge(Context::$context[$cid], $array);
|
||
else Context::$context[$cid] = $array;
|
||
foreach (Context::$context as $c => $v) {
|
||
if (!Co::exists($c)) unset(Context::$context[$c], ZMBuf::$context_class[$c]);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @return ContextInterface|null
|
||
*/
|
||
function context() {
|
||
return ctx();
|
||
}
|
||
|
||
/**
|
||
* @return ContextInterface|null
|
||
*/
|
||
function ctx() {
|
||
$cid = Co::getCid();
|
||
$c_class = ZMConfig::get("global", "context_class");
|
||
if (isset(Context::$context[$cid])) {
|
||
return ZMBuf::$context_class[$cid] ?? (ZMBuf::$context_class[$cid] = new $c_class($cid));
|
||
} else {
|
||
Console::debug("未找到当前协程的上下文($cid)，正在找父进程的上下文");
|
||
while (($pcid = Co::getPcid($cid)) !== -1) {
|
||
$cid = $pcid;
|
||
if (isset(Context::$context[$cid])) return ZMBuf::$context_class[$cid] ?? (ZMBuf::$context_class[$cid] = new $c_class($cid));
|
||
}
|
||
return null;
|
||
}
|
||
}
|
||
|
||
function debug($msg) { Console::debug($msg); }
|
||
|
||
function onebot_target_id_name($message_type) {
|
||
return ($message_type == "group" ? "group_id" : "user_id");
|
||
}
|
||
|
||
function zm_sleep($s = 1) {
|
||
if (Coroutine::getCid() != -1) System::sleep($s);
|
||
else usleep($s * 1000 * 1000);
|
||
return true;
|
||
}
|
||
|
||
function zm_exec($cmd): array { return System::exec($cmd); }
|
||
|
||
function zm_cid() { return Co::getCid(); }
|
||
|
||
function zm_yield() { Co::yield(); }
|
||
|
||
function zm_resume(int $cid) { Co::resume($cid); }
|
||
|
||
function zm_timer_after($ms, callable $callable) {
|
||
go(function () use ($ms, $callable) {
|
||
Swoole\Timer::after($ms, $callable);
|
||
});
|
||
}
|
||
|
||
function zm_timer_tick($ms, callable $callable) {
|
||
go(function () use ($ms, $callable) {
|
||
Console::debug("Adding extra timer tick of " . $ms . " ms");
|
||
Swoole\Timer::tick($ms, $callable);
|
||
});
|
||
}
|
||
|
||
function zm_data_hash($v) {
|
||
return md5($v["user_id"] . "^" . $v["self_id"] . "^" . $v["message_type"] . "^" . ($v[$v["message_type"] . "_id"] ?? $v["user_id"]));
|
||
}
|
||
|
||
function server() {
|
||
return Framework::$server;
|
||
}
|
||
|
||
|
||
|
||
|