@empiricalrun/test-gen 0.73.1 → 0.74.0

package/dist/file-info/{file-system.js → adapters/file-system/reader.js}

@@ -3,9 +3,25 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.FilesystemFileReader = void 0;
 exports.getFileInfoFromFS = getFileInfoFromFS;
+const fs_1 = __importDefault(require("fs"));
 const promises_1 = __importDefault(require("fs/promises"));
 const path_1 = __importDefault(require("path"));
+class FilesystemFileReader {
+    async readFile(filePath, absoluteFilePath) {
+        if (!fs_1.default.existsSync(absoluteFilePath)) {
+            return null;
+        }
+        if (fs_1.default.statSync(absoluteFilePath).isDirectory()) {
+            const files = fs_1.default.readdirSync(absoluteFilePath);
+            return { content: files.join("\n"), isDirectory: true };
+        }
+        const content = fs_1.default.readFileSync(absoluteFilePath, "utf8");
+        return { content, isDirectory: false };
+    }
+}
+exports.FilesystemFileReader = FilesystemFileReader;
 async function getFileInfoFromFS(directory) {
     // TODO: This traverses all node_modules - can be excluded with DEFAULT_EXCLUDE
     let currentDirName = path_1.default.basename(directory);

package/dist/file-info/adapters/github/index.d.ts

@@ -0,0 +1,11 @@
+import { IDashboardAPIClient, ToolResult } from "@empiricalrun/shared-types";
+import { type StrReplaceInputParams } from "../../../tools/file-operations/shared/helpers";
+export { getFileInfoFromGitHub } from "./reader";
+export declare function viewFileUsingGitHub({ input, filePath, repoName, apiClient, branchName, }: {
+    input: StrReplaceInputParams;
+    filePath: string;
+    repoName: string;
+    apiClient: IDashboardAPIClient;
+    branchName: string;
+}): Promise<ToolResult>;
+//# sourceMappingURL=index.d.ts.map

package/dist/file-info/adapters/github/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/file-info/adapters/github/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAE7E,OAAO,EAEL,KAAK,qBAAqB,EAC3B,MAAM,+CAA+C,CAAC;AAEvD,OAAO,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAC;AAIjD,wBAAsB,mBAAmB,CAAC,EACxC,KAAK,EACL,QAAQ,EACR,QAAQ,EACR,SAAS,EACT,UAAU,GACX,EAAE;IACD,KAAK,EAAE,qBAAqB,CAAC;IAC7B,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,mBAAmB,CAAC;IAC/B,UAAU,EAAE,MAAM,CAAC;CACpB,GAAG,OAAO,CAAC,UAAU,CAAC,CAsBtB"}

package/dist/file-info/adapters/github/index.js

@@ -0,0 +1,29 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getFileInfoFromGitHub = void 0;
+exports.viewFileUsingGitHub = viewFileUsingGitHub;
+const helpers_1 = require("../../../tools/file-operations/shared/helpers");
+const reader_1 = require("./reader");
+var reader_2 = require("./reader");
+Object.defineProperty(exports, "getFileInfoFromGitHub", { enumerable: true, get: function () { return reader_2.getFileInfoFromGitHub; } });
+const REPO_OWNER = "empirical-run";
+async function viewFileUsingGitHub({ input, filePath, repoName, apiClient, branchName, }) {
+    const githubReader = new reader_1.GitHubFileReader(repoName, apiClient, REPO_OWNER);
+    const fileData = await githubReader.readFile(filePath, branchName);
+    if (!fileData) {
+        return {
+            result: `Error: File or directory ${filePath} is not found.`,
+            isError: true,
+        };
+    }
+    if (fileData.isDirectory) {
+        return {
+            result: fileData.content,
+            isError: false,
+        };
+    }
+    return {
+        result: (0, helpers_1.formatLinesWithNumbers)(fileData.content, input.view_range),
+        isError: false,
+    };
+}

package/dist/file-info/adapters/github/reader.d.ts

@@ -0,0 +1,16 @@
+import { FileInfo, FileReadResult, IDashboardAPIClient } from "@empiricalrun/shared-types";
+export declare class GitHubFileReader {
+    private repoName;
+    private apiClient;
+    private repoOwner;
+    constructor(repoName: string, apiClient: IDashboardAPIClient, repoOwner: string);
+    private fetchFileFromGitHub;
+    private fetchTreeFromGitHub;
+    private resolveBranchName;
+    readFile(filePath: string, branchName: string): Promise<FileReadResult | null>;
+    private getFileContent;
+    private getContentsForDirectory;
+    getFileInfo(branchName: string): Promise<FileInfo>;
+}
+export declare function getFileInfoFromGitHub(repoName: string, apiClient: IDashboardAPIClient, branchName: string): Promise<FileInfo>;
+//# sourceMappingURL=reader.d.ts.map

package/dist/file-info/adapters/github/reader.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"reader.d.ts","sourceRoot":"","sources":["../../../../src/file-info/adapters/github/reader.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EACR,cAAc,EACd,mBAAmB,EACpB,MAAM,4BAA4B,CAAC;AA4CpC,qBAAa,gBAAgB;IAEzB,OAAO,CAAC,QAAQ;IAChB,OAAO,CAAC,SAAS;IACjB,OAAO,CAAC,SAAS;gBAFT,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,mBAAmB,EAC9B,SAAS,EAAE,MAAM;YAGb,mBAAmB;YAUnB,mBAAmB;YASnB,iBAAiB;IAkBzB,QAAQ,CACZ,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC;YAmCnB,cAAc;YAiBd,uBAAuB;IAuB/B,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC;CA+EzD;AAED,wBAAsB,qBAAqB,CACzC,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,mBAAmB,EAC9B,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,QAAQ,CAAC,CAGnB"}

package/dist/file-info/adapters/github/reader.js

@@ -0,0 +1,185 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GitHubFileReader = void 0;
+exports.getFileInfoFromGitHub = getFileInfoFromGitHub;
+const path_1 = __importDefault(require("path"));
+const dashboard_1 = require("../../../dashboard");
+const REPO_OWNER = "empirical-run";
+class GitHubFileReader {
+    repoName;
+    apiClient;
+    repoOwner;
+    constructor(repoName, apiClient, repoOwner) {
+        this.repoName = repoName;
+        this.apiClient = apiClient;
+        this.repoOwner = repoOwner;
+    }
+    async fetchFileFromGitHub(filePath, branch) {
+        return (await this.apiClient.callGitHubProxy({
+            method: "GET",
+            url: `/repos/${this.repoOwner}/${this.repoName}/contents/${filePath}?ref=${branch}`,
+        }));
+    }
+    async fetchTreeFromGitHub(branch) {
+        return (await this.apiClient.callGitHubProxy({
+            method: "GET",
+            url: `/repos/${this.repoOwner}/${this.repoName}/git/trees/${branch}?recursive=1`,
+        }));
+    }
+    async resolveBranchName(branchName) {
+        if (!branchName || branchName === "main") {
+            return "main";
+        }
+        try {
+            // Try to fetch the branch to see if it exists
+            await this.apiClient.callGitHubProxy({
+                method: "GET",
+                url: `/repos/${this.repoOwner}/${this.repoName}/branches/${branchName}`,
+            });
+            return branchName;
+        }
+        catch {
+            // Branch doesn't exist, fallback to main
+            return "main";
+        }
+    }
+    async readFile(filePath, branchName) {
+        const branch = await this.resolveBranchName(branchName);
+        const normalizedPath = filePath === "" ? "." : filePath;
+        try {
+            const response = await this.fetchFileFromGitHub(normalizedPath, branch);
+            const isDirectory = Array.isArray(response);
+            if (!isDirectory) {
+                if (response.type === "file" && response.content) {
+                    const content = Buffer.from(response.content, "base64").toString("utf-8");
+                    return {
+                        content,
+                        isDirectory: false,
+                    };
+                }
+            }
+            else {
+                return {
+                    content: response.map((item) => item.name).join("\n"),
+                    isDirectory: true,
+                };
+            }
+        }
+        catch (error) {
+            if (error instanceof dashboard_1.NonRetryableHTTPError && error.status === 404) {
+                return null; // File or directory not found
+            }
+            throw error; // Re-throw other errors
+        }
+        return null;
+    }
+    async getFileContent(path, branchName) {
+        const branch = await this.resolveBranchName(branchName);
+        const response = await this.fetchFileFromGitHub(path, branch);
+        const isDirectory = Array.isArray(response);
+        if (isDirectory) {
+            throw new Error(`${path} is a directory, not a file`);
+        }
+        if (response.type === "file" && response.content) {
+            return Buffer.from(response.content, "base64").toString("utf-8");
+        }
+        throw new Error(`Unable to fetch file content for ${path}`);
+    }
+    async getContentsForDirectory(path, branchName) {
+        const branch = await this.resolveBranchName(branchName);
+        const response = await this.fetchTreeFromGitHub(branch);
+        if (response.tree) {
+            // Convert to expected RepoTree format
+            return {
+                sha: response.sha,
+                truncated: false,
+                tree: response.tree.map((item) => ({
+                    path: item.path || "",
+                    mode: item.mode || "",
+                    type: item.type === "blob" ? "blob" : "tree",
+                    sha: item.sha,
+                })),
+            };
+        }
+        throw new Error(`Unable to fetch directory contents for ${path}`);
+    }
+    async getFileInfo(branchName) {
+        const files = await this.getContentsForDirectory("", branchName);
+        const root = [];
+        const nodeMap = {};
+        if (!files) {
+            return {
+                type: "directory",
+                path: this.repoName,
+                name: this.repoName,
+                children: [],
+            };
+        }
+        files.tree.forEach((file) => {
+            if (!file.path) {
+                return;
+            }
+            const pathParts = file.path.split("/");
+            const fileName = path_1.default.basename(file.path);
+            const isFile = file.type === "blob";
+            const currentPath = file.path;
+            const currentNode = isFile
+                ? {
+                    type: "file",
+                    path: currentPath,
+                    name: fileName,
+                    getContent: async () => this.getFileContent(currentPath, branchName),
+                }
+                : {
+                    type: "directory",
+                    path: currentPath,
+                    name: fileName,
+                    children: [],
+                };
+            nodeMap[currentPath] = currentNode;
+            if (pathParts.length === 1) {
+                root.push(currentNode);
+            }
+            else {
+                const parentPath = path_1.default.dirname(currentPath);
+                if (!nodeMap[parentPath]) {
+                    const parentNode = {
+                        type: "directory",
+                        path: parentPath,
+                        name: parentPath.split("/").pop() || "",
+                        children: [],
+                    };
+                    nodeMap[parentPath] = parentNode;
+                    const parentPathParts = parentPath.split("/");
+                    if (parentPathParts.length === 1) {
+                        root.push(parentNode);
+                    }
+                    else {
+                        const grandparentPath = path_1.default.dirname(parentPath);
+                        if (nodeMap[grandparentPath] &&
+                            nodeMap[grandparentPath].type === "directory") {
+                            nodeMap[grandparentPath].children.push(parentNode);
+                        }
+                    }
+                }
+                if (nodeMap[parentPath].type === "directory") {
+                    nodeMap[parentPath].children.push(currentNode);
+                }
+            }
+        });
+        return {
+            type: "directory",
+            path: this.repoName,
+            name: this.repoName,
+            children: root,
+        };
+    }
+}
+exports.GitHubFileReader = GitHubFileReader;
+async function getFileInfoFromGitHub(repoName, apiClient, branchName) {
+    const fileReader = new GitHubFileReader(repoName, apiClient, REPO_OWNER);
+    return fileReader.getFileInfo(branchName);
+}

package/dist/file-info/index.d.ts

@@ -0,0 +1,4 @@
+export { viewFileUsingFileSystem } from "./adapters/file-system";
+export { FilesystemFileReader, getFileInfoFromFS, } from "./adapters/file-system/reader";
+export { viewFileUsingGitHub } from "./adapters/github";
+//# sourceMappingURL=index.d.ts.map

package/dist/file-info/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/file-info/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EACL,oBAAoB,EACpB,iBAAiB,GAClB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/file-info/index.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.viewFileUsingGitHub = exports.getFileInfoFromFS = exports.FilesystemFileReader = exports.viewFileUsingFileSystem = void 0;
+var file_system_1 = require("./adapters/file-system");
+Object.defineProperty(exports, "viewFileUsingFileSystem", { enumerable: true, get: function () { return file_system_1.viewFileUsingFileSystem; } });
+var reader_1 = require("./adapters/file-system/reader");
+Object.defineProperty(exports, "FilesystemFileReader", { enumerable: true, get: function () { return reader_1.FilesystemFileReader; } });
+Object.defineProperty(exports, "getFileInfoFromFS", { enumerable: true, get: function () { return reader_1.getFileInfoFromFS; } });
+var github_1 = require("./adapters/github");
+Object.defineProperty(exports, "viewFileUsingGitHub", { enumerable: true, get: function () { return github_1.viewFileUsingGitHub; } });

package/dist/index.d.ts

@@ -1,6 +1,8 @@
 import { FrameLocator, Page } from "playwright";
 import { ScopeVars } from "./types";
 export { downloadBuild } from "./test-build";
+export { extractFileFromZipFromUrl, getFilenamesInZip, } from "./tools/trace-dot-zip/utils/extract-zip";
+export { generateNetworkTrace } from "./tools/trace-dot-zip/utils/network-trace";
 export declare function createTest(task: string, pageRef: Page | FrameLocator, scope?: ScopeVars): Promise<void>;
 export declare function recordTest(pageRef: Page): Promise<void>;
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAYhD,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAsB7C,wBAAsB,UAAU,CAC9B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,IAAI,GAAG,YAAY,EAC5B,KAAK,CAAC,EAAE,SAAS,iBA0DlB;AAED,wBAAsB,UAAU,CAAC,OAAO,EAAE,IAAI,iBAY7C"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAYhD,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EACL,yBAAyB,EACzB,iBAAiB,GAClB,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAsBjF,wBAAsB,UAAU,CAC9B,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,IAAI,GAAG,YAAY,EAC5B,KAAK,CAAC,EAAE,SAAS,iBA0DlB;AAED,wBAAsB,UAAU,CAAC,OAAO,EAAE,IAAI,iBAY7C"}

