121 lines
4.2 KiB
PHP
121 lines
4.2 KiB
PHP
<?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\Bundle\PropelBundle\Command;
|
|
|
|
use Propel\Runtime\Adapter\Pdo\MysqlAdapter;
|
|
use Propel\Runtime\Propel;
|
|
use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand;
|
|
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 ContainerAwareCommand
|
|
{
|
|
use FormattingHelpers;
|
|
|
|
/**
|
|
* {@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)
|
|
{
|
|
$connection = Propel::getConnection($input->getOption('connection'));
|
|
$adapter = Propel::getAdapter($connection->getName());
|
|
|
|
if (!$adapter instanceof MysqlAdapter) {
|
|
return $output->writeln('<error>This command is MySQL only.</error>');
|
|
}
|
|
|
|
if (!$input->getOption('force')) {
|
|
return $output->writeln('<error>You have to use the "--force" option to drop some tables.</error>');
|
|
}
|
|
|
|
$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 {
|
|
$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');
|
|
}
|
|
}
|
|
}
|