ThinkChat2.0新版上线,更智能更精彩,支持会话、画图、阅读、搜索等,送10W Token,即刻开启你的AI之旅 广告
* ## 一.先创建数据表 ~~~ ---------------------------------------------------------------------------- -- auth_rule,规则表, -- id:主键, -- src:规则唯一标识, title:规则中文名称 status 状态:为1正常,为0禁用 -- src 可以自定义名称,也可以是模块/控制器/方法、模块_控制器_方法、控制器/方法、控制器_方法、控制器-方法 -- condition:规则表达式,为空表示存在就验证,不为空表示按照条件验证 -- condition 简单来说,如果字段为空,则只验证name就行;如果字段不为空,则在验证了name的基础上,还要验证字段里面的条件 -- condition 条件,是user表的字段条件(准确来说,是auth_user配置的表),如 {score} > 10 ---------------------------------------------------------------------------- DROP TABLE IF EXISTS `auth_rule`; CREATE TABLE `auth_rule` ( `id` mediumint(8) unsigned NOT NULL AUTO_INCREMENT, `src` varchar(80) NOT NULL DEFAULT '', `title` varchar(20) NOT NULL DEFAULT '', `status` tinyint(1) NOT NULL DEFAULT 1, `condition` varchar(100) NOT NULL DEFAULT '', PRIMARY KEY (`id`), UNIQUE KEY `name` (`src`) ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; ---------------------------------------------------------------------------- -- auth_group 用户组表, -- id:主键, title:用户组中文名称, rules:用户组拥有的规则id, 多个规则","隔开,status 状态:为1正常,为0禁用 ---------------------------------------------------------------------------- DROP TABLE IF EXISTS `auth_group`; CREATE TABLE `auth_group` ( `id` mediumint(8) unsigned NOT NULL AUTO_INCREMENT, `title` varchar(100) NOT NULL DEFAULT '', `status` tinyint(1) NOT NULL DEFAULT 1, `rules` varchar(80) NOT NULL DEFAULT '', PRIMARY KEY (`id`) ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; ---------------------------------------------------------------------------- -- auth_group_access 用户-用户组关系表 -- uid:用户id,group_id:用户组id ---------------------------------------------------------------------------- DROP TABLE IF EXISTS `auth_group_access`; CREATE TABLE `auth_group_access` ( `uid` mediumint(8) unsigned NOT NULL, `group_id` mediumint(8) unsigned NOT NULL, UNIQUE KEY `uid_group_id` (`uid`,`group_id`), KEY `uid` (`uid`), KEY `group_id` (`group_id`) ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4; ~~~ * ## 二.在使用Auth类前需要配置config.php ~~~ // auth配置 自定义数据表位置在 ./config/auth.php里面 [ 'auth_on' => 1, // 权限开关 'auth_type' => 1, // 认证方式,1为实时认证;2为登录认证。 'auth_group' => 'auth_group', // 用户组数据表(不带前缀表名) 'auth_group_access' => 'auth_group_access', // 用户-用户组关系表(不带前缀表名) 'auth_rule' => 'auth_rule', // 权限规则表(不带前缀表名) 'auth_user' => 'user', // 用户信息表(不带前缀表名) ] // 注意:condition里的变量是用户表(配置auth_user的值的表,通常配置为用户表)的字段 // (当然,不配置auth_user的值为用户表,而是其他表,那也行的!但是,condition里的变量只能是配置的表里面的字段) // (如配置auth_user的值为integral积分表,那么condition里的变量就只能是integral表里面的字段) // condition里的变量用花括号括住,如:{score} // 表的前缀在框架的数据库配置那配置,这里的配置的表都是不带前缀的 ~~~ * ## 三.新建Auth类 ./extend/lib/Auth.php ``` <?php namespace lib; use think\facade\Db; use think\facade\Config; use think\facade\Session; use think\facade\Request; /** * 权限认证类 * 功能特性: * 1,是对规则进行认证,不是对节点进行认证。用户可以把节点当作规则名称实现对节点进行认证。 * $auth=new Auth(); $auth->check('规则名称','用户id') * 2,可以同时对多条规则进行认证,并设置多条规则的关系(or或者and) * $auth=new Auth(); $auth->check('规则1,规则2','用户id','and') * 第三个参数为and时表示,用户需要同时具有规则1和规则2的权限。 当第三个参数为or时,表示用户值需要具备其中一个条件即可。默认为or * 3,一个用户可以属于多个用户组(auth_group_access表 定义了用户所属用户组)。我们需要设置每个用户组拥有哪些规则(auth_group 定义了用户组权限) * * 4,支持规则表达式。 * 在auth_rule 表中定义一条规则时,如果type为1, condition字段就可以定义规则表达式。 如定义{score}>5 and {score}<100 表示用户的分数在5-100之间时这条规则才会通过。 */ class Auth { /** * var object 对象实例 */ protected static $instance; //默认配置 protected $config = [ 'auth_on' => 1, // 权限开关 'auth_type' => 1, // 认证方式,1为实时认证;2为登录认证。 'auth_group' => 'admin_group', // 用户组数据表名 'auth_group_access' => 'admin_group_access', // 用户-用户组关系表 'auth_rule' => 'menu', // 权限规则表 'auth_user' => 'manager', // 用户信息表 ]; /** * 类架构函数 * Auth constructor. */ public function __construct() { //可设置配置项 auth, 此配置项为数组。 if ($auth = Config::get('auth')) { $this->config = array_merge($this->config, $auth); } } /** * 初始化 * access public * @param array $options 参数 * return \think\Request */ public static function instance($options = []) { if (is_null(self::$instance)) { self::$instance = new static($options); } return self::$instance; } /** * 检查权限 * @param $name string|array 需要验证的规则列表,支持逗号分隔的权限规则或索引数组 * @param $uid int 认证用户的id * @param int $type 认证类型 * @param string $mode 执行check的模式 * @param string $relation 如果为 'or' 表示满足任一条规则即通过验证;如果为 'and'则表示需满足所有规则才能通过验证 * return bool 通过验证返回true;失败返回false */ public function check($name, $uid, $type = 1, $mode = 'url', $relation = 'or') { if (!$this->config['auth_on']) { return true; } // 获取用户需要验证的所有有效规则列表 $authList = $this->getAuthList($uid, $type); // dump($authList); if (is_string($name)) { $name = strtolower($name); if (strpos($name, ',') !== false) { $name = explode(',', $name); } else { $name = [$name]; } } $list = []; //保存验证通过的规则名 if ('url' == $mode) { $REQUEST = unserialize(strtolower(serialize(Request::param()))); } foreach ($authList as $auth) { $query = preg_replace('/^.+\?/U', '', $auth); if ('url' == $mode && $query != $auth) { parse_str($query, $param); //解析规则中的param $intersect = array_intersect_assoc($REQUEST, $param); $auth = preg_replace('/\?.*$/U', '', $auth); if (in_array($auth, $name) && $intersect == $param) { //如果节点相符且url参数满足 $list[] = $auth; } } else { if (in_array($auth, $name)) { $list[] = $auth; } } } // dump($list); // dump($name); if ('or' == $relation && !empty($list)) { return true; } $diff = array_diff($name, $list); if ('and' == $relation && empty($diff)) { return true; } return false; } /** * 根据用户id获取用户组,返回值为数组 * @param $uid int 用户id * return array 用户所属的用户组 array( * array('uid'=>'用户id','group_id'=>'用户组id','title'=>'用户组名称','rules'=>'用户组拥有的规则id,多个,号隔开'), * ...) */ public function getGroups($uid) { static $groups = []; if (isset($groups[$uid])) { return $groups[$uid]; } // 转换表名 $auth_group_access = $this->config['auth_group_access']; $auth_group = $this->config['auth_group']; // 执行查询 $user_groups = Db::view($auth_group_access, 'uid,group_id') ->view($auth_group, 'title,rules', "{$auth_group_access}.group_id={$auth_group}.id", 'LEFT') ->where("{$auth_group_access}.uid='{$uid}' and {$auth_group}.status='1'") ->select(); $groups[$uid] = $user_groups ?: []; return $groups[$uid]; } /** * 获得权限列表 * @param integer $uid 用户id * @param integer $type * return array */ protected function getAuthList($uid, $type) { static $_authList = []; //保存用户验证通过的权限列表 $t = implode(',', (array)$type); if (isset($_authList[$uid . $t])) { return $_authList[$uid . $t]; } if (2 == $this->config['auth_type'] && Session::has('_auth_list_' . $uid . $t)) { return Session::get('_auth_list_' . $uid . $t); } //读取用户所属用户组 $groups = $this->getGroups($uid); $ids = []; //保存用户所属用户组设置的所有权限规则id foreach ($groups as $g) { $ids = array_merge($ids, explode(',', trim($g['rules'], ','))); } $ids = array_unique($ids); if (empty($ids)) { $_authList[$uid . $t] = []; return []; } $map = [ ['type','=',$type], ['smid','in', $ids], // ['status','=',0], ]; //读取用户组所有权限规则 $rules = Db::name($this->config['auth_rule'])->where($map)->field('condition,src')->select(); //循环规则,判断结果。 $authList = []; // foreach ($rules as $rule) { if (!empty($rule['condition'])) { //根据condition进行验证 $user = $this->getUserInfo($uid); //获取用户信息,一维数组 $command = preg_replace('/\{(\w*?)\}/', '$user[\'\\1\']', $rule['condition']); //dump($command); //debug @(eval('$condition=(' . $command . ');')); if ($condition) { $authList[] = strtolower($rule['src']); } } else { //只要存在就记录 $authList[] = strtolower($rule['src']); } } $_authList[$uid . $t] = $authList; if (2 == $this->config['auth_type']) { //规则列表结果保存到session Session::set('_auth_list_' . $uid . $t, $authList); } return array_unique($authList); } /** * 获得用户资料,根据自己的情况读取数据库 */ function getUserInfo($uid) { static $userinfo = []; $user = Db::name($this->config['auth_user']); // 获取用户表主键 $_pk = is_string($user->getPk()) ? $user->getPk() : 'uid'; if (!isset($userinfo[$uid])) { $userinfo[$uid] = $user->where($_pk, $uid)->find(); } return $userinfo[$uid]; } //根据uid获取角色名称 //根据uid获取角色名称 function getRole($uid){ try{ $auth_group_access = Db::name($this->config['auth_group_access'])->where('uid',$uid)->find(); $title = Db::name($this->config['auth_group'])->where('id',$auth_group_access['group_id'])->value('title'); return $title; }catch (\Exception $e){ return '此用户未授予角色'; } } /** * 授予用户权限 */ public function setRole($uid,$group_id){ $res = Db::name('auth_group_access') ->where('uid',$uid) ->update(['group_id'=>$group_id]); return true; } } ``` * ## 四.使用 1. 在某个控制的方法里 ~~~ // 引入类库 use lib\Auth; // 获取auth实例 //$auth = Auth::instance(); //下面代码动态判断权限 $auth = new Auth(); // 检测权限 if ($auth->check('show_button', 1)) { // 第一个参数是规则名称, 第二个参数是用户UID //有显示操作按钮的权限 } else { //没有显示操作按钮的权限 } ~~~ ~~~ //会员信息编辑页面展示-demo public function edit(){ $module = strtolower(app('http')->getName()); //应用名 $controller = strtolower(request()->controller()); //控制器名 $action = strtolower(request()->action()); //方法名 // 请求到的规则名 AuthRule Name $url=$module . '/' . $controller . '/' . $action; $uid = session('userid'); //用户iD //下面代码动态判断权限 $auth = new Auth(); if(!$auth->check($url,$uid)){ echo '没有权限'; }else{ echo '有权限'; //todo... } return View::fetch(); } ~~~ 2. 公共控制器 ``` <?php namespace app\admin\controller; use app\BaseController; use think\facade\View; use think\facade\Session; use lib\Auth; class Common extends BaseController { public function initialize() { $uid = session('userid'); if(!Session::has('username') || !Session::has('userid')){ return redirect('/admin/login')->send(); exit; } else { $auth = new Auth(); //实例化Auth // 检测权限 $module = strtolower(app('http')->getName()); //应用名 $controller = strtolower(request()->controller()); //控制器名 $action = strtolower(request()->action()); //方法名 // 请求到的规则名 AuthRule Name $url=$module . '/' . $controller . '/' . $action; // dump($url);exit; if(!$auth->check($url,$uid)){// 第一个参数是规则名称,第二个参数是用户UID echo '<div style="text-align:center;color:red;margin-top:20%;">您没有权限,请联系超级管理员</div>'; exit; } } } } ``` 3. 中间件控制 ``` <?php namespace app\admin\middleware; use think\Request; use think\facade\Session; use lib\Auth; class CheckAdminLgoin { public function handle($request, \Closure $next) { $module = strtolower(app('http')->getName()); //应用名 $controller = strtolower(request()->controller()); //控制器名 $action = strtolower(request()->action()); //方法名 // 请求到的规则名 AuthRule Name $url=$module . '/' . $controller . '/' . $action; $uid = session('userid'); //用户iD //登录验证 if(!Session::has('username') || !Session::has('userid')){ return redirect('/admin/login'); } else { //实例化Auth $auth = new Auth(); // dump($auth);die; if (!$auth->check($url,$uid)) { echo '<div style="text-align:center;color:red;margin-top:20%;">您没有权限,请联系超级管理员</div>'; exit; } } return $next($request); } } ```