@empiricalrun/test-gen 0.66.0 → 0.66.2

package/dist/agent/cua/pw-codegen/pw-pause/types.d.ts

@@ -0,0 +1,14 @@
+export type SourcesPayload = {
+    isRecorded: boolean;
+    label: "library";
+    group: "Node.js";
+    id: "javascript";
+    text: string;
+    header: string;
+    footer: string;
+    actions: string[];
+    language: "javascript";
+    revealLine: number;
+    highlight: string[];
+};
+//# sourceMappingURL=types.d.ts.map

package/dist/agent/cua/pw-codegen/pw-pause/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../../src/agent/cua/pw-codegen/pw-pause/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,cAAc,GAAG;IAC3B,UAAU,EAAE,OAAO,CAAC;IACpB,KAAK,EAAE,SAAS,CAAC;IACjB,KAAK,EAAE,SAAS,CAAC;IACjB,EAAE,EAAE,YAAY,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,QAAQ,EAAE,YAAY,CAAC;IAEvB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,EAAE,CAAC;CACrB,CAAC"}

package/dist/agent/cua/pw-codegen/pw-pause/types.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/artifacts/utils.d.ts

@@ -5,7 +5,7 @@ import { JSONReport as PlaywrightJSONReport } from "@playwright/test/reporter";
  * @param report The Playwright JSON report to extract attachments from
  * @returns An array of objects containing path and contentType for each attachment
  */
-export declare function extractAttachmentsFromPlaywrightJSONReport(report: PlaywrightJSONReport): ArtifactInputPath[];
+export declare function extractAttachmentsFromPlaywrightJSONReport(report: PlaywrightJSONReport, testNameFilter?: string): ArtifactInputPath[];
 /**
  * Scans the given repository directory for Playwright artifacts (attachments) by:
  * - Constructing the path to the Playwright report directory using the provided repoDir.

package/dist/artifacts/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/artifacts/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC9E,OAAO,EACL,UAAU,IAAI,oBAAoB,EAGnC,MAAM,2BAA2B,CAAC;AAGnC;;;;GAIG;AACH,wBAAgB,0CAA0C,CACxD,MAAM,EAAE,oBAAoB,GAC3B,iBAAiB,EAAE,CAqErB;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,aAAa,EAAE,CAmBxE"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/artifacts/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC9E,OAAO,EACL,UAAU,IAAI,oBAAoB,EAGnC,MAAM,2BAA2B,CAAC;AAInC;;;;GAIG;AACH,wBAAgB,0CAA0C,CACxD,MAAM,EAAE,oBAAoB,EAC5B,cAAc,CAAC,EAAE,MAAM,GACtB,iBAAiB,EAAE,CA2DrB;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,aAAa,EAAE,CAmCxE"}

package/dist/artifacts/utils.js

@@ -6,12 +6,13 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.extractAttachmentsFromPlaywrightJSONReport = extractAttachmentsFromPlaywrightJSONReport;
 exports.findPlaywrightArtifacts = findPlaywrightArtifacts;
 const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
 /**
  * Extracts attachment information from a Playwright JSON report.
  * @param report The Playwright JSON report to extract attachments from
  * @returns An array of objects containing path and contentType for each attachment
  */
