@empiricalrun/playwright-utils 0.22.7 → 0.23.0

package/bundled/test-gen/index.js

@@ -0,0 +1,55 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createTest = void 0;
+const llm_1 = require("@empiricalrun/llm");
+const run_1 = require("./agent/master/run");
+const scenarios_1 = require("./bin/utils/scenarios");
+const client_1 = __importDefault(require("./file/client"));
+const reporter_1 = require("./reporter");
+const session_1 = require("./session");
+const pw_test_1 = require("./utils/pw-test");
+const flushEvents = async () => {
+    await (0, llm_1.flushAllTraces)();
+};
+process.on("beforeExit", async () => await flushEvents());
+process.on("exit", async () => await flushEvents());
+process.on("SIGINT", async () => await flushEvents());
+process.on("SIGTERM", async () => await flushEvents());
+async function createTest(task, page, scope) {
+    const testConfigArg = process.env.TEST_GEN_TOKEN;
+    const testGenConfig = (0, scenarios_1.loadTestConfigs)(testConfigArg);
+    (0, reporter_1.setReporterConfig)({
+        projectRepoName: testGenConfig.options?.metadata.projectRepoName,
+        testSessionId: testGenConfig.options?.metadata.testSessionId,
+        generationId: testGenConfig.options?.metadata.generationId,
+    });
+    (0, session_1.setSessionDetails)({
+        sessionId: testGenConfig.options?.metadata.testSessionId,
+        generationId: testGenConfig.options?.metadata.generationId,
+        testCaseId: testGenConfig.testCase.id,
+        projectRepoName: testGenConfig.options?.metadata.projectRepoName,
+    });
+    const fileService = new client_1.default();
+    const { testCase, specPath } = testGenConfig;
+    const { code, importPaths } = await (0, run_1.createTestUsingMasterAgent)({
+        testCase,
+        specPath,
+        page,
+        task,
+        options: {
+            ...testGenConfig.options,
+        },
+        scopeVars: scope,
+    });
+    await fileService.updateTest({
+        task,
+        generatedCode: code,
+        importPaths,
+    });
+    // skip the rest of the test once generation is over
+    await (0, pw_test_1.skipTest)();
+}
+exports.createTest = createTest;

package/bundled/test-gen/package.json

