* @since 2.0.5 */ class MongoDbManager extends BaseManager { /** * @var Connection|array|string the MongoDB connection object or the application component ID of the MongoDB connection. * After the MongoDbManager object is created, if you want to change this property, you should only assign it * with a MongoDB connection object. */ public $db = 'mongodb'; /** * @var Cache|array|string the cache used to improve RBAC performance. This can be one of the following: * * - an application component ID (e.g. `cache`) * - a configuration array * - a [[\yii\caching\Cache]] object * * When this is not set, it means caching is not enabled. * * Note that by enabling RBAC cache, all auth items, rules and auth item parent-child relationships will * be cached and loaded into memory. This will improve the performance of RBAC permission check. However, * it does require extra memory and as a result may not be appropriate if your RBAC system contains too many * auth items. You should seek other RBAC implementations (e.g. RBAC based on Redis storage) in this case. * * Also note that if you modify RBAC items, rules or parent-child relationships from outside of this component, * you have to manually call [[invalidateCache()]] to ensure data consistency. */ public $cache; /** * @var string the key used to store RBAC data in cache * @see cache */ public $cacheKey = 'rbac'; /** * @var string|array the name of the collection storing authorization items. Defaults to "auth_item". */ public $itemCollection = 'auth_item'; /** * @var string|array the name of the collection storing authorization item assignments. Defaults to "auth_assignment". */ public $assignmentCollection = 'auth_assignment'; /** * @var string|array the name of the collection storing rules. Defaults to "auth_rule". */ public $ruleCollection = 'auth_rule'; /** * @var Item[] all auth items (name => Item) */ protected $items; /** * @var Rule[] all auth rules (name => Rule) */ protected $rules; /** * Initializes the application component. * This method overrides the parent implementation by establishing the MongoDB connection. */ public function init() { parent::init(); $this->db = Instance::ensure($this->db, Connection::className()); if ($this->cache !== null) { $this->cache = Instance::ensure($this->cache, Cache::className()); } } /** * {@inheritdoc} */ public function checkAccess($userId, $permissionName, $params = []) { $assignments = $this->getAssignments($userId); $this->loadFromCache(); return $this->items !== null ? $this->checkAccessFromCache($userId, $permissionName, $params, $assignments) : $this->checkAccessRecursive($userId, $permissionName, $params, $assignments); } /** * Performs access check for the specified user based on the data loaded from cache. * This method is internally called by [[checkAccess()]] when [[cache]] is enabled. * @param string|int $user the user ID. This should can be either an integer or a string representing * the unique identifier of a user. See [[\yii\web\User::id]]. * @param string $itemName the name of the operation that need access check * @param array $params name-value pairs that would be passed to rules associated * with the tasks and roles assigned to the user. A param with name 'user' is added to this array, * which holds the value of `$userId`. * @param Assignment[] $assignments the assignments to the specified user * @return bool whether the operations can be performed by the user. */ protected function checkAccessFromCache($user, $itemName, $params, $assignments) { if (!isset($this->items[$itemName])) { return false; } $item = $this->items[$itemName]; Yii::trace($item instanceof Role ? "Checking role: $itemName" : "Checking permission: $itemName", __METHOD__); if (!$this->executeRule($user, $item, $params)) { return false; } if (isset($assignments[$itemName]) || in_array($itemName, $this->defaultRoles)) { return true; } if (!empty($item->parents)) { foreach ($item->parents as $parent) { if ($this->checkAccessFromCache($user, $parent, $params, $assignments)) { return true; } } } return false; } /** * Performs access check for the specified user. * This method is internally called by [[checkAccess()]]. * @param string|int $user the user ID. This should can be either an integer or a string representing * the unique identifier of a user. See [[\yii\web\User::id]]. * @param string $itemName the name of the operation that need access check * @param array $params name-value pairs that would be passed to rules associated * with the tasks and roles assigned to the user. A param with name 'user' is added to this array, * which holds the value of `$userId`. * @param Assignment[] $assignments the assignments to the specified user * @return bool whether the operations can be performed by the user. */ protected function checkAccessRecursive($user, $itemName, $params, $assignments) { if (($item = $this->getItem($itemName)) === null) { return false; } Yii::trace($item instanceof Role ? "Checking role: $itemName" : "Checking permission: $itemName", __METHOD__); if (!$this->executeRule($user, $item, $params)) { return false; } if (isset($assignments[$itemName]) || in_array($itemName, $this->defaultRoles)) { return true; } if (!empty($item->parents)) { foreach ($item->parents as $parent) { if ($this->checkAccessRecursive($user, $parent, $params, $assignments)) { return true; } } } return false; } /** * {@inheritdoc} */ protected function getItem($name) { if (empty($name)) { return null; } if (!empty($this->items[$name])) { return $this->items[$name]; } $row = (new Query)->from($this->itemCollection) ->where(['name' => $name]) ->one($this->db); if ($row === false) { return null; } return $this->populateItem($row); } /** * {@inheritdoc} */ protected function getItems($type) { $query = (new Query) ->from($this->itemCollection) ->where(['type' => $type]); $items = []; foreach ($query->all($this->db) as $row) { $items[$row['name']] = $this->populateItem($row); } return $items; } /** * {@inheritdoc} */ protected function addItem($item) { $time = time(); if ($item->createdAt === null) { $item->createdAt = $time; } if ($item->updatedAt === null) { $item->updatedAt = $time; } $this->db->getCollection($this->itemCollection) ->insert([ 'name' => $item->name, 'type' => $item->type, 'description' => $item->description, 'rule_name' => $item->ruleName, 'data' => $item->data === null ? null : serialize($item->data), 'created_at' => $item->createdAt, 'updated_at' => $item->updatedAt, ]); $this->invalidateCache(); return true; } /** * {@inheritdoc} */ protected function addRule($rule) { $time = time(); if ($rule->createdAt === null) { $rule->createdAt = $time; } if ($rule->updatedAt === null) { $rule->updatedAt = $time; } $this->db->getCollection($this->ruleCollection) ->insert([ 'name' => $rule->name, 'data' => serialize($rule), 'created_at' => $rule->createdAt, 'updated_at' => $rule->updatedAt, ]); $this->invalidateCache(); return true; } /** * {@inheritdoc} */ protected function removeItem($item) { $this->db->getCollection($this->assignmentCollection) ->remove(['item_name' => $item->name]); $this->db->getCollection($this->itemCollection) ->remove(['name' => $item->name]); $this->db->getCollection($this->itemCollection) ->update( [ 'parents' => [ '$in' => [$item->name] ], ], [ '$pull' => [ 'parents' => [ '$in' => [$item->name], ] ] ], [ 'multi' => true ] ); $this->invalidateCache(); return true; } /** * {@inheritdoc} */ protected function removeRule($rule) { $this->db->getCollection($this->itemCollection) ->update(['rule_name' => $rule->name], ['rule_name' => null]); $this->db->getCollection($this->ruleCollection) ->remove(['name' => $rule->name]); $this->invalidateCache(); return true; } /** * {@inheritdoc} */ protected function updateItem($name, $item) { if ($item->name !== $name) { $this->db->getCollection($this->assignmentCollection) ->update(['item_name' => $name], ['item_name' => $item->name]); $this->db->getCollection($this->itemCollection) ->update( [ 'parents' => [ '$in' => [$item->name] ], ], [ '$pull' => [ 'parents' => [ '$in' => [$item->name], ] ], ], [ 'multi' => true ] ); $this->db->getCollection($this->itemCollection) ->update( [ 'parents' => [ '$in' => [$item->name] ], ], [ '$push' => [ 'parents' => $name ] ], [ 'multi' => true ] ); } $item->updatedAt = time(); $this->db->getCollection($this->itemCollection) ->update( [ 'name' => $name, ], [ 'name' => $item->name, 'description' => $item->description, 'rule_name' => $item->ruleName, 'data' => $item->data === null ? null : serialize($item->data), 'updated_at' => $item->updatedAt, ] ); $this->invalidateCache(); return true; } /** * {@inheritdoc} */ protected function updateRule($name, $rule) { if ($rule->name !== $name) { $this->db->getCollection($this->itemCollection) ->update(['rule_name' => $name], ['rule_name' => $rule->name]); } $rule->updatedAt = time(); $this->db->getCollection($this->ruleCollection) ->update( [ 'name' => $name, ], [ 'name' => $rule->name, 'data' => serialize($rule), 'updated_at' => $rule->updatedAt, ] ); $this->invalidateCache(); return true; } /** * {@inheritdoc} */ public function getRolesByUser($userId) { if (!isset($userId) || $userId === '') { return []; } $roles = $this->instantiateDefaultRoles(); $rows = (new Query()) ->select(['item_name']) ->from($this->assignmentCollection) ->where(['user_id' => (string) $userId]) ->all($this->db); if (empty($rows)) { return $roles; } $itemNames = ArrayHelper::getColumn($rows, 'item_name'); $query = (new Query()) ->from($this->itemCollection) ->where(['name' => $itemNames]) ->andWhere(['type' => Item::TYPE_ROLE]); foreach ($query->all($this->db) as $row) { $roles[$row['name']] = $this->populateItem($row); } return $roles; } /** * {@inheritdoc} * @since 2.1.2 */ public function getChildRoles($roleName) { $role = $this->getRole($roleName); if (is_null($role)) { throw new InvalidParamException("Role '{$roleName}' not found."); } $result = []; $this->getChildrenRecursive($roleName, $this->getChildrenList(), $result); $roles = [$roleName => $role]; $roles += array_filter($this->getRoles(), function (Role $roleItem) use ($result) { return array_key_exists($roleItem->name, $result); }); return $roles; } /** * {@inheritdoc} */ public function getPermissionsByRole($roleName) { $childrenList = $this->getChildrenList(); $result = []; $this->getChildrenRecursive($roleName, $childrenList, $result); if (empty($result)) { return []; } $query = (new Query) ->from($this->itemCollection) ->where([ 'type' => Item::TYPE_PERMISSION, 'name' => array_keys($result), ]); $permissions = []; foreach ($query->all($this->db) as $row) { $permissions[$row['name']] = $this->populateItem($row); } return $permissions; } /** * {@inheritdoc} */ public function getPermissionsByUser($userId) { if (empty($userId)) { return []; } $this->getAssignments($userId); $rows = (new Query) ->select(['item_name']) ->from($this->assignmentCollection) ->where(['user_id' => (string) $userId]) ->all($this->db); if (empty($rows)) { return []; } $names = ArrayHelper::getColumn($rows, 'item_name'); $childrenList = $this->getChildrenList(); $result = []; foreach ($names as $roleName) { $this->getChildrenRecursive($roleName, $childrenList, $result); } $names = array_merge($names, array_keys($result)); $query = (new Query) ->from($this->itemCollection) ->where([ 'type' => Item::TYPE_PERMISSION, 'name' => $names, ]); $permissions = []; foreach ($query->all($this->db) as $row) { $permissions[$row['name']] = $this->populateItem($row); } return $permissions; } /** * {@inheritdoc} */ public function getRule($name) { if ($this->rules !== null) { return isset($this->rules[$name]) ? $this->rules[$name] : null; } $row = (new Query)->select(['data']) ->from($this->ruleCollection) ->where(['name' => $name]) ->one($this->db); return $row === false ? null : unserialize($row['data']); } /** * {@inheritdoc} */ public function getRules() { if ($this->rules !== null) { return $this->rules; } $query = (new Query)->from($this->ruleCollection); $rules = []; foreach ($query->all($this->db) as $row) { $rules[$row['name']] = unserialize($row['data']); } return $rules; } /** * {@inheritdoc} */ public function canAddChild($parent, $child) { return !$this->detectLoop($parent, $child); } /** * {@inheritdoc} */ public function addChild($parent, $child) { if ($parent->name === $child->name) { throw new InvalidParamException("Cannot add '{$parent->name}' as a child of itself."); } if ($parent instanceof Permission && $child instanceof Role) { throw new InvalidParamException('Cannot add a role as a child of a permission.'); } if ($this->detectLoop($parent, $child)) { throw new InvalidCallException("Cannot add '{$child->name}' as a child of '{$parent->name}'. A loop has been detected."); } $result = $this->db->getCollection($this->itemCollection) ->update( [ 'name' => $child->name, ], [ '$push' => [ 'parents' => $parent->name ] ], [ 'multi' => false ] ) > 0; $this->invalidateCache(); return $result; } /** * {@inheritdoc} */ public function removeChild($parent, $child) { $result = $this->db->getCollection($this->itemCollection) ->update( [ 'name' => $child->name, ], [ '$pull' => [ 'parents' => [ '$in' => [$parent->name] ] ] ], [ 'multi' => false ] ) > 0; $this->invalidateCache(); return $result; } /** * {@inheritdoc} */ public function removeChildren($parent) { $result = $this->db->getCollection($this->itemCollection) ->update( [], [ '$pull' => [ 'parents' => [ '$in' => [$parent->name] ] ] ], [ 'multi' => true ] ) > 0; $this->invalidateCache(); return $result; } /** * {@inheritdoc} */ public function hasChild($parent, $child) { return (new Query) ->from($this->itemCollection) ->where([ 'name' => $child->name ]) ->andWhere([ 'parents' => [ '$in' => [$parent->name] ] ]) ->one($this->db) !== false; } /** * {@inheritdoc} */ public function getChildren($name) { $query = (new Query) ->from($this->itemCollection) ->where([ 'parents' => [ '$in' => [$name] ] ]); $children = []; foreach ($query->all($this->db) as $row) { $children[$row['name']] = $this->populateItem($row); } return $children; } /** * {@inheritdoc} */ public function assign($role, $userId) { $assignment = new Assignment([ 'userId' => (string)$userId, 'roleName' => $role->name, 'createdAt' => time(), ]); $this->db->getCollection($this->assignmentCollection) ->insert([ 'user_id' => $assignment->userId, 'item_name' => $assignment->roleName, 'created_at' => $assignment->createdAt, ]); return $assignment; } /** * {@inheritdoc} */ public function revoke($role, $userId) { if (empty($userId)) { return false; } return $this->db->getCollection($this->assignmentCollection) ->remove(['user_id' => (string) $userId, 'item_name' => $role->name]) > 0; } /** * {@inheritdoc} */ public function revokeAll($userId) { if (empty($userId)) { return false; } return $this->db->getCollection($this->assignmentCollection) ->remove(['user_id' => (string) $userId]) > 0; } /** * {@inheritdoc} */ public function getAssignment($roleName, $userId) { if (empty($userId)) { return null; } $row = (new Query)->from($this->assignmentCollection) ->where(['user_id' => (string) $userId, 'item_name' => $roleName]) ->one($this->db); if ($row === false) { return null; } return new Assignment([ 'userId' => $row['user_id'], 'roleName' => $row['item_name'], 'createdAt' => $row['created_at'], ]); } /** * {@inheritdoc} */ public function getAssignments($userId) { if (empty($userId)) { return []; } $query = (new Query) ->from($this->assignmentCollection) ->where(['user_id' => (string) $userId]); $assignments = []; foreach ($query->all($this->db) as $row) { $assignments[$row['item_name']] = new Assignment([ 'userId' => $row['user_id'], 'roleName' => $row['item_name'], 'createdAt' => $row['created_at'], ]); } return $assignments; } /** * {@inheritdoc} */ public function getUserIdsByRole($roleName) { if (empty($roleName)) { return []; } $rows = (new Query)->select(['user_id']) ->from($this->assignmentCollection) ->where(['item_name' => $roleName]) ->all($this->db); return ArrayHelper::getColumn($rows, 'user_id'); } /** * {@inheritdoc} */ public function removeAll() { $this->removeAllAssignments(); $this->db->getCollection($this->itemCollection)->remove(); $this->db->getCollection($this->ruleCollection)->remove(); $this->invalidateCache(); } /** * {@inheritdoc} */ public function removeAllPermissions() { $this->removeAllItems(Item::TYPE_PERMISSION); } /** * {@inheritdoc} */ public function removeAllRoles() { $this->removeAllItems(Item::TYPE_ROLE); } /** * {@inheritdoc} */ public function removeAllRules() { $this->db->getCollection($this->itemCollection) ->update([], ['rule_name' => null]); $this->db->getCollection($this->ruleCollection)->remove(); $this->invalidateCache(); } /** * {@inheritdoc} */ public function removeAllAssignments() { $this->db->getCollection($this->assignmentCollection)->remove(); } /** * Invalidates RBAC related cache */ public function invalidateCache() { if ($this->cache !== null) { $this->cache->delete($this->cacheKey); $this->items = null; $this->rules = null; } } /** * Populates an auth item with the data fetched from collection * @param array $row the data from the auth item collection * @return Item the populated auth item instance (either Role or Permission) */ protected function populateItem($row) { $class = $row['type'] == Item::TYPE_PERMISSION ? Permission::className() : Role::className(); if (!isset($row['data']) || ($data = @unserialize($row['data'])) === false) { $data = null; } return new $class([ 'name' => $row['name'], 'type' => $row['type'], 'description' => $row['description'], 'ruleName' => $row['rule_name'], 'data' => $data, 'createdAt' => $row['created_at'], 'updatedAt' => $row['updated_at'], 'parents' => isset($row['parents']) ? $row['parents'] : null, ]); } /** * Removes all auth items of the specified type. * @param int $type the auth item type (either Item::TYPE_PERMISSION or Item::TYPE_ROLE) */ protected function removeAllItems($type) { $rows = (new Query) ->select(['name']) ->from($this->itemCollection) ->where(['type' => $type]) ->all($this->db); if (empty($rows)) { return; } $names = ArrayHelper::getColumn($rows, 'name'); $this->db->getCollection($this->assignmentCollection) ->remove(['item_name' => $names]); $this->db->getCollection($this->itemCollection) ->remove(['type' => $type]); $this->db->getCollection($this->itemCollection) ->update( [], [ '$pull' => [ 'parents' => [ '$in' => $names, ] ], ], [ 'multi' => true ] ); $this->invalidateCache(); } /** * Loads data from cache */ public function loadFromCache() { if ($this->items !== null || !$this->cache instanceof Cache) { return; } $data = $this->cache->get($this->cacheKey); if (is_array($data) && isset($data[0], $data[1])) { list ($this->items, $this->rules) = $data; return; } $query = (new Query)->from($this->itemCollection); $this->items = []; foreach ($query->all($this->db) as $row) { $this->items[$row['name']] = $this->populateItem($row); } $query = (new Query)->from($this->ruleCollection); $this->rules = []; foreach ($query->all($this->db) as $row) { $this->rules[$row['name']] = unserialize($row['data']); } $this->cache->set($this->cacheKey, [$this->items, $this->rules]); } /** * Returns the children for every parent. * @return array the children list. Each array key is a parent item name, * and the corresponding array value is a list of child item names. */ protected function getChildrenList() { $query = (new Query) ->select(['name', 'parents']) ->from($this->itemCollection); $children = []; foreach ($query->all($this->db) as $row) { if (!empty($row['parents'])) { foreach ($row['parents'] as $name) { $children[$name][] = $row['name']; } } } return $children; } /** * Recursively finds all children and grand children of the specified item. * @param string $name the name of the item whose children are to be looked for. * @param array $childrenList the child list built via [[getChildrenList()]] * @param array $result the children and grand children (in array keys) */ protected function getChildrenRecursive($name, $childrenList, &$result) { if (isset($childrenList[$name])) { foreach ($childrenList[$name] as $child) { $result[$child] = true; $this->getChildrenRecursive($child, $childrenList, $result); } } } /** * Checks whether there is a loop in the authorization item hierarchy. * @param Item $parent the parent item * @param Item $child the child item to be added to the hierarchy * @return bool whether a loop exists */ protected function detectLoop($parent, $child) { if ($child->name === $parent->name) { return true; } foreach ($this->getChildren($child->name) as $grandchild) { if ($this->detectLoop($parent, $grandchild)) { return true; } } return false; } /** * Returns defaultRoles as array of Role objects * @since 2.1.3 * @return Role[] default roles. The array is indexed by the role names */ private function instantiateDefaultRoles() { // this method can be removed in favor of `yii\rbac\BaseManager::getDefaultRoles()` in case // extension dependency on `yii2` is raised up to 2.0.12 $result = []; foreach ($this->defaultRoles as $roleName) { $result[$roleName] = $this->createRole($roleName); } return $result; } }