feat(hook): inject new template into create migration command

Signed-off-by: Tobias Gurtzick <[email protected]>

Author: wzrdtales

api.js

@@ -37,7 +37,6 @@ function dbmigrate (plugins, isModule, options, callback) {
   }
   var internals = this.internals;
 
-  console.log(load('fn/plugin'));
   this.internals.plugins = load('fn/plugin')(plugins);
 
   if (typeof callback === 'function') this.internals.onComplete = callback;

index.js

@@ -47,14 +47,13 @@ function loadPlugins (options) {
   for (; i < length; ++i) {
     var plugin = require(path.join(options.cwd, 'node_modules', plugins[i]));
 
-    if (
-      typeof plugin.name !== 'string' ||
-      !plugin.hooks ||
-      !plugin.loadPlugin
-    ) {
+    if (!plugin.hooks || !plugin.loadPlugin) {
       continue;
     }
 
+    // name is now derived from package name
+    plugin.name = plugins[i];
+
     plugin.hooks.forEach(function (hook) {
       hooks[hook] = hooks[hook] || [];
       hooks[hook].push(plugin);

lib/commands/create-migration.js

@@ -22,6 +22,10 @@ async function createMigrationDir (dir) {
 async function executeCreateMigration (internals, config) {
   let migrationsDir = internals.argv['migrations-dir'];
   let path;
+  let hooks = false;
+  let pluginTemplate = false;
+  let customWrite = false;
+  const { plugins } = internals;
 
   internals.runTimestamp = new Date();
 
@@ -79,19 +83,79 @@ async function executeCreateMigration (internals, config) {
   } else if (shouldCreateCoffeeFile(internals, config)) {
     templateType = Migration.TemplateType.DEFAULT_COFFEE;
   }
-  const migration = new Migration(
-    internals.argv.title +
-      (shouldCreateCoffeeFile(internals, config) ? '.coffee' : '.js'),
-    path,
-    internals.runTimestamp,
-    templateType,
-    internals.plugins
-  );
 
-  await migration.write();
-  log.info(util.format('Created migration at %s', migration.file.path));
-  if (shouldCreateSqlFiles(internals, config)) {
-    return createSqlFiles(internals, config);
+  if (plugins) {
+    hooks = plugins.hook('create:template') || [];
+  }
+
+  if (hooks !== false) {
+    for (const plugin of hooks) {
+      const template = plugin['init:template']();
+
+      if (internals.argv[template.option] || config[template.option]) {
+        pluginTemplate = plugin;
+        customWrite = plugin['create:template:custom:write'] === true;
+
+        if (customWrite) {
+          templateType = `${plugin.name}${templateType}`;
+        } else {
+          templateType = template.type;
+        }
+
+        // no reason to continue processing here
+        break;
+      }
+    }
+  }
+
+  if (!pluginTemplate && !customWrite) {
+    const migration = new Migration(
+      internals.argv.title +
+        (shouldCreateCoffeeFile(internals, config) ? '.coffee' : '.js'),
+      path,
+      internals.runTimestamp,
+      templateType,
+      internals.plugins
+    );
+
+    await migration.write();
+    log.info(util.format('Created migration at %s', migration.file.path));
+    if (shouldCreateSqlFiles(internals, config)) {
+      return createSqlFiles(internals, config);
+    }
+  } else {
+    const plugin = pluginTemplate;
+
+    if (typeof plugin['write:template'] !== 'function') {
+      log.error(`Plugin ${plugin.name} does not have function write:template`);
+      throw new Error(`write:template not existent!`);
+    }
+
+    await plugin['write:template'](
+      { argv: { ...internals.argv }, config: { ...config } },
+      async opts => {
+        let title = internals.argv.title;
+        let _path = path;
+        const extension = opts.extension || '.js';
+
+        if (opts.suffix) {
+          title += opts.suffix;
+        }
+
+        if (opts.pathExtension) {
+          _path += opts.pathExtension;
+        }
+
+        const migration = new Migration(
+          title + extension,
+          _path,
+          internals.runTimestamp,
+          opts.type,
+          internals.plugins
+        );
+        await migration.write();
+      }
+    );
   }
 
   return Promise.resolve();