Template by set

Zeyckler · Zeyckler · commit 44150a7c3922 · 2018-12-31T19:58:50.000+01:00
diff --git a/README.md b/README.md
@@ -391,11 +391,13 @@ $container['phpmig.sets'] = function ($container) {
     return array(
         'cms' => array(
             'adapter' => new Adapter\File\Flat('modules/migrationLogs/cms_migrations.log'),
-            'migrations_path' => 'migrations/cms'
+            'migrations_path' => 'migrations/cms',
+            'migrations_template_path' => 'PhpmigCmsTemplate.php'
         ),
         'blog' => array(
             'adapter' => new Adapter\File\Flat('modules/migrationLogs/blog_migrations.log'),
-            'migrations_path' => 'migrations/blog'
+            'migrations_path' => 'migrations/blog',
+            'migrations_template_path' => 'PhpmigBlogTemplate.php',
         )
     );
 };
diff --git a/src/Phpmig/Console/Command/GenerateCommand.php b/src/Phpmig/Console/Command/GenerateCommand.php
@@ -34,16 +34,17 @@ protected function configure()
         parent::configure();
 
         $this->setName('generate')
-             ->addArgument('name', InputArgument::REQUIRED, 'The name for the migration')
-             ->addArgument('path', InputArgument::OPTIONAL, 'The directory in which to put the migration ( optional if phpmig.migrations_path is setted )')
-             ->setDescription('Generate a new migration')
-             ->setHelp(<<<EOT
+            ->addArgument('name', InputArgument::REQUIRED, 'The name for the migration')
+            ->addArgument('path', InputArgument::OPTIONAL,
+                'The directory in which to put the migration ( optional if phpmig.migrations_path is setted )')
+            ->setDescription('Generate a new migration')
+            ->setHelp(<<<EOT
 The <info>generate</info> command creates a new migration with the name and path specified
 
 <info>phpmig generate Dave ./migrations</info>
 
 EOT
-        );
+            );
     }
 
     /**
@@ -55,18 +56,18 @@ protected function execute(InputInterface $input, OutputInterface $output)
 
         $path = $input->getArgument('path');
         $set = $input->getOption('set');
-        if( null === $path ){
-            if (isset($this->container['phpmig.migrations_path'])) {
+        if (null === $path) {
+            if (true === isset($this->container['phpmig.migrations_path'])) {
                 $path = $this->container['phpmig.migrations_path'];
             }
-            if (isset($this->container['phpmig.sets']) && isset($this->container['phpmig.sets'][$set]['migrations_path'])) {
+            if (true === isset($this->container['phpmig.sets'][$set]['migrations_path'])) {
                 $path = $this->container['phpmig.sets'][$set]['migrations_path'];
             }
         }
         $locator = new FileLocator(array());
-        $path    = $locator->locate($path, getcwd(), $first = true);
+        $path = $locator->locate($path, getcwd(), $first = true);
 
-        if (!is_writeable($path)) {
+        if (!is_writable($path)) {
             throw new \InvalidArgumentException(sprintf(
                 'The directory "%s" is not writeable',
                 $path
@@ -77,7 +78,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
 
         $migrationName = $this->transMigName($input->getArgument('name'));
 
-        $basename  = date('YmdHis') . '_' . $migrationName . '.php';
+        $basename = date('YmdHis') . '_' . $migrationName . '.php';
 
         $path = $path . DIRECTORY_SEPARATOR . $basename;
 
@@ -90,8 +91,13 @@ protected function execute(InputInterface $input, OutputInterface $output)
 
         $className = $this->migrationToClassName($migrationName);
 
-        if (isset($this->container['phpmig.migrations_template_path'])) {
-            $migrationsTemplatePath = $this->container['phpmig.migrations_template_path'];
+        if (isset($this->container['phpmig.migrations_template_path']) || isset($this->container['phpmig.sets'][$set]['migrations_template_path'])) {
+            if (true === isset($this->container['phpmig.migrations_template_path'])) {
+                $migrationsTemplatePath = $this->container['phpmig.migrations_template_path'];
+            } else {
+                $migrationsTemplatePath = $this->container['phpmig.sets'][$set]['migrations_template_path'];
+            }
+
             if (false === file_exists($migrationsTemplatePath)) {
                 throw new \RuntimeException(sprintf(
                     'The template file "%s" not found',
@@ -101,7 +107,7 @@ protected function execute(InputInterface $input, OutputInterface $output)
 
             if (preg_match('/\.php$/', $migrationsTemplatePath)) {
                 ob_start();
-                include($migrationsTemplatePath);
+                include $migrationsTemplatePath;
                 $contents = ob_get_clean();
             } else {
                 $contents = file_get_contents($migrationsTemplatePath);
