check file && class exist

ftwbzhao · ftwbzhao · commit 164dd2aee729 · 2015-05-20T18:20:58.000+08:00
diff --git a/src/Command/Generate/Migration.php b/src/Command/Generate/Migration.php
@@ -44,8 +44,36 @@ public function __invoke($type, $classname)
         }
 
         $migration_path = $this->config->item('migration_path');
+
+        $classname = ucfirst(strtolower($classname));
+
         $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 ($match[1] === $classname) {
+                    $this->stdio->errln(
+                        "<<red>>The Class \"$classname\" already exists<<reset>>"
+                    );
+                    return Status::FAILURE;
+                }
+            }
+        }
+
         $template = file_get_contents(__DIR__ . '/templates/Migration.txt');
         $search = [
             '@@classname@@',
