Retrieve username/password from database

Inject the current user & the repo to retrieve username/password from the database
This commit is contained in:
Jeremy Benoist 2017-05-01 22:13:17 +02:00
parent fc6d92c63d
commit 5a9bc00726
No known key found for this signature in database
GPG key ID: BCA73962457ACC3C
4 changed files with 49 additions and 13 deletions

View file

@ -60,6 +60,3 @@ parameters:
redis_port: 6379 redis_port: 6379
redis_path: null redis_path: null
redis_password: null redis_password: null
# sites credentials
sites_credentials: {}

View file

@ -6,6 +6,8 @@ use BD\GuzzleSiteAuthenticator\SiteConfig\SiteConfig;
use BD\GuzzleSiteAuthenticator\SiteConfig\SiteConfigBuilder; use BD\GuzzleSiteAuthenticator\SiteConfig\SiteConfigBuilder;
use Graby\SiteConfig\ConfigBuilder; use Graby\SiteConfig\ConfigBuilder;
use Psr\Log\LoggerInterface; use Psr\Log\LoggerInterface;
use Wallabag\CoreBundle\Repository\SiteCredentialRepository;
use Wallabag\UserBundle\Entity\User;
class GrabySiteConfigBuilder implements SiteConfigBuilder class GrabySiteConfigBuilder implements SiteConfigBuilder
{ {
@ -13,26 +15,36 @@ class GrabySiteConfigBuilder implements SiteConfigBuilder
* @var ConfigBuilder * @var ConfigBuilder
*/ */
private $grabyConfigBuilder; private $grabyConfigBuilder;
/** /**
* @var array * @var SiteCredentialRepository
*/ */
private $credentials; private $credentialRepository;
/** /**
* @var LoggerInterface * @var LoggerInterface
*/ */
private $logger; private $logger;
/**
* @var User
*/
private $currentUser;
/** /**
* GrabySiteConfigBuilder constructor. * GrabySiteConfigBuilder constructor.
* *
* @param ConfigBuilder $grabyConfigBuilder * @param ConfigBuilder $grabyConfigBuilder
* @param array $credentials * @param User $currentUser
* @param SiteCredentialRepository $credentialRepository
* @param LoggerInterface $logger * @param LoggerInterface $logger
*/ */
public function __construct(ConfigBuilder $grabyConfigBuilder, array $credentials, LoggerInterface $logger) public function __construct(ConfigBuilder $grabyConfigBuilder, User $currentUser, SiteCredentialRepository $credentialRepository, LoggerInterface $logger)
{ {
$this->grabyConfigBuilder = $grabyConfigBuilder; $this->grabyConfigBuilder = $grabyConfigBuilder;
$this->credentials = $credentials; $this->credentialRepository = $credentialRepository;
$this->currentUser = $currentUser;
$this->logger = $logger; $this->logger = $logger;
} }
@ -47,7 +59,9 @@ class GrabySiteConfigBuilder implements SiteConfigBuilder
$host = substr($host, 4); $host = substr($host, 4);
} }
if (empty($this->credentials[$host])) { $credentials = $this->credentialRepository->findOneByHostAndUser($host, $this->currentUser->getId());
if (null === $credentials) {
$this->logger->debug('Auth: no credentials available for host.', ['host' => $host]); $this->logger->debug('Auth: no credentials available for host.', ['host' => $host]);
return false; return false;
@ -62,8 +76,8 @@ class GrabySiteConfigBuilder implements SiteConfigBuilder
'passwordField' => $config->login_password_field ?: null, 'passwordField' => $config->login_password_field ?: null,
'extraFields' => $this->processExtraFields($config->login_extra_fields), 'extraFields' => $this->processExtraFields($config->login_extra_fields),
'notLoggedInXpath' => $config->not_logged_in_xpath ?: null, 'notLoggedInXpath' => $config->not_logged_in_xpath ?: null,
'username' => $this->credentials[$host]['username'], 'username' => $credentials['username'],
'password' => $this->credentials[$host]['password'], 'password' => $credentials['password'],
]; ];
$config = new SiteConfig($parameters); $config = new SiteConfig($parameters);

View file

@ -7,4 +7,22 @@ namespace Wallabag\CoreBundle\Repository;
*/ */
class SiteCredentialRepository extends \Doctrine\ORM\EntityRepository class SiteCredentialRepository extends \Doctrine\ORM\EntityRepository
{ {
/**
* Retrieve one username/password for the given host and userId.
*
* @param string $host
* @param int $userId
*
* @return null|array
*/
public function findOneByHostAndUser($host, $userId)
{
return $this->createQueryBuilder('s')
->select('s.username', 's.password')
->where('s.host = :hostname')->setParameter('hostname', $host)
->andWhere('s.user = :userId')->setParameter('userId', $userId)
->setMaxResults(1)
->getQuery()
->getOneOrNullResult();
}
} }

View file

@ -63,7 +63,8 @@ services:
class: Wallabag\CoreBundle\GuzzleSiteAuthenticator\GrabySiteConfigBuilder class: Wallabag\CoreBundle\GuzzleSiteAuthenticator\GrabySiteConfigBuilder
arguments: arguments:
- "@wallabag_core.graby.config_builder" - "@wallabag_core.graby.config_builder"
- "%sites_credentials%" - "@=service('security.token_storage').getToken().getUser()"
- "@wallabag_core.site_credential_repository"
- '@logger' - '@logger'
tags: tags:
- { name: monolog.logger, channel: graby } - { name: monolog.logger, channel: graby }
@ -120,6 +121,12 @@ services:
arguments: arguments:
- WallabagCoreBundle:Tag - WallabagCoreBundle:Tag
wallabag_core.site_credential_repository:
class: Wallabag\CoreBundle\Repository\SiteCredentialRepository
factory: [ "@doctrine.orm.default_entity_manager", getRepository ]
arguments:
- WallabagCoreBundle:SiteCredential
wallabag_core.helper.entries_export: wallabag_core.helper.entries_export:
class: Wallabag\CoreBundle\Helper\EntriesExport class: Wallabag\CoreBundle\Helper\EntriesExport
arguments: arguments: