Implemented the table drop command

2013-11-04 21:29:16 +00:00 · 2013-11-04 21:29:16 +00:00 · 53db41026b
commit 53db41026b
parent ba6b0b9ee2
2 changed files with 159 additions and 0 deletions
--- a/Command/TableDropCommand.php
+++ b/Command/TableDropCommand.php
@ -0,0 +1,129 @@
+<?php
+
+/**
+ * This file is part of the PropelBundle package.
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ *
+ * @license    MIT License
+ */
+
+namespace Propel\PropelBundle\Command;
+
+use Propel\Runtime\Propel;
+use Symfony\Component\Console\Input\InputArgument;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Input\InputOption;
+use Symfony\Component\Console\Output\OutputInterface;
+
+/**
+ * @author Kévin Gomez <contact@kevingomez.fr>
+ */
+class TableDropCommand extends AbstractCommand
+{
+    /**
+     * {@inheritdoc}
+     */
+    protected function configure()
+    {
+        $this
+            ->setName('propel:table:drop')
+            ->setDescription('Drop a given table or all tables in the database.')
+
+            ->addOption('connection',   null, InputOption::VALUE_OPTIONAL, 'Connection to use. Example: default, bookstore')
+            ->addOption('force',        null, InputOption::VALUE_NONE, 'Set this parameter to execute this action.')
+            ->addArgument('table',      InputArgument::IS_ARRAY, 'Set this parameter to défine which table to delete (default all the table in the database.')
+        ;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    protected function execute(InputInterface $input, OutputInterface $output)
+    {
+        if (!$input->getOption('force')) {
+            $output->writeln('<error>You have to use the "--force" option to drop some tables.</error>');
+            return;
+        }
+
+        $tablesToDelete = $input->getArgument('table');
+        $nbTable = count($tablesToDelete);
+        $tablePlural = (($nbTable > 1 || $nbTable == 0) ? 's' : '' );
+
+        if ('prod' === $this->getApplication()->getKernel()->getEnvironment()) {
+            $count = $nbTable ?: 'all';
+
+            $this->writeSection(
+                $output,
+                'WARNING: you are about to drop ' . $count . ' table' . $tablePlural . ' in production !',
+                'bg=red;fg=white'
+            );
+
+            if (false === $this->askConfirmation($output, 'Are you sure ? (y/n) ', false)) {
+                $output->writeln('<info>Aborted, nice decision !</info>');
+
+                return -2;
+            }
+        }
+
+        try {
+            $connection = Propel::getConnection($input->getOption('connection'));
+            $adapter = Propel::getAdapter($connection->getName());
+
+            $showStatement = $connection->prepare('SHOW TABLES;');
+            $showStatement->execute();
+
+            $allTables = $showStatement->fetchAll(\PDO::FETCH_COLUMN);
+
+            if ($nbTable) {
+                foreach ($tablesToDelete as $tableToDelete) {
+                    if (!array_search($tableToDelete, $allTables)) {
+                        throw new \InvalidArgumentException(sprintf('Table %s doesn\'t exist in the database.', $tableToDelete));
+                    }
+                }
+            } else {
+                $tablesToDelete = $allTables;
+            }
+
+            $connection->exec('SET FOREIGN_KEY_CHECKS = 0;');
+
+            array_walk($tablesToDelete, function(&$table, $key, $dbAdapter) {
+                $table = $dbAdapter->quoteIdentifierTable($table);
+            }, $adapter);
+
+            $tablesToDelete = join(', ', $tablesToDelete);
+
+            if ('' !== $tablesToDelete) {
+                $connection->exec('DROP TABLE ' . $tablesToDelete . ' ;');
+
+                $output->writeln(sprintf('Table' . $tablePlural . ' <info><comment>%s</comment> has been dropped.</info>', $tablesToDelete));
+            } else {
+                $output->writeln('<info>No tables have been dropped</info>');
+            }
+
+            $connection->exec('SET FOREIGN_KEY_CHECKS = 1;');
+        } catch (\Exception $e) {
+            $this->writeSection($output, array(
+                '[Propel] Exception caught',
+                '',
+                $e->getMessage()
+            ), 'fg=white;bg=red');
+        }
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    protected function createSubCommandInstance()
+    {
+        // useless for this command
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    protected function getSubCommandArguments(InputInterface $input)
+    {
+        // useless for this command
+    }
+}