fix: defined types for codemod

wheresrhys · Aug 31, 2024 · 0c3debf · 0c3debf
1 parent 4725251
commit 0c3debf
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 4 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/packages/codemods/package.json b/packages/codemods/package.json
@@ -30,6 +30,7 @@
 	},
 	"homepage": "http://www.wheresrhys.co.uk/fetch-mock",
 	"dependencies": {
+		"@types/jscodeshift": "^0.11.11",
 		"jscodeshift": "^17.0.0"
 	}
 }
diff --git a/packages/codemods/src/index.ts b/packages/codemods/src/index.ts
@@ -1,4 +1,7 @@
-export function codemod(source, j) {
+import type {
+  JSCodeshift, MemberExpression, Identifier, Transform, FileInfo, API
+} from 'jscodeshift';
+export function codemod(source: string, j: JSCodeshift) {
   const root = j(source);
   const fetchMockVariableName = root
     .find(j.CallExpression, {
@@ -30,16 +33,18 @@ export function codemod(source, j) {
       return paths;
     })
     .forEach((path) => {
-      const method = path.value.callee.property.name;
+      const callee = path.value.callee as MemberExpression;
+      const property = callee.property as Identifier;
+      const method = property.name;
       if (method === "mock") {
-        path.value.callee.property.name = "route";
+        property.name = "route";
       }
     });
 
   return root.toSource();
 }
 
 
-export default function transformer(file, api) {
+export default function transformer(file: FileInfo, api: API): string {
   return codemod(file.source, api.j);
 }