fix .gitignore, etc

Author: Awk34

.gitignore

@@ -4,5 +4,7 @@ bower_components/
 
 build/
 dist/
-generators
-test
+generators/
+test/
+!src/generators/
+!src/test/

src/generators/app/index.js

@@ -0,0 +1,46 @@
+import Generator from 'yeoman-generator';
+import chalk from 'chalk';
+
+class AppGenerator extends Generator {
+  initializing() {
+    if(!this.options['skip-message']) {
+      // this.log(chalk.magenta('You\'re using the fantastic NgComponent generator.\n'));
+      this.log(chalk.magenta('Initializing yo-rc.json configuration.\n'));
+    }
+  }
+
+  configuring() {
+    const config = {
+      appModulePath: this.options.appModulePath || '',
+      routeDirectory: this.options.routeDirectory || 'app/components/',
+      directiveDirectory: this.options.directiveDirectory || 'app/components/',
+      componentDirectory: this.options.componentDirectory || 'app/components/',
+      filterDirectory: this.options.filterDirectory || 'app/components/',
+      serviceDirectory: this.options.serviceDirectory || 'app/components/',
+      basePath: this.options.basePath || 'app',
+      moduleName: this.options.moduleName || '',
+      modulePrompt: {}.hasOwnProperty.call(this.options, 'modulePrompt') ? this.options.modulePrompt : true,
+      filters: this.options.filters || ['uirouter', 'jasmine'],
+      
+      extensions: this.options.extensions || ['js', 'html', 'scss'],
+      directiveSimpleTemplates: this.options.directiveSimple || '',
+      directiveComplexTemplates: this.options.directiveComplex || '',
+      filterTemplates: this.options.filter || '',
+      serviceTemplates: this.options.service || '',
+      factoryTemplates: this.options.factory || '',
+      controllerTemplates: this.options.controller || '',
+      componentTemplates: this.options.component || '',
+      decoratorTemplates: this.options.decorator || '',
+      providerTemplates: this.options.provider || '',
+      routeTemplates: this.options.route || ''
+    };
+
+    if(this.options.forceConfig) {
+      this.config.set(config);
+      this.config.save();
+    } else {
+      this.config.defaults(config);
+    }
+  }
+}
+module.exports = AppGenerator;

src/generators/base.js

@@ -179,8 +179,6 @@ export class BaseGenerator extends Generator {
 
     this.filters = this.config.get('filters');
     this.extensions = this.config.get('extensions');
-    this.hasFilter = filter => this.filters.indexOf(filter) !== -1;
-    this.hasExtension = ext => this.extensions.indexOf(ext) !== -1;
 
     this.scriptExt = this.hasExtension('ts') ? 'ts' : 'js';
     this.templateExt = this.hasExtension('pug') ? 'pug' : 'html';
@@ -205,6 +203,14 @@ export class BaseGenerator extends Generator {
     // return yoCheckPromise;
   }
 
+
+  hasFilter(filter) {
+    return this.filters.indexOf(filter) !== -1;
+  }
+  hasExtension(ext) {
+    return this.extensions.indexOf(ext) !== -1;
+  }
+
   /**
    * Copy templates from `source` to `destination` whily applying name transformations
    */

src/generators/route/module-transform.js

@@ -0,0 +1,60 @@
+import jscodeshift, { ImportSpecifier } from 'jscodeshift';
+const moduleName = 'Test1Module';
+const modulePath = './test0/test1.module';
+
+class NoModulesError extends Error {
+	constructor() {
+		super(`No NgModules found in app module.
+  Are you sure you have the correct path registered in 'appModulePath'?`)
+	}
+}
+class TooManyModulesError extends Error {
+	constructor() {
+		super(`More than one NgModule found in app module.
+  There should be only one.`)
+	}
+}
+
+/**
+ * @param {string} source
+ * @param {string} moduleName - ex 'MyModule'
+ * @param {string} modulePath - module path relative to appModulePath, ex './thing/my.module'
+ */
+export function addModule(sourceText, moduleName, modulePath) {
+    const source = jscodeshift.withParser('flow')(sourceText);
+
+    const ngModules = source
+        .find(jscodeshift.ClassDeclaration, path => path.decorators.some(decorator => decorator.callee.name === 'NgModule'));
+
+    if(ngModules.size() === 0) {
+        throw new NoModulesError();
+    }
+    if(ngModules.size() > 1) {
+        throw new TooManyModulesError();
+    }
+
+    const ngModuleClass = ngModules.get();
+    const ngModule = ngModuleClass.value.decorators.find(decorator => decorator.callee.name === 'NgModule');
+    const imports = ngModule.arguments[0].properties.find(prop => prop.key.name === 'imports');
+
+    if(!imports) {
+        console.info('No \'imports\' property? Strange..');
+        // TODO: create
+    }
+
+    // Push module to `imports` array
+    const MyModuleIdentifier = jscodeshift.identifier(moduleName);
+    imports.value.elements.push(MyModuleIdentifier);
+
+    const existingImports = source.find(ImportSpecifier);
+    if(existingImports.size() === 0) {
+        // TODO: Must be using some other module format
+    }
+
+    const MyModuleImport = jscodeshift.importDeclaration([jscodeshift.importSpecifier(jscodeshift.identifier(moduleName))], jscodeshift.literal(modulePath));
+
+    // Insert after last `import {...} from '...'` statement
+    jscodeshift(existingImports.at(-1).get().parent.insertAfter(MyModuleImport));
+
+    return source.toSource({quote: 'single'});
+}