thinkphp5.* auth 權限驗證

composer require 5ini99/think-auth  安裝

 

// auth配置
'auth'  => [
    'auth_on'           => 1, // 權限開關
    'auth_type'         => 1, // 認證方式，1爲實時認證；2爲登錄認證。
    'auth_group'        => 'auth_group', // 用戶組數據不帶前綴表名
    'auth_group_access' => 'auth_group_access', // 用戶-用戶組關係不帶前綴表名
    'auth_rule'         => 'auth_rule', // 權限規則不帶前綴表名
    'auth_user'         => 'member', // 用戶信息不帶前綴表名
],

 

-- think_auth_rule，規則表，
-- id:主鍵，name：規則唯一標識, title：規則中文名稱 status 狀態：爲1正常，爲0禁用，condition：規則表達式，爲空表示存在就驗證，不爲空表示按照條件驗證
------------------------------
 DROP TABLE IF EXISTS `think_auth_rule`;
CREATE TABLE `think_auth_rule` (  
    `id` mediumint(8) unsigned NOT NULL AUTO_INCREMENT,  
    `name` char(80) NOT NULL DEFAULT '',  
    `title` char(20) NOT NULL DEFAULT '',  
    `status` tinyint(1) NOT NULL DEFAULT '1',  
    `condition` char(100) NOT NULL DEFAULT '',  
    PRIMARY KEY (`id`),  
    UNIQUE KEY `name` (`name`)
) ENGINE=MyISAM  DEFAULT CHARSET=utf8;
------------------------------
-- think_auth_group 用戶組表， 
-- id：主鍵， title:用戶組中文名稱， rules：用戶組擁有的規則id， 多個規則","隔開，status 狀態：爲1正常，爲0禁用
------------------------------
 DROP TABLE IF EXISTS `think_auth_group`;
CREATE TABLE `think_auth_group` ( 
    `id` mediumint(8) unsigned NOT NULL AUTO_INCREMENT, 
    `title` char(100) NOT NULL DEFAULT '', 
    `status` tinyint(1) NOT NULL DEFAULT '1', 
    `rules` char(80) NOT NULL DEFAULT '', 
    PRIMARY KEY (`id`)
) ENGINE=MyISAM  DEFAULT CHARSET=utf8;
------------------------------
-- think_auth_group_access 用戶組明細表
-- uid:用戶id，group_id：用戶組id
------------------------------
DROP TABLE IF EXISTS `think_auth_group_access`;
CREATE TABLE `think_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=MyISAM DEFAULT CHARSET=utf8;

think_auth_group  加一個字段

更改auth.php

 

 

<?php
// +----------------------------------------------------------------------
// | ThinkPHP 5 [ WE CAN DO IT JUST THINK IT ]
// +----------------------------------------------------------------------
// | Copyright (c) 2016 http://www.zzstudio.net All rights reserved.
// +----------------------------------------------------------------------
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
// +----------------------------------------------------------------------
// | Author: Byron Sampson <[email protected]>
// +----------------------------------------------------------------------
namespace think\auth;

use think\Db;
use think\Loader;
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，一個用戶可以屬於多個用戶組(think_auth_group_access表 定義了用戶所屬用戶組)。我們需要設置每個用戶組擁有哪些規則(think_auth_group 定義了用戶組權限)
 *
 * 4，支持規則表達式。
 *      在think_auth_rule 表中定義一條規則時，如果type爲1， condition字段就可以定義規則表達式。 如定義{score}>5  and {score}<100  表示用戶的分數在5-100之間時這條規則纔會通過。
 */
//數據庫
/*
-- ----------------------------
-- think_auth_rule，規則表，
-- id:主鍵，name：規則唯一標識, title：規則中文名稱 status 狀態：爲1正常，爲0禁用，condition：規則表達式，爲空表示存在就驗證，不爲空表示按照條件驗證
-- ----------------------------
DROP TABLE IF EXISTS `think_auth_rule`;
CREATE TABLE `think_auth_rule` (
    `id` mediumint(8) unsigned NOT NULL AUTO_INCREMENT,
    `name` char(80) NOT NULL DEFAULT '',
    `title` char(20) NOT NULL DEFAULT '',
    `type` tinyint(1) NOT NULL DEFAULT '1',
    `status` tinyint(1) NOT NULL DEFAULT '1',
    `condition` char(100) NOT NULL DEFAULT '',  # 規則附件條件,滿足附加條件的規則,才認爲是有效的規則
    PRIMARY KEY (`id`),
    UNIQUE KEY `name` (`name`)
) ENGINE=MyISAM  DEFAULT CHARSET=utf8;
-- ----------------------------
-- think_auth_group 用戶組表，
-- id：主鍵， title:用戶組中文名稱， rules：用戶組擁有的規則id， 多個規則","隔開，status 狀態：爲1正常，爲0禁用
-- ----------------------------
DROP TABLE IF EXISTS `think_auth_group`;
    CREATE TABLE `think_auth_group` (
    `id` mediumint(8) unsigned NOT NULL AUTO_INCREMENT,
    `title` char(100) NOT NULL DEFAULT '',
    `status` tinyint(1) NOT NULL DEFAULT '1',
    `rules` char(80) NOT NULL DEFAULT '',
    PRIMARY KEY (`id`)
) ENGINE=MyISAM  DEFAULT CHARSET=utf8;
-- ----------------------------
-- think_auth_group_access 用戶組明細表
-- uid:用戶id，group_id：用戶組id
-- ----------------------------
DROP TABLE IF EXISTS `think_auth_group_access`;
    CREATE TABLE `think_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=MyISAM DEFAULT CHARSET=utf8;
 */

