chore(): auto import of generated components into AppModule

jkuri · jkuri · commit 136b4ebae47c · 2016-07-29T08:38:20.000+02:00
diff --git a/addon/ng2/commands/generate.ts b/addon/ng2/commands/generate.ts
@@ -2,10 +2,14 @@ import * as EmberGenerateCommand from 'ember-cli/lib/commands/generate';
 import * as fs from 'fs';
 import * as path from 'path';
 import * as SilentError from 'silent-error';
+import * as _ from 'lodash';
+import * as stringUtils from 'ember-cli-string-utils';
 var chalk = require('chalk');
 import * as Blueprint from 'ember-cli/lib/models/blueprint';
 var EOL = require('os').EOL;
 
+import * as DynamicPathParser from '../utilities/dynamic-path-parser';
+
 const GenerateCommand = EmberGenerateCommand.extend({
   name: 'generate',
 
@@ -40,6 +44,45 @@ const GenerateCommand = EmberGenerateCommand.extend({
     };
     
     return EmberGenerateCommand.prototype.beforeRun.apply(this, arguments);
+  },
+
+  run: function(commandOptions, rawArgs) {
+    var blueprintName = rawArgs[0];
+
+    if (!blueprintName) {
+      return Promise.reject(new SilentError('The `ember generate` command requires a ' +
+                                            'blueprint name to be specified. ' +
+                                            'For more details, use `ember help`.'));
+    }
+
+    var Task = this.tasks.GenerateFromBlueprint;
+    var task = new Task({
+      ui: this.ui,
+      analytics: this.analytics,
+      project: this.project,
+      testing: this.testing,
+      settings: this.settings
+    });
+
+    var taskArgs = {
+      args: rawArgs
+    };
+
+    if (this.settings && this.settings.usePods && !commandOptions.classic) {
+      commandOptions.pod = !commandOptions.pod;
+    }
+
+    var taskOptions = _.merge(taskArgs, commandOptions || {});
+
+    if (this.project.initializeAddons) {
+      this.project.initializeAddons();
+    }
+
+    if (rawArgs[0] === 'component') {
+      return Promise.all([task.run(taskOptions), importIntoAppModule(rawArgs, this.project)])
+    } else {
+      return task.run(taskOptions);
+    }
   }
 });
 
@@ -48,6 +91,32 @@ function mapBlueprintName(name) {
   return mappedName ? mappedName : name;
 }
 
+function importIntoAppModule(args, project, task) {
+  const pathObj = DynamicPathParser(project, args[1]);
+  const name = `${pathObj.name}.component`;
+  const camelizedName = `${name.slice(0, 1).toUpperCase()}${stringUtils.camelize(name).slice(1)}`;
+  const importPath = `./${pathObj.base}/${name}`;
+  const appModulePath = path.join(pathObj.appRoot, 'app.module.ts');
+  const appModuleContents = fs.readFileSync(appModulePath, 'utf8');
+
+  return new Promise((resolve, reject) => {
+    let parts = appModuleContents.split(EOL);
+    let imports = parts.filter(l => l.match(/import (.*)/));    
+    let declarationsStr = parts.filter(l => l.match(/declarations/))[0];
+    let idd = parts.indexOf(declarationsStr);
+    let currentDeclarations = declarationsStr.trim().match(/\[([^)]+)\]/)[1].split(',').map(d => d.trim());
+    let declarations = [].concat(currentDeclarations);
+
+    parts[imports.length] = `import { ${camelizedName} } from '${importPath}';${EOL}`;
+    declarations.push(camelizedName);
+
+    parts[idd] = `  declarations: [${declarations.join(', ')}],`;
+
+    fs.writeFileSync(appModulePath, parts.join(EOL), 'utf8');
+    resolve();
+  });
+}
+
 const aliasMap = {
   'cl': 'class',
   'c': 'component',
