chore: Scaffolded the update-classes step

src/migration/ember-app/index.ts

@@ -1,8 +1,9 @@
 import type { CodemodOptions } from '../../types/index.js';
 import { createOptions } from '../../utils/steps/create-options.js';
+import { updateClasses } from '../../utils/steps/update-classes.js';
 
 export function migrateEmberApp(codemodOptions: CodemodOptions): void {
   const options = createOptions(codemodOptions);
 
-  console.log(options);
+  updateClasses(['app/**/*.{js,ts}'], options);
 }

src/migration/ember-v1-addon/index.ts

@@ -1,8 +1,12 @@
 import type { CodemodOptions } from '../../types/index.js';
 import { createOptions } from '../../utils/steps/create-options.js';
+import { updateClasses } from '../../utils/steps/update-classes.js';
 
 export function migrateEmberV1Addon(codemodOptions: CodemodOptions): void {
   const options = createOptions(codemodOptions);
 
-  console.log(options);
+  updateClasses(
+    ['addon/**/*.{js,ts}', 'tests/dummy/app/**/*.{js,ts}'],
+    options,
+  );
 }

src/migration/ember-v2-addon/index.ts

@@ -1,8 +1,9 @@
 import type { CodemodOptions } from '../../types/index.js';
 import { createOptions } from '../../utils/steps/create-options.js';
+import { updateClasses } from '../../utils/steps/update-classes.js';
 
 export function migrateEmberV2Addon(codemodOptions: CodemodOptions): void {
   const options = createOptions(codemodOptions);
 
-  console.log(options);
+  updateClasses(['src/**/*.{js,ts}'], options);
 }

src/utils/steps/update-classes.ts

@@ -0,0 +1,38 @@
+import { readFileSync, writeFileSync } from 'node:fs';
+import { join } from 'node:path';
+
+import { findFiles } from '@codemod-utils/files';
+
+import { Options } from '../../types/index.js';
+import { findLocalName } from './update-classes/find-local-name.js';
+import { renameImports } from './update-classes/rename-imports.js';
+
+export function updateClasses(src: string[], options: Options): void {
+  const { projectRoot } = options;
+
+  const filePaths = findFiles(src, {
+    projectRoot,
+  });
+
+  filePaths.forEach((filePath) => {
+    const oldPath = join(projectRoot, filePath);
+    const oldFile = readFileSync(oldPath, 'utf8');
+
+    const isTypeScript = filePath.endsWith('.ts');
+
+    const localName = findLocalName(oldFile, {
+      isTypeScript,
+    });
+
+    if (localName === undefined) {
+      return;
+    }
+
+    const newFile = renameImports(oldFile, {
+      isTypeScript,
+      localName,
+    });
+
+    writeFileSync(oldPath, newFile, 'utf8');
+  });
+}

src/utils/steps/update-classes/find-local-name.ts

@@ -0,0 +1,17 @@
+import { AST } from '@codemod-utils/ast-javascript';
+
+type Data = {
+  isTypeScript: boolean;
+};
+
+export function findLocalName(file: string, data: Data): string | undefined {
+  let localName: string | undefined;
+
+  const traverse = AST.traverse(data.isTypeScript);
+
+  traverse(file, {
+    // ...
+  });
+
+  return localName;
+}

src/utils/steps/update-classes/rename-imports.ts

@@ -0,0 +1,16 @@
+import { AST } from '@codemod-utils/ast-javascript';
+
+type Data = {
+  isTypeScript: boolean;
+  localName: string;
+};
+
+export function renameImports(file: string, data: Data): string {
+  const traverse = AST.traverse(data.isTypeScript);
+
+  const ast = traverse(file, {
+    // ...
+  });
+
+  return AST.print(ast);
+}