@eko-ai/eko-nodejs 2.1.6 → 2.1.7-alpha.2

package/dist/index.d.ts

@@ -1,3 +1,5 @@
+import { getCdpWsEndpoint } from "./utils";
+export { getCdpWsEndpoint };
 export { BrowserAgent } from "./browser";
 export { FileAgent } from "./file";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAE3C,OAAO,EAAE,gBAAgB,EAAE,CAAC;AAC5B,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC"}

package/dist/index.esm.js

@@ -1,4 +1,4 @@
-import { BaseBrowserLabelsAgent, BaseFileAgent } from '@eko-ai/eko';
+import { Log, BaseBrowserLabelsAgent, BaseFileAgent } from '@eko-ai/eko';
 import { chromium } from 'playwright';
 import * as fs from 'fs/promises';
 import * as path$1 from 'path';
@@ -11,6 +11,14 @@ import { EventEmitter } from 'node:events';
 import Stream from 'node:stream';
 import { StringDecoder } from 'node:string_decoder';
 
+async function getCdpWsEndpoint(port) {
+    // Example => ws://localhost:9222/devtools/browser/{session-id}
+    const response = await fetch(`http://localhost:${port}/json/version`);
+    const browserInfo = await response.json();
+    Log.info("browserInfo: ", browserInfo);
+    return browserInfo.webSocketDebuggerUrl;
+}
+
 class BrowserAgent extends BaseBrowserLabelsAgent {
     constructor() {
         super(...arguments);
@@ -22,6 +30,12 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
     setHeadless(headless) {
         this.headless = headless;
     }
+    setCdpWsEndpoint(cdpWsEndpoint) {
+        this.cdpWsEndpoint = cdpWsEndpoint;
+    }
+    setOptions(options) {
+        this.options = options;
+    }
     async screenshot(agentContext) {
         let page = await this.currentPage();
         let screenshotBuffer = await page.screenshot({
@@ -167,17 +181,28 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
         return new Promise((resolve) => setTimeout(() => resolve(), time));
     }
     async getBrowserContext() {
-        if (!this.browser) {
-            this.current_page = null;
-            this.browser_context = null;
-            this.browser = await chromium.launch({
-                headless: this.headless,
-                args: ["--no-sandbox"],
-            });
-        }
         if (!this.browser_context) {
             this.current_page = null;
-            this.browser_context = await this.browser.newContext();
+            this.browser_context = null;
+            if (this.cdpWsEndpoint) {
+                this.browser = await chromium.connectOverCDP(this.cdpWsEndpoint, this.options);
+                this.browser_context = await this.browser.newContext();
+            }
+            else if (this.userDataDir) {
+                this.browser_context = await chromium.launchPersistentContext(this.userDataDir, {
+                    headless: this.headless,
+                    // channel: 'chrome',
+                    ...this.options,
+                });
+            }
+            else {
+                this.browser = await chromium.launch({
+                    headless: this.headless,
+                    args: ["--no-sandbox"],
+                    ...this.options,
+                });
+                this.browser_context = await this.browser.newContext();
+            }
             // Anti-crawling detection website:
             // https://bot.sannysoft.com/
             let init_script = await this.initScript();
@@ -8202,5 +8227,5 @@ class FileAgent extends BaseFileAgent {
     }
 }
 
-export { BrowserAgent, FileAgent };
+export { BrowserAgent, FileAgent, getCdpWsEndpoint };
 //# sourceMappingURL=index.esm.js.map