* * @copyright Copyright (c) 2017, ownCloud GmbH * @license AGPL-3.0 * * This code is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License, version 3, * as published by the Free Software Foundation. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License, version 3, * along with this program. If not, see * */ namespace OC\Core\Command\Db\Migrations; use OC\DB\MigrationService; use OC\Migration\ConsoleOutput; use OCP\IDBConnection; use Symfony\Component\Console\Command\Command; use Symfony\Component\Console\Exception\RuntimeException; use Symfony\Component\Console\Input\InputArgument; use Symfony\Component\Console\Input\InputInterface; use Symfony\Component\Console\Output\OutputInterface; class GenerateCommand extends Command { private static $_templateSimple = '; use OCP\Migration\SimpleMigrationStep; use OCP\Migration\IOutput; /** * Auto-generated migration step: Please modify to your needs! */ class Version extends SimpleMigrationStep { /** * @param IOutput $output * @since 13.0.0 */ public function preSchemaChange(IOutput $output) { } /** * @param \Closure $schema The `\Closure` returns a `Schema` * @param array $options * @return null|Schema * @since 13.0.0 */ public function changeSchema(\Closure $schema, array $options) { return null; } /** * @param IOutput $output * @since 13.0.0 */ public function postSchemaChange(IOutput $output) { } } '; /** @var IDBConnection */ private $connection; /** * @param IDBConnection $connection */ public function __construct(IDBConnection $connection) { $this->connection = $connection; parent::__construct(); } protected function configure() { $this ->setName('migrations:generate') ->addArgument('app', InputArgument::REQUIRED, 'Name of the app this migration command shall work on') ; parent::configure(); } public function execute(InputInterface $input, OutputInterface $output) { $appName = $input->getArgument('app'); $ms = new MigrationService($appName, $this->connection, new ConsoleOutput($output)); $version = date('YmdHis'); $path = $this->generateMigration($ms, $version); $output->writeln("New migration class has been generated to $path"); } /** * @param MigrationService $ms * @param string $version * @return string */ private function generateMigration(MigrationService $ms, $version) { $placeHolders = [ '', '', ]; $replacements = [ $ms->getMigrationsNamespace(), $version, ]; $code = str_replace($placeHolders, $replacements, self::$_templateSimple); $dir = $ms->getMigrationsDirectory(); $path = $dir . '/Version' . $version . '.php'; if (file_put_contents($path, $code) === false) { throw new RuntimeException('Failed to generate new migration step.'); } return $path; } }