security - Sonata User - 安全地自定义管理员查询

Question

我已经扩展了 SonataUserBundle，我想自定义管理查询以限制列表：

class UserAdmin extends BaseUserAdmin
{
// ...
    public function createQuery($context = 'list')
    {
        $query = parent::createQuery($context);
        $query->andWhere( 
            $query->expr()->eq($query->getRootAlias().'.company', ':comp')
        );
        $query->setParameter('comp', $securityContext->user->getCompany());
        return $query;
    }
// ...
}

在这里，我试图让用户只能看到他公司的用户。
但是$securityContext没有设置。

有人能告诉我如何将安全上下文注入我的管理类吗？

Answer 1

您必须在 UserAdmin 服务中注入安全上下文服务。

为此，您必须更新捆绑包的 services.yml：

services:
    # ...
    sonata.admin.user:
        class: My\ProjectBundle\Admin\UserAdmin
        tags:
            - {name: sonata.admin, manager_type: orm, group: users, label: users}
        arguments:
            - null
            - Application\Sonata\UserBundle\Entity\User
            - SonataAdminBundle:CRUD
            - @security.context #forth argument
        calls:
            - [setTranslationDomain, [MyProjectBundle]]
            - [setUserManager, [@fos_user.user_manager]]

在您的 UserAdmin 类中，覆盖构造函数：

namespace My\ProjectBundle\Admin;

class UserAdmin extends Admin
{

    private $securityContext = null;

    public function __construct($code, $class, $baseControllerName, $secutiryContext=null)
    {
        parent::__construct($code, $class, $baseControllerName);
        $this->securityContext = $securityContext;
    }

    public function createQuery($context = 'list')
    {
        $query = parent::createQuery($context);
        $query->andWhere( 
            $query->expr()->eq($query->getRootAlias().'.company', ':comp')
        );
        $query->setParameter('comp', $this->securityContext->user->getCompany());
        return $query;
    }
}

我没有测试此代码，但我使用此方法在奏鸣曲管理员中注入 service_container 以使用 Gedmo Uploadable 管理文件上传。

希望这可以帮助。

Answer 2

我知道这是一个非常古老的问题，但是呢？

$user = $this->getConfigurationPool()->getContainer()->get('security.context')->getToken()->getUser();

第一件事public function createQuery($context = 'list') {

然后setParameter你可以使用$user->getCompany()

像这样;

public function createQuery($context = 'list')
{
    $user = $this->getConfigurationPool()->getContainer()->get('security.context')->getToken()->getUser();

    $query = parent::createQuery($context);
    $query->andWhere( 
        $query->expr()->eq($query->getRootAlias().'.company', ':comp')
    );
    $query->setParameter('comp', $user->getCompany());
    return $query;
}