Merge branch 'ftwbzhao-develop'

kenjis · kenjis · commit 805bedca9787 · 2015-05-21T14:59:00.000+09:00
diff --git a/src/Command/Generate/Migration.php b/src/Command/Generate/Migration.php
@@ -22,7 +22,8 @@
  */
 class Migration extends Command
 {
-    public function __construct(Context $context, Stdio $stdio, CI_Controller $ci) {
+    public function __construct(Context $context, Stdio $stdio, CI_Controller $ci)
+    {
         parent::__construct($context, $stdio, $ci);
         $this->load->config('migration');
     }
@@ -44,8 +45,32 @@ public function __invoke($type, $classname)
         }
 
         $migration_path = $this->config->item('migration_path');
+
         $file_path = $migration_path . date('YmdHis') . '_' . $classname . '.php';
 
+        // check file exist
+        if (file_exists($file_path)) {
+            $this->stdio->errln(
+                "<<red>>The file \"$file_path\" already exists<<reset>>"
+            );
+            return Status::FAILURE;
+        }
+
+        // check class exist
+        foreach (glob($migration_path . '*_*.php') as $file) {
+            $name = basename($file, '.php');
+
+            // use date('YmdHis') so...
+            if (preg_match('/^\d{14}_(\w+)$/', $name, $match)) {
+                if (strtolower($match[1]) === strtolower($classname)) {
+                    $this->stdio->errln(
+                        "<<red>>The Class \"$classname\" already exists<<reset>>"
+                    );
+                    return Status::FAILURE;
+                }
+            }
+        }
+
         $template = file_get_contents(__DIR__ . '/templates/Migration.txt');
         $search = [
             '@@classname@@',
diff --git a/tests/Command/GenerateTest.php b/tests/Command/GenerateTest.php
@@ -50,7 +50,22 @@ public function test_migration_generate()
         $migration_path = __DIR__ . '/../Fake/migrations/';
         $this->ci->config->set_item('migration_path', $migration_path);
         $status = $this->cmd->__invoke('migration', 'Test_of_generate_migration');
-        $this->assertEquals(0, $status);
+        $this->assertEquals(Status::SUCCESS, $status);
+
+        foreach (glob($migration_path . '*_Test_of_generate_migration.php') as $file) {
+            unlink($file);
+        }
+    }
+
+    public function test_migration_generate_class_exist()
+    {
+        $migration_path = __DIR__ . '/../Fake/migrations/';
+        $this->ci->config->set_item('migration_path', $migration_path);
+        $status = $this->cmd->__invoke('migration', 'Test_of_generate_migration');
+        $status = $this->cmd->__invoke('migration', 'Test_of_generate_migration');
+        $this->assertEquals(Status::FAILURE, $status);
+        $status = $this->cmd->__invoke('migration', 'Test_of_Generate_Migration');
+        $this->assertEquals(Status::FAILURE, $status);
 
         foreach (glob($migration_path . '*_Test_of_generate_migration.php') as $file) {
             unlink($file);
@@ -63,7 +78,7 @@ public function test_migration_cannot_write_to_file()
         $this->ci->config->set_item('migration_path', $migration_path);
         $status = $this->cmd->__invoke('migration', 'Test_of_generate_migration');
         $this->assertEquals(Status::FAILURE, $status);
-        
+
         $this->stderr->rewind();
         $actual = $this->stderr->fread(8192);
         $this->assertContains("Can't write to ", $actual);
