Added authentication during import command for paywalled websites

diff --git a/src/Wallabag/ImportBundle/Command/ImportCommand.php b/src/Wallabag/ImportBundle/Command/ImportCommand.php
index f9ffe994db821fafb815b546871d4f1eb04ae034..26903e9cbbc13a56e7ca32e418181fccba153d15 100644 (file)
--- a/src/Wallabag/ImportBundle/Command/ImportCommand.php
+++ b/src/Wallabag/ImportBundle/Command/ImportCommand.php
@@ -8,6 +8,7 @@ use Symfony\Component\Console\Input\InputArgument;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Input\InputOption;
 use Symfony\Component\Console\Output\OutputInterface;
+use Symfony\Component\Security\Core\Authentication\Token\UsernamePasswordToken;
 
 class ImportCommand extends ContainerAwareCommand
 {
@@ -38,15 +39,25 @@ class ImportCommand extends ContainerAwareCommand
         $em->getConnection()->getConfiguration()->setSQLLogger(null);
 
         if ($input->getOption('useUserId')) {
-            $user = $em->getRepository('WallabagUserBundle:User')->findOneById($input->getArgument('username'));
+            $entityUser = $em->getRepository('WallabagUserBundle:User')->findOneById($input->getArgument('username'));
         } else {
-            $user = $em->getRepository('WallabagUserBundle:User')->findOneByUsername($input->getArgument('username'));
+            $entityUser = $em->getRepository('WallabagUserBundle:User')->findOneByUsername($input->getArgument('username'));
         }
 
-        if (!\is_object($user)) {
+        if (!\is_object($entityUser)) {
             throw new Exception(sprintf('User "%s" not found', $input->getArgument('username')));
         }
 
+        // Authenticate user for paywalled websites
+        $token = new UsernamePasswordToken(
+            $entityUser,
+            null,
+            'main',
+            $entityUser->getRoles());
+
+        $this->getContainer()->get('security.token_storage')->setToken($token);
+        $user = $this->getContainer()->get('security.token_storage')->getToken()->getUser();
+
         switch ($input->getOption('importer')) {
             case 'v2':
                 $import = $this->getContainer()->get('wallabag_import.wallabag_v2.import');