@empiricalrun/test-gen 0.65.0 → 0.66.0

package/dist/tools/commit-and-create-pr.js

@@ -1,25 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createPullRequestTool = void 0;
-exports.parseGitHubUrl = parseGitHubUrl;
 const child_process_1 = require("child_process");
 const zod_1 = require("zod");
 const git_1 = require("../utils/git");
 const utils_1 = require("./utils");
-function parseGitHubUrl(url) {
-    const githubIndex = url.indexOf("github.com");
-    if (githubIndex === -1) {
-        throw new Error("Invalid GitHub repository URL");
-    }
-    const [owner, repo] = url
-        .substring(githubIndex + "github.com/".length)
-        .replace(".git", "")
-        .split("/");
-    if (!owner || !repo) {
-        throw new Error("Invalid GitHub repository URL format - missing owner or repo");
-    }
-    return { owner, repo };
-}
 const createPullRequestSchema = zod_1.z.object({
     pullRequestTitle: zod_1.z
         .string()
@@ -61,25 +46,22 @@ Don't ask the user for this information, just come up with it yourself.
             })
                 .toString()
                 .trim();
-            const { owner, repo } = parseGitHubUrl(repoUrl);
-            const existingPRs = (await (0, utils_1.callGitHubProxy)({
-                method: "GET",
-                url: `https://api.github.com/repos/${owner}/${repo}/pulls`,
-                body: {
-                    head: `${owner}:${branchName}`,
-                    state: "open",
-                },
-                apiKey,
-            }));
+            const { owner, repo } = (0, utils_1.parseGitHubUrl)(repoUrl);
             (0, child_process_1.execSync)(`git push origin ${branchName} --set-upstream`, {
                 cwd: repoPath,
             });