@@ -0,0 +1,106 @@
+{
+  "name": "@empiricalrun/test-gen",
+  "version": "0.46.11",
+  "publishConfig": {
+    "registry": "https://registry.npmjs.org/",
+    "access": "public"
+  },
+  "bin": {
+    "@empiricalrun/test-gen": "dist/bin/index.js"
+  },
+  "main": "dist/index.js",
+  "exports": {
+    "./agent/infer-agent": {
+      "types": "./dist/agent/infer-agent/index.d.ts",
+      "default": "./dist/agent/infer-agent/index.js"
+    },
+    "./agent/master/run": {
+      "types": "./dist/agent/master/run.d.ts",
+      "default": "./dist/agent/master/run.js"
+    },
+    "./agent/master/planner": {
+      "types": "./dist/agent/master/planner.d.ts",
+      "default": "./dist/agent/master/planner.js"
+    },
+    "./agent/enrich-prompt": {
+      "types": "./dist/agent/enrich-prompt/index.d.ts",
+      "default": "./dist/agent/enrich-prompt/index.js"
+    },
+    "./types": {
+      "types": "./dist/types/index.d.ts",
+      "default": "./dist/types/index.js"
+    },
+    "./utils": {
+      "types": "./dist/utils/index.d.ts",
+      "default": "./dist/utils/index.js"
+    },
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    }
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/empirical-run/empirical.git"
+  },
+  "author": "Empirical Team <hey@empirical.run>",
+  "scripts": {
+    "dev": "tspc --build --watch",
+    "build": "tspc --build && cp -r src/browser-injected-scripts dist",
+    "clean": "tspc --build --clean",
+    "lint": "eslint .",
+    "test": "vitest run",
+    "test-browser": "pnpm build && RUNNING_BROWSER_TESTS_FOR_TEST_GEN=1 npx playwright test",
+    "test:watch": "vitest",
+    "test:watch-files": "vitest $0 --watch"
+  },
+  "dependencies": {
+    "@actions/core": "^1.10.1",
+    "@babel/parser": "^7.26.3",
+    "@empiricalrun/llm": "workspace:^",
+    "@empiricalrun/r2-uploader": "workspace:^",
+    "@empiricalrun/reporter": "workspace:^",
+    "@types/sanitize-html": "^2.11.0",
+    "commander": "^12.1.0",
+    "detect-port": "^1.6.1",
+    "dotenv": "^16.4.5",
+    "eslint": "^8.57.0",
+    "express": "^4.19.2",
+    "fs-extra": "^11.2.0",
+    "google-auth-library": "^9.10.0",
+    "google-spreadsheet": "^4.1.2",
+    "ignore": "^5.3.1",
+    "inquirer": "^12.4.2",
+    "jsdom": "^26.0.0",
+    "lodash.isequal": "^4.5.0",
+    "md5": "^2.3.0",
+    "mime": "^4.0.4",
+    "minimatch": "^10.0.1",
+    "nanoid": "^5.0.7",
+    "openai": "4.67.0",
+    "picocolors": "^1.0.1",
+    "prettier": "^3.2.5",
+    "remove-markdown": "^0.5.5",
+    "sanitize-html": "^2.13.0",
+    "slugify": "^1.6.6",
+    "ts-morph": "^23.0.0",
+    "tsx": "^4.16.2",
+    "typescript": "^5.3.3"
+  },
+  "devDependencies": {
+    "@empiricalrun/shared-types": "workspace:*",
+    "@playwright/test": "1.47.1",
+    "@types/detect-port": "^1.3.5",
+    "@types/express": "^4.17.21",
+    "@types/fs-extra": "^11.0.4",
+    "@types/http-server": "^0.12.4",
+    "@types/js-levenshtein": "^1.1.3",
+    "@types/jsdom": "^21.1.7",
+    "@types/lodash.isequal": "^4.5.8",
+    "@types/md5": "^2.3.5",
+    "http-server": "^14.1.1",
+    "js-levenshtein": "^1.1.6",
+    "playwright": "1.47.1",
+    "ts-patch": "^3.3.0"
+  }
+}

package/bundled/test-gen/page/index.d.ts

@@ -0,0 +1,11 @@
+import type { Page } from "playwright";
+export declare class TestGenPage {
+    pwPageInstance: Page;
+    name: string;
+    constructor(pwPageInstance: Page, name: string);
+    updatePage({ page, name }: {
+        page: Page;
+        name: string;
+    }): void;
+}
+//# sourceMappingURL=index.d.ts.map

package/bundled/test-gen/page/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/page/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAEvC,qBAAa,WAAW;IAEb,cAAc,EAAE,IAAI;IACpB,IAAI,EAAE,MAAM;gBADZ,cAAc,EAAE,IAAI,EACpB,IAAI,EAAE,MAAM;IAGrB,UAAU,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE;CAIxD"}

package/bundled/test-gen/page/index.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TestGenPage = void 0;
+class TestGenPage {
+    pwPageInstance;
+    name;
+    constructor(pwPageInstance, name) {
+        this.pwPageInstance = pwPageInstance;
+        this.name = name;
+    }
+    updatePage({ page, name }) {
+        this.pwPageInstance = page;
+        this.name = name;
+    }
+}
+exports.TestGenPage = TestGenPage;

package/bundled/test-gen/prompts/lib/ts-transformer.d.ts

@@ -0,0 +1,4 @@
+import * as ts from "typescript";
+declare function createHandlebarsTransformer(): ts.TransformerFactory<ts.SourceFile>;
+export default createHandlebarsTransformer;
+//# sourceMappingURL=ts-transformer.d.ts.map

package/bundled/test-gen/prompts/lib/ts-transformer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ts-transformer.d.ts","sourceRoot":"","sources":["../../../src/prompts/lib/ts-transformer.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,MAAM,YAAY,CAAC;AAEjC,iBAAS,2BAA2B,IAAI,EAAE,CAAC,kBAAkB,CAAC,EAAE,CAAC,UAAU,CAAC,CAsG3E;AAED,eAAe,2BAA2B,CAAC"}

package/bundled/test-gen/prompts/lib/ts-transformer.js

