refactor: move component ast-utils to a separate file

Author: sis0k0

src/ast-utils.ts

@@ -22,22 +22,6 @@ class RemoveContent implements Node {
   }
 }
 
-export function addSymbolToComponentMetadata(
-  source: ts.SourceFile,
-  componentPath: string,
-  metadataField: string,
-  symbolName: string,
-): Change[] {
-  return addSymbolToDecoratorMetadata(
-    source,
-    componentPath,
-    metadataField,
-    symbolName,
-    'Component',
-    '@angular/core'
-  );
-}
-
 // Almost identical to addSymbolToNgModuleMetadata from @schematics/angular/utility/ast-utils
 // the method can be refactored so that it can be used with custom decorators
 export function addSymbolToDecoratorMetadata(
@@ -630,19 +614,6 @@ export function findImportPath(source: ts.Node, name) {
   return moduleSpecifier.text;
 }
 
-export const insertModuleId = (tree: Tree, component: string) => {
-  const componentSource = getSourceFile(tree, component);
-  const recorder = tree.beginUpdate(component);
-
-  const metadataChange = addSymbolToComponentMetadata(
-    componentSource, component, 'moduleId', 'module.id');
-
-  metadataChange.forEach((change: InsertChange) =>
-    recorder.insertRight(change.pos, change.toAdd)
-  );
-  tree.commitUpdate(recorder);
-
-};
 
 export const updateNodeText = (tree: Tree, node: ts.Node, newText: string) => {
   const recorder = tree.beginUpdate(node.getSourceFile().fileName);

src/generate/component/ast-utils.ts

@@ -0,0 +1,87 @@
+import * as ts from 'typescript';
+
+import { SchematicsException, Tree } from '@angular-devkit/schematics';
+import { classify } from '@angular-devkit/core/src/utils/strings';
+import { buildRelativePath } from '@schematics/angular/utility/find-module';
+import { InsertChange, Change } from '@schematics/angular/utility/change';
+import { addEntryComponentToModule, addExportToModule, addDeclarationToModule } from '@schematics/angular/utility/ast-utils';
+import { Schema as ComponentOptions } from './schema';
+import { getSourceFile } from '../../utils';
+import { addSymbolToDecoratorMetadata } from '../../ast-utils';
+
+export const insertModuleId = (tree: Tree, component: string) => {
+  const componentSource = getSourceFile(tree, component);
+  const recorder = tree.beginUpdate(component);
+
+  const metadataChange = addSymbolToComponentMetadata(
+    componentSource, component, 'moduleId', 'module.id');
+
+  metadataChange.forEach((change: InsertChange) =>
+    recorder.insertRight(change.pos, change.toAdd)
+  );
+  tree.commitUpdate(recorder);
+};
+
+function addSymbolToComponentMetadata(
+  source: ts.SourceFile,
+  componentPath: string,
+  metadataField: string,
+  symbolName: string,
+): Change[] {
+  return addSymbolToDecoratorMetadata(
+    source,
+    componentPath,
+    metadataField,
+    symbolName,
+    'Component',
+    '@angular/core'
+  );
+}
+
+export function addDeclarationToNgModule(tree: Tree, options: ComponentOptions, componentPath: string, modulePath: string) {
+  const source = readIntoSourceFile(tree, modulePath);
+  const relativePath = buildRelativePath(modulePath, componentPath);
+  const classifiedName = classify(`${options.name}Component`);
+  const declarationChanges = addDeclarationToModule(source, modulePath, classifiedName, relativePath);
+  const declarationRecorder = tree.beginUpdate(modulePath);
+  for (const change of declarationChanges) {
+    if (change instanceof InsertChange) {
+      declarationRecorder.insertLeft(change.pos, change.toAdd);
+    }
+  }
+  tree.commitUpdate(declarationRecorder);
+  if (options.export) {
+    // Need to refresh the AST because we overwrote the file in the host.
+    const source = readIntoSourceFile(tree, modulePath);
+    const exportRecorder = tree.beginUpdate(modulePath);
+    const exportChanges = addExportToModule(source, modulePath, classify(`${options.name}Component`), relativePath);
+    for (const change of exportChanges) {
+      if (change instanceof InsertChange) {
+        exportRecorder.insertLeft(change.pos, change.toAdd);
+      }
+    }
+    tree.commitUpdate(exportRecorder);
+  }
+  if (options.entryComponent) {
+    // Need to refresh the AST because we overwrote the file in the host.
+    const source = readIntoSourceFile(tree, modulePath);
+    const entryComponentRecorder = tree.beginUpdate(modulePath);
+    const entryComponentChanges = addEntryComponentToModule(source, modulePath, classify(`${options.name}Component`), relativePath);
+    for (const change of entryComponentChanges) {
+      if (change instanceof InsertChange) {
+        entryComponentRecorder.insertLeft(change.pos, change.toAdd);
+      }
+    }
+    tree.commitUpdate(entryComponentRecorder);
+  }
+}
+
+function readIntoSourceFile(host: Tree, modulePath: string): ts.SourceFile {
+  const text = host.read(modulePath);
+  if (text === null) {
+    throw new SchematicsException(`File ${modulePath} does not exist.`);
+  }
+  const sourceText = text.toString('utf-8');
+
+  return ts.createSourceFile(modulePath, sourceText, ts.ScriptTarget.Latest, true);
+}