@kamaalio/codemod-kit 0.0.38 → 0.0.40

package/dist/codemods/types.d.ts

@@ -10,14 +10,18 @@ export type RunCodemodOkResult = {
     root: string;
 };
 export type RunCodemodResult = Result<RunCodemodOkResult, Error>;
+type CodemodOptions = {
+    postTransform?: Record<string, unknown>;
+};
 export type Codemod = {
     name: string;
     languages: Set<NapiLang> | Array<NapiLang>;
-    transformer: (content: string, filename?: types.Optional<string>) => Promise<string>;
+    options?: CodemodOptions;
+    transformer: (content: string, filename?: types.Optional<string>, codemod?: types.Optional<Codemod>) => Promise<string>;
     postTransform?: (results: {
         root: string;
         results: Array<RunCodemodOkResult>;
-    }) => Promise<void>;
+    }, codemod: Codemod) => Promise<void>;
     targetFiltering?: (filepath: string, codemod: Codemod) => boolean;
 };
 export type CodemodRunnerCodemod<Tag = string, C extends Codemod = Codemod> = C & {
@@ -42,3 +46,4 @@ export type FindAndReplaceConfig = {
     rule: Rule<TypesMap>;
     transformer: ((node: SgNode<TypesMap, Kinds<TypesMap>>, rule: Rule<TypesMap>) => types.Optional<Edit | string> | Array<Edit | string>) | string;
 };
+export {};

package/dist/github/utils.d.ts

@@ -4,7 +4,7 @@ import type { Codemod, CodemodRunnerCodemod } from '../codemods/index.js';
 export declare function makePullRequestsForCodemodResults<Tag = string, C extends Codemod = Codemod>(codemods: Array<CodemodRunnerCodemod<Tag, C>>, codemodResults: Record<string, Array<Result<{
     hasChanges: boolean;
     content: string;
-}, Error>>>, repositories: Array<Repository<Tag>>): Promise<void>;
+}, Error>>>, repositoriesMappedByCodemodName: Record<string, Array<Repository<Tag>>>): Promise<void>;
 export declare function makePullRequestsForCodemodResult<Tag = string, C extends Codemod = Codemod>(codemod: CodemodRunnerCodemod<Tag, C>, codemodResult: Array<Result<{
     hasChanges: boolean;
     content: string;

package/dist/index.cjs

@@ -456,10 +456,12 @@ function dedupeRepositoriesToClone(repositories, cwd) {
     }, initialDedupeResult).result;
     return dedupedRepos;
 }
