ztt
1 year ago
parent
af01b39dd5
commit
b27f40cddd
@ -0,0 +1,236 @@ |
|||||||
|
<?php |
||||||
|
|
||||||
|
namespace app\admin\command; |
||||||
|
|
||||||
|
use app\admin\model\AuthRule; |
||||||
|
use app\admin\model\User; |
||||||
|
use ReflectionClass; |
||||||
|
use ReflectionMethod; |
||||||
|
use think\Cache; |
||||||
|
use think\Config; |
||||||
|
use think\console\Command; |
||||||
|
use think\console\Input; |
||||||
|
use think\console\input\Option; |
||||||
|
use think\console\Output; |
||||||
|
use think\Exception; |
||||||
|
use think\Loader; |
||||||
|
|
||||||
|
/** |
||||||
|
* 用户升级 |
||||||
|
*/ |
||||||
|
class UserLevel extends Command |
||||||
|
{ |
||||||
|
protected $model = null; |
||||||
|
|
||||||
|
protected function configure() |
||||||
|
{ |
||||||
|
$this |
||||||
|
->setName('level_up') |
||||||
|
->setDescription('用户升级'); |
||||||
|
//要执行的controller必须一样，不适用模糊查询 |
||||||
|
} |
||||||
|
|
||||||
|
protected function execute(Input $input, Output $output) |
||||||
|
{ |
||||||
|
|
||||||
|
$output->info("Build Successed!"); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 递归扫描文件夹 |
||||||
|
* @param string $dir |
||||||
|
* @return array |
||||||
|
*/ |
||||||
|
public function scandir($dir) |
||||||
|
{ |
||||||
|
$result = []; |
||||||
|
$cdir = scandir($dir); |
||||||
|
foreach ($cdir as $value) { |
||||||
|
if (!in_array($value, array(".", ".."))) { |
||||||
|
if (is_dir($dir . DS . $value)) { |
||||||
|
$result[$value] = $this->scandir($dir . DS . $value); |
||||||
|
} else { |
||||||
|
$result[] = $value; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
return $result; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 导入规则节点 |
||||||
|
* @param array $dirarr |
||||||
|
* @param array $parentdir |
||||||
|
* @return array |
||||||
|
*/ |
||||||
|
public function import($dirarr, $parentdir = []) |
||||||
|
{ |
||||||
|
$menuarr = []; |
||||||
|
foreach ($dirarr as $k => $v) { |
||||||
|
if (is_array($v)) { |
||||||
|
//当前是文件夹 |
||||||
|
$nowparentdir = array_merge($parentdir, [$k]); |
||||||
|
$this->import($v, $nowparentdir); |
||||||
|
} else { |
||||||
|
//只匹配PHP文件 |
||||||
|
if (!preg_match('/^(\w+)\.php$/', $v, $matchone)) { |
||||||
|
continue; |
||||||
|
} |
||||||
|
//导入文件 |
||||||
|
$controller = ($parentdir ? implode('/', $parentdir) . '/' : '') . $matchone[1]; |
||||||
|
$this->importRule($controller); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return $menuarr; |
||||||
|
} |
||||||
|
|
||||||
|
protected function importRule($controller) |
||||||
|
{ |
||||||
|
$controller = str_replace('\\', '/', $controller); |
||||||
|
if (stripos($controller, '/') !== false) { |
||||||
|
$controllerArr = explode('/', $controller); |
||||||
|
end($controllerArr); |
||||||
|
$key = key($controllerArr); |
||||||
|
$controllerArr[$key] = ucfirst($controllerArr[$key]); |
||||||
|
} else { |
||||||
|
$key = 0; |
||||||
|
$controllerArr = [ucfirst($controller)]; |
||||||
|
} |
||||||
|
$classSuffix = Config::get('controller_suffix') ? ucfirst(Config::get('url_controller_layer')) : ''; |
||||||
|
$className = "\\app\\admin\\controller\\" . implode("\\", $controllerArr) . $classSuffix; |
||||||
|
|
||||||
|
$pathArr = $controllerArr; |
||||||
|
array_unshift($pathArr, '', 'application', 'admin', 'controller'); |
||||||
|
$classFile = ROOT_PATH . implode(DS, $pathArr) . $classSuffix . ".php"; |
||||||
|
$classContent = file_get_contents($classFile); |
||||||
|
$uniqueName = uniqid("FastAdmin") . $classSuffix; |
||||||
|
$classContent = str_replace("class " . $controllerArr[$key] . $classSuffix . " ", 'class ' . $uniqueName . ' ', $classContent); |
||||||
|
$classContent = preg_replace("/namespace\s(.*);/", 'namespace ' . __NAMESPACE__ . ";", $classContent); |
||||||
|
|
||||||
|
//临时的类文件 |
||||||
|
$tempClassFile = __DIR__ . DS . $uniqueName . ".php"; |
||||||
|
file_put_contents($tempClassFile, $classContent); |
||||||
|
$className = "\\app\\admin\\command\\" . $uniqueName; |
||||||
|
|
||||||
|
//删除临时文件 |
||||||
|
register_shutdown_function(function () use ($tempClassFile) { |
||||||
|
if ($tempClassFile) { |
||||||
|
//删除临时文件 |
||||||
|
@unlink($tempClassFile); |
||||||
|
} |
||||||
|
}); |
||||||
|
|
||||||
|
//反射机制调用类的注释和方法名 |
||||||
|
$reflector = new ReflectionClass($className); |
||||||
|
|
||||||
|
//只匹配公共的方法 |
||||||
|
$methods = $reflector->getMethods(ReflectionMethod::IS_PUBLIC); |
||||||
|
$classComment = $reflector->getDocComment(); |
||||||
|
//判断是否有启用软删除 |
||||||
|
$softDeleteMethods = ['destroy', 'restore', 'recyclebin']; |
||||||
|
$withSofeDelete = false; |
||||||
|
$modelRegexArr = ["/\\\$this\->model\s*=\s*model\(['|\"](\w+)['|\"]\);/", "/\\\$this\->model\s*=\s*new\s+([a-zA-Z\\\]+);/"]; |
||||||
|
$modelRegex = preg_match($modelRegexArr[0], $classContent) ? $modelRegexArr[0] : $modelRegexArr[1]; |
||||||
|
preg_match_all($modelRegex, $classContent, $matches); |
||||||
|
if (isset($matches[1]) && isset($matches[1][0]) && $matches[1][0]) { |
||||||
|
\think\Request::instance()->module('admin'); |
||||||
|
$model = model($matches[1][0]); |
||||||
|
if (in_array('trashed', get_class_methods($model))) { |
||||||
|
$withSofeDelete = true; |
||||||
|
} |
||||||
|
} |
||||||
|
//忽略的类 |
||||||
|
if (stripos($classComment, "@internal") !== false) { |
||||||
|
return; |
||||||
|
} |
||||||
|
preg_match_all('#(@.*?)\n#s', $classComment, $annotations); |
||||||
|
$controllerIcon = 'fa fa-circle-o'; |
||||||
|
$controllerRemark = ''; |
||||||
|
//判断注释中是否设置了icon值 |
||||||
|
if (isset($annotations[1])) { |
||||||
|
foreach ($annotations[1] as $tag) { |
||||||
|
if (stripos($tag, '@icon') !== false) { |
||||||
|
$controllerIcon = substr($tag, stripos($tag, ' ') + 1); |
||||||
|
} |
||||||
|
if (stripos($tag, '@remark') !== false) { |
||||||
|
$controllerRemark = substr($tag, stripos($tag, ' ') + 1); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
//过滤掉其它字符 |
||||||
|
$controllerTitle = trim(preg_replace(array('/^\/\*\*(.*)[\n\r\t]/u', '/[\s]+\*\//u', '/\*\s@(.*)/u', '/[\s|\*]+/u'), '', $classComment)); |
||||||
|
|
||||||
|
//导入中文语言包 |
||||||
|
\think\Lang::load(dirname(__DIR__) . DS . 'lang/zh-cn.php'); |
||||||
|
|
||||||
|
//先导入菜单的数据 |
||||||
|
$pid = 0; |
||||||
|
foreach ($controllerArr as $k => $v) { |
||||||
|
$key = $k + 1; |
||||||
|
//驼峰转下划线 |
||||||
|
$controllerNameArr = array_slice($controllerArr, 0, $key); |
||||||
|
foreach ($controllerNameArr as &$val) { |
||||||
|
$val = strtolower(trim(preg_replace("/[A-Z]/", "_\\0", $val), "_")); |
||||||
|
} |
||||||
|
unset($val); |
||||||
|
$name = implode('/', $controllerNameArr); |
||||||
|
$title = (!isset($controllerArr[$key]) ? $controllerTitle : ''); |
||||||
|
$icon = (!isset($controllerArr[$key]) ? $controllerIcon : 'fa fa-list'); |
||||||
|
$remark = (!isset($controllerArr[$key]) ? $controllerRemark : ''); |
||||||
|
$title = $title ? $title : $v; |
||||||
|
$rulemodel = $this->model->get(['name' => $name]); |
||||||
|
if (!$rulemodel) { |
||||||
|
$this->model |
||||||
|
->data(['pid' => $pid, 'name' => $name, 'title' => $title, 'icon' => $icon, 'remark' => $remark, 'ismenu' => 1, 'status' => 'normal']) |
||||||
|
->isUpdate(false) |
||||||
|
->save(); |
||||||
|
$pid = $this->model->id; |
||||||
|
} else { |
||||||
|
$pid = $rulemodel->id; |
||||||
|
} |
||||||
|
} |
||||||
|
$ruleArr = []; |
||||||
|
foreach ($methods as $m => $n) { |
||||||
|
//过滤特殊的类 |
||||||
|
if (substr($n->name, 0, 2) == '__' || $n->name == '_initialize') { |
||||||
|
continue; |
||||||
|
} |
||||||
|
//未启用软删除时过滤相关方法 |
||||||
|
if (!$withSofeDelete && in_array($n->name, $softDeleteMethods)) { |
||||||
|
continue; |
||||||
|
} |
||||||
|
//只匹配符合的方法 |
||||||
|
if (!preg_match('/^(\w+)' . Config::get('action_suffix') . '/', $n->name, $matchtwo)) { |
||||||
|
unset($methods[$m]); |
||||||
|
continue; |
||||||
|
} |
||||||
|
$comment = $reflector->getMethod($n->name)->getDocComment(); |
||||||
|
//忽略的方法 |
||||||
|
if (stripos($comment, "@internal") !== false) { |
||||||
|
continue; |
||||||
|
} |
||||||
|
//过滤掉其它字符 |
||||||
|
$comment = preg_replace(array('/^\/\*\*(.*)[\n\r\t]/u', '/[\s]+\*\//u', '/\*\s@(.*)/u', '/[\s|\*]+/u'), '', $comment); |
||||||
|
|
||||||
|
$title = $comment ? $comment : ucfirst($n->name); |
||||||
|
|
||||||
|
//获取主键，作为AuthRule更新依据 |
||||||
|
$id = $this->getAuthRulePK($name . "/" . strtolower($n->name)); |
||||||
|
|
||||||
|
$ruleArr[] = array('id' => $id, 'pid' => $pid, 'name' => $name . "/" . strtolower($n->name), 'icon' => 'fa fa-circle-o', 'title' => $title, 'ismenu' => 0, 'status' => 'normal'); |
||||||
|
} |
||||||
|
$this->model->isUpdate(false)->saveAll($ruleArr); |
||||||
|
} |
||||||
|
|
||||||
|
//获取主键 |
||||||
|
protected function getAuthRulePK($name) |
||||||
|
{ |
||||||
|
if (!empty($name)) { |
||||||
|
$id = $this->model |
||||||
|
->where('name', $name) |
||||||
|
->value('id'); |
||||||
|
return $id ? $id : null; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue