src/Wallabag/CoreBundle/Command/InstallCommand.php

   1 <?php
   2
   3 namespace Wallabag\CoreBundle\Command;
   4
   5 use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand;
   6 use Symfony\Component\Console\Input\InputInterface;
   7 use Symfony\Component\Console\Input\InputOption;
   8 use Symfony\Component\Console\Input\ArrayInput;
   9 use Symfony\Component\Console\Output\OutputInterface;
  10 use Symfony\Component\Console\Output\NullOutput;
  11 use Wallabag\CoreBundle\Entity\User;
  12 use Wallabag\CoreBundle\Entity\Config;
  13
  14 class InstallCommand extends ContainerAwareCommand
  15 {
  16     /**
  17      * @var InputInterface
  18      */
  19     protected $defaultInput;
  20
  21     /**
  22      * @var OutputInterface
  23      */
  24     protected $defaultOutput;
  25
  26     protected function configure()
  27     {
  28         $this
  29             ->setName('wallabag:install')
  30             ->setDescription('Wallabag installer.')
  31             ->addOption(
  32                'reset',
  33                null,
  34                InputOption::VALUE_NONE,
  35                'Reset current database'
  36             )
  37         ;
  38     }
  39
  40     protected function execute(InputInterface $input, OutputInterface $output)
  41     {
  42         $this->defaultInput = $input;
  43         $this->defaultOutput = $output;
  44
  45         $output->writeln('<info>Installing Wallabag...</info>');
  46         $output->writeln('');
  47
  48         $this
  49             ->checkRequirements()
  50             ->setupDatabase()
  51             ->setupAdmin()
  52             ->setupAsset()
  53         ;
  54
  55         $output->writeln('<info>Wallabag has been successfully installed.</info>');
  56         $output->writeln('<comment>Just execute `php app/console server:run` for using wallabag: http://localhost:8000</comment>');
  57     }
  58
  59     protected function checkRequirements()
  60     {
  61         $this->defaultOutput->writeln('<info><comment>Step 1 of 4.</comment> Checking system requirements.</info>');
  62
  63         $fulfilled = true;
  64
  65         // @TODO: find a better way to check requirements
  66         $label = '<comment>PCRE</comment>';
  67         if (extension_loaded('pcre')) {
  68             $status = '<info>OK!</info>';
  69             $help = '';
  70         } else {
  71             $fulfilled = false;
  72             $status = '<error>ERROR!</error>';
  73             $help = 'You should enabled PCRE extension';
  74         }
  75         $rows[] = array($label, $status, $help);
  76
  77         $label = '<comment>DOM</comment>';
  78         if (extension_loaded('DOM')) {
  79             $status = '<info>OK!</info>';
  80             $help = '';
  81         } else {
  82             $fulfilled = false;
  83             $status = '<error>ERROR!</error>';
  84             $help = 'You should enabled DOM extension';
  85         }
  86         $rows[] = array($label, $status, $help);
  87
  88         $this->getHelper('table')
  89             ->setHeaders(array('Checked', 'Status', 'Recommendation'))
  90             ->setRows($rows)
  91             ->render($this->defaultOutput);
  92
  93         if (!$fulfilled) {
  94             throw new \RuntimeException('Some system requirements are not fulfilled. Please check output messages and fix them.');
  95         } else {
  96             $this->defaultOutput->writeln('<info>Success! Your system can run Wallabag properly.</info>');
  97         }
  98
  99         $this->defaultOutput->writeln('');
 100
 101         return $this;
 102     }
 103
 104     protected function setupDatabase()
 105     {
 106         $this->defaultOutput->writeln('<info><comment>Step 2 of 4.</comment> Setting up database.</info>');
 107
 108         // user want to reset everything? Don't care about what is already here
 109         if (true === $this->defaultInput->getOption('reset')) {
 110             $this->defaultOutput->writeln('Droping database, creating database and schema');
 111
 112             $this
 113                 ->runCommand('doctrine:database:drop', array('--force' => true))
 114                 ->runCommand('doctrine:database:create')
 115                 ->runCommand('doctrine:schema:create')
 116             ;
 117
 118             return $this;
 119         }
 120
 121         if (!$this->isDatabasePresent()) {
 122             $this->defaultOutput->writeln('Creating database and schema, clearing the cache');
 123
 124             $this
 125                 ->runCommand('doctrine:database:create')
 126                 ->runCommand('doctrine:schema:create')
 127                 ->runCommand('cache:clear')
 128             ;
 129
 130             return $this;
 131         }
 132
 133         $dialog = $this->getHelper('dialog');
 134
 135         if ($dialog->askConfirmation($this->defaultOutput, '<question>It appears that your database already exists. Would you like to reset it? (y/N)</question> ', false)) {
 136             $this->defaultOutput->writeln('Droping database, creating database and schema');
 137
 138             $this
 139                 ->runCommand('doctrine:database:drop', array('--force' => true))
 140                 ->runCommand('doctrine:database:create')
 141                 ->runCommand('doctrine:schema:create')
 142             ;
 143         } elseif ($this->isSchemaPresent()) {
 144             if ($dialog->askConfirmation($this->defaultOutput, '<question>Seems like your database contains schema. Do you want to reset it? (y/N)</question> ', false)) {
 145                 $this->defaultOutput->writeln('Droping schema and creating schema');
 146
 147                 $this
 148                     ->runCommand('doctrine:schema:drop', array('--force' => true))
 149                     ->runCommand('doctrine:schema:create')
 150                 ;
 151             }
 152         } else {
 153             $this->defaultOutput->writeln('Creating schema');
 154
 155             $this
 156                 ->runCommand('doctrine:schema:create')
 157             ;
 158         }
 159
 160         $this->defaultOutput->writeln('Clearing the cache');
 161         $this->runCommand('cache:clear');
 162
 163         /*
 164         if ($this->getHelperSet()->get('dialog')->askConfirmation($this->defaultOutput, '<question>Load fixtures (Y/N)?</question>', false)) {
 165             $doctrineConfig = $this->getContainer()->get('doctrine.orm.entity_manager')->getConnection()->getConfiguration();
 166             $logger = $doctrineConfig->getSQLLogger();
 167             // speed up fixture load
 168             $doctrineConfig->setSQLLogger(null);
 169             $this->runCommand('doctrine:fixtures:load');
 170             $doctrineConfig->setSQLLogger($logger);
 171         }
 172         */
 173
 174         $this->defaultOutput->writeln('');
 175
 176         return $this;
 177     }
 178
 179     protected function setupAdmin()
 180     {
 181         $this->defaultOutput->writeln('<info><comment>Step 3 of 4.</comment> Administration setup.</info>');
 182
 183         $dialog = $this->getHelperSet()->get('dialog');
 184
 185         if (false === $dialog->askConfirmation($this->defaultOutput, '<question>Would you like to create a new user ? (y/N)</question>', true)) {
 186             return $this;
 187         }
 188
 189         $em = $this->getContainer()->get('doctrine.orm.entity_manager');
 190
 191         $user = new User();
 192         $user->setUsername($dialog->ask($this->defaultOutput, '<question>Username</question> <comment>(default: wallabag)</comment> :', 'wallabag'));
 193         $user->setPassword($dialog->ask($this->defaultOutput, '<question>Password</question> <comment>(default: wallabag)</comment> :', 'wallabag'));
 194         $user->setEmail($dialog->ask($this->defaultOutput, '<question>Email:</question>', ''));
 195         $user->setEnabled(true);
 196
 197         $em->persist($user);
 198
 199         $config = new Config($user);
 200         $config->setTheme($this->getContainer()->getParameter('theme'));
 201         $config->setItemsPerPage($this->getContainer()->getParameter('items_on_page'));
 202         $config->setRssLimit($this->getContainer()->getParameter('rss_limit'));
 203         $config->setLanguage($this->getContainer()->getParameter('language'));
 204
 205         $em->persist($config);
 206
 207         $em->flush();
 208
 209         $this->defaultOutput->writeln('');
 210
 211         return $this;
 212     }
 213
 214     protected function setupAsset()
 215     {
 216         $this->defaultOutput->writeln('<info><comment>Step 4 of 4.</comment> Installing assets.</info>');
 217
 218         $this
 219             ->runCommand('assets:install')
 220             ->runCommand('assetic:dump')
 221         ;
 222
 223         $this->defaultOutput->writeln('');
 224
 225         return $this;
 226     }
 227
 228     /**
 229      * Run a command.
 230      *
 231      * @param string $command
 232      * @param array  $parameters Parameters to this command (usually 'force' => true)
 233      */
 234     protected function runCommand($command, $parameters = array())
 235     {
 236         $parameters = array_merge(
 237             array('command' => $command),
 238             $parameters,
 239             array(
 240                 '--no-debug' => true,
 241                 '--env' => $this->defaultInput->getOption('env') ?: 'dev',
 242             )
 243         );
 244
 245         if ($this->defaultInput->getOption('no-interaction')) {
 246             $parameters = array_merge($parameters, array('--no-interaction' => true));
 247         }
 248
 249         $this->getApplication()->setAutoExit(false);
 250         $exitCode = $this->getApplication()->run(new ArrayInput($parameters), new NullOutput());
 251
 252         if (0 !== $exitCode) {
 253             $this->getApplication()->setAutoExit(true);
 254
 255             $errorMessage = sprintf('The command "%s" terminated with an error code: %u.', $command, $exitCode);
 256             $this->defaultOutput->writeln("<error>$errorMessage</error>");
 257             $exception = new \Exception($errorMessage, $exitCode);
 258
 259             throw $exception;
 260         }
 261
 262         // PDO does not always close the connection after Doctrine commands.
 263         // See https://github.com/symfony/symfony/issues/11750.
 264         $this->getContainer()->get('doctrine')->getManager()->getConnection()->close();
 265
 266         return $this;
 267     }
 268
 269     /**
 270      * Check if the database already exists.
 271      *
 272      * @return bool
 273      */
 274     private function isDatabasePresent()
 275     {
 276         $connection = $this->getContainer()->get('doctrine')->getManager()->getConnection();
 277         $databaseName = $connection->getDatabase();
 278
 279         try {
 280             $schemaManager = $connection->getSchemaManager();
 281         } catch (\Exception $exception) {
 282             if (false !== strpos($exception->getMessage(), sprintf("Unknown database '%s'", $databaseName))) {
 283                 return false;
 284             }
 285
 286             throw $exception;
 287         }
 288
 289         // custom verification for sqlite, since `getListDatabasesSQL` doesn't work for sqlite
 290         if ('sqlite' == $schemaManager->getDatabasePlatform()->getName()) {
 291             $params = $this->getContainer()->get('doctrine.dbal.default_connection')->getParams();
 292
 293             if (isset($params['path']) && file_exists($params['path'])) {
 294                 return true;
 295             }
 296
 297             return false;
 298         }
 299
 300         return in_array($databaseName, $schemaManager->listDatabases());
 301     }
 302
 303     /**
 304      * Check if the schema is already created.
 305      * If we found at least oen table, it means the schema exists.
 306      *
 307      * @return bool
 308      */
 309     private function isSchemaPresent()
 310     {
 311         $schemaManager = $this->getContainer()->get('doctrine')->getManager()->getConnection()->getSchemaManager();
 312
 313         return count($schemaManager->listTableNames()) > 0 ? true : false;
 314     }
 315 }