-            const existingPR = existingPRs?.find((pr) => pr.head.ref === branchName);
+            const existingPR = await (0, utils_1.findExistingPR)({
+                owner,
+                repo,
+                branchName,
+                apiKey,
+            });
             if (existingPR) {
-                await (0, utils_1.callGitHubProxy)({
-                    method: "PATCH",
-                    url: `https://api.github.com/repos/${owner}/${repo}/pulls/${existingPR.number}`,
-                    body: { body: descriptionWithTimestamp(pullRequestDescription) },
+                await (0, utils_1.updatePullRequest)({
+                    owner,
+                    repo,
+                    prNumber: existingPR.number,
+                    body: descriptionWithTimestamp(pullRequestDescription),
                     apiKey,
                 });
                 return {
@@ -87,17 +69,15 @@ Don't ask the user for this information, just come up with it yourself.
                     result: `Committed and pushed changes to existing PR: ${existingPR.html_url}`,
                 };
             }
-            const pr = (await (0, utils_1.callGitHubProxy)({
-                method: "POST",
-                url: `https://api.github.com/repos/${owner}/${repo}/pulls`,
-                body: {
-                    title: pullRequestTitle,
-                    head: branchName,
-                    base: "main",
-                    body: descriptionWithTimestamp(pullRequestDescription),
-                },
+            const pr = await (0, utils_1.createPullRequest)({
+                owner,
+                repo,
+                title: pullRequestTitle,
+                head: branchName,
+                base: "main",
+                body: descriptionWithTimestamp(pullRequestDescription),
                 apiKey,
-            }));
+            });
             return {
                 isError: false,
                 result: `Committed and pushed changes to new PR: ${pr.html_url}`,

package/dist/tools/str_replace_editor.d.ts

@@ -1,4 +1,4 @@
-import type { Tool, ToolResult } from "@empiricalrun/shared-types";
+import type { CollectArtifacts, Tool, ToolResult } from "@empiricalrun/shared-types";
 interface StrReplaceInput {
     command: string;
     path: string;
@@ -12,9 +12,10 @@ interface StrReplaceInput {
  * Our implementation of Claude's built-in text editor tool
  * https://docs.anthropic.com/en/docs/build-with-claude/tool-use/text-editor-tool
  */
-export declare function strReplaceEditorExecutor({ input, repoPath, }: {
+export declare function strReplaceEditorExecutor({ input, repoPath, collectArtifacts, }: {
     input: StrReplaceInput;
     repoPath: string;
+    collectArtifacts?: CollectArtifacts;
 }): Promise<ToolResult>;
 export declare const textEditorTools: Tool[];
 export {};

package/dist/tools/str_replace_editor.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"str_replace_editor.d.ts","sourceRoot":"","sources":["../../src/tools/str_replace_editor.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,IAAI,EAAe,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAOhF,UAAU,eAAe;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AA6UD;;;GAGG;AACH,wBAAsB,wBAAwB,CAAC,EAC7C,KAAK,EACL,QAAQ,GACT,EAAE;IACD,KAAK,EAAE,eAAe,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC;CAClB,GAAG,OAAO,CAAC,UAAU,CAAC,CA2CtB;AAkID,eAAO,MAAM,eAAe,EAAE,IAAI,EAKjC,CAAC"}
+{"version":3,"file":"str_replace_editor.d.ts","sourceRoot":"","sources":["../../src/tools/str_replace_editor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAEV,gBAAgB,EAChB,IAAI,EAEJ,UAAU,EACX,MAAM,4BAA4B,CAAC;AAsCpC,UAAU,eAAe;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAiWD;;;GAGG;AACH,wBAAsB,wBAAwB,CAAC,EAC7C,KAAK,EACL,QAAQ,EACR,gBAAgB,GACjB,EAAE;IACD,KAAK,EAAE,eAAe,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC;IACjB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;CACrC,GAAG,OAAO,CAAC,UAAU,CAAC,CA8CtB;AAqID,eAAO,MAAM,eAAe,EAAE,IAAI,EAKjC,CAAC"}

package/dist/tools/str_replace_editor.js

@@ -9,6 +9,31 @@ const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 const zod_1 = require("zod");
 const web_1 = require("../bin/utils/platform/web");
+const git_1 = require("../utils/git");
+/**
+ * Helper function to collect git patch artifacts for file modifications
+ */
+async function collectGitPatchArtifact(filePath, repoDir, operation, collectArtifacts) {
+    if (!collectArtifacts)
+        return;
+    try {
+        const gitPatch = operation === "create"
+            ? (0, git_1.getGitDiffForNewFile)(filePath, repoDir)
+            : (0, git_1.getGitDiff)(filePath, repoDir);
+        if (gitPatch.trim()) {
+            const patchArtifact = {
+                name: `${path_1.default.basename(filePath, path_1.default.extname(filePath))}_${operation}.patch`,
+                contentType: "text/plain",
+                data: Buffer.from(gitPatch, "utf-8"),
+            };
+            collectArtifacts([patchArtifact]);
+        }
+    }
+    catch (error) {
+        // Ignore git diff errors, don't fail the operation
+        console.warn(`Failed to generate git patch for ${filePath}:`, error);
+    }
+}
 /**
  * While running str_replace command, we've seen LLM can struggle to send unique old_str.
  * This function tries to find unique contexts for each occurrence of old_str, so that the error
@@ -115,7 +140,7 @@ async function fileViewExecutor({ input, filePath, absoluteFilePath, }) {
         isError: false,
     };
 }
-async function fileCreateExecutor({ input, filePath, absoluteFilePath, repoDir, }) {
+async function fileCreateExecutor({ input, filePath, absoluteFilePath, repoDir, collectArtifacts, }) {
     if (input.file_text === undefined || input.file_text === null) {
         return {
             result: "Error: file_text is required for create command",
@@ -149,6 +174,8 @@ the required directories recursively for the new file.`,
         fs_1.default.mkdirSync(parentDir, { recursive: true });
     }
     fs_1.default.writeFileSync(absoluteFilePath, input.file_text);
+    // Collect git patch artifact
+    await collectGitPatchArtifact(filePath, repoDir, "create", collectArtifacts);
     let createTypescriptResult = await (0, web_1.runTypescriptCompiler)(repoDir);
     if (!createTypescriptResult.success) {
         return {
@@ -163,7 +190,7 @@ the required directories recursively for the new file.`,
         };
     }
 }
-async function fileStrReplaceExecutor({ input, filePath, absoluteFilePath, repoDir, }) {
+async function fileStrReplaceExecutor({ input, filePath, absoluteFilePath, repoDir, collectArtifacts, }) {
     if (!fs_1.default.existsSync(absoluteFilePath)) {
         return {
             result: `Error: File ${filePath} not found. Please provide relative file path to the Repository.`,
@@ -216,6 +243,8 @@ async function fileStrReplaceExecutor({ input, filePath, absoluteFilePath, repoD
         }
         const newContent = normalizedContent.replace(normalizedOldStr, input.new_str);
         fs_1.default.writeFileSync(absoluteFilePath, newContent);
+        // Collect git patch artifact
+        await collectGitPatchArtifact(filePath, repoDir, "replace", collectArtifacts);
         let strReplaceTypescriptResult = await (0, web_1.runTypescriptCompiler)(repoDir);
         if (!strReplaceTypescriptResult.success) {
             return {
@@ -231,7 +260,7 @@ async function fileStrReplaceExecutor({ input, filePath, absoluteFilePath, repoD
         }
     }
 }
-async function fileInsertExecutor({ input, filePath, absoluteFilePath, repoDir, }) {
+async function fileInsertExecutor({ input, filePath, absoluteFilePath, repoDir, collectArtifacts, }) {
     if (!fs_1.default.existsSync(absoluteFilePath)) {
         return {
             result: `Error: File ${filePath} not found. Please provide relative file path to the Repository.`,
@@ -260,6 +289,8 @@ async function fileInsertExecutor({ input, filePath, absoluteFilePath, repoDir,
     }
     lines.splice(input.insert_line - 1, 0, input.new_str);
     fs_1.default.writeFileSync(absoluteFilePath, lines.join("\n"));
+    // Collect git patch artifact
+    await collectGitPatchArtifact(filePath, repoDir, "insert", collectArtifacts);
     let insertTypescriptResult = await (0, web_1.runTypescriptCompiler)(repoDir);
     if (!insertTypescriptResult.success) {
         return {
@@ -278,7 +309,7 @@ async function fileInsertExecutor({ input, filePath, absoluteFilePath, repoDir,
  * Our implementation of Claude's built-in text editor tool
  * https://docs.anthropic.com/en/docs/build-with-claude/tool-use/text-editor-tool
  */
-async function strReplaceEditorExecutor({ input, repoPath, }) {
+async function strReplaceEditorExecutor({ input, repoPath, collectArtifacts, }) {
     const repoDir = repoPath;
     const { path: filePath } = input;
     const absoluteFilePath = path_1.default.join(repoDir, filePath);
@@ -292,6 +323,7 @@ async function strReplaceEditorExecutor({ input, repoPath, }) {
                     filePath,
                     absoluteFilePath,
                     repoDir,
+                    collectArtifacts,
                 });
             case "str_replace":
                 return fileStrReplaceExecutor({
@@ -299,6 +331,7 @@ async function strReplaceEditorExecutor({ input, repoPath, }) {
                     filePath,
                     absoluteFilePath,
                     repoDir,
+                    collectArtifacts,
                 });
             case "insert":
                 return fileInsertExecutor({
@@ -306,6 +339,7 @@ async function strReplaceEditorExecutor({ input, repoPath, }) {
                     filePath,
                     absoluteFilePath,
                     repoDir,
+                    collectArtifacts,
                 });
             case "undo_edit":
                 return {
@@ -368,7 +402,7 @@ directories "tests", "tests/example", and "tests/example/foo.spec.ts".`,
     needsBrowser: false,
     execute: async (params) => {
         const { input } = params;
-        const { repoPath } = params;
+        const { repoPath, collectArtifacts } = params;
         return strReplaceEditorExecutor({
             input: {
                 command: "create",
@@ -376,6 +410,7 @@ directories "tests", "tests/example", and "tests/example/foo.spec.ts".`,
                 file_text: input.file_text,
             },
             repoPath,
+            collectArtifacts,
         });
     },
 };
@@ -393,7 +428,7 @@ in the file. If old_str is not unique, the tool will return an error.`,
     needsBrowser: false,
     execute: async (params) => {
         const { input } = params;
-        const { repoPath } = params;
+        const { repoPath, collectArtifacts } = params;
         return strReplaceEditorExecutor({
             input: {
                 command: "str_replace",
@@ -402,6 +437,7 @@ in the file. If old_str is not unique, the tool will return an error.`,
                 new_str: input.new_str,
             },
             repoPath,
+            collectArtifacts,
         });
     },
 };
@@ -424,7 +460,7 @@ To insert a string at the end of the file, you should use insert_line = (total l
     needsBrowser: false,
     execute: async (params) => {
         const { input } = params;
-        const { repoPath } = params;
+        const { repoPath, collectArtifacts } = params;
         return strReplaceEditorExecutor({
             input: {
                 command: "insert",
@@ -433,6 +469,7 @@ To insert a string at the end of the file, you should use insert_line = (total l
                 new_str: input.new_str,
             },
             repoPath,
+            collectArtifacts,
         });
     },
 };

package/dist/tools/test-gen-browser.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"test-gen-browser.d.ts","sourceRoot":"","sources":["../../src/tools/test-gen-browser.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,IAAI,EAGL,MAAM,4BAA4B,CAAC;AAoGpC,eAAO,MAAM,4BAA4B,EAAE,IA2L1C,CAAC"}
+{"version":3,"file":"test-gen-browser.d.ts","sourceRoot":"","sources":["../../src/tools/test-gen-browser.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,IAAI,EAGL,MAAM,4BAA4B,CAAC;AAoGpC,eAAO,MAAM,4BAA4B,EAAE,IAoM1C,CAAC"}

package/dist/tools/test-gen-browser.js

@@ -160,6 +160,14 @@ exports.generateTestWithBrowserAgent = {
                 BUILD_URL: input.buildUrl,
             };
         }
+        if (featureFlags.includes("headed_codegen_in_browser_agent")) {
+            // Running browser agent in headed implies using page.pause method for codegen
+            // which improves codegen quality when script injection fails (iframes, etc.)
+            envOverrides = {
+                ...envOverrides,
+                RUN_PLAYWRIGHT_HEADED: "true",
+            };
+        }
         const testGenToken = (0, scenarios_1.buildTokenFromOptions)({ name: testName, file: filePath, prompt: input.changeToMake }, { useComputerUseAgent: true });
         console.log("[generateTestWithBrowserAgent] Validations passed, starting agent");
         const agentResult = await (0, run_1.runBrowsingAgent)({

package/dist/tools/upgrade-packages/index.d.ts

@@ -0,0 +1,3 @@
+import type { Tool } from "@empiricalrun/shared-types";
+export declare const upgradePackagesTool: Tool;
+//# sourceMappingURL=index.d.ts.map

package/dist/tools/upgrade-packages/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/upgrade-packages/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,4BAA4B,CAAC;AA+BvD,eAAO,MAAM,mBAAmB,EAAE,IA+GjC,CAAC"}

package/dist/tools/upgrade-packages/index.js

@@ -0,0 +1,124 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.upgradePackagesTool = void 0;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const zod_1 = require("zod");
+const git_1 = require("../../utils/git");
+const utils_1 = require("../utils");
+const utils_2 = require("./utils");
+const pkgs = [
+    "@empiricalrun/playwright-utils",
+    "@empiricalrun/eslint-config",
+    "@empiricalrun/typescript-config",
+];
+const packageSpecSchema = zod_1.z.union([
+    zod_1.z.string(),
+    zod_1.z.object({
+        name: zod_1.z.string(),
+        version: zod_1.z.string().optional(),
+    }),
+]);
+const upgradePackagesSchema = zod_1.z.object({
+    packages: zod_1.z.array(packageSpecSchema),
+});
+exports.upgradePackagesTool = {
+    schema: {
+        name: "upgradePackages",
+        description: `Automatically upgrades specific packages. You can optionally specify a version for each package. If a version is not provided, the package will be upgraded to the latest version. The tool handles the entire workflow: updating package files, committing changes, creating/updating PRs, and managing merges. For patch updates (e.g., 1.0.1 → 1.0.2), it automatically merges the PR and cleans up. For minor/major updates, it leaves the PR open for review. Returns a success message with the PR URL and merge status.`,
+        parameters: upgradePackagesSchema,
+    },
+    needsBrowser: false,
+    execute: async ({ input, repoPath, apiKey, }) => {
+        const { packages } = input;
+        let packagesToUpdate = packages.length ? packages : pkgs;
+        const packageSpecs = packagesToUpdate.map((p) => typeof p === "string" ? { name: p, version: undefined } : p);
+        try {
+            const repoName = path_1.default.basename(repoPath);
+            const branchName = await (0, git_1.getCurrentBranchName)(repoPath);
+            const packageJsonPath = path_1.default.join(repoPath, "package.json");
+            const packageJson = JSON.parse(fs_1.default.readFileSync(packageJsonPath, "utf-8"));
+            for (let { name: pkgName, version } of packageSpecs) {
+                const isDevDep = !!packageJson.devDependencies?.[pkgName];
+                try {
+                    await (0, utils_2.upgradeAndStagePackage)({
+                        repoPath,
+                        pkgName,
+                        version,
+                        isDevDep,
+                    });
+                }
+                catch (err) {
+                    console.error(`Failed to upgrade ${pkgName}:`, err);
+                }
+            }
+            await (0, git_1.commitFilesAndPushBranch)({
+                commitMessage: "[create-pull-request] automated change [skip ci]",
+                branchName,
+                files: ["package.json", "package-lock.json"],
+                repoPath,
+            });
+            await new Promise((resolve) => setTimeout(resolve, 5_000));
+            let shouldMerge = false;
+            let prNumber = null;
+            let merged = false;
+            try {
+                const { pr } = await (0, utils_1.findOrCreatePullRequest)({
+                    owner: "empirical-run",
+                    repo: repoName,
+                    apiKey,
+                    branchName,
+                    title: "Upgrade packages",
+                    body: "Upgrade packages to specified versions",
+                    labels: ["automated"],
+                });
+                prNumber = pr.number;
+                if (!pr.number) {
+                    throw new Error("Failed to create PR");
+                }
+                shouldMerge = await (0, utils_2.shouldMergePR)({ repoName, prNumber, apiKey });
+                if (shouldMerge) {
+                    console.log("All changes are patch updates, proceeding with merge");
+                    merged = await (0, git_1.mergePullRequest)({
+                        repoName,
+                        prNumber,
+                        apiKey,
+                    });
+                }
+                else {
+                    console.log(`PR #${prNumber} created but not merged - contains non-patch updates`);
+                }
+            }
+            catch (error) {
+                console.error("Failed to handle PR operations:", error);
+                const prLink = prNumber
+                    ? ` PR: https://github.com/empirical-run/${repoName}/pull/${prNumber}`
+                    : "";
+                return {
+                    result: `Failed to handle PR operations: ${error}.${prLink}`,
+                    isError: true,
+                };
+            }
+            const prLink = `https://github.com/empirical-run/${repoName}/pull/${prNumber}`;
+            const status = merged
+                ? "merged successfully"
+                : shouldMerge
+                    ? "created but merge failed"
+                    : "created but not merged (non-patch updates)";
+            return {
+                result: `Success: PR ${status}. Link: ${prLink}`,
+                isError: false,
+            };
+        }
+        catch (error) {
+            console.error("Failed to upgrade packages:", error);
+            return {
+                result: `Failed to upgrade packages: ${error.message}`,
+                isError: true,
+            };
+        }
+    },
+};

package/dist/tools/upgrade-packages/utils.d.ts

@@ -0,0 +1,13 @@
+export declare function getLatestVersion(packageName: string): Promise<any>;
+export declare function shouldMergePR({ repoName, prNumber, apiKey, }: {
+    repoName: string;
+    prNumber: number;
+    apiKey: string;
+}): Promise<boolean>;
+export declare function upgradeAndStagePackage({ repoPath, pkgName, version, isDevDep, }: {
+    repoPath: string;
+    pkgName: string;
+    version?: string;
+    isDevDep: boolean;
+}): Promise<void>;
+//# sourceMappingURL=utils.d.ts.map

package/dist/tools/upgrade-packages/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/tools/upgrade-packages/utils.ts"],"names":[],"mappings":"AAYA,wBAAsB,gBAAgB,CAAC,WAAW,EAAE,MAAM,gBAczD;AAmED,wBAAsB,aAAa,CAAC,EAClC,QAAQ,EACR,QAAQ,EACR,MAAM,GACP,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;CAChB,oBA6BA;AAED,wBAAsB,sBAAsB,CAAC,EAC3C,QAAQ,EACR,OAAO,EACP,OAAO,EACP,QAAQ,GACT,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,OAAO,CAAC;CACnB,iBAUA"}

package/dist/tools/upgrade-packages/utils.js

@@ -0,0 +1,106 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getLatestVersion = getLatestVersion;
+exports.shouldMergePR = shouldMergePR;
+exports.upgradeAndStagePackage = upgradeAndStagePackage;
+const child_process_1 = require("child_process");
+const utils_1 = require("../utils");
+function isPatchUpdate(currentVersion, newVersion) {
+    const current = currentVersion.replace(/^[\^~]/, "").split(".");
+    const next = newVersion.replace(/^[\^~]/, "").split(".");
+    return (current[0] === next[0] && current[1] === next[1] && current[2] !== next[2]);
+}
+async function getLatestVersion(packageName) {
+    const url = `https://registry.npmjs.org/${packageName}/latest`;
+    try {
+        const response = await fetch(url);
+        if (!response.ok) {
+            throw new Error(`Failed to fetch data for package: ${packageName}`);
+        }
+        const packageInfo = await response.json();
+        console.log(`Latest version of ${packageName}: ${packageInfo.version}`);
+        return packageInfo.version;
+    }
+    catch (error) {
+        console.error(`Error fetching package info: ${error.message}`);
+        throw error;
+    }
+}
+async function getPackageJsonChanges({ repoName, prNumber, apiKey, }) {
+    const url = `${utils_1.GITHUB_API_BASE}/${repoName}/pulls/${prNumber}/files`;
+    try {
+        const files = await (0, utils_1.callGitHubProxy)({
+            method: "GET",
+            url: url,
+            apiKey,
+        });
+        if (!files) {
+            throw new Error("Failed to fetch PR files via proxy");
+        }
+        const packageJsonChanges = files.find((file) => file.filename === "package.json");
+        if (!packageJsonChanges) {
+            console.log("No package.json changes found in PR");
+            return null;
+        }
+        const patchLines = packageJsonChanges.patch.split("\n");
+        const changes = {};
+        // Process the patch lines to find version changes
+        for (let i = 0; i < patchLines.length; i++) {
+            const line = patchLines[i];
+            // Look for lines that change version numbers
+            const match = line.match(/^[-+]\s*"([^"]+)":\s*"([^"]+)"/);
+            if (match) {
+                const [, pkg, version] = match;
+                const changeType = line.startsWith("+") ? "new" : "old";
+                if (!changes[pkg]) {
+                    changes[pkg] = {};
+                }
+                changes[pkg][changeType] = version;
+            }
+        }
+        // Filter out any packages that don't have both old and new versions
+        const validChanges = {};
+        for (const [pkg, versions] of Object.entries(changes)) {
+            if (versions.old && versions.new) {
+                validChanges[pkg] = {
+                    oldVersion: versions.old,
+                    newVersion: versions.new,
+                };
+            }
+        }
+        return validChanges;
+    }
+    catch (error) {
+        console.error(`Error fetching PR changes: ${error.message}`);
+        throw error;
+    }
+}
+async function shouldMergePR({ repoName, prNumber, apiKey, }) {
+    const changes = await getPackageJsonChanges({
+        repoName,
+        prNumber,
+        apiKey,
+    });
+    if (!changes) {
+        console.log("No package.json changes found, skipping merge");
+        return false;
+    }
+    let allPatchUpdates = true;
+    for (const [pkg, versions] of Object.entries(changes)) {
+        const { oldVersion, newVersion } = versions;
+        if (!isPatchUpdate(oldVersion, newVersion)) {
+            console.log(`${pkg}: ${oldVersion} -> ${newVersion} is not a patch update`);
+            allPatchUpdates = false;
+        }
+        else {
+            console.log(`${pkg}: ${oldVersion} -> ${newVersion} is a patch update`);
+        }
+    }
+    return allPatchUpdates;
+}
+async function upgradeAndStagePackage({ repoPath, pkgName, version, isDevDep, }) {
+    const pkgVersion = version ?? (await getLatestVersion(pkgName));
+    (0, child_process_1.execSync)(`npm i ${pkgName}@${pkgVersion} ${isDevDep ? "--save-dev" : "--save"}`, { cwd: repoPath });
+    console.log(`Updated package: ${pkgName} version to ${pkgVersion} in ${repoPath}`);
+    (0, child_process_1.execSync)("git add package.json package-lock.json", { cwd: repoPath });
+}

package/dist/tools/utils/index.d.ts

@@ -4,10 +4,58 @@ export declare function makeDashboardRequest<T>({ path, method, body, apiKey, }:
     body?: any;
     apiKey: string;
 }): Promise<T>;
-export declare function callGitHubProxy({ method, url, body, apiKey, }: {
+export declare function callGitHubProxy<T>({ method, url, body, apiKey, }: {
     method: string;
     url: string;
     body?: any;
     apiKey: string;
-}): Promise<unknown>;
+}): Promise<T>;
+export declare const GITHUB_API_BASE = "https://api.github.com/repos/empirical-run";
+export interface GitHubRepoInfo {
+    owner: string;
+    repo: string;
+}
+export declare function parseGitHubUrl(url: string): GitHubRepoInfo;
+export interface PullRequestInfo {
+    number: number;
+    html_url: string;
+    head: {
+        ref: string;
+    };
+}
+export declare function findExistingPR({ owner, repo, branchName, apiKey, }: {
+    owner: string;
+    repo: string;
+    branchName: string;
+    apiKey: string;
+}): Promise<PullRequestInfo | null>;
+export declare function createPullRequest({ owner, repo, title, head, base, body, labels, apiKey, }: {
+    owner: string;
+    repo: string;
+    title: string;
+    head: string;
+    base?: string;
+    body: string;
+    labels?: string[];
+    apiKey: string;
+}): Promise<PullRequestInfo>;
+export declare function updatePullRequest({ owner, repo, prNumber, body, apiKey, }: {
+    owner: string;
+    repo: string;
+    prNumber: number;
+    body?: string;
+    apiKey: string;
+}): Promise<PullRequestInfo>;
+export declare function findOrCreatePullRequest({ owner, repo, branchName, title, body, labels, apiKey, }: {
+    owner: string;
+    repo: string;
+    branchName: string;
+    title: string;
+    body: string;
+    labels?: string[];
+    apiKey: string;
+}): Promise<{
+    pr: PullRequestInfo;
+    isNew: boolean;
+}>;
 //# sourceMappingURL=index.d.ts.map

package/dist/tools/utils/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/utils/index.ts"],"names":[],"mappings":"AAAA,wBAAsB,oBAAoB,CAAC,CAAC,EAAE,EAC5C,IAAI,EACJ,MAAc,EACd,IAAI,EACJ,MAAM,GACP,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,GAAG,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,CAAC,CAAC,CAmBb;AAED,wBAAsB,eAAe,CAAC,EACpC,MAAM,EACN,GAAG,EACH,IAAI,EACJ,MAAM,GACP,EAAE;IACD,MAAM,EAAE,MAAM,CAAC;IACf,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,GAAG,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;CAChB,oBAYA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tools/utils/index.ts"],"names":[],"mappings":"AAAA,wBAAsB,oBAAoB,CAAC,CAAC,EAAE,EAC5C,IAAI,EACJ,MAAc,EACd,IAAI,EACJ,MAAM,GACP,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,GAAG,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,CAAC,CAAC,CAmBb;AAED,wBAAsB,eAAe,CAAC,CAAC,EAAE,EACvC,MAAM,EACN,GAAG,EACH,IAAI,EACJ,MAAM,GACP,EAAE;IACD,MAAM,EAAE,MAAM,CAAC;IACf,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,GAAG,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,CAAC,CAAC,CAYb;AAGD,eAAO,MAAM,eAAe,+CAA+C,CAAC;AAG5E,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;CACd;AAED,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,cAAc,CAiB1D;AAGD,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE;QAAE,GAAG,EAAE,MAAM,CAAA;KAAE,CAAC;CACvB;AAED,wBAAsB,cAAc,CAAC,EACnC,KAAK,EACL,IAAI,EACJ,UAAU,EACV,MAAM,GACP,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,CAYlC;AAED,wBAAsB,iBAAiB,CAAC,EACtC,KAAK,EACL,IAAI,EACJ,KAAK,EACL,IAAI,EACJ,IAAa,EACb,IAAI,EACJ,MAAM,EACN,MAAM,GACP,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,eAAe,CAAC,CAqB3B;AAED,wBAAsB,iBAAiB,CAAC,EACtC,KAAK,EACL,IAAI,EACJ,QAAQ,EACR,IAAI,EACJ,MAAM,GACP,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC,eAAe,CAAC,CAU3B;AAED,wBAAsB,uBAAuB,CAAC,EAC5C,KAAK,EACL,IAAI,EACJ,UAAU,EACV,KAAK,EACL,IAAI,EACJ,MAAM,EACN,MAAM,GACP,EAAE;IACD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,OAAO,CAAC;IAAE,EAAE,EAAE,eAAe,CAAC;IAAC,KAAK,EAAE,OAAO,CAAA;CAAE,CAAC,CAyBnD"}