Update CreateMigration.php

Author: usernane

webfiori/framework/cli/helpers/CreateMigration.php

@@ -26,27 +26,55 @@ class CreateMigration extends CreateClassHelper {
      * @var DatabaseMigrationWriter
      */
     private $writer;
+    private $isConfigured;
     /**
      * Creates new instance of the class.
      *
      * @param CreateCommand $command A command that is used to call the class.
      */
     public function __construct(CreateCommand $command) {
         $ns = APP_DIR.'\\database\\migrations';
-
+        $this->isConfigured = false;
         if (!$command->isArgProvided('--defaults')) {
             $ns = CLIUtils::readNamespace($command, $ns , 'Migration namespace:');
         }
 
-        $runner = new MigrationsRunner(ROOT_PATH.DS. str_replace('\\', DS, $ns), $ns, null);
-        parent::__construct($command, new DatabaseMigrationWriter($runner));
-        $this->writer = $this->getWriter();
-        $this->setNamespace($ns);
-        
-        if (!$command->isArgProvided('--defaults')) {
-            $this->setClassName($command->readClassName('Provide an optional name for the class that will have migration logic:', null));
-            $this->readClassInfo();
+        $runner = $this->initRunner($ns, $command);
+        if ($runner === null) {
+            $command->error("Unable to set migrations path.");
+        } else {
+            parent::__construct($command, new DatabaseMigrationWriter($runner));
+            $this->writer = $this->getWriter();
+            $this->setNamespace($ns);
+
+            if (!$command->isArgProvided('--defaults')) {
+                $this->setClassName($command->readClassName('Provide an optional name for the class that will have migration logic:', null));
+                $this->readClassInfo();
+                $this->isConfigured = true;
+            }
+        }
+    }
+    public function isConfigured() : bool {
+        return $this->isConfigured;
+    }
+    private function initRunner($ns, $command) {
+        $path = ROOT_PATH.DS. str_replace('\\', DS, $ns);
+        if (!is_dir($path)) {
+            $command->warning("The path '$path' does not exist.");
+            $create = $command->confirm("Would you like to create it?", true);
+            
+            if ($create) {
+                if (!mkdir($path)) {
+                    $command->error("Unable to create directory.");
+                    return null;
+                }
+            } else {
+                return null;
+            }
+        } else {
+            return null;
         }
+        return new MigrationsRunner($path, $ns, null);
     }
 
     private function readClassInfo() {