package/dist/index.js

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.downloadBuild = void 0;
+exports.generateNetworkTrace = exports.getFilenamesInZip = exports.extractFileFromZipFromUrl = exports.downloadBuild = void 0;
 exports.createTest = createTest;
 exports.recordTest = recordTest;
 const llm_1 = require("@empiricalrun/llm");
@@ -16,6 +16,11 @@ const client_1 = __importDefault(require("./file/client"));
 const logger_1 = require("./logger");
 var test_build_1 = require("./test-build");
 Object.defineProperty(exports, "downloadBuild", { enumerable: true, get: function () { return test_build_1.downloadBuild; } });
+var extract_zip_1 = require("./tools/trace-dot-zip/utils/extract-zip");
+Object.defineProperty(exports, "extractFileFromZipFromUrl", { enumerable: true, get: function () { return extract_zip_1.extractFileFromZipFromUrl; } });
+Object.defineProperty(exports, "getFilenamesInZip", { enumerable: true, get: function () { return extract_zip_1.getFilenamesInZip; } });
+var network_trace_1 = require("./tools/trace-dot-zip/utils/network-trace");
+Object.defineProperty(exports, "generateNetworkTrace", { enumerable: true, get: function () { return network_trace_1.generateNetworkTrace; } });
 const flushEvents = async () => {
     await (0, llm_1.flushAllTraces)();
 };

package/dist/recorder/index.d.ts

@@ -1,4 +1,8 @@
-export declare function runRecorder({ name }: {
-    name: string;
-}): Promise<void>;
+interface RecorderCLIOptions {
+    testName: string | undefined;
+    project: string | undefined;
+    url: string | undefined;
+}
+export declare function runRecorder({ testName, project, url, }: RecorderCLIOptions): Promise<void>;
+export {};
 //# sourceMappingURL=index.d.ts.map

package/dist/recorder/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/recorder/index.ts"],"names":[],"mappings":"AAgCA,wBAAsB,WAAW,CAAC,EAAE,IAAI,EAAE,EAAE;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,iBA+I3D"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/recorder/index.ts"],"names":[],"mappings":"AAsEA,UAAU,kBAAkB;IAC1B,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B,OAAO,EAAE,MAAM,GAAG,SAAS,CAAC;IAC5B,GAAG,EAAE,MAAM,GAAG,SAAS,CAAC;CACzB;AAqTD,wBAAsB,WAAW,CAAC,EAChC,QAAQ,EACR,OAAO,EACP,GAAG,GACJ,EAAE,kBAAkB,iBAGpB"}