Auth.php 9.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257
  1. <?php
  2. // +----------------------------------------------------------------------
  3. // | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
  4. // +----------------------------------------------------------------------
  5. // | Copyright (c) 2011 http://thinkphp.cn All rights reserved.
  6. // +----------------------------------------------------------------------
  7. // | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
  8. // +----------------------------------------------------------------------
  9. // | Author: luofei614 <weibo.com/luofei614>
  10. // +----------------------------------------------------------------------
  11. // | 修改者: anuo (本权限类在原3.2.3的基础上修改过来的)
  12. // +----------------------------------------------------------------------
  13. namespace fast;
  14. use think\Db;
  15. use think\Config;
  16. use think\Session;
  17. use think\Request;
  18. /**
  19. * 权限认证类
  20. * 功能特性:
  21. * 1,是对规则进行认证,不是对节点进行认证。用户可以把节点当作规则名称实现对节点进行认证。
  22. * $auth=new Auth(); $auth->check('规则名称','用户id')
  23. * 2,可以同时对多条规则进行认证,并设置多条规则的关系(or或者and)
  24. * $auth=new Auth(); $auth->check('规则1,规则2','用户id','and')
  25. * 第三个参数为and时表示,用户需要同时具有规则1和规则2的权限。 当第三个参数为or时,表示用户值需要具备其中一个条件即可。默认为or
  26. * 3,一个用户可以属于多个用户组(think_auth_group_access表 定义了用户所属用户组)。我们需要设置每个用户组拥有哪些规则(think_auth_group 定义了用户组权限)
  27. * 4,支持规则表达式。
  28. * 在think_auth_rule 表中定义一条规则,condition字段就可以定义规则表达式。 如定义{score}>5 and {score}<100
  29. * 表示用户的分数在5-100之间时这条规则才会通过。
  30. */
  31. class Auth
  32. {
  33. /**
  34. * @var object 对象实例
  35. */
  36. protected static $instance;
  37. protected $rules = [];
  38. /**
  39. * 当前请求实例
  40. * @var Request
  41. */
  42. protected $request;
  43. //默认配置
  44. protected $config = [
  45. 'auth_on' => 1, // 权限开关
  46. 'auth_type' => 1, // 认证方式,1为实时认证;2为登录认证。
  47. 'auth_group' => 'auth_group', // 用户组数据表名
  48. 'auth_group_access' => 'auth_group_access', // 用户-用户组关系表
  49. 'auth_rule' => 'auth_rule', // 权限规则表
  50. 'auth_user' => 'user', // 用户信息表
  51. ];
  52. public function __construct()
  53. {
  54. if ($auth = Config::get('auth')) {
  55. $this->config = array_merge($this->config, $auth);
  56. }
  57. // 初始化request
  58. $this->request = Request::instance();
  59. }
  60. /**
  61. * 初始化
  62. * @access public
  63. * @param array $options 参数
  64. * @return Auth
  65. */
  66. public static function instance($options = [])
  67. {
  68. if (is_null(self::$instance)) {
  69. self::$instance = new static($options);
  70. }
  71. return self::$instance;
  72. }
  73. /**
  74. * 检查权限
  75. * @param string|array $name 需要验证的规则列表,支持逗号分隔的权限规则或索引数组
  76. * @param int $uid 认证用户的id
  77. * @param string $relation 如果为 'or' 表示满足任一条规则即通过验证;如果为 'and'则表示需满足所有规则才能通过验证
  78. * @param string $mode 执行验证的模式,可分为url,normal
  79. * @return bool 通过验证返回true;失败返回false
  80. */
  81. public function check($name, $uid, $relation = 'or', $mode = 'url')
  82. {
  83. if (!$this->config['auth_on']) {
  84. return true;
  85. }
  86. // 获取用户需要验证的所有有效规则列表
  87. $rulelist = $this->getRuleList($uid);
  88. if (in_array('*', $rulelist)) {
  89. return true;
  90. }
  91. if (is_string($name)) {
  92. $name = strtolower($name);
  93. if (strpos($name, ',') !== false) {
  94. $name = explode(',', $name);
  95. } else {
  96. $name = [$name];
  97. }
  98. }
  99. $list = []; //保存验证通过的规则名
  100. if ('url' == $mode) {
  101. $REQUEST = unserialize(strtolower(serialize($this->request->param())));
  102. }
  103. foreach ($rulelist as $rule) {
  104. $query = preg_replace('/^.+\?/U', '', $rule);
  105. if ('url' == $mode && $query != $rule) {
  106. parse_str($query, $param); //解析规则中的param
  107. $intersect = array_intersect_assoc($REQUEST, $param);
  108. $rule = preg_replace('/\?.*$/U', '', $rule);
  109. if (in_array($rule, $name) && $intersect == $param) {
  110. //如果节点相符且url参数满足
  111. $list[] = $rule;
  112. }
  113. } else {
  114. if (in_array($rule, $name)) {
  115. $list[] = $rule;
  116. }
  117. }
  118. }
  119. if ('or' == $relation && !empty($list)) {
  120. return true;
  121. }
  122. $diff = array_diff($name, $list);
  123. if ('and' == $relation && empty($diff)) {
  124. return true;
  125. }
  126. return false;
  127. }
  128. /**
  129. * 根据用户id获取用户组,返回值为数组
  130. * @param int $uid 用户id
  131. * @return array 用户所属的用户组 array(
  132. * array('uid'=>'用户id','group_id'=>'用户组id','name'=>'用户组名称','rules'=>'用户组拥有的规则id,多个,号隔开'),
  133. * ...)
  134. */
  135. public function getGroups($uid)
  136. {
  137. static $groups = [];
  138. if (isset($groups[$uid])) {
  139. return $groups[$uid];
  140. }
  141. // 执行查询
  142. $user_groups = Db::name($this->config['auth_group_access'])
  143. ->alias('aga')
  144. ->join('__' . strtoupper($this->config['auth_group']) . '__ ag', 'aga.group_id = ag.id', 'LEFT')
  145. ->field('aga.uid,aga.group_id,ag.id,ag.pid,ag.name,ag.rules')
  146. ->where("aga.uid='{$uid}' and ag.status='normal'")
  147. ->select();
  148. $groups[$uid] = $user_groups ?: [];
  149. return $groups[$uid];
  150. }
  151. /**
  152. * 获得权限规则列表
  153. * @param int $uid 用户id
  154. * @return array
  155. */
  156. public function getRuleList($uid)
  157. {
  158. static $_rulelist = []; //保存用户验证通过的权限列表
  159. if (isset($_rulelist[$uid])) {
  160. return $_rulelist[$uid];
  161. }
  162. if (2 == $this->config['auth_type'] && Session::has('_rule_list_' . $uid)) {
  163. return Session::get('_rule_list_' . $uid);
  164. }
  165. // 读取用户规则节点
  166. $ids = $this->getRuleIds($uid);
  167. if (empty($ids)) {
  168. $_rulelist[$uid] = [];
  169. return [];
  170. }
  171. // 筛选条件
  172. $where = [
  173. 'status' => 'normal'
  174. ];
  175. if (!in_array('*', $ids)) {
  176. $where['id'] = ['in', $ids];
  177. }
  178. //读取用户组所有权限规则
  179. $this->rules = Db::name($this->config['auth_rule'])->where($where)->field('id,pid,condition,icon,name,title,ismenu')->select();
  180. //循环规则,判断结果。
  181. $rulelist = []; //
  182. if (in_array('*', $ids)) {
  183. $rulelist[] = "*";
  184. }
  185. foreach ($this->rules as $rule) {
  186. //超级管理员无需验证condition
  187. if (!empty($rule['condition']) && !in_array('*', $ids)) {
  188. //根据condition进行验证
  189. $user = $this->getUserInfo($uid); //获取用户信息,一维数组
  190. $command = preg_replace('/\{(\w*?)\}/', '$user[\'\\1\']', $rule['condition']);
  191. @(eval('$condition=(' . $command . ');'));
  192. if ($condition) {
  193. $rulelist[$rule['id']] = strtolower($rule['name']);
  194. }
  195. } else {
  196. //只要存在就记录
  197. $rulelist[$rule['id']] = strtolower($rule['name']);
  198. }
  199. }
  200. $_rulelist[$uid] = $rulelist;
  201. //登录验证则需要保存规则列表
  202. if (2 == $this->config['auth_type']) {
  203. //规则列表结果保存到session
  204. Session::set('_rule_list_' . $uid, $rulelist);
  205. }
  206. return array_unique($rulelist);
  207. }
  208. public function getRuleIds($uid)
  209. {
  210. //读取用户所属用户组
  211. $groups = $this->getGroups($uid);
  212. $ids = []; //保存用户所属用户组设置的所有权限规则id
  213. foreach ($groups as $g) {
  214. $ids = array_merge($ids, explode(',', trim($g['rules'], ',')));
  215. }
  216. $ids = array_unique($ids);
  217. return $ids;
  218. }
  219. /**
  220. * 获得用户资料
  221. * @param int $uid 用户id
  222. * @return mixed
  223. */
  224. protected function getUserInfo($uid)
  225. {
  226. static $user_info = [];
  227. $user = Db::name($this->config['auth_user']);
  228. // 获取用户表主键
  229. $_pk = is_string($user->getPk()) ? $user->getPk() : 'uid';
  230. if (!isset($user_info[$uid])) {
  231. $user_info[$uid] = $user->where($_pk, $uid)->find();
  232. }
  233. return $user_info[$uid];
  234. }
  235. }