@@ -0,0 +1,92 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const ts = __importStar(require("typescript"));
+function createHandlebarsTransformer() {
+    return (context) => {
+        // Map to store the mapping between original import identifiers and generated variable names
+        const importToVarMap = new Map();
+        const visitor = (node) => {
+            // Handle references to the imported identifiers
+            if (ts.isIdentifier(node) &&
+                importToVarMap.has(node.text) &&
+                !ts.isPropertyAccessExpression(node.parent)) {
+                return ts.factory.createIdentifier(importToVarMap.get(node.text));
+            }
+            return ts.visitEachChild(node, visitor, context);
+        };
+        return (sourceFile) => {
+            // Reset the import map for each source file
+            importToVarMap.clear();
+            // Keep track of generated variable names to avoid duplicates
+            let templateCounter = 0;
+            // Create a new statements array by iterating over the existing ones
+            const newStatements = sourceFile.statements.flatMap((stmt) => {
+                // Check if the statement is an import declaration for a .handlebars file
+                if (ts.isImportDeclaration(stmt) &&
+                    ts.isStringLiteral(stmt.moduleSpecifier) &&
+                    stmt.moduleSpecifier.text.endsWith(".handlebars")) {
+                    // Ensure we have a default import (an identifier)
+                    if (stmt.importClause &&
+                        stmt.importClause.name &&
+                        ts.isIdentifier(stmt.importClause.name)) {
+                        const localIdentifier = stmt.importClause.name;
+                        // Generate a unique variable name
+                        const generatedVarName = `promptTemplate_${templateCounter++}`;
+                        // Store the mapping between import identifier and generated variable name
+                        importToVarMap.set(localIdentifier.text, generatedVarName);
+                        // Resolve the path to the handlebars file
+                        const containingDir = path.dirname(sourceFile.fileName);
+                        const templatePath = path.resolve(containingDir, stmt.moduleSpecifier.text);
+                        try {
+                            // Read the handlebars file content
+                            const content = fs.readFileSync(templatePath, "utf-8");
+                            // Create a variable declaration that binds the generated name to the content
+                            const variableStmt = ts.factory.createVariableStatement(undefined, ts.factory.createVariableDeclarationList([
+                                ts.factory.createVariableDeclaration(ts.factory.createIdentifier(generatedVarName), undefined, undefined, ts.factory.createStringLiteral(content)),
+                            ], ts.NodeFlags.Const));
+                            // Replace the import declaration with the variable statement
+                            return [variableStmt];
+                        }
+                        catch (error) {
+                            console.error(`Error reading handlebars file ${templatePath}:`, error);
+                            // If reading the file fails, keep the original import
+                            return [stmt];
+                        }
+                    }
+                }
+                // For all other statements, keep them as is
+                return [stmt];
+            });
+            // Create updated source file with the new statements
+            const updatedSourceFile = ts.factory.updateSourceFile(sourceFile, newStatements);
+            // Visit the entire source file to transform references
+            return ts.visitNode(updatedSourceFile, visitor);
+        };
+    };
+}
+exports.default = createHandlebarsTransformer;

package/bundled/test-gen/reporter/index.d.ts

@@ -0,0 +1,33 @@
+import { Reporter } from "@empiricalrun/reporter";
+type ReporterConfigType = {
+    testSessionId: number;
+    generationId: number;
+    projectRepoName: string;
+};
+export declare function getReporter(): Reporter | undefined;
+/**
+ * function will upload videos and json summary of test results to r2 and report them to reporter.
+ * method won't throw error if it fails to report
+ * @param {{
+ *   projectRepoName: string;
+ * }} {
+ *   projectRepoName
+ * }
+ * @returns Promise<void> returns void
+ */
+export declare function setReporterConfig(config: ReporterConfigType): void;
+export declare class TestGenUpdatesReporter {
+    private repoDir;
+    constructor();
+    sendGenTrace(trace: string): Promise<void>;
+    reportGenAssets({ projectRepoName, testName, }: {
+        projectRepoName: string;
+        testName: string;
+    }): Promise<void>;
+    sendCurrentView(buffer: Buffer): Promise<void>;
+    sendMessage(message: string): Promise<void>;
+    sendLogUrl(message: string): Promise<void>;
+    sendAgentTraceUrl(message: string): Promise<void>;
+}
+export {};
+//# sourceMappingURL=index.d.ts.map