-function extractAttachmentsFromPlaywrightJSONReport(report) {
+function extractAttachmentsFromPlaywrightJSONReport(report, testNameFilter) {
     const attachments = [];
     if (!report || !report.suites || report.suites.length === 0) {
         return attachments;
@@ -25,6 +26,9 @@ function extractAttachmentsFromPlaywrightJSONReport(report) {
         if (!spec.tests) {
             return;
         }
+        if (testNameFilter && !testTitle.includes(testNameFilter)) {
+            return;
+        }
         for (const test of spec.tests) {
             if (!test.results) {
                 continue;
@@ -37,7 +41,7 @@ function extractAttachmentsFromPlaywrightJSONReport(report) {
                     // Only collect attachments that have a 'path' (i.e., not embedded content)
                     if (attachment.path && attachment.contentType) {
                         attachments.push({
-                            name: `${testTitle} ${attachment.name}`,
+                            name: `${testTitle} - ${attachment.name}`,
                             path: attachment.path,
                             contentType: attachment.contentType,
                         });
@@ -55,19 +59,16 @@ function extractAttachmentsFromPlaywrightJSONReport(report) {
             return;
         }
         for (const suite of suites) {
-            // Process specs directly within the current suite
             if (suite.specs) {
                 for (const spec of suite.specs) {
                     processSpec(spec);
                 }
             }
-            // Recursively call for any nested suites
             if (suite.suites) {
                 traverseSuites(suite.suites);
             }
         }
     }
-    // Start the recursive traversal from the top-level suites
     traverseSuites(report.suites);
     return attachments;
 }
@@ -84,9 +85,19 @@ function extractAttachmentsFromPlaywrightJSONReport(report) {
  */
 function findPlaywrightArtifacts(repoDir) {
     const artifacts = [];
-    const reportPath = require("path").join(repoDir, "playwright-report");
-    const summaryJsonPath = require("path").join(reportPath, "summary.json");
-    if (fs_1.default.existsSync(summaryJsonPath)) {
+    const pathForPlaywright147 = path_1.default.join(repoDir, "playwright-report", "summary.json");
+    const pathForPlaywright153 = path_1.default.join(repoDir, "summary.json");
+    let summaryJsonPath;
+    if (fs_1.default.existsSync(pathForPlaywright147)) {
+        summaryJsonPath = pathForPlaywright147;
+    }
+    else if (fs_1.default.existsSync(pathForPlaywright153)) {
+        summaryJsonPath = pathForPlaywright153;
+    }
+    else {
+        console.warn(`[findPlaywrightArtifacts] No Playwright report found in ${repoDir}`);
+    }
+    if (summaryJsonPath) {
         try {
             const summaryJson = JSON.parse(fs_1.default.readFileSync(summaryJsonPath, "utf-8"));
             const found = extractAttachmentsFromPlaywrightJSONReport(summaryJson);

package/dist/bin/index.js

@@ -16,6 +16,7 @@ const diagnosis_agent_1 = require("../agent/diagnosis-agent");
 const enrich_prompt_1 = require("../agent/enrich-prompt");
 const infer_agent_1 = require("../agent/infer-agent");
 const run_3 = require("../agent/planner/run");
+const recorder_1 = require("../recorder");
 const reporter_1 = require("../reporter");
 const session_1 = require("../session");
 const test_build_1 = require("../test-build");
@@ -200,7 +201,7 @@ async function runAgentsWorkflow(testGenConfig, testGenToken) {
 }
 async function main() {
     const removeListeners = setupProcessListeners(flushEvents);
-    (0, utils_2.printBanner)();
+    await (0, utils_2.printBanner)();
     const program = new commander_1.Command();
     program
         .option("--token <token>", "Test generation token")
@@ -209,6 +210,7 @@ async function main() {
         .option("--file <test-file>", "File path of the test case (inside tests dir)")
         .option("--suites <suites>", "Comma separated list of describe blocks")
         .option("--use-chat", "Use chat agent (and not the workflow)")
+        .option("--use-recorder", "Run the recorder flow to create a request")
         .option("--chat-session-id <chat-session-id>", "Identifier for chat session (fetched from dash.empirical.run)")
         .option("--use-disk-for-chat-state", "Save and load chat state from disk")
         .option("--chat-model <model>", "Chat model to use (claude-3-7-sonnet-20250219 or claude-3-5-sonnet-20241022 or gemini-2.5-pro-preview-06-05)")
@@ -242,6 +244,10 @@ async function main() {
             apiKey: process.env.EMPIRICALRUN_API_KEY,
         });
     }
+    if (completedOptions.useRecorder) {
+        await (0, recorder_1.runRecorder)({ name: completedOptions.name });
+        return;
+    }
     if (completedOptions.useChat) {
         await runChatAgent({
             chatSessionId: completedOptions.chatSessionId,

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

@@ -11,7 +11,8 @@ export interface CLIOptions {
     initialPrompt?: string;
     chatSessionId?: string;
     chatModel?: (typeof ARGS_TO_MODEL_MAP)[keyof typeof ARGS_TO_MODEL_MAP];
+    useRecorder?: boolean;
 }
 export declare function validateAndCompleteCliOptions(options: CLIOptions): Promise<CLIOptions>;
-export declare function printBanner(): void;
+export declare function printBanner(): Promise<void>;
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/bin/utils/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAGjE,eAAO,MAAM,iBAAiB,EAAE,MAAM,CAAC,MAAM,EAAE,mBAAmB,CAgBjE,CAAC;AAEF,MAAM,WAAW,UAAU;IACzB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,CAAC,OAAO,iBAAiB,CAAC,CAAC,MAAM,OAAO,iBAAiB,CAAC,CAAC;CACxE;AAQD,wBAAsB,6BAA6B,CACjD,OAAO,EAAE,UAAU,GAClB,OAAO,CAAC,UAAU,CAAC,CAyDrB;AAED,wBAAgB,WAAW,SAgC1B"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/bin/utils/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAKjE,eAAO,MAAM,iBAAiB,EAAE,MAAM,CAAC,MAAM,EAAE,mBAAmB,CAgBjE,CAAC;AAEF,MAAM,WAAW,UAAU;IACzB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAGhB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,CAAC,OAAO,iBAAiB,CAAC,CAAC,MAAM,OAAO,iBAAiB,CAAC,CAAC;IAGvE,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;AAQD,wBAAsB,6BAA6B,CACjD,OAAO,EAAE,UAAU,GAClB,OAAO,CAAC,UAAU,CAAC,CA8DrB;AAeD,wBAAsB,WAAW,kBAgDhC"}

package/dist/bin/utils/index.js

@@ -7,6 +7,7 @@ exports.ARGS_TO_MODEL_MAP = void 0;
 exports.validateAndCompleteCliOptions = validateAndCompleteCliOptions;
 exports.printBanner = printBanner;
 const inquirer_1 = __importDefault(require("inquirer"));
+const PACKAGE_NAME = "@empiricalrun/test-gen";
 exports.ARGS_TO_MODEL_MAP = {
     "claude-3-5": "claude-3-5-sonnet-20241022",
     "claude-3-7": "claude-3-7-sonnet-20250219",
@@ -32,6 +33,9 @@ async function validateAndCompleteCliOptions(options) {
         // Chat agent can prompt the user directly, nothing is required in CLI args
         requiredFields = [];
     }
+    if (options.useRecorder) {
+        requiredFields = ["name"];
+    }
     const questions = [];
     if (!options.name && requiredFields.includes("name")) {
         questions.push({
@@ -76,7 +80,19 @@ async function validateAndCompleteCliOptions(options) {
     }
     return options;
 }
-function printBanner() {
+async function getLatestVersion(packageName) {
+    try {
+        const response = await fetch(`https://registry.npmjs.org/${packageName}/latest`);
+        if (!response.ok)
+            return null;
+        const data = await response.json();
+        return data.version;
+    }
+    catch {
+        return null;
+    }
+}
+async function printBanner() {
     const gray = "\x1b[90m";
     const reset = "\x1b[0m";
     const asciiArtRaw = `
@@ -85,7 +101,17 @@ function printBanner() {
  /)__)
 --"-"-`;
     const version = require("../../../package.json").version;
-    const logLine1 = `Running test-gen v${version}`;
+    const latestVersion = await getLatestVersion(PACKAGE_NAME);
+    let versionSuffix = "";
+    if (latestVersion) {
+        if (version === latestVersion) {
+            versionSuffix = ` ${gray}(latest)${reset}`;
+        }
+        else {
+            versionSuffix = ` (latest is ${latestVersion})`;
+        }
+    }
+    const logLine1 = `Running ${PACKAGE_NAME} v${version}${versionSuffix}`;
     const logLine2 = `from ${__dirname.split("/bin/utils")[0]}`;
     // Process ASCII art
     const asciiLines = asciiArtRaw
@@ -94,20 +120,23 @@ function printBanner() {
     const maxWidth = Math.max(...asciiLines.map((line) => line.length));
     const middleIndex = Math.floor(asciiLines.length / 2);
     const spacing = "    "; // 4 spaces for padding
+    const leftPadding = "  "; // Left padding
+    console.log(""); // Top padding
     // Print formatted output (art default, logs gray)
     asciiLines.forEach((line, index) => {
         const paddedLine = line.padEnd(maxWidth, " ");
         if (index === middleIndex) {
             // Print art line (default color) + first log line (gray)
-            console.log(`${paddedLine}${spacing}${gray}${logLine1}${reset}`);
+            console.log(`${leftPadding}${paddedLine}${spacing}${gray}${logLine1}${reset}`);
         }
         else if (index === middleIndex + 1) {
             // Print corresponding art line (default color) + second log line (gray)
-            console.log(`${paddedLine}${spacing}${gray}${logLine2}${reset}`);
+            console.log(`${leftPadding}${paddedLine}${spacing}${gray}${logLine2}${reset}`);
         }
         else {
             // Print other art lines (default color)
-            console.log(paddedLine);
+            console.log(`${leftPadding}${paddedLine}`);
         }
     });
+    console.log(""); // Bottom padding
 }

package/dist/browser-injected-scripts/annotate-elements.js

@@ -1,5 +1,3 @@
-/* eslint-disable autofix/no-unused-vars */
-
 /**
  * Annotates all clickable elements on the page with unique hint markers.
  * Returns an object containing annotations and methods to enable/disable them.
@@ -10,6 +8,7 @@
  * @param {string} options.markerClass - CSS class to apply to hint markers.
  * @returns {Object} An object containing annotations map and enable/disable methods.
  */
+// eslint-disable-next-line no-unused-vars, autofix/no-unused-vars
 function annotateElementsWithPreference({
   options = {},
   preference = {},
@@ -32,8 +31,6 @@ function annotateElementsWithPreference({
   // Check if the element is not blocked and visible for clicking
   function isElementClickNotBlocked(element, windowToAnnotate) {
     const rect = element.getBoundingClientRect();
-    const originalScrollX = windowToAnnotate.scrollX;
-    const originalScrollY = windowToAnnotate.scrollY;
 
     // Calculate the center point of the element
     const centerX = rect.left + rect.width / 2;

package/dist/file/client.d.ts

@@ -5,6 +5,7 @@ declare class FileServiceClient {
     constructor();
     static isAvailable(): boolean;
     sendAgentResult(payload: BrowserAgentIPCPayload): Promise<any>;
+    sendCodegenSources(payload: any): Promise<any>;
     post(path: string, body: any): Promise<any>;
 }
 export default FileServiceClient;

package/dist/file/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/file/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAElD,cAAM,iBAAiB;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;;IAUzB,MAAM,CAAC,WAAW;IAIZ,eAAe,CAAC,OAAO,EAAE,sBAAsB;IAI/C,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG;CAgBnC;AAED,eAAe,iBAAiB,CAAC"}
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/file/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAElD,cAAM,iBAAiB;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;;IAUzB,MAAM,CAAC,WAAW;IAIZ,eAAe,CAAC,OAAO,EAAE,sBAAsB;IAI/C,kBAAkB,CAAC,OAAO,EAAE,GAAG;IAI/B,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG;CAgBnC;AAED,eAAe,iBAAiB,CAAC"}

package/dist/file/client.js

@@ -16,6 +16,9 @@ class FileServiceClient {
     async sendAgentResult(payload) {
         return this.post("/agent-results", payload);
     }
+    async sendCodegenSources(payload) {
+        return this.post("/codegen-sources", payload);
+    }
     async post(path, body) {
         const resp = await fetch(`${this.baseUrl}${path}`, {
             method: "POST",

package/dist/file/server.d.ts

@@ -17,6 +17,7 @@ export declare class FileServiceServer {
     private artifactsInputs;
     private result;
     private usage;
+    private codegenSources;
     constructor({ port, repoDir, updateFile, onComplete, }: {
         port: number;
         repoDir: string;
@@ -28,6 +29,7 @@ export declare class FileServiceServer {
         usage: Usage | undefined;
     };
     getArtifactInputsFromServer(): ArtifactInput[];
+    getCodegenSources(): string | undefined;
     setFilePath(filePath: string): void;
     startFileService(): Promise<number>;
     stop(): Promise<void>;

package/dist/file/server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/file/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,4BAA4B,CAAC;AAKlE,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAQlD,MAAM,MAAM,sBAAsB,GAAG;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,kBAAkB,CAAC;IAC3B,KAAK,EAAE,KAAK,CAAC;IACb,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,EAAE,CAAC;CACvB,CAAC;AAEF,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,IAAI,CAAa;IACzB,OAAO,CAAC,QAAQ,CAAc;IAC9B,OAAO,CAAC,OAAO,CAAc;IAC7B,OAAO,CAAC,MAAM,CAA4C;IAC1D,OAAO,CAAC,UAAU,CAAkB;IACpC,OAAO,CAAC,UAAU,CAAC,CAAa;IAEhC,OAAO,CAAC,eAAe,CAAuB;IAC9C,OAAO,CAAC,MAAM,CAAiC;IAC/C,OAAO,CAAC,KAAK,CAAoB;gBAErB,EACV,IAAI,EACJ,OAAO,EACP,UAAU,EACV,UAAU,GACX,EAAE;QACD,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,UAAU,EAAE,OAAO,CAAC;QACpB,UAAU,CAAC,EAAE,MAAM,IAAI,CAAC;KACzB;IAOD,iBAAiB;;;;IAIjB,2BAA2B;IAI3B,WAAW,CAAC,QAAQ,EAAE,MAAM;IAItB,gBAAgB,IAAI,OAAO,CAAC,MAAM,CAAC;IA4DnC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;CAgB5B"}
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/file/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,4BAA4B,CAAC;AAKlE,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AASlD,MAAM,MAAM,sBAAsB,GAAG;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,kBAAkB,CAAC;IAC3B,KAAK,EAAE,KAAK,CAAC;IACb,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,EAAE,CAAC;CACvB,CAAC;AAEF,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,IAAI,CAAa;IACzB,OAAO,CAAC,QAAQ,CAAc;IAC9B,OAAO,CAAC,OAAO,CAAc;IAC7B,OAAO,CAAC,MAAM,CAA4C;IAC1D,OAAO,CAAC,UAAU,CAAkB;IACpC,OAAO,CAAC,UAAU,CAAC,CAAa;IAEhC,OAAO,CAAC,eAAe,CAAuB;IAC9C,OAAO,CAAC,MAAM,CAAiC;IAC/C,OAAO,CAAC,KAAK,CAAoB;IAEjC,OAAO,CAAC,cAAc,CAAqB;gBAE/B,EACV,IAAI,EACJ,OAAO,EACP,UAAU,EACV,UAAU,GACX,EAAE;QACD,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,UAAU,EAAE,OAAO,CAAC;QACpB,UAAU,CAAC,EAAE,MAAM,IAAI,CAAC;KACzB;IAOD,iBAAiB;;;;IAIjB,2BAA2B;IAI3B,iBAAiB;IAIjB,WAAW,CAAC,QAAQ,EAAE,MAAM;IAItB,gBAAgB,IAAI,OAAO,CAAC,MAAM,CAAC;IAqEnC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;CAgB5B"}

package/dist/file/server.js

@@ -19,6 +19,7 @@ class FileServiceServer {
     artifactsInputs = [];
     result;
     usage;
+    codegenSources;
     constructor({ port, repoDir, updateFile, onComplete, }) {
         this.port = port;
         this.repoDir = repoDir;
@@ -31,6 +32,9 @@ class FileServiceServer {
     getArtifactInputsFromServer() {
         return this.artifactsInputs;
     }
+    getCodegenSources() {
+        return this.codegenSources;
+    }
     setFilePath(filePath) {
         this.filePath = path_1.default.resolve(this.repoDir, filePath);
     }
@@ -38,6 +42,11 @@ class FileServiceServer {
         const app = (0, express_1.default)();
         app.use(express_1.default.json({ limit: "50mb" }));
         (0, ipc_1.humanLoopRoute)(app);
+        app.post("/codegen-sources", async (req) => {
+            const payload = req.body;
+            this.codegenSources = payload.map((c) => c.actions.join("\n")).join("\n");
+            console.log("[FileServiceServer] Received codegen sources", this.codegenSources);
+        });
         app.post("/agent-results", async (req, res) => {
             const { generatedCode, importPaths, result, usage } = req.body;
             this.result = result;

package/dist/index.d.ts

@@ -2,4 +2,5 @@ import { FrameLocator, Page } from "playwright";
 import { ScopeVars } from "./types";
 export { downloadBuild } from "./test-build";
 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;AAWhD,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,iBAwElB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAahD,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,iBAwElB;AAED,wBAAsB,UAAU,CAAC,OAAO,EAAE,IAAI,iBAY7C"}

package/dist/index.js

@@ -5,8 +5,11 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.downloadBuild = void 0;
 exports.createTest = createTest;
+exports.recordTest = recordTest;
 const llm_1 = require("@empiricalrun/llm");
 const cua_1 = require("./agent/cua");
+const pw_pause_1 = require("./agent/cua/pw-codegen/pw-pause");
+const for_recorder_1 = require("./agent/cua/pw-codegen/pw-pause/for-recorder");
 const run_1 = require("./agent/master/run");
 const scenarios_1 = require("./bin/utils/scenarios");
 const client_1 = __importDefault(require("./file/client"));
@@ -94,3 +97,16 @@ async function createTest(task, pageRef, scope) {
         await flushEvents();
     }
 }
+async function recordTest(pageRef) {
+    const fileServiceClient = new client_1.default();
+    const repoDir = process.cwd();
+    const canUsePwPause = await (0, pw_pause_1.canUsePauseCodegen)(repoDir);
+    if (!canUsePwPause) {
+        return;
+    }
+    console.log("[getCodegen] using PlaywrightPauseCodegen");
+    const codegen = new for_recorder_1.PlaywrightPauseCodegenForRecorder(async (sources) => {
+        await fileServiceClient.sendCodegenSources(sources);
+    });
+    await codegen.initialize(pageRef);
+}

package/dist/recorder/display.d.ts

@@ -0,0 +1,2 @@
+export declare function displayResultsAndConfirm(name: string, codegenResult: string): Promise<string>;
+//# sourceMappingURL=display.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"display.d.ts","sourceRoot":"","sources":["../../src/recorder/display.ts"],"names":[],"mappings":"AAEA,wBAAsB,wBAAwB,CAC5C,IAAI,EAAE,MAAM,EACZ,aAAa,EAAE,MAAM,mBA+CtB"}

package/dist/recorder/display.js

@@ -0,0 +1,50 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.displayResultsAndConfirm = displayResultsAndConfirm;
+const inquirer_1 = __importDefault(require("inquirer"));
+async function displayResultsAndConfirm(name, codegenResult) {
+    let finalResult = codegenResult || "No code generated";
+    let approved = false;
+    while (!approved) {
+        // Display results in a box format
+        console.log("\n" + "=".repeat(80));
+        console.log("📝 TEST GENERATION RESULTS");
+        console.log("=".repeat(80));
+        console.log(`Test Name: ${name}`);
+        console.log("-".repeat(80));
+        console.log("Generated Code:");
+        console.log("-".repeat(80));
+        console.log(finalResult);
+        console.log("=".repeat(80));
+        // Ask user for confirmation
+        const { confirmed } = await inquirer_1.default.prompt([
+            {
+                type: "confirm",
+                name: "confirmed",
+                message: "Does this look good to you?",
+                default: true,
+            },
+        ]);
+        if (confirmed) {
+            console.log("✅ Great! The test generation was successful.");
+            approved = true;
+        }
+        else {
+            // Open multiline editor for user to modify
+            const { editedCode } = await inquirer_1.default.prompt([
+                {
+                    type: "editor",
+                    name: "editedCode",
+                    message: "Edit the generated code:",
+                    default: finalResult,
+                },
+            ]);
+            finalResult = editedCode;
+            console.log("Code updated. Let's review the changes...");
+        }
+    }
+    return finalResult;
+}

package/dist/recorder/index.d.ts

@@ -0,0 +1,4 @@
+export declare function runRecorder({ name }: {
+    name: string;
+}): Promise<void>;
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/recorder/index.ts"],"names":[],"mappings":"AAyCA,wBAAsB,WAAW,CAAC,EAAE,IAAI,EAAE,EAAE;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,iBAiF3D"}

package/dist/recorder/index.js

@@ -0,0 +1,108 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runRecorder = runRecorder;
+const test_run_1 = require("@empiricalrun/test-run");
+const detect_port_1 = __importDefault(require("detect-port"));
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const utils_1 = require("../agent/browsing/utils");
+const chat_1 = require("../agent/chat");
+const pw_pause_1 = require("../agent/cua/pw-codegen/pw-pause");
+const utils_2 = require("../artifacts/utils");
+const server_1 = require("../file/server");
+const display_1 = require("./display");
+const request_1 = require("./request");
+const temp_files_1 = require("./temp-files");
+const upload_1 = require("./upload");
+const validation_1 = require("./validation");
+function extractVideoAttachments(repoDir) {
+    try {
+        const summaryPath = path_1.default.join(repoDir, "summary.json");
+        if (!fs_1.default.existsSync(summaryPath)) {
+            console.log("summary.json not found");
+            return [];
+        }
+        const summaryContent = JSON.parse(fs_1.default.readFileSync(summaryPath, "utf-8"));
+        const attachments = (0, utils_2.extractAttachmentsFromPlaywrightJSONReport)(summaryContent, "temp test");
+        const videoPaths = attachments
+            .filter((attachment) => attachment.contentType === "video/webm")
+            .map((attachment) => attachment.path);
+        return videoPaths;
+    }
+    catch (error) {
+        console.warn("Error processing summary.json:", error);
+        return [];
+    }
+}
+async function runRecorder({ name }) {
+    console.log(`Recording for test name: ${name}`);
+    const repoDir = process.cwd();
+    try {
+        await (0, validation_1.validate)(repoDir);
+    }
+    catch (error) {
+        console.error("Error running recorder:", error);
+        process.exit(1);
+    }
+    if (!process.env.EMPIRICALRUN_API_KEY) {
+        console.error("EMPIRICALRUN_API_KEY is not set. Please set it in your environment variables.");
+        process.exit(1);
+    }
+    try {
+        // Prepare playwright for codegen
+        console.log("[generateTestWithBrowserAgent] Preparing playwright for codegen");
+        await (0, pw_pause_1.preparePlaywrightForCodegen)(repoDir);
+    }
+    catch (err) {
+        console.warn("[generateTestWithBrowserAgent] Error preparing playwright for codegen", err);
+    }
+    const envVariables = await (0, chat_1.fetchEnvironmentVariables)();
+    await (0, temp_files_1.createTempTestFile)();
+    const absFilePath = path_1.default.join(process.cwd(), "tests", "temp-test.spec.ts");
+    await (0, utils_1.addImportForMethod)(absFilePath, "recordTest");
+    // Start a file service for IPC with the agent (which runs in a different process)
+    const availablePort = await (0, detect_port_1.default)(3030);
+    const fileServer = new server_1.FileServiceServer({
+        port: availablePort,
+        repoDir,
+        updateFile: false,
+    });
+    await fileServer.startFileService();
+    await (0, test_run_1.runSingleTest)({
+        testName: "temp test",
+        suites: [],
+        filePath: "tests/temp-test.spec.ts",
+        projects: ["chromium"],
+        repoDir: process.cwd(),
+        envOverrides: {
+            ...envVariables,
+            RUN_PLAYWRIGHT_HEADED: "true",
+            PW_CODEGEN_NO_INSPECTOR: "1",
+            IPC_FILE_SERVICE_PORT: availablePort.toString(),
+        },
+    });
+    const videoPaths = extractVideoAttachments(repoDir);
+    let attachments = [];
+    if (videoPaths.length === 0) {
+        console.warn("No video attachments found for temp test");
+    }
+    else {
+        const videoUrls = await (0, upload_1.uploadVideosWithSpinner)(videoPaths, name);
+        if (videoUrls) {
+            attachments = [...attachments, ...videoUrls];
+        }
+    }
+    await (0, temp_files_1.deleteTempTestFile)();
+    await (0, pw_pause_1.revertToOriginalPwCode)(repoDir);
+    const codegenResult = fileServer.getCodegenSources();
+    await fileServer.stop();
+    const finalCode = await (0, display_1.displayResultsAndConfirm)(name, codegenResult);
+    await (0, request_1.sendToDashboardAsRequest)({
+        testName: name,
+        codegenResult: finalCode,
+        attachments,
+    });
+}

package/dist/recorder/request.d.ts

@@ -0,0 +1,6 @@
+export declare function sendToDashboardAsRequest({ testName, codegenResult, attachments, }: {
+    testName: string;
+    codegenResult: string;
+    attachments: string[];
+}): Promise<void>;
+//# sourceMappingURL=request.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"request.d.ts","sourceRoot":"","sources":["../../src/recorder/request.ts"],"names":[],"mappings":"AAeA,wBAAsB,wBAAwB,CAAC,EAC7C,QAAQ,EACR,aAAa,EACb,WAAW,GACZ,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,EAAE,CAAC;CACvB,iBAKA"}

package/dist/recorder/request.js

@@ -0,0 +1,55 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sendToDashboardAsRequest = sendToDashboardAsRequest;
+const DASHBOARD_DOMAIN = process.env.DASHBOARD_DOMAIN || "https://dash.empirical.run";
+const title = (name) => `Add a test: ${name}`;
+function description(codegenResult, attachments) {
+    return [
+        `To add this test, refer to the following code that was captured using playwright codegen. Make modifications that you need to make to convert this into a test case that sits well with other tests in this repository.`,
+        `Codegen result:`,
+        codegenResult,
+        `Other attachments:`,
+        ...attachments,
+    ].join("\n\n");
+}
+async function sendToDashboardAsRequest({ testName, codegenResult, attachments, }) {
+    return createRequest({
+        title: title(testName),
+        description: description(codegenResult, attachments),
+    });
+}
+async function createRequest({ title, description, }) {
+    if (!DASHBOARD_DOMAIN) {
+        console.warn("DASHBOARD_DOMAIN not set, skipping request creation");
+        return;
+    }
+    if (!process.env.EMPIRICALRUN_API_KEY) {
+        console.warn("EMPIRICALRUN_API_KEY not set, skipping request creation");
+        return;
+    }
+    try {
+        const source = "cli";
+        const sourceIdentifier = "random-string";
+        const response = await fetch(`${DASHBOARD_DOMAIN}/api/requests`, {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${process.env.EMPIRICALRUN_API_KEY}`,
+            },
+            body: JSON.stringify({
+                source,
+                source_identifier: sourceIdentifier,
+                title,
+                description,
+            }),
+        });
+        if (!response.ok) {
+            throw new Error(`Failed to create request: ${response.statusText}`);
+        }
+        const data = await response.json();
+        console.log("Request created successfully:", data);
+    }
+    catch (error) {
+        console.error("Failed to create request:", error);
+    }
+}

package/dist/recorder/temp-files.d.ts

@@ -0,0 +1,3 @@
+export declare function createTempTestFile(): Promise<void>;
+export declare function deleteTempTestFile(): Promise<void>;
+//# sourceMappingURL=temp-files.d.ts.map