]>
Commit | Line | Data |
---|---|---|
1 | <?php | |
2 | ||
3 | namespace Wallabag\ImportBundle\Command; | |
4 | ||
5 | use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand; | |
6 | use Symfony\Component\Config\Definition\Exception\Exception; | |
7 | use Symfony\Component\Console\Input\InputArgument; | |
8 | use Symfony\Component\Console\Input\InputInterface; | |
9 | use Symfony\Component\Console\Input\InputOption; | |
10 | use Symfony\Component\Console\Output\OutputInterface; | |
11 | use Symfony\Component\Security\Core\Authentication\Token\UsernamePasswordToken; | |
12 | ||
13 | class ImportCommand extends ContainerAwareCommand | |
14 | { | |
15 | protected function configure() | |
16 | { | |
17 | $this | |
18 | ->setName('wallabag:import') | |
19 | ->setDescription('Import entries from a JSON export') | |
20 | ->addArgument('username', InputArgument::REQUIRED, 'User to populate') | |
21 | ->addArgument('filepath', InputArgument::REQUIRED, 'Path to the JSON file') | |
22 | ->addOption('importer', null, InputOption::VALUE_OPTIONAL, 'The importer to use: v1, v2, instapaper, pinboard, readability, firefox or chrome', 'v1') | |
23 | ->addOption('markAsRead', null, InputOption::VALUE_OPTIONAL, 'Mark all entries as read', false) | |
24 | ->addOption('useUserId', null, InputOption::VALUE_NONE, 'Use user id instead of username to find account') | |
25 | ->addOption('disableContentUpdate', null, InputOption::VALUE_NONE, 'Disable fetching updated content from URL') | |
26 | ; | |
27 | } | |
28 | ||
29 | protected function execute(InputInterface $input, OutputInterface $output) | |
30 | { | |
31 | $output->writeln('Start : ' . (new \DateTime())->format('d-m-Y G:i:s') . ' ---'); | |
32 | ||
33 | if (!file_exists($input->getArgument('filepath'))) { | |
34 | throw new Exception(sprintf('File "%s" not found', $input->getArgument('filepath'))); | |
35 | } | |
36 | ||
37 | $em = $this->getContainer()->get('doctrine')->getManager(); | |
38 | // Turning off doctrine default logs queries for saving memory | |
39 | $em->getConnection()->getConfiguration()->setSQLLogger(null); | |
40 | ||
41 | if ($input->getOption('useUserId')) { | |
42 | $entityUser = $em->getRepository('WallabagUserBundle:User')->findOneById($input->getArgument('username')); | |
43 | } else { | |
44 | $entityUser = $em->getRepository('WallabagUserBundle:User')->findOneByUsername($input->getArgument('username')); | |
45 | } | |
46 | ||
47 | if (!\is_object($entityUser)) { | |
48 | throw new Exception(sprintf('User "%s" not found', $input->getArgument('username'))); | |
49 | } | |
50 | ||
51 | // Authenticate user for paywalled websites | |
52 | $token = new UsernamePasswordToken( | |
53 | $entityUser, | |
54 | null, | |
55 | 'main', | |
56 | $entityUser->getRoles()); | |
57 | ||
58 | $this->getContainer()->get('security.token_storage')->setToken($token); | |
59 | $user = $this->getContainer()->get('security.token_storage')->getToken()->getUser(); | |
60 | ||
61 | switch ($input->getOption('importer')) { | |
62 | case 'v2': | |
63 | $import = $this->getContainer()->get('wallabag_import.wallabag_v2.import'); | |
64 | break; | |
65 | case 'firefox': | |
66 | $import = $this->getContainer()->get('wallabag_import.firefox.import'); | |
67 | break; | |
68 | case 'chrome': | |
69 | $import = $this->getContainer()->get('wallabag_import.chrome.import'); | |
70 | break; | |
71 | case 'readability': | |
72 | $import = $this->getContainer()->get('wallabag_import.readability.import'); | |
73 | break; | |
74 | case 'instapaper': | |
75 | $import = $this->getContainer()->get('wallabag_import.instapaper.import'); | |
76 | break; | |
77 | case 'pinboard': | |
78 | $import = $this->getContainer()->get('wallabag_import.pinboard.import'); | |
79 | break; | |
80 | default: | |
81 | $import = $this->getContainer()->get('wallabag_import.wallabag_v1.import'); | |
82 | } | |
83 | ||
84 | $import->setMarkAsRead($input->getOption('markAsRead')); | |
85 | $import->setDisableContentUpdate($input->getOption('disableContentUpdate')); | |
86 | $import->setUser($user); | |
87 | ||
88 | $res = $import | |
89 | ->setFilepath($input->getArgument('filepath')) | |
90 | ->import(); | |
91 | ||
92 | if (true === $res) { | |
93 | $summary = $import->getSummary(); | |
94 | $output->writeln('<info>' . $summary['imported'] . ' imported</info>'); | |
95 | $output->writeln('<comment>' . $summary['skipped'] . ' already saved</comment>'); | |
96 | } | |
97 | ||
98 | $em->clear(); | |
99 | ||
100 | $output->writeln('End : ' . (new \DateTime())->format('d-m-Y G:i:s') . ' ---'); | |
101 | } | |
102 | } |