package/bundled/test-gen/reporter/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/reporter/index.ts"],"names":[],"mappings":"AACA,OAAO,EAA4B,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAa5E,KAAK,kBAAkB,GAAG;IACxB,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,MAAM,CAAC;CACzB,CAAC;AAKF,wBAAgB,WAAW,IAAI,QAAQ,GAAG,SAAS,CAUlD;AAED;;;;;;;;;GASG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,kBAAkB,GAAG,IAAI,CAGlE;AAED,qBAAa,sBAAsB;IACjC,OAAO,CAAC,OAAO,CAAS;;IAKlB,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAK1C,eAAe,CAAC,EACpB,eAAe,EACf,QAAQ,GACT,EAAE;QACD,eAAe,EAAE,MAAM,CAAC;QACxB,QAAQ,EAAE,MAAM,CAAC;KAClB;IAiDK,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA8C9C,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAY3C,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAY1C,iBAAiB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAWxD"}

package/bundled/test-gen/reporter/index.js

@@ -0,0 +1,161 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TestGenUpdatesReporter = exports.setReporterConfig = exports.getReporter = void 0;
+const r2_uploader_1 = require("@empiricalrun/r2-uploader");
+const reporter_1 = require("@empiricalrun/reporter");
+const fs_extra_1 = __importDefault(require("fs-extra"));
+const path_1 = __importDefault(require("path"));
+const logger_1 = require("../bin/logger");
+const uploader_1 = require("../uploader");
+let reporterInstance = undefined;
+let reporterConfig = undefined;
+function getReporter() {
+    // if no config is set, return undefined
+    if (!reporterConfig) {
+        console.warn("initialising reporter without any config");
+    }
+    // initialise once config is set
+    if (!reporterInstance && reporterConfig) {
+        reporterInstance = new reporter_1.Reporter(reporterConfig);
+    }
+    return reporterInstance;
+}
+exports.getReporter = getReporter;
+/**
+ * function will upload videos and json summary of test results to r2 and report them to reporter.
+ * method won't throw error if it fails to report
+ * @param {{
+ *   projectRepoName: string;
+ * }} {
+ *   projectRepoName
+ * }
+ * @returns Promise<void> returns void
+ */
+function setReporterConfig(config) {
+    console.info("initialised reporter config");
+    reporterConfig = config;
+}
+exports.setReporterConfig = setReporterConfig;
+class TestGenUpdatesReporter {
+    repoDir;
+    constructor() {
+        this.repoDir = process.cwd();
+    }
+    async sendGenTrace(trace) {
+        console.log("trace", trace);
+        // upload trace to r2 and report it to reporter
+    }
+    async reportGenAssets({ projectRepoName, testName, }) {
+        const logger = new logger_1.CustomLogger();
+        try {
+            if (!(0, uploader_1.checkIfResultsUploadAllowed)()) {
+                logger.log("Skipped uploading generated test video");
+                return;
+            }
+            console.log("Uploading assets for the test genearation");
+            const { videoUrls, traceFiles } = await (0, uploader_1.uploadTestResultsUsingPrjRepo)({
+                projectRepoName,
+                testName,
+                repoDir: this.repoDir,
+            });
+            const reporter = getReporter();
+            const message = {
+                type: "video",
+                videoUrls,
+            };
+            await Promise.allSettled([
+                ...(videoUrls.length
+                    ? [
+                        reporter?.report(new reporter_1.ProcessLogMessageBuilder({
+                            type: "video",
+                            message: JSON.stringify(message),
+                        })),
+                    ]
+                    : []),
+                ...(traceFiles.length
+                    ? [
+                        reporter?.report(new reporter_1.ProcessLogMessageBuilder({
+                            type: "pw-trace",
+                            message: JSON.stringify({
+                                type: "trace",
+                                traceFiles,
+                            }),
+                        })),
+                    ]
+                    : []),
+            ]);
+        }
+        catch (err) {
+            logger.error("Failed to report test results");
+            console.error(err);
+        }
+    }
+    async sendCurrentView(buffer) {
+        const logger = new logger_1.CustomLogger();
+        if (!(0, uploader_1.checkIfResultsUploadAllowed)()) {
+            logger.log("Skipped uploading current view screenshot");
+            return;
+        }
+        if (!reporterConfig?.projectRepoName) {
+            logger.log("Reporter config is not set skipping screenshot uploads.");
+            return;
+        }
+        try {
+            // upload current screenshot to r2 and report it to reporter
+            if (!fs_extra_1.default.existsSync(path_1.default.join(this.repoDir, "gen-assets"))) {
+                await fs_extra_1.default.mkdir(this.repoDir, "gen-assets");
+            }
+            await fs_extra_1.default.writeFile(path_1.default.join(this.repoDir, "gen-assets", `current-view-${Date.now()}.png`), buffer);
+            const uploadDir = (0, uploader_1.getUploadPathForRun)(reporterConfig?.projectRepoName);
+            const files = await (0, r2_uploader_1.uploadDirectory)({
+                sourceDir: path_1.default.join(this.repoDir, "gen-assets"),
+                destinationDir: uploadDir,
+                uploadBucket: uploader_1.UPLOAD_BUCKET,
+            });
+            const filePath = Object.keys(files)[0];
+            const relativeFilePath = filePath.replace(path_1.default.join(this.repoDir, "gen-assets"), "");
+            const url = `${uploader_1.UPLOAD_DOMAIN}/${uploadDir}${relativeFilePath}`;
+            await getReporter()?.report(new reporter_1.ProcessLogMessageBuilder({
+                type: "current-snapshot",
+                message: JSON.stringify({ type: "current-view", url }),
+            }));
+            await fs_extra_1.default.rmdir(path_1.default.join(this.repoDir, "gen-assets"), {
+                recursive: true,
+            });
+        }
+        catch (e) {
+            console.warn("Failed to upload current view screenshot", e);
+        }
+    }
+    async sendMessage(message) {
+        const reporter = getReporter();
+        if (reporter) {
+            await reporter.report(new reporter_1.ProcessLogMessageBuilder({
+                type: "message",
+                message,
+            }));
+        }
+    }
+    async sendLogUrl(message) {
+        const reporter = getReporter();
+        if (reporter) {
+            await reporter.report(new reporter_1.ProcessLogMessageBuilder({
+                type: "log-url",
+                message,
+            }));
+        }
+    }
+    async sendAgentTraceUrl(message) {
+        const reporter = getReporter();
+        if (reporter) {
+            await reporter.report(new reporter_1.ProcessLogMessageBuilder({
+                type: "agent-trace",
+                message,
+            }));
+        }
+    }
+}
+exports.TestGenUpdatesReporter = TestGenUpdatesReporter;

