Lab #8

thinktecture · Oct 7, 2024 · 4897280 · 4897280
1 parent b1caf67
commit 4897280
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tools/my-plugin/src/generators/my-generator/generator.ts b/tools/my-plugin/src/generators/my-generator/generator.ts
@@ -1,5 +1,8 @@
 import { applicationGenerator } from '@nx/angular/generators';
+import { addImportToComponent } from '@nx/angular/src/utils';
 import { generateFiles, joinPathFragments, readProjectConfiguration, Tree } from '@nx/devkit';
+import { insertImport } from '@nx/js';
+import { getSourceFile } from '../utils/get-source-file';
 import { MyGeneratorGeneratorSchema } from './schema';
 
 export async function myGeneratorGenerator(
@@ -17,6 +20,10 @@ export async function myGeneratorGenerator(
   const projectDir = readProjectConfiguration(tree, options.name).root;
   const destinationDir = joinPathFragments(projectDir, 'src', 'app');
   generateFiles(tree, sourceDir, destinationDir, {});
+
+  const appComponentPath = joinPathFragments(destinationDir, 'app.component.ts');
+  addImportToComponent(tree, getSourceFile(tree, appComponentPath), appComponentPath, 'MyLibraryComponent');
+  insertImport(tree, getSourceFile(tree, appComponentPath), appComponentPath, 'MyLibraryComponent', '@my-workspace/my-library');
 }
 
 export default myGeneratorGenerator;