chore(generators): allow multiple component types in getNgModules

imhoffd · imhoffd · commit 4e6e372d2e3a · 2017-03-14T15:33:56.000-05:00
diff --git a/src/generators/util.ts b/src/generators/util.ts
@@ -86,11 +86,10 @@ function createDirAndWriteFile(filePath: string, fileContent: string) {
   });
 }
 
-export function getNgModules(context: BuildContext, type: string): Promise<GlobResult[]> {
+export function getNgModules(context: BuildContext, types: string[]): Promise<GlobResult[]> {
   const ngModuleSuffix = getStringPropertyValue(Constants.ENV_NG_MODULE_FILE_NAME_SUFFIX);
-  const genDir = getDirToWriteToByType(context, type);
-
-  return globAll([join(genDir, '**', `*${ngModuleSuffix}`)]);
+  const patterns = types.map((type) => join(getDirToWriteToByType(context, type), '**', `*${ngModuleSuffix}`));
+  return globAll(patterns);
 }
 
 export function getDirToWriteToByType(context: BuildContext, type: string) {

Original file line number	Diff line number	Diff line change
`@@ -86,11 +86,10 @@ function createDirAndWriteFile(filePath: string, fileContent: string) {`
`86`	`86`	`});`
`87`	`87`	`}`
`88`	`88`
`89`		`-export function getNgModules(context: BuildContext, type: string): Promise<GlobResult[]> {`
	`89`	`+export function getNgModules(context: BuildContext, types: string[]): Promise<GlobResult[]> {`
`90`	`90`	`const ngModuleSuffix = getStringPropertyValue(Constants.ENV_NG_MODULE_FILE_NAME_SUFFIX);`
`91`		`- const genDir = getDirToWriteToByType(context, type);`
`92`		`-`
`93`		- return globAll([join(genDir, '*', `${ngModuleSuffix}`)]);
	`91`	+ const patterns = types.map((type) => join(getDirToWriteToByType(context, type), '*', `${ngModuleSuffix}`));
	`92`	`+ return globAll(patterns);`
`94`	`93`	`}`
`95`	`94`
`96`	`95`	`export function getDirToWriteToByType(context: BuildContext, type: string) {`