-async function makePullRequestsForCodemodResults(codemods, codemodResults, repositories) {
+async function makePullRequestsForCodemodResults(codemods, codemodResults, repositoriesMappedByCodemodName) {
     for (const [codemodName, codemodResult] of Object.entries(codemodResults)){
         const codemod = codemods.find((c)=>c.name === codemodName);
-        if (null == codemod) throw new Error('Invariant found codemod should be present');
+        kamaal_namespaceObject.asserts.invariant(null != codemod, 'Codemod should have been present');
+        const repositories = repositoriesMappedByCodemodName[codemodName];
+        kamaal_namespaceObject.asserts.invariant(null != repositories, 'Repositories should have been present');
         await makePullRequestsForCodemodResult(codemod, codemodResult, repositories);
     }
 }
@@ -482,13 +484,14 @@ async function makePullRequestsForCodemodResult(codemod, codemodResult, reposito
 async function makePullRequest(params) {
     return tryCatchAsync(()=>(0, external_execa_namespaceObject.$)({
             cwd: params.workingDirectory
-        })`gh pr create --title ${params.title} --fill`);
+        })`gh pr create --draft --title ${params.title} --fill`);
 }
 async function runCodemodsOnProjects(repositoriesToClone, codemods, options) {
     const clonedRepositories = await cloneRepositories(repositoriesToClone, options.workingDirectory);
     console.log(`\u{1F5A8}\u{FE0F} cloned ${clonedRepositories.length} ${1 === clonedRepositories.length ? 'repository' : 'repositories'}`);
-    const codemodResults = await runCodemodRunner(codemods, clonedRepositories, options.workingDirectory);
-    if (options.pushChanges) await makePullRequestsForCodemodResults(codemods, codemodResults, clonedRepositories);
+    const codemodRepositoriesMappedByCodemodName = await prepareRepositoriesForCodemods(clonedRepositories, codemods, options.workingDirectory);
+    const codemodResults = await runCodemodRunner(codemods, codemodRepositoriesMappedByCodemodName, options.workingDirectory);
+    if (options.pushChanges) await makePullRequestsForCodemodResults(codemods, codemodResults, codemodRepositoriesMappedByCodemodName);
 }
 function utils_codemodTargetFiltering(repositories, failedRepositoryAddressesMappedByCodemodNames) {
     return (filepath, codemod)=>{
@@ -514,10 +517,9 @@ async function codemodPostTransform(transformedContent) {
     return transformedContent;
 }
 async function runCodemodRunner(codemods, repositories, workingDirectory) {
-    const codemodRepositoriesMappedByCodemodName = await prepareRepositoriesForCodemods(repositories, codemods, workingDirectory);
     const results = Object.fromEntries(await Promise.all(codemods.map(async (codemod)=>{
         const start = performance.now();
-        const codemodRepositories = codemodRepositoriesMappedByCodemodName[codemod.name];
+        const codemodRepositories = repositories[codemod.name];
         kamaal_namespaceObject.asserts.invariant(null != codemodRepositories, 'Codemod repositories should be present');
         const codemodWorkingDirectory = external_node_path_default().resolve(workingDirectory, codemod.name.replace(/\//g, '-'));
         const failedRepositoryAddressesMappedByCodemodNames = {};
@@ -525,7 +527,7 @@ async function runCodemodRunner(codemods, repositories, workingDirectory) {
             rootPaths: codemodRepositories.map((repository)=>repository.path),
             hooks: {
                 preCodemodRun: async (codemod)=>{
-                    failedRepositoryAddressesMappedByCodemodNames[codemod.name] = await codemodPreCodemodRun(repositories, codemod);
+                    failedRepositoryAddressesMappedByCodemodNames[codemod.name] = await codemodPreCodemodRun(codemodRepositories, codemod);
                 },
                 targetFiltering: utils_codemodTargetFiltering(groupByFlat(codemodRepositories, 'name'), failedRepositoryAddressesMappedByCodemodNames),
                 postTransform: codemodPostTransform
@@ -556,19 +558,7 @@ async function prepareRepositoriesForCodemods(repositories, codemods, workingDir
     }));
     console.log(`\u{1F4CB} prepared the following repos for codemods:
 \xb7 ${updatedRepositories.map((repo)=>repo.address).join("\n\xb7 ")}`);
-    return Object.fromEntries(await Promise.all(codemods.map(async (codemod)=>{
-        const codemodWorkingDirectory = external_node_path_default().resolve(workingDirectory, codemod.name.replace(/\//g, '-'));
-        await (0, external_execa_namespaceObject.$)`mkdir -p ${codemodWorkingDirectory}`;
-        const codeRepositories = await Promise.all(updatedRepositories.map((repo)=>{
-            const newPath = external_node_path_default().join(codemodWorkingDirectory, repo.name);
-            console.log(`\xa9\u{FE0F} copying ${repo.name} -> ${newPath}`);
-            return repo.copy(newPath);
-        }));
-        return [
-            codemod.name,
-            codeRepositories
-        ];
-    })));
+    return Object.fromEntries(await Promise.all(codemods.map(copyRepo(workingDirectory, updatedRepositories))));
 }
 async function runCodemods(codemods, transformationPath, options) {
     const results = {};
@@ -631,7 +621,7 @@ async function runCodemod(codemod, transformationPath, options) {
             root,
             results: successesGroupedByRoot[root] ?? []
         }));
-    await Promise.all(rootPathsWithResults.map((r)=>(codemod.postTransform ?? (async ()=>{}))(r)));
+    await Promise.all(rootPathsWithResults.map((r)=>(codemod.postTransform ?? (async ()=>{}))(r, codemod)));
     return results;
 }
 function traverseUp(node, until) {
@@ -744,6 +734,21 @@ function findAndReplace(content, rule, transformer) {
     const edits = findAndReplaceEdits(content, rule, transformer);
     return root.commitEdits(edits);
 }
+function copyRepo(workingDirectory, repositories) {
+    return async (codemod)=>{
+        const codemodWorkingDirectory = external_node_path_default().resolve(workingDirectory, codemod.name.replace(/\//g, '-'));
+        await (0, external_execa_namespaceObject.$)`mkdir -p ${codemodWorkingDirectory}`;
+        const codeRepositories = await Promise.all(repositories.map((repo)=>{
+            const newPath = external_node_path_default().join(codemodWorkingDirectory, repo.name);
+            console.log(`\xa9\u{FE0F} copying ${repo.name} -> ${newPath}`);
+            return repo.copy(newPath);
+        }));
+        return [
+            codemod.name,
+            codeRepositories
+        ];
+    };
+}
 async function commitEditModifications(edits, modifications) {
     if (0 === edits.length) return modifications;
     const root = modifications.ast.root();

package/dist/index.js

@@ -407,10 +407,12 @@ function dedupeRepositoriesToClone(repositories, cwd) {
     }, initialDedupeResult).result;
     return dedupedRepos;
 }
-async function makePullRequestsForCodemodResults(codemods, codemodResults, repositories) {
+async function makePullRequestsForCodemodResults(codemods, codemodResults, repositoriesMappedByCodemodName) {
     for (const [codemodName, codemodResult] of Object.entries(codemodResults)){
         const codemod = codemods.find((c)=>c.name === codemodName);
-        if (null == codemod) throw new Error('Invariant found codemod should be present');
+        asserts.invariant(null != codemod, 'Codemod should have been present');
+        const repositories = repositoriesMappedByCodemodName[codemodName];
+        asserts.invariant(null != repositories, 'Repositories should have been present');
         await makePullRequestsForCodemodResult(codemod, codemodResult, repositories);
     }
 }
@@ -433,13 +435,14 @@ async function makePullRequestsForCodemodResult(codemod, codemodResult, reposito
 async function makePullRequest(params) {
     return tryCatchAsync(()=>$({
             cwd: params.workingDirectory
-        })`gh pr create --title ${params.title} --fill`);
+        })`gh pr create --draft --title ${params.title} --fill`);
 }
 async function runCodemodsOnProjects(repositoriesToClone, codemods, options) {
     const clonedRepositories = await cloneRepositories(repositoriesToClone, options.workingDirectory);
     console.log(`\u{1F5A8}\u{FE0F} cloned ${clonedRepositories.length} ${1 === clonedRepositories.length ? 'repository' : 'repositories'}`);
-    const codemodResults = await runCodemodRunner(codemods, clonedRepositories, options.workingDirectory);
-    if (options.pushChanges) await makePullRequestsForCodemodResults(codemods, codemodResults, clonedRepositories);
+    const codemodRepositoriesMappedByCodemodName = await prepareRepositoriesForCodemods(clonedRepositories, codemods, options.workingDirectory);
+    const codemodResults = await runCodemodRunner(codemods, codemodRepositoriesMappedByCodemodName, options.workingDirectory);
+    if (options.pushChanges) await makePullRequestsForCodemodResults(codemods, codemodResults, codemodRepositoriesMappedByCodemodName);
 }
 function utils_codemodTargetFiltering(repositories, failedRepositoryAddressesMappedByCodemodNames) {
     return (filepath, codemod)=>{
@@ -465,10 +468,9 @@ async function codemodPostTransform(transformedContent) {
     return transformedContent;
 }
 async function runCodemodRunner(codemods, repositories, workingDirectory) {
-    const codemodRepositoriesMappedByCodemodName = await prepareRepositoriesForCodemods(repositories, codemods, workingDirectory);
     const results = Object.fromEntries(await Promise.all(codemods.map(async (codemod)=>{
         const start = performance.now();
-        const codemodRepositories = codemodRepositoriesMappedByCodemodName[codemod.name];
+        const codemodRepositories = repositories[codemod.name];
         asserts.invariant(null != codemodRepositories, 'Codemod repositories should be present');
         const codemodWorkingDirectory = node_path.resolve(workingDirectory, codemod.name.replace(/\//g, '-'));
         const failedRepositoryAddressesMappedByCodemodNames = {};
@@ -476,7 +478,7 @@ async function runCodemodRunner(codemods, repositories, workingDirectory) {
             rootPaths: codemodRepositories.map((repository)=>repository.path),
             hooks: {
                 preCodemodRun: async (codemod)=>{
-                    failedRepositoryAddressesMappedByCodemodNames[codemod.name] = await codemodPreCodemodRun(repositories, codemod);
+                    failedRepositoryAddressesMappedByCodemodNames[codemod.name] = await codemodPreCodemodRun(codemodRepositories, codemod);
                 },
                 targetFiltering: utils_codemodTargetFiltering(groupByFlat(codemodRepositories, 'name'), failedRepositoryAddressesMappedByCodemodNames),
                 postTransform: codemodPostTransform
@@ -507,19 +509,7 @@ async function prepareRepositoriesForCodemods(repositories, codemods, workingDir
     }));
     console.log(`\u{1F4CB} prepared the following repos for codemods:
 \xb7 ${updatedRepositories.map((repo)=>repo.address).join("\n\xb7 ")}`);
-    return Object.fromEntries(await Promise.all(codemods.map(async (codemod)=>{
-        const codemodWorkingDirectory = node_path.resolve(workingDirectory, codemod.name.replace(/\//g, '-'));
-        await $`mkdir -p ${codemodWorkingDirectory}`;
-        const codeRepositories = await Promise.all(updatedRepositories.map((repo)=>{
-            const newPath = node_path.join(codemodWorkingDirectory, repo.name);
-            console.log(`\xa9\u{FE0F} copying ${repo.name} -> ${newPath}`);
-            return repo.copy(newPath);
-        }));
-        return [
-            codemod.name,
-            codeRepositories
-        ];
-    })));
+    return Object.fromEntries(await Promise.all(codemods.map(copyRepo(workingDirectory, updatedRepositories))));
 }
 async function runCodemods(codemods, transformationPath, options) {
     const results = {};
@@ -582,7 +572,7 @@ async function runCodemod(codemod, transformationPath, options) {
             root,
             results: successesGroupedByRoot[root] ?? []
         }));
-    await Promise.all(rootPathsWithResults.map((r)=>(codemod.postTransform ?? (async ()=>{}))(r)));
+    await Promise.all(rootPathsWithResults.map((r)=>(codemod.postTransform ?? (async ()=>{}))(r, codemod)));
     return results;
 }
 function traverseUp(node, until) {
@@ -695,6 +685,21 @@ function findAndReplace(content, rule, transformer) {
     const edits = findAndReplaceEdits(content, rule, transformer);
     return root.commitEdits(edits);
 }
+function copyRepo(workingDirectory, repositories) {
+    return async (codemod)=>{
+        const codemodWorkingDirectory = node_path.resolve(workingDirectory, codemod.name.replace(/\//g, '-'));
+        await $`mkdir -p ${codemodWorkingDirectory}`;
+        const codeRepositories = await Promise.all(repositories.map((repo)=>{
+            const newPath = node_path.join(codemodWorkingDirectory, repo.name);
+            console.log(`\xa9\u{FE0F} copying ${repo.name} -> ${newPath}`);
+            return repo.copy(newPath);
+        }));
+        return [
+            codemod.name,
+            codeRepositories
+        ];
+    };
+}
 async function commitEditModifications(edits, modifications) {
     if (0 === edits.length) return modifications;
     const root = modifications.ast.root();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kamaalio/codemod-kit",
-  "version": "0.0.38",
+  "version": "0.0.40",
   "type": "module",
   "author": "Kamaal Farah",
   "license": "MIT",