@empiricalrun/test-gen 0.67.0 → 0.69.0

package/dist/bin/index.js

@@ -17,26 +17,18 @@ const enrich_prompt_1 = require("../agent/enrich-prompt");
 const infer_agent_1 = require("../agent/infer-agent");
 const run_3 = require("../agent/planner/run");
 const auth_1 = require("../auth");
+const api_client_1 = require("../auth/api-client");
 const recorder_1 = require("../recorder");
 const reporter_1 = require("../reporter");
 const session_1 = require("../session");
 const test_build_1 = require("../test-build");
 const logger_1 = require("./logger");
+const setup_1 = require("./setup");
 const utils_2 = require("./utils");
 const scenarios_1 = require("./utils/scenarios");
 dotenv_1.default.config({
     path: [".env.local", ".env"],
 });
-const flushEvents = async () => {
-    await (0, llm_1.flushAllTraces)();
-};
-function setupProcessListeners(cleanup) {
-    const events = ["beforeExit", "exit", "SIGINT", "SIGTERM"];
-    events.forEach((event) => process.once(event, cleanup));
-    return () => {
-        events.forEach((event) => process.removeListener(event, cleanup));
-    };
-}
 async function runChatAgent({ modelInput, chatSessionId, useDiskForChatState, initialPromptPath, }) {
     if (modelInput && !utils_2.ARGS_TO_MODEL_MAP[modelInput]) {
         throw new Error(`Invalid chat model: ${modelInput}`);
@@ -201,13 +193,11 @@ async function runAgentsWorkflow(testGenConfig, testGenToken) {
     return agent;
 }
 async function main() {
-    const removeListeners = setupProcessListeners(flushEvents);
     await (0, utils_2.printBanner)();
     const program = new commander_1.Command();
-    // Add authentication commands
     program
         .command("login")
-        .description("Authenticate with your Empirical Run account")
+        .description("Authenticate with your Empirical account")
         .action(async () => {
         console.log("🔐 Starting authentication...\n");
         try {
@@ -232,7 +222,7 @@ async function main() {
     });
     program
         .command("logout")
-        .description("Sign out of your Empirical Run account")
+        .description("Sign out of your Empirical account")
         .action(async () => {
         try {
             await (0, auth_1.logout)();
@@ -265,118 +255,124 @@ async function main() {
             console.error("❌ Error checking auth status:", error.message);
             process.exit(1);
         }
-        removeListeners();
-        await (0, llm_1.flushAllTraces)();
-        await (0, logger_1.waitForLogsToFlush)();
         process.exit(0);
     });
     program
-        .command("auth-test")
-        .description("Test authentication by making an API call")
+        .command("record")
+        .description("Record a new test case")
+        .option("--name <string>", "Name of the test case")
+        .action(async (opts) => {
+        const options = await (0, utils_2.validateAndCompleteCliOptions)(opts, ["name"]);
+        await (0, recorder_1.runRecorder)({ name: options.name });
+        process.exit(0);
+    });
+    program
+        .command("repos")
+        .description("List your projects and repositories")
         .action(async () => {
         try {
-            console.log("🔍 Testing authentication with API call...\n");
-            const data = await (0, auth_1.authenticatedFetch)("/requests");
-            console.log("✅ API call successful!");
-            console.log("📄 Response:", JSON.stringify(data, null, 2));
+            const response = await api_client_1.apiClient.request("/api/projects");
+            if (!response.ok) {
+                console.error("❌ Failed to fetch projects:", response.statusText);
+                process.exit(1);
+            }
+            const result = await response.json();
+            result.data.projects.forEach((project) => {
+                console.log(`  ${project.repo_name}`);
+            });
         }
         catch (error) {
-            console.error("❌ API call failed:", error.message);
-            if (error.message.includes("Authentication required")) {
-                console.log('\n💡 Tip: Run "npx @empiricalrun/test-gen login" to authenticate');
-            }
+            console.error("❌ Error fetching projects:", error.message);
             process.exit(1);
         }
-        removeListeners();
-        await (0, llm_1.flushAllTraces)();
-        await (0, logger_1.waitForLogsToFlush)();
         process.exit(0);
     });
     program
-        .option("--token <token>", "Test generation token")
-        .option("--prompt <prompt>", "Prompt for the chat agent")
-        .option("--name <test-name>", "Name of the test case")
-        .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)")
+        .command("chat-agent")
+        .description("Run the chat agent")
+        .option("--chat-model <model>", "LLM to use (claude-3-7, claude-4 or gemini-2.5)")
         .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)")
         .option("--initial-prompt <path>", "Path to an initial prompt file (e.g. prompt.md)")
-        .option("--with-retry", "Use the retry strategy")
-        .parse(process.argv);
-    // Check if a command was executed (auth commands exit early)
-    const options = program.opts();
-    // Only proceed with main workflow if no auth command was executed
-    const completedOptions = await (0, utils_2.validateAndCompleteCliOptions)(options);
-    const testGenConfig = completedOptions.token
-        ? (0, scenarios_1.loadTestConfigs)(completedOptions.token)
-        : (0, scenarios_1.buildTestConfigFromOptions)(completedOptions);
-    const testGenToken = completedOptions.token
-        ? completedOptions.token
-        : (0, scenarios_1.buildTokenFromOptions)(completedOptions);
-    (0, reporter_1.setReporterConfig)({
-        projectRepoName: testGenConfig.options?.metadata.projectRepoName,
-        testSessionId: testGenConfig.options?.metadata.testSessionId,
-        generationId: testGenConfig.options?.metadata.generationId,
+        .option("--chat-session-id <chat-session-id>", "Identifier for chat session (fetched from dash.empirical.run)")
+        .action(async (options) => {
+        await runChatAgent({
+            chatSessionId: options.chatSessionId,
+            modelInput: options.chatModel,
+            useDiskForChatState: options.useDiskForChatState,
+            initialPromptPath: options.initialPrompt,
+        });
+        process.exit(0);
     });
-    (0, session_1.setSessionDetails)({
-        testCaseId: testGenConfig.testCase.id,
-        sessionId: testGenConfig.options?.metadata.testSessionId,
-        generationId: testGenConfig.options?.metadata.generationId,
-        projectRepoName: testGenConfig.options?.metadata.projectRepoName,
+    program
+        .command("setup")
+        .description("Clone a repository from your projects")
+        .option("--repo-name <string>", "Name of the repository to clone")
+        .action(async (opts) => {
+        const options = await (0, utils_2.validateAndCompleteCliOptions)(opts, ["repoName"]);
+        await (0, setup_1.runSetup)({ repoName: options.repoName });
+        process.exit(0);
     });
-    if (testGenConfig.build?.url) {
-        // Download the build if repo has a download script
-        await (0, test_build_1.downloadBuild)({
-            buildUrl: testGenConfig.build.url,
-            repoPath: process.cwd(),
-            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,
-            modelInput: completedOptions.chatModel,
-            useDiskForChatState: completedOptions.useDiskForChatState,
-            initialPromptPath: completedOptions.initialPrompt,
+    program
+        .command("legacy")
+        .description("Run the legacy workflows")
+        .option("-t, --token <token>", "Test generation token")
+        .action(async (opts) => {
+        const options = await (0, utils_2.validateAndCompleteCliOptions)(opts);
+        const testGenConfig = options.token
+            ? (0, scenarios_1.loadTestConfigs)(options.token)
+            : (0, scenarios_1.buildTestConfigFromOptions)(options);
+        const testGenToken = options.token
+            ? options.token
+            : (0, scenarios_1.buildTokenFromOptions)(options);
+        (0, reporter_1.setReporterConfig)({
+            projectRepoName: testGenConfig.options?.metadata.projectRepoName,
+            testSessionId: testGenConfig.options?.metadata.testSessionId,
+            generationId: testGenConfig.options?.metadata.generationId,
         });
-        return;
-    }
-    let agentUsed;
-    let testGenFailed = false;
-    try {
-        agentUsed = await runAgentsWorkflow(testGenConfig, testGenToken);
-    }
-    catch (e) {
-        testGenFailed = true;
-        new logger_1.CustomLogger().error(`Failed to generate test for the scenario. ${process.env.LOG_URL ? `[view log](${process.env.LOG_URL})` : ""}`, e?.message, e?.stack);
-    }
-    if (agentUsed &&
-        agentUsed !== "code" &&
-        agentUsed !== "plan" &&
-        testGenConfig.testCase.name &&
-        testGenConfig.options) {
-        await new reporter_1.TestGenUpdatesReporter().reportGenAssets({
-            projectRepoName: testGenConfig.options.metadata.projectRepoName,
-            testName: testGenConfig.testCase.name,
+        (0, session_1.setSessionDetails)({
+            testCaseId: testGenConfig.testCase.id,
+            sessionId: testGenConfig.options?.metadata.testSessionId,
+            generationId: testGenConfig.options?.metadata.generationId,
+            projectRepoName: testGenConfig.options?.metadata.projectRepoName,
         });
-    }
-    removeListeners();
-    await (0, llm_1.flushAllTraces)();
-    await (0, logger_1.waitForLogsToFlush)();
-    await (0, session_1.endSession)();
-    if (testGenFailed) {
-        process.exit(1);
-    }
-    else {
-        process.exit(0);
-    }
+        if (testGenConfig.build?.url) {
+            // Download the build if repo has a download script
+            await (0, test_build_1.downloadBuild)({
+                buildUrl: testGenConfig.build.url,
+                repoPath: process.cwd(),
+                apiKey: process.env.EMPIRICALRUN_API_KEY,
+            });
+        }
+        let agentUsed;
+        let testGenFailed = false;
+        try {
+            agentUsed = await runAgentsWorkflow(testGenConfig, testGenToken);
+        }
+        catch (e) {
+            testGenFailed = true;
+            new logger_1.CustomLogger().error(`Failed to generate test for the scenario. ${process.env.LOG_URL ? `[view log](${process.env.LOG_URL})` : ""}`, e?.message, e?.stack);
+        }
+        if (agentUsed &&
+            agentUsed !== "code" &&
+            agentUsed !== "plan" &&
+            testGenConfig.testCase.name &&
+            testGenConfig.options) {
+            await new reporter_1.TestGenUpdatesReporter().reportGenAssets({
+                projectRepoName: testGenConfig.options.metadata.projectRepoName,
+                testName: testGenConfig.testCase.name,
+            });
+        }
+        await (0, llm_1.flushAllTraces)();
+        await (0, logger_1.waitForLogsToFlush)();
+        await (0, session_1.endSession)();
+        if (testGenFailed) {
+            process.exit(1);
+        }
+        else {
+            process.exit(0);
+        }
+    });
+    program.parse(process.argv);
 }
 main().catch((error) => {
     console.error("Unhandled error in main function:", error);

package/dist/bin/setup.d.ts

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

package/dist/bin/setup.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup.d.ts","sourceRoot":"","sources":["../../src/bin/setup.ts"],"names":[],"mappings":"AA+EA,wBAAsB,QAAQ,CAAC,EAAE,QAAQ,EAAE,EAAE;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,iBA2BhE"}

package/dist/bin/setup.js

@@ -0,0 +1,86 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runSetup = runSetup;
+const child_process_1 = require("child_process");
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const util_1 = require("util");
+const api_client_1 = require("../auth/api-client");
+const execAsync = (0, util_1.promisify)(child_process_1.exec);
+async function cloneRepository(cloneUrl, repoName) {
+    const targetDir = path_1.default.join(process.cwd(), repoName);
+    if (fs_1.default.existsSync(targetDir)) {
+        console.error(`Directory '${repoName}' already exists in current location`);
+        process.exit(1);
+    }
+    console.log(`📥 Cloning repository to ./${repoName}...`);
+    const { stderr } = await execAsync(`git clone ${cloneUrl} ${repoName}`, {
+        cwd: process.cwd(),
+    });
+    if (stderr && !stderr.includes("Cloning into")) {
+        console.error(`Git clone error: ${stderr}`);
+        process.exit(1);
+    }
+    console.log(`✅ Repository cloned successfully to ./${repoName}`);
+    return targetDir;
+}
+async function installDependencies(targetDir) {
+    console.log(`📦 Installing dependencies...`);
+    try {
+        const { stderr: installStderr } = await execAsync(`npm install`, {
+            cwd: targetDir,
+        });
+        if (installStderr && !installStderr.toLowerCase().includes("npm warn")) {
+            console.error(`npm install error: ${installStderr}`);
+            process.exit(1);
+        }
+        console.log(`✅ Dependencies installed successfully`);
+    }
+    catch (installError) {
+        console.error(`❌ Failed to install dependencies: ${installError.message}`);
+        process.exit(1);
+    }
+}
+async function installPlaywrightBrowser(targetDir) {
+    console.log(`🎭 Installing Playwright browser...`);
+    try {
+        const { stderr: playwrightStderr } = await execAsync(`npx playwright install chromium --with-deps`, {
+            cwd: targetDir,
+        });
+        if (playwrightStderr && !playwrightStderr.includes("Installing")) {
+            console.error(`Playwright install error: ${playwrightStderr}`);
+            process.exit(1);
+        }
+        console.log(`✅ Playwright browser installed successfully`);
+    }
+    catch (playwrightError) {
+        console.error(`❌ Failed to install Playwright browser: ${playwrightError.message}`);
+        process.exit(1);
+    }
+}
+async function runSetup({ repoName }) {
+    try {
+        console.log(`🔍 Fetching details for repository: ${repoName}`);
+        const response = await api_client_1.apiClient.request(`/api/projects/clone?project_repo_name=${repoName}`);
+        if (!response.ok) {
+            if (response.status === 404) {
+                throw new Error(`Repository '${repoName}' not found in your projects`);
+            }
+            throw new Error(`Failed to fetch repository details: ${response.statusText}`);
+        }
+        const { data: repoDetails } = await response.json();
+        const { clone_url } = repoDetails;
+        console.log(`📂 Clone URL: ${clone_url}`);
+        const targetDir = await cloneRepository(clone_url, repoName);
+        await installDependencies(targetDir);
+        await installPlaywrightBrowser(targetDir);
+        console.log(`\n🎉 Setup complete! Repository is ready at ./${repoName}`);
+    }
+    catch (error) {
+        console.error(`❌ Setup failed: ${error.message}`);
+        process.exit(1);
+    }
+}

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

@@ -6,13 +6,8 @@ export interface CLIOptions {
     file?: string;
     prompt?: string;
     suites?: string;
-    useChat?: boolean;
-    useDiskForChatState?: boolean;
-    initialPrompt?: string;
-    chatSessionId?: string;
-    chatModel?: (typeof ARGS_TO_MODEL_MAP)[keyof typeof ARGS_TO_MODEL_MAP];
-    useRecorder?: boolean;
+    repoName?: string;
 }
-export declare function validateAndCompleteCliOptions(options: CLIOptions): Promise<CLIOptions>;
+export declare function validateAndCompleteCliOptions(options: CLIOptions, requiredFields?: string[]): Promise<CLIOptions>;
 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;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"}
+{"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;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAQD,wBAAsB,6BAA6B,CACjD,OAAO,EAAE,UAAU,EACnB,cAAc,GAAE,MAAM,EAA+B,GACpD,OAAO,CAAC,UAAU,CAAC,CAqDrB;AAeD,wBAAsB,WAAW,kBAgDhC"}

package/dist/bin/utils/index.js

@@ -22,20 +22,12 @@ exports.ARGS_TO_MODEL_MAP = {
     "o4-mini": "o4-mini-2025-04-16",
     "o4-mini-2025-04-16": "o4-mini-2025-04-16",
 };
-async function validateAndCompleteCliOptions(options) {
+async function validateAndCompleteCliOptions(options, requiredFields = ["name", "file", "prompt"]) {
     // For existing flow between dashboard <> test-gen (via ci-worker)
     const hasToken = !!options.token;
     if (hasToken) {
         return options;
     }
-    let requiredFields = ["name", "file", "prompt"];
-    if (options.useChat) {
-        // 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({

package/dist/file/server.js

@@ -45,7 +45,10 @@ class FileServiceServer {
         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);
+            // console.log(
+            //   "[FileServiceServer] Received codegen sources",
+            //   this.codegenSources,
+            // );
         });
         app.post("/agent-results", async (req, res) => {
             const { generatedCode, importPaths, result, usage } = req.body;

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;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"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAchD,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

@@ -13,6 +13,7 @@ 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"));
+const logger_1 = require("./logger");
 const reporter_1 = require("./reporter");
 const session_1 = require("./session");
 var test_build_1 = require("./test-build");
@@ -104,7 +105,7 @@ async function recordTest(pageRef) {
     if (!canUsePwPause) {
         return;
     }
-    console.log("[getCodegen] using PlaywrightPauseCodegen");
+    logger_1.logger.debug("[getCodegen] using PlaywrightPauseCodegen");
     const codegen = new for_recorder_1.PlaywrightPauseCodegenForRecorder(async (sources) => {
         await fileServiceClient.sendCodegenSources(sources);
     });

package/dist/logger.d.ts

@@ -0,0 +1,3 @@
+import Logger from "console-log-level";
+export declare const logger: Logger.Logger;
+//# sourceMappingURL=logger.d.ts.map

package/dist/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":"AAAA,OAAO,MAAyB,MAAM,mBAAmB,CAAC;AAE1D,eAAO,MAAM,MAAM,eAGjB,CAAC"}

package/dist/logger.js

@@ -0,0 +1,11 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.logger = void 0;
+const console_log_level_1 = __importDefault(require("console-log-level"));
+exports.logger = (0, console_log_level_1.default)({
+    prefix: "test-gen",
+    level: process.env.LOG_LEVEL || "info",
+});

package/dist/recorder/env-variables.d.ts

@@ -0,0 +1,2 @@
+export declare function fetchEnvironmentVariables(repoName: string): Promise<Record<string, string>>;
+//# sourceMappingURL=env-variables.d.ts.map

package/dist/recorder/env-variables.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"env-variables.d.ts","sourceRoot":"","sources":["../../src/recorder/env-variables.ts"],"names":[],"mappings":"AAEA,wBAAsB,yBAAyB,CAC7C,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAgCjC"}

package/dist/recorder/env-variables.js

@@ -0,0 +1,29 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fetchEnvironmentVariables = fetchEnvironmentVariables;
+const api_client_1 = require("../auth/api-client");
+async function fetchEnvironmentVariables(repoName) {
+    try {
+        const response = await api_client_1.apiClient.request(`/api/environment-variables?project_repo_name=${encodeURIComponent(repoName)}`, {
+            method: "GET",
+        });
+        if (!response.ok) {
+            const errorMessage = await response.text();
+            throw new Error(`Failed to fetch environment variables: ${errorMessage}`);
+        }
+        const data = await response.json();
+        if (!data.data) {
+            console.error("Failed to fetch environment variables:", data);
+            throw new Error("Failed to fetch environment variables");
+        }
+        const envVars = data.data.environment_variables.reduce((acc, envVar) => {
+            acc[envVar.name] = envVar.value;
+            return acc;
+        }, {});
+        return envVars;
+    }
+    catch (error) {
+        console.error("Failed to fetch environment variables:", error);
+        return {};
+    }
+}

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

@@ -1 +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"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/recorder/index.ts"],"names":[],"mappings":"AAkBA,wBAAsB,WAAW,CAAC,EAAE,IAAI,EAAE,EAAE;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,iBAyF3D"}

package/dist/recorder/index.js

@@ -6,66 +6,56 @@ 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 logger_1 = require("../logger");
 const display_1 = require("./display");
+const env_variables_1 = require("./env-variables");
 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();
+    let repoName = "";
+    let fileServer = null;
+    process.on("SIGINT", async () => {
+        try {
+            await (0, temp_files_1.deleteTempTestFile)();
+            await (0, pw_pause_1.revertToOriginalPwCode)(repoDir);
+            if (fileServer) {
+                await fileServer.stop();
+            }
+        }
+        catch (error) {
+            console.error("Error during cleanup:", error);
+        }
+        process.exit(0);
+    });
     try {
-        await (0, validation_1.validate)(repoDir);
+        repoName = await (0, validation_1.validatePackageJson)(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");
+        logger_1.logger.debug("[generateTestWithBrowserAgent] Preparing playwright for codegen");
         await (0, pw_pause_1.preparePlaywrightForCodegen)(repoDir);
     }
     catch (err) {
-        console.warn("[generateTestWithBrowserAgent] Error preparing playwright for codegen", err);
+        logger_1.logger.warn("[generateTestWithBrowserAgent] Error preparing playwright for codegen", err);
     }
-    const envVariables = await (0, chat_1.fetchEnvironmentVariables)();
+    const envVariables = await (0, env_variables_1.fetchEnvironmentVariables)(repoName);
     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({
+    fileServer = new server_1.FileServiceServer({
         port: availablePort,
         repoDir,
         updateFile: false,
@@ -84,10 +74,10 @@ async function runRecorder({ name }) {
             IPC_FILE_SERVICE_PORT: availablePort.toString(),
         },
     });
-    const videoPaths = extractVideoAttachments(repoDir);
+    const videoPaths = (0, upload_1.extractVideoAttachments)(repoDir);
     let attachments = [];
     if (videoPaths.length === 0) {
-        console.warn("No video attachments found for temp test");
+        logger_1.logger.warn("No video attachments found for temp test");
     }
     else {
         const videoUrls = await (0, upload_1.uploadVideosWithSpinner)(videoPaths, name);
@@ -101,6 +91,7 @@ async function runRecorder({ name }) {
     await fileServer.stop();
     const finalCode = await (0, display_1.displayResultsAndConfirm)(name, codegenResult);
     await (0, request_1.sendToDashboardAsRequest)({
+        repoName,
         testName: name,
         codegenResult: finalCode,
         attachments,

package/dist/recorder/request.d.ts

@@ -1,4 +1,5 @@
-export declare function sendToDashboardAsRequest({ testName, codegenResult, attachments, }: {
+export declare function sendToDashboardAsRequest({ repoName, testName, codegenResult, attachments, }: {
+    repoName: string;
     testName: string;
     codegenResult: string;
     attachments: string[];

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

@@ -1 +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"}
+{"version":3,"file":"request.d.ts","sourceRoot":"","sources":["../../src/recorder/request.ts"],"names":[],"mappings":"AAaA,wBAAsB,wBAAwB,CAAC,EAC7C,QAAQ,EACR,QAAQ,EACR,aAAa,EACb,WAAW,GACZ,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,EAAE,CAAC;CACvB,iBAMA"}