ezmigrationbundle/Core/AuthenticatedUserSetterTrait.php

<?php

declare(strict_types=1);

namespace Kaliop\eZMigrationBundle\Core;

use eZ\Publish\API\Repository\Exceptions\NotFoundException;
use eZ\Publish\API\Repository\PermissionResolver;
use eZ\Publish\API\Repository\UserService;
use eZ\Publish\API\Repository\Values\User\UserReference;
use Kaliop\eZMigrationBundle\API\Exception\InvalidUserAccountException;

trait AuthenticatedUserSetterTrait
{
    /** @var \eZ\Publish\API\Repository\PermissionResolver */
    private $permissionResolver;

    /** @var \eZ\Publish\API\Repository\UserService */
    private $userService;

    /**
     * Helper method to authenticate a user that can make changes to the system.
     *
     * @throws \Kaliop\eZMigrationBundle\API\Exception\InvalidUserAccountException
     */
    protected function authenticateUserByLogin(string $userLogin): void
    {
        try {
            $user = $this->userService->loadUserByLogin($userLogin);
            $this->permissionResolver->setCurrentUserReference($user);
        } catch (NotFoundException $e) {
            throw new InvalidUserAccountException(
                sprintf(
                    'Could not find the required user account to be used for logging in: "%s". UserService says: %s',
                    $userLogin,
                    $e->getMessage()
                ),
                $e->getCode(),
                $e
            );
        }
    }

    protected function authenticateUserByReference(UserReference $userReference): void
    {
        $this->permissionResolver->setCurrentUserReference($userReference);
    }

    protected function getCurrentUser(): UserReference
    {
        return $this->permissionResolver->getCurrentUserReference();
    }

    public function setPermissionResolver(PermissionResolver $permissionResolver): void
    {
        $this->permissionResolver = $permissionResolver;
    }

    public function setUserService(UserService $userService): void
    {
        $this->userService = $userService;
    }
}