@eko-ai/eko-nodejs 4.0.9-alpha.1 → 4.1.0

package/dist/index.esm.js

@@ -2408,6 +2408,7 @@ function requirePuppeteerExtraPluginStealth () {
 var puppeteerExtraPluginStealthExports = requirePuppeteerExtraPluginStealth();
 var StealthPlugin = /*@__PURE__*/getDefaultExportFromCjs(puppeteerExtraPluginStealthExports);
 
+var tabId = 1000;
 class BrowserAgent extends BaseBrowserLabelsAgent {
     constructor() {
         super(...arguments);
@@ -2415,6 +2416,8 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
         this.browser_context = null;
         this.current_page = null;
         this.headless = false;
+        this.pageMap = new Map();
+        this.activePage = null;
     }
     setHeadless(headless) {
         this.headless = headless;
@@ -2436,12 +2439,12 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
         const page = await this.currentPage();
         const screenshotBuffer = await page.screenshot({
             fullPage: false,
-            type: "jpeg",
+            type: 'jpeg',
             quality: 60,
         });
-        const base64 = screenshotBuffer.toString("base64");
+        const base64 = screenshotBuffer.toString('base64');
         return {
-            imageType: "image/jpeg",
+            imageType: 'image/jpeg',
             imageBase64: base64,
         };
     }
@@ -2453,45 +2456,67 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
             title: await page.title(),
         };
     }