package/bundled/test-gen/session/index.d.ts

@@ -0,0 +1,20 @@
+declare function getSessionDetails(): {
+    generationId: number | undefined;
+    sessionId: number | undefined;
+    testUrl: string | undefined;
+    version: string;
+};
+export declare function setSessionDetails({ sessionId, generationId, testCaseId, projectRepoName, }: {
+    sessionId: number;
+    generationId: number;
+    testCaseId: number;
+    projectRepoName: string;
+}): void;
+export declare function shouldStopSession(): Promise<boolean>;
+export declare function getSessionState(): Promise<"started" | "completed" | "request_complete">;
+export declare function endSession(): Promise<void>;
+export { getSessionDetails };
+export declare function updateSessionStatus(sessionId: number, payload: {
+    status: string;
+}): Promise<void>;
+//# sourceMappingURL=index.d.ts.map

package/bundled/test-gen/session/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/session/index.ts"],"names":[],"mappings":"AAmBA,iBAAS,iBAAiB;;;;;EAOzB;AAED,wBAAgB,iBAAiB,CAAC,EAChC,SAAS,EACT,YAAY,EACZ,UAAU,EACV,eAAe,GAChB,EAAE;IACD,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,eAAe,EAAE,MAAM,CAAC;CACzB,QAIA;AAED,wBAAsB,iBAAiB,qBAItC;AAED,wBAAsB,eAAe,0DA0BpC;AAED,wBAAsB,UAAU,kBAyB/B;AAED,OAAO,EAAE,iBAAiB,EAAE,CAAC;AAE7B,wBAAsB,mBAAmB,CACvC,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE;IACP,MAAM,EAAE,MAAM,CAAC;CAChB,iBAmBF"}

package/bundled/test-gen/session/index.js

