qimall/services/rbac/AuthItemChildService.php

<?php

namespace services\rbac;

use common\models\rbac\AuthRole;
use Yii;
use yii\web\UnprocessableEntityHttpException;
use common\components\Service;
use common\models\rbac\AuthItemChild;
use common\enums\AddonDefaultRouteEnum;
use common\enums\AppEnum;
use common\enums\AuthMenuEnum;
use common\enums\WhetherEnum;
use common\helpers\ArrayHelper;
use common\helpers\StringHelper;
use common\helpers\TreeHelper;
use common\models\rbac\AuthItem;

/**
 *
 * 已授权权限
 *
 * Class AuthItemChildService
 * @package services\rbac
 * @author qimall
 */
class AuthItemChildService extends Service
{
    /**
     * 当前的角色所有权限
     *
     * @var array
     */
    protected $allAuthNames = [];

    /**
     * 获取用户所有的权限 - 包含插件
     *
     * @param $role
     * @return array
     */
    public function getAuthByRole($role, $app_id, $addons_name = '')
    {
        if (!empty($this->allAuthNames)) {
            return $this->allAuthNames;
        }

        // 获取当前角色的权限
//        $allAuth = AuthItemChild::find()
//            ->select(['addons_name', 'name'])
//            ->where(['role_id' => $role['role_id']])
//            ->andWhere(['app_id' => $app_id])
//            ->andFilterWhere(['addons_name' => $addons_name])
//            ->asArray()
//            ->all();
        $i = 1;
        $page_size = 1000;
        $data = [];
        $query = AuthItemChild::find()->select(['addons_name', 'name'])
            ->where(['role_id' => $role['role_id'] ?? ''])
            ->andWhere(['app_id' => $app_id])
            ->andFilterWhere(['addons_name' => $addons_name])
            ->asArray();
        while ($list =$query->offset(($i - 1) * $page_size)->limit($page_size)->all()) {
            $data = array_merge($data, $list);
            $i++;

        }
        $allAuth = $data;



        $addonsName = [];
        foreach ($allAuth as $item) {
            !isset($addonsName[$item['addons_name']]) && $this->allAuthNames[] = $item['addons_name'];

            $this->allAuthNames[] = $item['name'];
            $addonsName[$item['addons_name']] = true;
        }

        return $this->allAuthNames;
    }

    /**
     * @param $allAuthItem
     * @param $name
     * @throws UnprocessableEntityHttpException
     * @throws \yii\db\Exception
     */
    public function accreditByAddon($allAuthItem, $name)
    {
        // 卸载权限
        Yii::$app->services->rbacAuthItem->delByAddonsName($name);
        // 重组
        foreach ($allAuthItem as &$val) {
            $val = ArrayHelper::regroupMapToArr($val, 'name');
        }

        $defaultAuth = [];
//        // 默认后台权限
//        if (!empty($allAuthItem[AppEnum::BACKEND])) {
//            $defaultAuth = ArrayHelper::merge(AddonDefaultRouteEnum::route(AppEnum::BACKEND, $name), $defaultAuth);
//        }
//
//        // 默认商家权限
//        if (!empty($allAuthItem[AppEnum::MERCHANT])) {
//            $defaultAuth = ArrayHelper::merge(AddonDefaultRouteEnum::route(AppEnum::MERCHANT, $name), $defaultAuth);
//        }

        // 重组路由
        $allAuth = [];
        foreach ($allAuthItem as $key => $item) {
            $allAuth = ArrayHelper::merge($allAuth, $this->regroupByAddonsData($item, $name, $key));
        }

        // 创建权限
        $rows = $this->createByAddonsData(ArrayHelper::merge($defaultAuth, $allAuth));
        // 批量写入数据
        $field = ['title', 'name', 'app_id', 'is_addon', 'addons_name', 'pid', 'level', 'sort', 'tree', 'created_at', 'updated_at'];
        !empty($rows) && Yii::$app->db->createCommand()->batchInsert(AuthItem::tableName(), $field, $rows)->execute();

        unset($data, $allAuth, $installData, $defaultAuth);
    }

