feat: add ability to resolve path for custom generator file

CSNW · Jun 29, 2020 · aee7d7a · aee7d7a
1 parent 1485388
commit aee7d7a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/bin/migrate-create b/bin/migrate-create
@@ -53,7 +53,12 @@ function create (name) {
   }
 
   // Load the template generator
-  var gen = require(program.generator)
+  var gen
+  try {
+    gen = require(program.generator)
+  } catch (e) {
+    gen = require(path.resolve(program.generator))
+  }
   gen({
     name: name,
     dateFormat: program.dateFormat,