@@ -0,0 +1,105 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.updateSessionStatus = exports.getSessionDetails = exports.endSession = exports.getSessionState = exports.shouldStopSession = exports.setSessionDetails = void 0;
+const package_json_1 = __importDefault(require("../../package.json"));
+const env_1 = require("../utils/env");
+const sessionDetails = {
+    sessionId: undefined,
+    version: package_json_1.default.version,
+    generationId: undefined,
+    testUrl: undefined,
+};
+const DASHBOARD_DOMAIN = process.env.DASHBOARD_DOMAIN ||
+    ((0, env_1.isRunningOnCloud)() ? "https://dash.empirical.run" : "");
+function getSessionDetails() {
+    return {
+        generationId: sessionDetails.generationId,
+        sessionId: sessionDetails.sessionId,
+        testUrl: sessionDetails.testUrl,
+        version: package_json_1.default.version,
+    };
+}
+exports.getSessionDetails = getSessionDetails;
+function setSessionDetails({ sessionId, generationId, testCaseId, projectRepoName, }) {
+    sessionDetails.sessionId = sessionId;
+    sessionDetails.generationId = generationId;
+    sessionDetails.testUrl = `${process.env.DASHBOARD_DOMAIN ?? ""}/${projectRepoName}/test-cases/${testCaseId}`;
+}
+exports.setSessionDetails = setSessionDetails;
+async function shouldStopSession() {
+    const sessionState = await getSessionState();
+    const shouldStopSession = sessionState !== "started";
+    return shouldStopSession;
+}
+exports.shouldStopSession = shouldStopSession;
+async function getSessionState() {
+    // if there is no dashboard domain to connect to or session details missing, then continue the generation session.
+    if (!DASHBOARD_DOMAIN ||
+        !sessionDetails.sessionId ||
+        !sessionDetails.generationId) {
+        return "started";
+    }
+    const apiPath = `${DASHBOARD_DOMAIN}/api/sessions/${sessionDetails.sessionId}/generations/${sessionDetails.generationId}/state`;
+    const response = await fetch(apiPath, {
+        method: "GET",
+        headers: {
+            "Content-Type": "application/json",
+            Authorization: "weQPMWKT",
+        },
+    });
+    const generationStateData = (await response.json());
+    // TODO: fix this heirarchy
+    return generationStateData.data.state.state;
+}
+exports.getSessionState = getSessionState;
+async function endSession() {
+    if (!DASHBOARD_DOMAIN) {
+        return;
+    }
+    if (!sessionDetails.sessionId || !sessionDetails.generationId) {
+        console.warn("Session details missing, skipping session end");
+        return;
+    }
+    const apiPath = `${DASHBOARD_DOMAIN}/api/sessions/${sessionDetails.sessionId}/generations/${sessionDetails.generationId}/state`;
+    try {
+        await fetch(apiPath, {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json",
+                Authorization: "weQPMWKT",
+            },
+            body: JSON.stringify({
+                state: {
+                    state: "completed",
+                },
+            }),
+        });
+    }
+    catch (error) {
+        console.error("Failed to end session:", error);
+    }
+}
+exports.endSession = endSession;
+async function updateSessionStatus(sessionId, payload) {
+    if (!DASHBOARD_DOMAIN) {
+        return;
+    }
+    try {
+        const body = JSON.stringify(payload);
+        await fetch(`${DASHBOARD_DOMAIN}/api/sessions/${sessionId}`, {
+            method: "PATCH",
+            headers: {
+                "Content-Type": "application/json",
+                Authorization: "weQPMWKT",
+            },
+            body,
+        });
+    }
+    catch (error) {
+        console.error("Failed to update session status:", error);
+    }
+}
+exports.updateSessionStatus = updateSessionStatus;

package/bundled/test-gen/test-build/index.d.ts

@@ -0,0 +1,10 @@
+import { Build } from "@empiricalrun/shared-types";
+/**
+ * method to download the build from the URL provided in the build object
+ * this is only used in cases like chrome extension as of yet.
+ * @export
+ * @param {Build} build
+ * @return {*}  {Promise<void>}
+ */
+export declare function downloadBuild(build: Build): Promise<void>;
+//# sourceMappingURL=index.d.ts.map

package/bundled/test-gen/test-build/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/test-build/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,4BAA4B,CAAC;AAUnD;;;;;;GAMG;AACH,wBAAsB,aAAa,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,CAY/D"}