@eko-ai/eko-nodejs 2.0.8 → 2.0.9

package/dist/index.esm.js

@@ -17,6 +17,10 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
         this.browser = null;
         this.browser_context = null;
         this.current_page = null;
+        this.headless = false;
+    }
+    setHeadless(headless) {
+        this.headless = headless;
     }
     async screenshot(agentContext) {
         let page = await this.currentPage();
@@ -112,19 +116,8 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
         return await page.evaluate(func, args);
     }
     async open_url(agentContext, url) {
-        if (!this.browser) {
-            this.current_page = null;
-            this.browser_context = null;
-            this.browser = await chromium.launch({
-                headless: false,
-                args: ["--no-sandbox"],
-            });
-        }
-        if (!this.browser_context) {
-            this.current_page = null;
-            this.browser_context = await this.browser.newContext();
-        }
-        const page = await this.browser_context.newPage();
+        let browser_context = await this.getBrowserContext();
+        const page = await browser_context.newPage();
         // await page.setViewportSize({ width: 1920, height: 1080 });
         await page.setViewportSize({ width: 1536, height: 864 });
         try {
@@ -173,6 +166,62 @@ class BrowserAgent extends BaseBrowserLabelsAgent {
     sleep(time) {
         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();
+            // Anti-crawling detection website:
+            // https://bot.sannysoft.com/
+            let init_script = await this.initScript();
+            this.browser_context.addInitScript(init_script);
+        }
+        return this.browser_context;
+    }
+    async initScript() {
+        return {
+            content: `
+      // Webdriver property
+			Object.defineProperty(navigator, 'webdriver', {
+				get: () => undefined
+			});
+
+			// Languages
+			Object.defineProperty(navigator, 'languages', {
+				get: () => ['en-US']
+			});
+
+			// Plugins
+			Object.defineProperty(navigator, 'plugins', {
+				get: () => [{name:"1"}, {name:"2"}, {name:"3"}, {name:"4"}, {name:"5"}]
+			});
+
+			// Chrome runtime
+			window.chrome = { runtime: {} };
+
+			// Permissions
+			const originalQuery = window.navigator.permissions.query;
+			window.navigator.permissions.query = (parameters) => (
+				parameters.name === 'notifications' ?
+					Promise.resolve({ state: Notification.permission }) :
+					originalQuery(parameters)
+			);
+			(function () {
+				const originalAttachShadow = Element.prototype.attachShadow;
+				Element.prototype.attachShadow = function attachShadow(options) {
+					return originalAttachShadow.call(this, { ...options, mode: "open" });
+				};
+			})();
+      `,
+        };
+    }
 }
 
 function getDefaultExportFromCjs (x) {
@@ -8099,11 +8148,11 @@ class FileAgent extends BaseFileAgent {
                 name: file,
                 path: filePath,
                 isDirectory: stats.isDirectory(),
-                size: stats.size,
-                modified: stats.mtime,
+                size: this.formatFileSize(stats.size),
+                modified: stats.mtime.toLocaleString(),
             };
         }));
-        return fileDetails.map((s) => s.path);
+        return fileDetails;
     }
     async file_read(agentContext, filePath) {
         return await fs.readFile(filePath, "utf-8");
@@ -8136,11 +8185,20 @@ class FileAgent extends BaseFileAgent {
                 name: path$1.basename(file),
                 path: file,
                 isDirectory: stats.isDirectory(),
-                size: stats.size,
-                modified: stats.mtime,
+                size: this.formatFileSize(stats.size),
+                modified: stats.mtime.toLocaleString(),
             };
         }));
-        return fileDetails.map((s) => s.path);
+        return fileDetails;
+    }
+    formatFileSize(size) {
+        if (size < 1024) {
+            return size + " B";
+        }
+        if (size < 1024 * 1024) {
+            return (size / 1024).toFixed(1) + " KB";
+        }
+        return (size / 1024 / 1024).toFixed(1) + " MB";
     }
 }