<?php
/**
 * Copyright © 2016 Magento. All rights reserved.
 * See COPYING.txt for license details.
 */
namespace Magento\Framework\Acl\Role;

/**
 * Acl role registry. Contains list of roles and their relations.
 */
class Registry extends \Zend_Acl_Role_Registry
{
    /**
     * Add parent to the $role node
     *
     * @param \Zend_Acl_Role_Interface|string $role
     * @param array|\Zend_Acl_Role_Interface|string $parents
     * @return $this
     * @throws \Zend_Acl_Role_Registry_Exception
     */
    public function addParent($role, $parents)
    {
        try {
            if ($role instanceof \Zend_Acl_Role_Interface) {
                $roleId = $role->getRoleId();
            } else {
                $roleId = $role;
                $role = $this->get($role);
            }
        } catch (\Zend_Acl_Role_Registry_Exception $e) {
            throw new \Zend_Acl_Role_Registry_Exception("Child Role id '{$roleId}' does not exist");
        }

        if (!is_array($parents)) {
            $parents = [$parents];
        }
        foreach ($parents as $parent) {
            try {
                if ($parent instanceof \Zend_Acl_Role_Interface) {
                    $roleParentId = $parent->getRoleId();
                } else {
                    $roleParentId = $parent;
                }
                $roleParent = $this->get($roleParentId);
            } catch (\Zend_Acl_Role_Registry_Exception $e) {
                throw new \Zend_Acl_Role_Registry_Exception("Parent Role id '{$roleParentId}' does not exist");
            }
            $this->_roles[$roleId]['parents'][$roleParentId] = $roleParent;
            $this->_roles[$roleParentId]['children'][$roleId] = $role;
        }
        return $this;
    }
}