automation_model 1.0.436-dev → 1.0.436

package/lib/auto_page.js

@@ -1,7 +1,8 @@
 import { browserManager } from "./browser_manager.js";
 import { getContext } from "./init_browser.js";
-import fs from "fs";
+import fs, { existsSync } from "fs";
 import path from "path";
+import { _getDataFile, decrypt } from "./utils.js";
 let context = null;
 let reportFolder = "";
 const navigate = async (path = "") => {
@@ -15,9 +16,9 @@ const navigate = async (path = "") => {
     else {
         url = new URL(path, context.environment.baseUrl).href;
     }
-    await context.stable.goto(url);
+    await context.web.goto(url);
     context.navigate = true;
-    await context.stable.waitForPageLoad();
+    await context.web.waitForPageLoad();
 };
 const _findEmptyFolder = (folder) => {
     if (!folder) {
@@ -26,68 +27,97 @@ const _findEmptyFolder = (folder) => {
     if (!fs.existsSync(folder)) {
         fs.mkdirSync(folder);
     }
+    if (process.env.REPORT_FOLDER) {
+        return process.env.REPORT_FOLDER;
+    }
+    if (process.env.REPORT_ID) {
+        return path.join(folder, process.env.REPORT_ID);
+    }
     let nextIndex = 1;
     while (fs.existsSync(path.join(folder, nextIndex.toString()))) {
         nextIndex++;
     }
     return path.join(folder, nextIndex.toString());
 };
-const initContext = async (path, doNavigate = true, headless = false, world = null, moveToRight = -1) => {
-    if (context) {
+const initContext = async (path, doNavigate = true, headless = false, world = null, moveToRight = -1, initScript = null, envName = null) => {
+    if (context && context.playContext && context.playContext.isClosed !== true) {
         return context;
     }
-    if (world && world.context) {
+    if (world && world.context && world.context.playContext && world.context.playContext.isClosed !== true) {
         return world.context;
     }
-    context = await getContext(null, headless);
-    if (world) {
-        world.context = context;
-        world.screenshot = true;
-        if (!reportFolder) {
-            reportFolder = _findEmptyFolder();
-            if (world.attach) {
-                world.attach(reportFolder, { mediaType: "text/plain" });
-                world.attach(JSON.stringify(context.environment), {
-                    mediaType: "application/json+env",
-                });
-            }
+    if (!reportFolder) {
+        reportFolder = _findEmptyFolder();
+        if (world && world.attach) {
+            world.attach(reportFolder, { mediaType: "text/plain" });
         }
-        world.screenshotPath = reportFolder + "/screenshots/";
-        if (!fs.existsSync(world.screenshotPath)) {
-            fs.mkdirSync(world.screenshotPath, { recursive: true });
+    }
+    const globalTestDataFile = process.env.GLOBAL_TEST_DATA_FILE;
+    if (globalTestDataFile) {
+        // check if file exists
+        if (!fs.existsSync(globalTestDataFile)) {
+            console.log("GLOBAL_TEST_DATA_FILE not found: " + process.env.GLOBAL_TEST_DATA_FILE);
+        }
+        else {
+            // if report folder does not exist, create it
+            if (!fs.existsSync(reportFolder)) {
+                fs.mkdirSync(reportFolder, { recursive: true });
+            }
+            // copy the test data file to the report folder as data.json
+            fs.copyFileSync(globalTestDataFile, reportFolder + "/data.json");
         }
+    }
+    const screenshotPath = reportFolder + "/screenshots/";
+    if (!fs.existsSync(screenshotPath)) {
+        fs.mkdirSync(screenshotPath, { recursive: true });
+    }
+    if (world) {
         world.reportFolder = reportFolder;
-        context.reportFolder = reportFolder;
-    }
-    if (moveToRight > 0) {
-        // move the borwser to the top right corner of the screen
-        // create a cdp session
-        // Get CDP session
-        const playContext = context.playContext;
-        const client = await playContext.newCDPSession(context.page);
-        // Get window ID for the current target
-        const { windowId } = await client.send("Browser.getWindowForTarget");
-        console.log(windowId);
-        // get the window for the current target
-        const window = await client.send("Browser.getWindowBounds", {
-            windowId,
-        });
-        console.log(window);
-        await client.send("Browser.setWindowBounds", {
-            windowId,
-            bounds: {
-                left: window.bounds.left + moveToRight,
-            },
-        });
-        // close cdp
-        await client.detach();
+        world.screenshotPath = screenshotPath;
+        world.screenshot = true;
+    }
+    context = await getContext(null, headless, world, null, null, true, null, moveToRight, reportFolder, initScript);
+    if (world) {
+        world.context = context;
+        if (world.attach) {
+            world.attach(JSON.stringify(context.environment), {
+                mediaType: "application/json+env",
+            });
+        }
     }
+    context.reportFolder = reportFolder;
     if (doNavigate) {
         await navigate(path);
     }
+    if (context) {
+        const env = getEnv(envName);
+        if (env && !process.env.CDP_CONNECT_URL) {
+            await getTestData(env, world);
+        }
+    }
     return context;
 };
+const getEnv = (envName) => {
+    let env = process.env.BLINQ_ENV;
+    if (envName) {
+        env = envName;
+    }
+    if (env) {
+        try {
+            const content = JSON.parse(fs.readFileSync(env, "utf8"));
+            return content.name;
+        }
+        catch (e) {
+            console.log("Error reading env file: " + e);
+            return null;
+        }
+    }
+    return null;
+};
 const closeContext = async () => {
+    if (process.env.TEMP_RUN) {
+        return;
+    }
     try {
         if (context && context.browser) {
             await browserManager.closeBrowser(context.browser);
@@ -98,6 +128,107 @@ const closeContext = async () => {
         // ignore
     }
     context = null;
+    reportFolder = "";
+};
+const getTestData = async (currentEnv, world, dataFile, feature, scenario) => {
+    // copy the global test data located in data/data.json to the report folder
+    try {
+        if (fs.existsSync(path.join("data", "data.json"))) {
+            const filterFeatureScenario = feature || scenario;
+            const data = fs.readFileSync(path.join("data", "data.json"), "utf8");
+            const jsonData = JSON.parse(data);
+            const testData = {};
+            const allEnvData = jsonData["*"];
+            const currentEnvData = jsonData[currentEnv];
+            // Process all environment data first as a baseline
+            if (allEnvData) {
+                for (let i = 0; i < allEnvData.length; i++) {
+                    const item = allEnvData[i];
+                    if (process.env[item.key] && item.key.toLowerCase() !== "username" && item.key.toLowerCase() !== "user") {
+                        testData[item.key] = process.env[item.key];
+                        continue;
+                    }
+                    // Filter by feature/scenario if specified
+                    if (filterFeatureScenario) {
+                        if (feature && item.feature && item.feature !== feature) {
+                            continue;
+                        }
+                        if (scenario && item.scenario && item.scenario !== scenario) {
+                            continue;
+                        }
+                    }
+                    else if (item.feature || item.scenario) {
+                        // Skip feature/scenario specific items when not filtering
+                        continue;
+                    }
+                    let useValue = item.value;
+                    if (item.DataType === "secret") {
+                        testData[item.key] = "secret:" + item.value;
+                        // decrypt the secret
+                        useValue = decrypt("secret:" + item.value);
+                    }
+                    else if (item.DataType === "totp") {
+                        testData[item.key] = "totp:" + item.value;
+                        useValue = "totp:" + item.value;
+                    }
+                    else {
+                        testData[item.key] = item.value;
+                    }
+                }
+            }
+            // Then process currentEnvData to override the base values
+            if (currentEnvData) {
+                for (let i = 0; i < currentEnvData.length; i++) {
+                    const item = currentEnvData[i];
+                    if (process.env[item.key] && item.key.toLowerCase() !== "username" && item.key.toLowerCase() !== "user") {
+                        testData[item.key] = process.env[item.key];
+                        continue;
+                    }
+                    // Filter by feature/scenario if specified
+                    if (filterFeatureScenario) {
+                        if (feature && item.feature && item.feature !== feature) {
+                            continue;
+                        }
+                        if (scenario && item.scenario && item.scenario !== scenario) {
+                            continue;
+                        }
+                    }
+                    else if (item.feature || item.scenario) {
+                        // Skip feature/scenario specific items when not filtering
+                        continue;
+                    }
+                    let useValue = item.value;
+                    if (item.DataType === "secret") {
+                        testData[item.key] = "secret:" + item.value;
+                        // decrypt the secret
+                        useValue = decrypt("secret:" + item.value);
+                    }
+                    else if (item.DataType === "totp") {
+                        testData[item.key] = "totp:" + item.value;
+                        useValue = "totp:" + item.value;
+                    }
+                    else {
+                        testData[item.key] = item.value;
+                    }
+                }
+            }
+            if (dataFile && !existsSync(path.dirname(dataFile))) {
+                fs.mkdirSync(path.dirname(dataFile), { recursive: true });
+            }
+            if (!dataFile)
+                dataFile = _getDataFile(world, context, context?.web);
+            fs.writeFileSync(dataFile, JSON.stringify(testData, null, 2));
+        }
+    }
+    catch (e) {
+        console.log("Error reading data.json file: " + e);
+    }
+};
+const resetTestData = async (envPath, world) => {
+    const envName = getEnv(envPath);
+    if (envName) {
+        getTestData(envName, world);
+    }
 };
-export { initContext, navigate, closeContext };
+export { initContext, navigate, closeContext, resetTestData, getTestData };
 //# sourceMappingURL=auto_page.js.map

package/lib/auto_page.js.map

@@ -1 +1 @@
-{"version":3,"file":"auto_page.js","sourceRoot":"","sources":["../../src/auto_page.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AAExB,IAAI,OAAO,GAAuB,IAAI,CAAC;AACvC,IAAI,YAAY,GAAG,EAAE,CAAC;AACtB,MAAM,QAAQ,GAAG,KAAK,EAAE,IAAI,GAAG,EAAE,EAAE,EAAE;IACnC,IAAI,IAAI,KAAK,IAAI,IAAI,OAAQ,CAAC,QAAQ,KAAK,IAAI,EAAE;QAC/C,OAAO;KACR;IACD,IAAI,GAAG,GAAG,IAAI,CAAC;IACf,IAAI,IAAI,KAAK,IAAI,EAAE;QACjB,GAAG,GAAG,OAAQ,CAAC,WAAY,CAAC,OAAQ,CAAC;KACtC;SAAM;QACL,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,EAAE,OAAQ,CAAC,WAAY,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC;KACzD;IACD,MAAM,OAAQ,CAAC,MAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACjC,OAAQ,CAAC,QAAQ,GAAG,IAAI,CAAC;IACzB,MAAM,OAAQ,CAAC,MAAO,CAAC,eAAe,EAAE,CAAC;AAC3C,CAAC,CAAC;AACF,MAAM,gBAAgB,GAAG,CAAC,MAAe,EAAE,EAAE;IAC3C,IAAI,CAAC,MAAM,EAAE;QACX,MAAM,GAAG,QAAQ,CAAC;KACnB;IACD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE;QAC1B,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;KACtB;IACD,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,OAAO,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE;QAC7D,SAAS,EAAE,CAAC;KACb;IACD,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC;AACjD,CAAC,CAAC;AACF,MAAM,WAAW,GAAG,KAAK,EAAE,IAAY,EAAE,UAAU,GAAG,IAAI,EAAE,QAAQ,GAAG,KAAK,EAAE,QAAa,IAAI,EAAE,WAAW,GAAG,CAAC,CAAC,EAAE,EAAE;IACnH,IAAI,OAAO,EAAE;QACX,OAAO,OAAO,CAAC;KAChB;IACD,IAAI,KAAK,IAAI,KAAK,CAAC,OAAO,EAAE;QAC1B,OAAO,KAAK,CAAC,OAAO,CAAC;KACtB;IACD,OAAO,GAAG,MAAM,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAC3C,IAAI,KAAK,EAAE;QACT,KAAK,CAAC,OAAO,GAAG,OAAO,CAAC;QACxB,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,YAAY,EAAE;YACjB,YAAY,GAAG,gBAAgB,EAAE,CAAC;YAClC,IAAI,KAAK,CAAC,MAAM,EAAE;gBAChB,KAAK,CAAC,MAAM,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC,CAAC;gBACxD,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE;oBAChD,SAAS,EAAE,sBAAsB;iBAClC,CAAC,CAAC;aACJ;SACF;QACD,KAAK,CAAC,cAAc,GAAG,YAAY,GAAG,eAAe,CAAC;QACtD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE;YACxC,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,cAAc,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;SACzD;QACD,KAAK,CAAC,YAAY,GAAG,YAAY,CAAC;QAClC,OAAO,CAAC,YAAY,GAAG,YAAY,CAAC;KACrC;IAED,IAAI,WAAW,GAAG,CAAC,EAAE;QACnB,yDAAyD;QACzD,uBAAuB;QACvB,kBAAkB;QAClB,MAAM,WAAW,GAAQ,OAAO,CAAC,WAAW,CAAC;QAC7C,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAE7D,uCAAuC;QACvC,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QACrE,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAEtB,wCAAwC;QACxC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE;YAC1D,QAAQ;SACT,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpB,MAAM,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE;YAC3C,QAAQ;YACR,MAAM,EAAE;gBACN,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,GAAG,WAAW;aACvC;SACF,CAAC,CAAC;QACH,YAAY;QACZ,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;KACvB;IACD,IAAI,UAAU,EAAE;QACd,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAC;KACtB;IAED,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE;IAC9B,IAAI;QACF,IAAI,OAAO,IAAI,OAAO,CAAC,OAAO,EAAE;YAC9B,MAAM,cAAc,CAAC,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;SACpD;KACF;IAAC,OAAO,CAAC,EAAE;QACV,sCAAsC;QACtC,SAAS;KACV;IACD,OAAO,GAAG,IAAI,CAAC;AACjB,CAAC,CAAC;AACF,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC"}
+{"version":3,"file":"auto_page.js","sourceRoot":"","sources":["../../src/auto_page.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AACpC,OAAO,IAAI,MAAM,MAAM,CAAC;AAIxB,OAAO,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AACnD,IAAI,OAAO,GAAuB,IAAI,CAAC;AACvC,IAAI,YAAY,GAAG,EAAE,CAAC;AACtB,MAAM,QAAQ,GAAG,KAAK,EAAE,IAAI,GAAG,EAAE,EAAE,EAAE;IACnC,IAAI,IAAI,KAAK,IAAI,IAAI,OAAQ,CAAC,QAAQ,KAAK,IAAI,EAAE;QAC/C,OAAO;KACR;IACD,IAAI,GAAG,GAAG,IAAI,CAAC;IACf,IAAI,IAAI,KAAK,IAAI,EAAE;QACjB,GAAG,GAAG,OAAQ,CAAC,WAAY,CAAC,OAAQ,CAAC;KACtC;SAAM;QACL,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,EAAE,OAAQ,CAAC,WAAY,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC;KACzD;IACD,MAAM,OAAQ,CAAC,GAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC9B,OAAQ,CAAC,QAAQ,GAAG,IAAI,CAAC;IACzB,MAAM,OAAQ,CAAC,GAAI,CAAC,eAAe,EAAE,CAAC;AACxC,CAAC,CAAC;AACF,MAAM,gBAAgB,GAAG,CAAC,MAAe,EAAE,EAAE;IAC3C,IAAI,CAAC,MAAM,EAAE;QACX,MAAM,GAAG,QAAQ,CAAC;KACnB;IACD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE;QAC1B,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;KACtB;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE;QAC7B,OAAO,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;KAClC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE;QACzB,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;KACjD;IACD,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,OAAO,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE;QAC7D,SAAS,EAAE,CAAC;KACb;IACD,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC;AACjD,CAAC,CAAC;AACF,MAAM,WAAW,GAAG,KAAK,EACvB,IAAY,EACZ,UAAU,GAAG,IAAI,EACjB,QAAQ,GAAG,KAAK,EAChB,QAAa,IAAI,EACjB,WAAW,GAAG,CAAC,CAAC,EAChB,aAAiC,IAAI,EACrC,UAAyB,IAAI,EAC7B,EAAE;IACF,IAAI,OAAO,IAAI,OAAO,CAAC,WAAW,IAAK,OAAO,CAAC,WAAmB,CAAC,QAAQ,KAAK,IAAI,EAAE;QACpF,OAAO,OAAO,CAAC;KAChB;IACD,IAAI,KAAK,IAAI,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,KAAK,IAAI,EAAE;QACtG,OAAO,KAAK,CAAC,OAAO,CAAC;KACtB;IACD,IAAI,CAAC,YAAY,EAAE;QACjB,YAAY,GAAG,gBAAgB,EAAE,CAAC;QAClC,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,EAAE;YACzB,KAAK,CAAC,MAAM,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC,CAAC;SACzD;KACF;IACD,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;IAC7D,IAAI,kBAAkB,EAAE;QACtB,uBAAuB;QACvB,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,kBAAkB,CAAC,EAAE;YACtC,OAAO,CAAC,GAAG,CAAC,mCAAmC,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;SACtF;aAAM;YACL,6CAA6C;YAC7C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE;gBAChC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;aACjD;YACD,4DAA4D;YAC5D,EAAE,CAAC,YAAY,CAAC,kBAAkB,EAAE,YAAY,GAAG,YAAY,CAAC,CAAC;SAClE;KACF;IACD,MAAM,cAAc,GAAG,YAAY,GAAG,eAAe,CAAC;IACtD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE;QAClC,EAAE,CAAC,SAAS,CAAC,cAAc,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;KACnD;IACD,IAAI,KAAK,EAAE;QACT,KAAK,CAAC,YAAY,GAAG,YAAY,CAAC;QAClC,KAAK,CAAC,cAAc,GAAG,cAAc,CAAC;QACtC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC;KACzB;IACD,OAAO,GAAG,MAAM,UAAU,CAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;IACjH,IAAI,KAAK,EAAE;QACT,KAAK,CAAC,OAAO,GAAG,OAAO,CAAC;QACxB,IAAI,KAAK,CAAC,MAAM,EAAE;YAChB,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE;gBAChD,SAAS,EAAE,sBAAsB;aAClC,CAAC,CAAC;SACJ;KACF;IACD,OAAO,CAAC,YAAY,GAAG,YAAY,CAAC;IAEpC,IAAI,UAAU,EAAE;QACd,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAC;KACtB;IACD,IAAI,OAAO,EAAE;QACX,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;QAC5B,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE;YACvC,MAAM,WAAW,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;SAC/B;KACF;IAED,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,MAAM,GAAG,CAAC,OAAsB,EAAE,EAAE;IACxC,IAAI,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC;IAChC,IAAI,OAAO,EAAE;QACX,GAAG,GAAG,OAAO,CAAC;KACf;IACD,IAAI,GAAG,EAAE;QACP,IAAI;YACF,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC;YACzD,OAAO,OAAO,CAAC,IAAI,CAAC;SACrB;QAAC,OAAO,CAAC,EAAE;YACV,OAAO,CAAC,GAAG,CAAC,0BAA0B,GAAG,CAAC,CAAC,CAAC;YAC5C,OAAO,IAAI,CAAC;SACb;KACF;IACD,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AACF,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE;IAC9B,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE;QACxB,OAAO;KACR;IACD,IAAI;QACF,IAAI,OAAO,IAAI,OAAO,CAAC,OAAO,EAAE;YAC9B,MAAM,cAAc,CAAC,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;SACpD;KACF;IAAC,OAAO,CAAC,EAAE;QACV,sCAAsC;QACtC,SAAS;KACV;IACD,OAAO,GAAG,IAAI,CAAC;IACf,YAAY,GAAG,EAAE,CAAC;AACpB,CAAC,CAAC;AASF,MAAM,WAAW,GAAG,KAAK,EAAE,UAAkB,EAAE,KAAU,EAAE,QAAiB,EAAE,OAAgB,EAAE,QAAiB,EAAE,EAAE;IACnH,2EAA2E;IAC3E,IAAI;QACF,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC,EAAE;YACjD,MAAM,qBAAqB,GAAG,OAAO,IAAI,QAAQ,CAAC;YAClD,MAAM,IAAI,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,CAAC,CAAC;YACrE,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAoD,CAAC;YACrF,MAAM,QAAQ,GAA2B,EAAE,CAAC;YAC5C,MAAM,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;YACjC,MAAM,cAAc,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;YAE5C,mDAAmD;YACnD,IAAI,UAAU,EAAE;gBACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;oBAC1C,MAAM,IAAI,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;oBAC3B,IAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,KAAK,UAAU,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,KAAK,MAAM,EAAE;wBACtG,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAE,CAAC;wBAC5C,SAAS;qBACV;oBACD,0CAA0C;oBAC1C,IAAI,qBAAqB,EAAE;wBACzB,IAAI,OAAO,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,KAAK,OAAO,EAAE;4BACvD,SAAS;yBACV;wBACD,IAAI,QAAQ,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,EAAE;4BAC3D,SAAS;yBACV;qBACF;yBAAM,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ,EAAE;wBACxC,0DAA0D;wBAC1D,SAAS;qBACV;oBAED,IAAI,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC;oBAE1B,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,EAAE;wBAC9B,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC;wBAC5C,qBAAqB;wBACrB,QAAQ,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;qBAC5C;yBAAM,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,EAAE;wBACnC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC;wBAC1C,QAAQ,GAAG,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC;qBACjC;yBAAM;wBACL,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;qBACjC;iBACF;aACF;YAED,0DAA0D;YAC1D,IAAI,cAAc,EAAE;gBAClB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;oBAC9C,MAAM,IAAI,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;oBAC/B,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,KAAK,UAAU,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,KAAK,MAAM,EAAE;wBACvG,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAE,CAAC;wBAC5C,SAAS;qBACV;oBAED,0CAA0C;oBAC1C,IAAI,qBAAqB,EAAE;wBACzB,IAAI,OAAO,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,KAAK,OAAO,EAAE;4BACvD,SAAS;yBACV;wBACD,IAAI,QAAQ,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,EAAE;4BAC3D,SAAS;yBACV;qBACF;yBAAM,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ,EAAE;wBACxC,0DAA0D;wBAC1D,SAAS;qBACV;oBAED,IAAI,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC;oBAE1B,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,EAAE;wBAC9B,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC;wBAC5C,qBAAqB;wBACrB,QAAQ,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;qBAC5C;yBAAM,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,EAAE;wBACnC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC;wBAC1C,QAAQ,GAAG,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC;qBACjC;yBAAM;wBACL,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;qBACjC;iBACF;aACF;YAED,IAAI,QAAQ,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE;gBACnD,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;aAC3D;YAED,IAAI,CAAC,QAAQ;gBAAE,QAAQ,GAAG,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;YACrE,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;SAC/D;KACF;IAAC,OAAO,CAAC,EAAE;QACV,OAAO,CAAC,GAAG,CAAC,gCAAgC,GAAG,CAAC,CAAC,CAAC;KACnD;AACH,CAAC,CAAC;AAEF,MAAM,aAAa,GAAG,KAAK,EAAE,OAAe,EAAE,KAAU,EAAE,EAAE;IAC1D,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;IAChC,IAAI,OAAO,EAAE;QACX,WAAW,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;KAC7B;AACH,CAAC,CAAC;AACF,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,CAAC"}

package/lib/browser_manager.d.ts

@@ -1,5 +1,6 @@
 import { Browser as PlaywrightBrowser, BrowserContext, Page } from "playwright";
 import type { Cookie, LocalStorage } from "./environment.js";
+import { InitScripts } from "./generation_scripts.js";
 type StorageState = {
     cookies: Cookie[];
     origins: {
@@ -12,15 +13,18 @@ declare class BrowserManager {
     constructor(browsers?: Browser[]);
     closeAll(): Promise<void>;
     closeBrowser(browser?: PlaywrightBrowser | Browser): Promise<void>;
-    createBrowser(headless?: boolean, storageState?: StorageState, extensionPath?: string, userDataDirPath?: string): Promise<Browser>;
-    getBrowser(headless?: boolean, storageState?: StorageState, extensionPath?: string, userDataDirPath?: string): Promise<Browser>;
+    createBrowser(headless?: boolean, storageState?: StorageState, extensionPath?: string, userDataDirPath?: string, reportFolder?: string, userAgent?: string, channel?: string, aiConfig?: any, initScripts?: InitScripts | null): Promise<Browser>;
 }
 declare class Browser {
     browser: PlaywrightBrowser | null;
     context: BrowserContext | null;
     page: Page | null;
+    headless: boolean;
+    reportFolder: string | null;
+    trace: boolean;
+    traceFolder: string | null;
     constructor();
-    init(headless?: boolean, storageState?: StorageState, extensionPath?: string, userDataDirPath?: string): Promise<void>;
+    init(headless?: boolean, storageState?: StorageState, extensionPath?: string, userDataDirPath?: string, reportFolder?: string, userAgent?: string, channel?: string, aiConfig?: any, initScripts?: InitScripts | null): Promise<void>;
     close(): Promise<void>;
 }
 declare const browserManager: BrowserManager;

package/lib/browser_manager.js

@@ -1,5 +1,12 @@
 import { chromium, firefox, webkit, } from "playwright";
+import fs from "fs";
+import path from "path";
+import { fileURLToPath } from "url";
+// Get __filename and __dirname in ESM
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
 class BrowserManager {
+    browsers;
     constructor(browsers = []) {
         this.browsers = browsers;
     }
@@ -9,9 +16,12 @@ class BrowserManager {
     }
     async closeBrowser(browser) {
         if (!browser && this.browsers.length > 0) {
-            browser = this.browsers[0];
+            for (let i = 0; i < this.browsers.length; i++) {
+                await this.browsers[i].close();
+            }
+            this.browsers = [];
         }
-        if (browser) {
+        if (browser && !process.env.IGNORE_BROWSER_CLOSE) {
             await browser.close();
             for (let i = 0; i < this.browsers.length; i++) {
                 if (this.browsers[i].browser === browser || this.browsers[i] === browser) {
@@ -22,67 +32,71 @@ class BrowserManager {
             }
         }
     }
-    async createBrowser(headless = false, storageState, extensionPath, userDataDirPath) {
+    async createBrowser(headless = false, storageState, extensionPath, userDataDirPath, reportFolder, userAgent, channel, aiConfig, initScripts = null) {
         const browser = new Browser();
-        await browser.init(headless, storageState, extensionPath, userDataDirPath);
+        await browser.init(headless, storageState, extensionPath, userDataDirPath, reportFolder, userAgent, channel, aiConfig, initScripts);
         this.browsers.push(browser);
         return browser;
     }
-    async getBrowser(headless = false, storageState, extensionPath, userDataDirPath) {
-        if (this.browsers.length === 0) {
-            return await this.createBrowser(headless, storageState, extensionPath, userDataDirPath);
-        }
-        return this.browsers[0];
-    }
 }
 class Browser {
+    browser;
+    context;
+    page;
+    headless = false;
+    reportFolder = null;
+    trace = false;
+    traceFolder = null;
     constructor() {
         this.browser = null;
         this.context = null;
         this.page = null;
     }
-    async init(headless = false, storageState, extensionPath, userDataDirPath) {
+    async init(headless = false, storageState, extensionPath, userDataDirPath, reportFolder, userAgent, channel, aiConfig, initScripts = null) {
+        if (!aiConfig) {
+            aiConfig = {};
+        }
+        // if (!downloadsPath) {
+        //   downloadsPath = "downloads";
+        // }
+        // // check if downloads path exists
+        // if (!fs.existsSync(downloadsPath)) {
+        //   fs.mkdirSync(downloadsPath, { recursive: true });
+        // }
+        this.headless = headless;
+        if (reportFolder) {
+            this.reportFolder = reportFolder;
+        }
         let viewport = null;
         if (process.env.HEADLESS === "true") {
             headless = true;
+            this.headless = true;
         }
         else if (process.env.HEADLESS === "false") {
             headless = false;
+            this.headless = false;
         }
         if (process.env.VIEWPORT) {
             let viewportParts = process.env.VIEWPORT.split(",");
             viewport = { width: parseInt(viewportParts[0]), height: parseInt(viewportParts[1]) };
         }
+        else if (!aiConfig.noViewport) {
+            viewport = { width: 1280, height: 800 };
+        }
+        const args = ["--ignore-https-errors", "--ignore-certificate-errors"];
+        if (process.env.CDP_LISTEN_PORT) {
+            args.push(`--remote-debugging-port=${process.env.CDP_LISTEN_PORT}`);
+        }
         if (!extensionPath && userDataDirPath) {
             this.context = await chromium.launchPersistentContext(userDataDirPath, {
                 headless: false,
                 timeout: 0,
                 bypassCSP: true,
-                args: ["--ignore-https-errors", "--no-incognito", "--ignore-certificate-errors", "--disable-site-isolation-trials"],
+                args: ["--ignore-https-errors", "--no-incognito", "--ignore-certificate-errors"],
             });
-            // this.browser = await chromium.connectOverCDP({
-            //   endpointURL: `http://localhost:${cdpPort}`,
-            // });
-            // if (!this.browser) {
-            //   throw new Error("Could not connect to browser");
-            // }
-            // this.context = await this.browser.newContext();
-            //this.page = await this.context.newPage();
-            // if (this.browser.contexts?.length > 0) {
-            //   this.context =  (this.browser.contexts as BrowserContext[])[0];
-            //   //this.context = await this.browser.contexts[0];
-            //   if (this.context.pages.length > 0) {
-            //     this.page = await this.context.pages[0];
-            //   } else {
-            //     this.page = await this.context.newPage();
-            //   }
-            // } else {
-            //   this.context = await this.browser.newContext();
-            //   this.page = await this.context.newPage();
-            // }
         }
         else if (extensionPath) {
-            this.context = await chromium.launchPersistentContext(userDataDirPath !== null && userDataDirPath !== void 0 ? userDataDirPath : "", {
+            this.context = await chromium.launchPersistentContext(userDataDirPath ?? "", {
                 headless: headless,
                 timeout: 0,
                 bypassCSP: true,
@@ -92,7 +106,6 @@ class Browser {
                     "--load-extension=" + extensionPath,
                     "--no-incognito",
                     "--ignore-certificate-errors",
-                    "--disable-site-isolation-trials"
                 ],
             });
         }
@@ -101,24 +114,51 @@ class Browser {
                 this.browser = await firefox.launch({
                     headless: headless,
                     timeout: 0,
-                    args: ["--ignore-https-errors", "--ignore-certificate-errors", "--disable-site-isolation-trials"],
+                    args,
+                    //downloadsPath: downloadsPath,
                 });
             }
             else if (process.env.BROWSER === "webkit") {
                 this.browser = await webkit.launch({
                     headless: headless,
                     timeout: 0,
-                    args: ["--ignore-https-errors", "--ignore-certificate-errors", "--disable-site-isolation-trials"],
+                    args,
+                    //downloadsPath: downloadsPath,
                 });
             }
+            else if (channel) {
+                {
+                    this.browser = await chromium.launch({
+                        headless: headless,
+                        timeout: 0,
+                        args,
+                        channel: channel,
+                        //downloadsPath: downloadsPath,
+                    });
+                }
+            }
             else {
-                this.browser = await chromium.launch({
-                    headless: headless,
-                    timeout: 0,
-                    args: ["--ignore-https-errors", "--ignore-certificate-errors", "--disable-site-isolation-trials"],
-                });
+                if (process.env.CDP_CONNECT_URL) {
+                    this.browser = await chromium.connectOverCDP(process.env.CDP_CONNECT_URL);
+                }
+                else {
+                    this.browser = await chromium.launch({
+                        headless: headless,
+                        timeout: 0,
+                        args,
+                        //downloadsPath: downloadsPath,
+                    });
+                }
             }
+            // downloadsPath
             let contextOptions = {};
+            if (aiConfig.contextOptions) {
+                contextOptions = aiConfig.contextOptions;
+                console.log("contextOptions: " + JSON.stringify(contextOptions));
+            }
+            if (!contextOptions["acceptDownloads"]) {
+                contextOptions["acceptDownloads"] = true;
+            }
             if (storageState) {
                 contextOptions.storageState = storageState;
                 contextOptions.bypassCSP = true;
@@ -127,12 +167,86 @@ class Browser {
             if (viewport) {
                 contextOptions.viewport = viewport;
             }
-            this.context = await this.browser.newContext(contextOptions);
+            else {
+                if (!this.headless) {
+                    contextOptions.viewport = null;
+                }
+            }
+            if (userAgent) {
+                contextOptions.userAgent = userAgent;
+            }
+            if (!this.context && this.browser) {
+                if (this.browser.contexts().length > 0) {
+                    this.context = this.browser.contexts()[this.browser.contexts().length - 1];
+                }
+                else {
+                    this.context = await this.browser.newContext(contextOptions);
+                }
+            }
+        }
+        if ((process.env.TRACE === "true" || aiConfig.trace === true) && this.context) {
+            this.trace = true;
+            const traceFolder = path.join(this.reportFolder, "trace");
+            //const traceFile = path.join(traceFolder, "trace.zip");
+            if (!fs.existsSync(traceFolder)) {
+                fs.mkdirSync(traceFolder, { recursive: true });
+            }
+            this.traceFolder = traceFolder;
+            await this.context.tracing.start({ screenshots: true, snapshots: true });
+        }
+        if (initScripts && this.context) {
+            if (initScripts.recorderCjs) {
+                await this.context.addInitScript({
+                    content: `
+            (() => {
+            const module = {};
+            ${initScripts.recorderCjs}
+            const sss = new (module.exports.InjectedScript())(
+              window,
+              true,
+              "javascript",
+              [],
+              ${0},
+              "${this.browser?.browserType().name()}",
+              []
+            );
+          })();`,
+                });
+            }
+            if (initScripts.scripts) {
+                for (let script of initScripts.scripts) {
+                    await this.context.addInitScript({
+                        content: script,
+                    });
+                }
+            }
+        }
+        let axeMinJsPath = path.join(__dirname, "..", "scripts", "axe.mini.js");
+        // Check if the file exists
+        if (!fs.existsSync(axeMinJsPath)) {
+            axeMinJsPath = path.join(__dirname, "scripts", "axe.mini.js");
+        }
+        // Read the content of axe.min.js synchronously
+        const axeMinJsContent = fs.readFileSync(axeMinJsPath, "utf-8");
+        await this.context?.addInitScript({
+            content: axeMinJsContent,
+        });
+        if (this.context && this.context.pages().length > 0) {
+            this.page = this.context.pages()[this.context.pages().length - 1];
+        }
+        else {
+            this.page = await this.context.newPage();
+        }
+        if (this.context) {
+            this.context.on("close", () => {
+                this.context.isClosed = true;
+            });
         }
-        // }
-        this.page = await this.context.newPage();
     }
     async close() {
+        if (process.env.IGNORE_BROWSER_CLOSE === "true") {
+            return;
+        }
         if (this.browser !== null) {
             await this.browser.close();
             this.browser = null;
@@ -143,11 +257,4 @@ class Browser {
 }
 const browserManager = new BrowserManager();
 export { browserManager };
-// let browser = await browserManager.createBrowser();
-// browser.page.goto("https://www.cnn.com");
-// let browser2 = await browserManager.createBrowser();
-// await browser2.page.goto("https://www.google.com");
-// // sleep for 2 seconds
-// await new Promise((r) => setTimeout(r, 1000));
-// await browserManager.closeAll();
 //# sourceMappingURL=browser_manager.js.map