class Auth
{
    /**
     * @var object 對象實例
     */
    protected static $instance;
    /**
     * 當前請求實例
     * @var Request
     */
    protected $request;

    //默認配置
    protected $config = [
        'auth_on' => 1, // 權限開關
        'auth_type' => 1, // 認證方式，1爲實時認證；2爲登錄認證。
        'auth_group' => 'auth_group', // 用戶組數據表名
        'auth_group_access' => 'auth_group_access', // 用戶-用戶組關係表
        'auth_rule' => 'auth_rule', // 權限規則表
        'auth_user' => 'member', // 用戶信息表
    ];

    /**
     * 類架構函數
     * Auth constructor.
     */
    public function __construct()
    {
        //可設置配置項 auth, 此配置項爲數組。
        if ($auth = Config::get('auth')) {
            $this->config = array_merge($this->config, $auth);
        }
        // 初始化request
        $this->request = Request::instance();
    }

    /**
     * 初始化
     * @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;
        }
        //判斷是否是super
        $group=Db::name($this->config['auth_group_access'])->where(['uid'=>$uid])->field('group_id')->find();
        //判斷pid是否爲0
        $pid=Db::name($this->config['auth_group'])->where(['id'=>$group['group_id']])->field('pid')->find();
        if($pid['pid']==0){
            return true;
        }
        
        // 獲取用戶需要驗證的所有有效規則列表
        $authList = $this->getAuthList($uid, $type);
        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($this->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;
                }
            }
        }
        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 = Loader::parseName($this->config['auth_group_access'], 1);
        $auth_group = Loader::parseName($this->config['auth_group'], 1);
        // 執行查詢
        $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 = array(
            // 'id' => ['in', $ids],
            'id' => ['in', '1,2,3'],
            'status' => 1,
        );
        //讀取用戶組所有權限規則
        $rules = Db::name($this->config['auth_rule'])->where('id','in',$ids)->where('status',1)->field('condition,name')->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['name']);
                }
            } else {
                //只要存在就記錄
                $authList[] = strtolower($rule['name']);
            }
        }
        $_authList[$uid . $t] = $authList;

        if (2 == $this->config['auth_type']) {
            //規則列表結果保存到session
            Session::set('_auth_list_' . $uid . $t, $authList);
        }
        return array_unique($authList);
    }

    /**
     * 獲得用戶資料,根據自己的情況讀取數據庫
     */
    protected function getUserInfo($uid)
    {
        static $userinfo = [];

        $user = Db::name($this->config['auth_user']);
        // 獲取用戶表主鍵
        $_pk = is_string($user->getPk()) ? $user->getPk() : 'id';
        if (!isset($userinfo[$uid])) {
            $userinfo[$uid] = $user->where($_pk, $uid)->find();
        }

        return $userinfo[$uid];
    }
}