+    async loadTabs() {
+        return await this.get_all_tabs(undefined);
+    }
+    getPageMap() {
+        return this.pageMap;
+    }
     async get_all_tabs(agentContext) {
         if (!this.browser_context) {
             return [];
         }
         const result = [];
-        const pages = await this.browser_context.pages();
+        const pages = this.browser_context.pages();
         for (let i = 0; i < pages.length; i++) {
             let page = pages[i];
+            let pageInfo = this.pageMap.get(page);
+            if (!pageInfo) {
+                pageInfo = {
+                    tabId: tabId++,
+                    lastAccessed: Date.now(),
+                };
+                this.pageMap.set(page, pageInfo);
+            }
             result.push({
-                tabId: i,
+                tabId: pageInfo.tabId,
                 url: page.url(),
                 title: await page.title(),
+                lastAccessed: pageInfo.lastAccessed,
+                active: page === this.activePage,
             });
         }
         return result;
     }
     async switch_tab(agentContext, tabId) {
         if (!this.browser_context) {
-            throw new Error("tabId does not exist: " + tabId);
+            throw new Error('tabId does not exist: ' + tabId);
         }
-        const pages = await this.browser_context.pages();
-        const page = pages[tabId];
-        if (!page) {
-            throw new Error("tabId does not exist: " + tabId);
+        let switchPage = null;
+        this.pageMap.forEach((pageInfo, page) => {
+            if (pageInfo.tabId === tabId) {
+                switchPage = page;
+                return;
+            }
+        });
+        if (!switchPage) {
+            throw new Error('tabId does not exist: ' + tabId);
         }
-        this.current_page = page;
+        this.current_page = switchPage;
+        this.activePage = switchPage;
         return {
             tabId: tabId,
-            url: page.url(),
-            title: await page.title(),
+            url: switchPage.url(),
+            title: await switchPage.title(),
         };
     }
     async input_text(agentContext, index, text, enter) {
         try {
             const elementHandle = await this.get_element(index, true);
-            await elementHandle.fill("");
+            await elementHandle.fill('');
             await elementHandle.fill(text);
             if (enter) {
-                await elementHandle.press("Enter");
+                await elementHandle.press('Enter');
                 await this.sleep(200);
             }
         }
@@ -2540,13 +2565,13 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
             await this.autoLoadCookies(url);
             await this.autoLoadLocalStorage(page, url);
             await page.goto(url, {
-                waitUntil: "domcontentloaded",
+                waitUntil: 'domcontentloaded',
                 timeout: 10000,
             });
-            await page.waitForLoadState("networkidle", { timeout: 5000 });
+            await page.waitForLoadState('networkidle', { timeout: 5000 });
         }
         catch (e) {
-            if ((e + "").indexOf("Timeout") == -1) {
+            if ((e + '').indexOf('Timeout') == -1) {
                 throw e;
             }
         }
@@ -2555,11 +2580,11 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
     }
     async currentPage() {
         if (this.current_page == null) {
-            throw new Error("There is no page, please call navigate_to first");
+            throw new Error('There is no page, please call navigate_to first');
         }
         const page = this.current_page;
         try {
-            await page.waitForLoadState("domcontentloaded", { timeout: 10000 });
+            await page.waitForLoadState('domcontentloaded', { timeout: 10000 });
         }
         catch (e) { }
         return page;
@@ -2569,12 +2594,12 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
         return await page.evaluateHandle((params) => {
             let element = window.get_highlight_element(params.index);
             if (element && params.findInput) {
-                if (element.tagName != "INPUT" &&
-                    element.tagName != "TEXTAREA" &&
+                if (element.tagName != 'INPUT' &&
+                    element.tagName != 'TEXTAREA' &&
                     element.childElementCount != 0) {
                     element =
-                        element.querySelector("input") ||
-                            element.querySelector("textarea") ||
+                        element.querySelector('input') ||
+                            element.querySelector('textarea') ||
                             element;
                 }
             }
@@ -2598,7 +2623,7 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
             else if (this.userDataDir) {
                 this.browser_context = (await chromium.launchPersistentContext(this.userDataDir, {
                     headless: this.headless,
-                    channel: "chrome",
+                    channel: 'chrome',
                     args: this.getChromiumArgs(),
                     ...this.options,
                 }));
@@ -2622,16 +2647,24 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
             if (init_script.content || init_script.path) {
                 this.browser_context.addInitScript(init_script);
             }
-            this.browser_context.on("page", async (page) => {
-                page.on("framenavigated", async (frame) => {
+            this.browser_context.on('page', async (page) => {
+                this.activePage = page;
+                this.pageMap.set(page, {
+                    tabId: tabId++,
+                    lastAccessed: Date.now(),
+                });
+                page.on('framenavigated', async (frame) => {
                     if (frame === page.mainFrame()) {
                         const url = frame.url();
-                        if (url.startsWith("http")) {
+                        if (url.startsWith('http')) {
                             await this.autoLoadCookies(url);
                             await this.autoLoadLocalStorage(page, url);
                         }
                     }
                 });
+                page.on('close', () => {
+                    this.pageMap.delete(page);
+                });
             });
         }
         if (this.cookies) {
@@ -2648,7 +2681,7 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
             }
         }
         catch (e) {
-            Log.error("Failed to auto load cookies: " + url, e);
+            Log.error('Failed to auto load cookies: ' + url, e);
         }
     }
     async autoLoadLocalStorage(page, url) {
@@ -2661,12 +2694,12 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
                     }
                 }
                 catch (e) {
-                    Log.error("Failed to inject localStorage: " + url, e);
+                    Log.error('Failed to inject localStorage: ' + url, e);
                 }
             }, localStorageData);
         }
         catch (e) {
-            Log.error("Failed to auto load localStorage: " + url, e);
+            Log.error('Failed to auto load localStorage: ' + url, e);
         }
     }
     async loadCookiesWithUrl(url) {
@@ -2677,18 +2710,18 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
     }
     getChromiumArgs() {
         return [
-            "--no-sandbox",
-            "--remote-allow-origins=*",
-            "--disable-dev-shm-usage",
-            "--disable-popup-blocking",
-            "--ignore-ssl-errors",
-            "--ignore-certificate-errors",
-            "--ignore-certificate-errors-spki-list",
-            "--disable-blink-features=AutomationControlled",
-            "--disable-infobars",
-            "--disable-notifications",
-            "--disable-web-security",
-            "--disable-features=IsolateOrigins,site-per-process",
+            '--no-sandbox',
+            '--remote-allow-origins=*',
+            '--disable-dev-shm-usage',
+            '--disable-popup-blocking',
+            '--ignore-ssl-errors',
+            '--ignore-certificate-errors',
+            '--ignore-certificate-errors-spki-list',
+            '--disable-blink-features=AutomationControlled',
+            '--disable-infobars',
+            '--disable-notifications',
+            '--disable-web-security',
+            '--disable-features=IsolateOrigins,site-per-process',
         ];
     }
     getUserAgent() {
@@ -2704,6 +2737,9 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
     async initScript() {
         return {};
     }
+    async getActivePage() {
+        return this.activePage;
+    }
 }
 
 class SimpleStdioMcpClient {