mirror of
https://gitee.com/zoujingli/ThinkAdmin.git
synced 2025-04-06 03:58:04 +08:00
111 lines
4.4 KiB
PHP
111 lines
4.4 KiB
PHP
<?php
|
||
|
||
// +----------------------------------------------------------------------
|
||
// | ThinkAdmin
|
||
// +----------------------------------------------------------------------
|
||
// | 版权所有 2014~2019 广州楚才信息科技有限公司 [ http://www.cuci.cc ]
|
||
// +----------------------------------------------------------------------
|
||
// | 官方网站: http://demo.thinkadmin.top
|
||
// +----------------------------------------------------------------------
|
||
// | 开源协议 ( https://mit-license.org )
|
||
// +----------------------------------------------------------------------
|
||
// | gitee 代码仓库：https://gitee.com/zoujingli/ThinkAdmin
|
||
// | github 代码仓库：https://github.com/zoujingli/ThinkAdmin
|
||
// +----------------------------------------------------------------------
|
||
|
||
namespace think\admin\service;
|
||
|
||
use think\admin\extend\DataExtend;
|
||
use think\admin\Service;
|
||
|
||
/**
|
||
* 系统权限管理服务
|
||
* Class AdminService
|
||
* @package think\admin\service
|
||
*/
|
||
class AdminService extends Service
|
||
{
|
||
|
||
/**
|
||
* 判断是否已经登录
|
||
* @return boolean
|
||
*/
|
||
public function isLogin()
|
||
{
|
||
return $this->app->session->get('user.id') ? true : false;
|
||
}
|
||
|
||
/**
|
||
* 检查指定节点授权
|
||
* --- 需要读取缓存或扫描所有节点
|
||
* @param string $node
|
||
* @return boolean
|
||
* @throws \ReflectionException
|
||
*/
|
||
public function check($node = '')
|
||
{
|
||
$service = NodeService::instance();
|
||
if ($this->app->session->get('user.username') === 'admin') return true;
|
||
list($real, $nodes) = [$service->fullnode($node), $service->getMethods()];
|
||
if (!empty($nodes[$real]['isauth'])) {
|
||
return in_array($real, $this->app->session->get('user.nodes', []));
|
||
} else {
|
||
return !(!empty($nodes[$real]['islogin']) && !$this->isLogin());
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获取授权节点列表
|
||
* @param array $checkeds
|
||
* @return array
|
||
* @throws \ReflectionException
|
||
*/
|
||
public function getTree($checkeds = [])
|
||
{
|
||
list($nodes, $pnodes) = [[], []];
|
||
$methods = array_reverse(NodeService::instance()->getMethods());
|
||
foreach ($methods as $node => $method) {
|
||
$count = substr_count($node, '/');
|
||
$pnode = substr($node, 0, strripos($node, '/'));
|
||
if ($count === 2 && !empty($method['isauth'])) {
|
||
in_array($pnode, $pnodes) or array_push($pnodes, $pnode);
|
||
$nodes[$node] = ['node' => $node, 'title' => $method['title'], 'pnode' => $pnode, 'checked' => in_array($node, $checkeds)];
|
||
} elseif ($count === 1 && in_array($pnode, $pnodes)) {
|
||
$nodes[$node] = ['node' => $node, 'title' => $method['title'], 'pnode' => $pnode, 'checked' => in_array($node, $checkeds)];
|
||
}
|
||
}
|
||
foreach (array_keys($nodes) as $key) foreach ($methods as $node => $method) if (stripos($key, "{$node}/") !== false) {
|
||
$pnode = substr($node, 0, strripos($node, '/'));
|
||
$nodes[$node] = ['node' => $node, 'title' => $method['title'], 'pnode' => $pnode, 'checked' => in_array($node, $checkeds)];
|
||
$nodes[$pnode] = ['node' => $pnode, 'title' => ucfirst($pnode), 'pnode' => '', 'checked' => in_array($pnode, $checkeds)];
|
||
}
|
||
return DataExtend::arr2tree(array_reverse($nodes), 'node', 'pnode', '_sub_');
|
||
}
|
||
|
||
/**
|
||
* 初始化用户权限
|
||
* @param boolean $force 强刷权限
|
||
* @return $this
|
||
* @throws \think\db\exception\DataNotFoundException
|
||
* @throws \think\db\exception\DbException
|
||
* @throws \think\db\exception\ModelNotFoundException
|
||
*/
|
||
public function apply($force = false)
|
||
{
|
||
if ($force) $this->app->cache->delete('system_auth_node');
|
||
if (($uid = $this->app->session->get('user.id'))) {
|
||
$user = $this->app->db->name('SystemUser')->where(['id' => $uid])->find();
|
||
if (($aids = $user['authorize'])) {
|
||
$where = [['status', '=', '1'], ['id', 'in', explode(',', $aids)]];
|
||
$subsql = $this->app->db->name('SystemAuth')->field('id')->where($where)->buildSql();
|
||
$user['nodes'] = array_unique($this->app->db->name('SystemAuthNode')->whereRaw("auth in {$subsql}")->column('node'));
|
||
$this->app->session->set('user', $user);
|
||
} else {
|
||
$user['nodes'] = [];
|
||
$this->app->session->set('user', $user);
|
||
}
|
||
}
|
||
return $this;
|
||
}
|
||
|
||
} |