    /**
     * 授权
     *
     * @param int $role_id 角色ID
     * @param array $data 数据
     * @param string $is_addon 是否插件
     * @param string $app_id 应用ID
     * @throws \yii\db\Exception
     */
    public function accredit(int $role_id, array $data, int $is_addon, string $app_id)
    {
        // 删除原先所有权限
        AuthItemChild::deleteAll(['role_id' => $role_id, 'is_addon' => $is_addon]);

        if (empty($data)) {
            return;
        }

        $rows = [];
        $items = Yii::$app->services->rbacAuthItem->findByAppId($app_id, $data);

        foreach ($items as $value) {
            $rows[] = [
                $role_id,
                $value['id'],
                $value['name'],
                $value['app_id'],
                $value['is_addon'],
                $value['addons_name'],
            ];
        }

        $field = ['role_id', 'item_id', 'name', 'app_id', 'is_addon', 'addons_name'];
        !empty($rows) && Yii::$app->db->createCommand()->batchInsert(AuthItemChild::tableName(), $field, $rows)->execute();
    }

    /**
     * 根据角色信息去授权
     *
     * @param AuthRole $role
     * @param $itemChilds
     * @throws \yii\db\Exception
     */
    public function accreditByDefault(AuthRole $role, $itemChilds)
    {
        $rows = [];
        foreach ($itemChilds as $child) {
            $rows[] = [
                $role->id,
                $child['item_id'],
                $child['name'],
                $child['app_id'],
                $child['is_addon'],
                $child['addons_name'],
            ];
        }

        $field = ['role_id', 'item_id', 'name', 'app_id', 'is_addon', 'addons_name'];
        !empty($rows) && Yii::$app->db->createCommand()->batchInsert(AuthItemChild::tableName(), $field, $rows)->execute();
    }

    /**
     * 获取某角色的所有权限
     *
     * @param $role_id
     * @return array
     */
    public function findItemByRoleId($role_id)
    {
        $auth = AuthItemChild::find()
            ->where(['role_id' => $role_id])
            ->with(['item'])
            ->asArray()
            ->all();

        return array_column($auth, 'item');
    }

    /**
     * @param $item
     * @param $name
     * @param $app_id
     * @return mixed
     */
    protected function regroupByAddonsData($item, $name, $app_id)
    {
        foreach ($item as &$value) {
            $value['app_id'] = $app_id;
            $value['is_addon'] = WhetherEnum::ENABLED;
            $value['addons_name'] = $name;

            // 组合子级
            if (isset($value['child']) && !empty($value['child'])) {
                $value['child'] = $this->regroupByAddonsData($value['child'], $name, $app_id);
            }
        }

        return $item;
    }

    /**
     * @param array $data
     * @param int $pid
     * @param int $level
     * @param AuthItem $parent
     * @throws UnprocessableEntityHttpException
     * @throws \yii\db\Exception
     */
    protected function createByAddonsData(array $data, $pid = 0, $level = 1, $parent = '')
    {
        $rows = [];

        foreach ($data as $datum) {
            $model = new AuthItem();
            $model = $model->loadDefaultValues();
            $model->attributes = $datum;
            // 增加父级
            !empty($parent) && $model->setParent($parent);
            $model->pid = $pid;
            $model->level = $level;
            $model->name = StringHelper::toUnderScore($model->addons_name) . '/' . $model->name;
            $model->setScenario('addonsBatchCreate');
            if (!$model->validate()) {
                throw new UnprocessableEntityHttpException($model->getErrorMessage());
            }

            // 创建子权限
            if (isset($datum['child']) && !empty($datum['child'])) {
                // 有子权限的直接写入
                if (!$model->save()) {
                    throw new UnprocessableEntityHttpException($model->getErrorMessage());
                }

                $rows = array_merge($rows, $this->createByAddonsData($datum['child'], $model->id, $level++, $model));
            } else {
                $model->tree = !empty($parent) ?  $parent->tree . TreeHelper::prefixTreeKey($parent->id) : TreeHelper::defaultTreeKey();

                $rows[] = [
                    $model->title,
                    $model->name,
                    $model->app_id,
                    $model->is_addon,
                    $model->addons_name,
                    $pid,
                    $level,
                    $model->sort ?? 9999,
                    $model->tree,
                    time(),
                    time(),
                ];

                unset($model);
            }
        }

        return $rows;
    }
}