nothing-browser 0.0.8 → 0.0.9

package/dist/client/index.js

@@ -889,6 +889,9 @@ class PiggyClient {
   async evaluate(js, tabId = "default") {
     return this.send("evaluate", { js, tabId });
   }
+  async addInitScript(js, tabId = "default") {
+    await this.send("addInitScript", { js, tabId });
+  }
   async click(selector, tabId = "default") {
     return this.send("click", { selector, tabId });
   }
@@ -1039,9 +1042,8 @@ class PiggyClient {
   }
   async unexposeFunction(name, tabId = "default") {
     const handlers = this.eventHandlers.get(tabId);
-    if (handlers) {
+    if (handlers)
       handlers.delete(name);
-    }
     logger_default.info(`[${tabId}] unexposed function: ${name}`);
   }
   async clearExposedFunctions(tabId = "default") {

package/dist/piggy.js

@@ -6417,6 +6417,9 @@ class PiggyClient {
   async evaluate(js, tabId = "default") {
     return this.send("evaluate", { js, tabId });
   }
+  async addInitScript(js, tabId = "default") {
+    await this.send("addInitScript", { js, tabId });
+  }
   async click(selector, tabId = "default") {
     return this.send("click", { selector, tabId });
   }
@@ -6567,9 +6570,8 @@ class PiggyClient {
   }
   async unexposeFunction(name, tabId = "default") {
     const handlers = this.eventHandlers.get(tabId);
-    if (handlers) {
+    if (handlers)
       handlers.delete(name);
-    }
     logger_default.info(`[${tabId}] unexposed function: ${name}`);
   }
   async clearExposedFunctions(tabId = "default") {
@@ -21261,6 +21263,12 @@ function createSiteObject(name, registeredUrl, client, tabId) {
     },
     waitForVisible: (selector, timeout = 30000) => client.waitForSelector(selector, timeout, tabId),
     waitForResponse: (pattern, timeout = 30000) => client.waitForResponse(pattern, timeout, tabId),
+    addInitScript: async (js) => {
+      const code = typeof js === "function" ? `(${js.toString()})();` : js;
+      await client.addInitScript(code, tabId);
+      logger_default.success(`[${name}] init script added`);
+      return site;
+    },
     click: (selector, opts) => withErrScreen(() => retry(name, async () => {
       if (humanMode)
         await randomDelay(80, 220);
@@ -21370,9 +21378,9 @@ function createSiteObject(name, registeredUrl, client, tabId) {
       logger_default.info(`[${name}] images unblocked`);
     },
     cookies: {
-      set: async (name2, value, domain, path = "/") => {
-        await client.setCookie(name2, value, domain, path, tabId);
-        logger_default.info(`[${name2}] cookie set: ${name2} @ ${domain}`);
+      set: async (cookieName, value, domain, path = "/") => {
+        await client.setCookie(cookieName, value, domain, path, tabId);
+        logger_default.info(`[${name}] cookie set: ${cookieName} @ ${domain}`);
       },
       get: (cookieName) => client.getCookie(cookieName, tabId),
       delete: async (cookieName) => {

package/dist/register/index.js

@@ -16775,6 +16775,12 @@ function createSiteObject(name, registeredUrl, client, tabId) {
     },
     waitForVisible: (selector, timeout = 30000) => client.waitForSelector(selector, timeout, tabId),
     waitForResponse: (pattern, timeout = 30000) => client.waitForResponse(pattern, timeout, tabId),
+    addInitScript: async (js) => {
+      const code = typeof js === "function" ? `(${js.toString()})();` : js;
+      await client.addInitScript(code, tabId);
+      logger_default.success(`[${name}] init script added`);
+      return site;
+    },
     click: (selector, opts) => withErrScreen(() => retry(name, async () => {
       if (humanMode)
         await randomDelay(80, 220);
@@ -16884,9 +16890,9 @@ function createSiteObject(name, registeredUrl, client, tabId) {
       logger_default.info(`[${name}] images unblocked`);
     },
     cookies: {
-      set: async (name2, value, domain, path = "/") => {
-        await client.setCookie(name2, value, domain, path, tabId);
-        logger_default.info(`[${name2}] cookie set: ${name2} @ ${domain}`);
+      set: async (cookieName, value, domain, path = "/") => {
+        await client.setCookie(cookieName, value, domain, path, tabId);
+        logger_default.info(`[${name}] cookie set: ${cookieName} @ ${domain}`);
       },
       get: (cookieName) => client.getCookie(cookieName, tabId),
       delete: async (cookieName) => {
@@ -16999,9 +17005,8 @@ function createExposedAPI(site, apiName, handlers) {
   const wrappedHandler = async (call) => {
     const { method, args } = call;
     const handler = handlers[method];
-    if (!handler) {
+    if (!handler)
       throw new Error(`Unknown method: ${method}`);
-    }
     try {
       return await handler(args);
     } catch (err) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "nothing-browser",
-  "version": "0.0.8",
+  "version": "0.0.9",
   "description": "Browser automation library powered by Nothing Browser",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",

package/piggy/client/index.ts

@@ -45,13 +45,11 @@ export class PiggyClient {
           try {
             const msg = JSON.parse(line);
             
-            // Handle events
             if (msg.type === "event") {
               this.handleEvent(msg);
               continue;
             }
             
-            // Handle command responses
             const p = this.pending.get(msg.id);
             if (p) {
               this.pending.delete(msg.id);
@@ -103,7 +101,6 @@ export class PiggyClient {
                 }).catch(e => logger.error(`Failed to send exposed error: ${e}`));
               }
             } else {
-              // Handler returned raw value
               this.send("exposed.result", {
                 tabId: effectiveTabId,
                 callId,
@@ -192,6 +189,12 @@ export class PiggyClient {
     return this.send("evaluate", { js, tabId });
   }
 
+  // ── Init Script ───────────────────────────────────────────────────────────────
+  // HERE IT IS - ADD THIS METHOD TO THE CLIENT
+  async addInitScript(js: string, tabId = "default"): Promise<void> {
+    await this.send("addInitScript", { js, tabId });
+  }
+
   // ── Interactions ──────────────────────────────────────────────────────────────
 
   async click(selector: string, tabId = "default"): Promise<boolean> {
@@ -318,13 +321,7 @@ export class PiggyClient {
 
   // ── Cookies ───────────────────────────────────────────────────────────────────
 
-  async setCookie(
-    name: string,
-    value: string,
-    domain: string,
-    path = "/",
-    tabId = "default"
-  ): Promise<void> {
+  async setCookie(name: string, value: string, domain: string, path = "/", tabId = "default"): Promise<void> {
     await this.send("cookie.set", { name, value, domain, path, tabId });
   }
 
@@ -342,12 +339,7 @@ export class PiggyClient {
 
   // ── Interception ──────────────────────────────────────────────────────────────
 
-  async addInterceptRule(
-    action: "block" | "redirect" | "modifyHeaders",
-    pattern: string,
-    options: { redirectUrl?: string; headers?: Record<string, string> } = {},
-    tabId = "default"
-  ): Promise<void> {
+  async addInterceptRule(action: "block" | "redirect" | "modifyHeaders", pattern: string, options: { redirectUrl?: string; headers?: Record<string, string> } = {}, tabId = "default"): Promise<void> {
     await this.send("intercept.rule.add", { action, pattern, ...options, tabId });
   }
 
@@ -397,15 +389,10 @@ export class PiggyClient {
 
   // ── Expose Function ───────────────────────────────────────────────────────────
 
-  async exposeFunction(
-    name: string,
-    handler: (data: any) => Promise<any> | any,
-    tabId = "default"
-  ): Promise<void> {
+  async exposeFunction(name: string, handler: (data: any) => Promise<any> | any, tabId = "default"): Promise<void> {
     if (!this.eventHandlers.has(tabId)) {
       this.eventHandlers.set(tabId, new Map());
     }
-    
     this.eventHandlers.get(tabId)!.set(name, async (data: any) => {
       try {
         const result = await handler(data);
@@ -417,16 +404,13 @@ export class PiggyClient {
         return { success: false, error: err.message || String(err) };
       }
     });
-    
     await this.send("expose.function", { name, tabId });
     logger.success(`[${tabId}] exposed function: ${name}`);
   }
 
   async unexposeFunction(name: string, tabId = "default"): Promise<void> {
     const handlers = this.eventHandlers.get(tabId);
-    if (handlers) {
-      handlers.delete(name);
-    }
+    if (handlers) handlers.delete(name);
     logger.info(`[${tabId}] unexposed function: ${name}`);
   }
 

package/piggy/register/index.ts

@@ -10,12 +10,7 @@ export let humanMode = false;
 export function setClient(c: PiggyClient | null) { globalClient = c; }
 export function setHumanMode(v: boolean) { humanMode = v; }
 
-async function retry<T>(
-  label: string,
-  fn: () => Promise<T>,
-  retries = 2,
-  backoff = 150
-): Promise<T> {
+async function retry<T>(label: string, fn: () => Promise<T>, retries = 2, backoff = 150): Promise<T> {
   let last!: Error;
   for (let i = 0; i <= retries; i++) {
     try { return await fn(); } catch (e: any) {
@@ -29,12 +24,7 @@ async function retry<T>(
   throw last;
 }
 
-export function createSiteObject(
-  name: string,
-  registeredUrl: string,
-  client: PiggyClient,
-  tabId: string
-) {
+export function createSiteObject(name: string, registeredUrl: string, client: PiggyClient, tabId: string) {
   let _currentUrl: string = registeredUrl;
 
   const withErrScreen = async <T>(fn: () => Promise<T>, label: string): Promise<T> => {
@@ -51,7 +41,6 @@ export function createSiteObject(
     _tabId: tabId,
 
     // ── Navigation ─────────────────────────────────────────────────────────────
-
     navigate: (url?: string, opts?: { retries?: number }) => {
       const target = url ?? registeredUrl;
       return retry(name, async () => {
@@ -86,8 +75,16 @@ export function createSiteObject(
     waitForVisible:  (selector: string, timeout = 30000) => client.waitForSelector(selector, timeout, tabId),
     waitForResponse: (pattern: string, timeout = 30000)  => client.waitForResponse(pattern, timeout, tabId),
 
-    // ── Interactions ───────────────────────────────────────────────────────────
+    // ── Init Script ─────────────────────────────────────────────────────────────
+    // HERE IT IS - ADD THIS METHOD TO THE SITE OBJECT
+    addInitScript: async (js: string | (() => void)) => {
+      const code = typeof js === 'function' ? `(${js.toString()})();` : js;
+      await client.addInitScript(code, tabId);
+      logger.success(`[${name}] init script added`);
+      return site;
+    },
 
+    // ── Interactions ───────────────────────────────────────────────────────────
     click: (selector: string, opts?: { retries?: number; timeout?: number }) =>
       withErrScreen(() =>
         retry(name, async () => {
@@ -173,7 +170,6 @@ export function createSiteObject(
     },
 
     // ── Fetch ──────────────────────────────────────────────────────────────────
-
     fetchText:   (selector: string) => client.fetchText(selector, tabId),
     fetchLinks:  async (selector: string) => {
       const links = await client.fetchLinks(selector, tabId);
@@ -192,7 +188,6 @@ export function createSiteObject(
     },
 
     // ── Screenshot / PDF ───────────────────────────────────────────────────────
-
     screenshot: async (filePath?: string) => {
       const r = await client.screenshot(filePath, tabId);
       logger.success(`[${name}] screenshot → ${filePath ?? "base64"}`);
@@ -208,11 +203,10 @@ export function createSiteObject(
     unblockImages: async () => { await client.unblockImages(tabId); logger.info(`[${name}] images unblocked`); },
 
     // ── Cookies ────────────────────────────────────────────────────────────────
-
     cookies: {
-      set: async (name: string, value: string, domain: string, path = "/") => {
-        await client.setCookie(name, value, domain, path, tabId);
-        logger.info(`[${name}] cookie set: ${name} @ ${domain}`);
+      set: async (cookieName: string, value: string, domain: string, path = "/") => {
+        await client.setCookie(cookieName, value, domain, path, tabId);
+        logger.info(`[${name}] cookie set: ${cookieName} @ ${domain}`);
       },
       get: (cookieName: string) => client.getCookie(cookieName, tabId),
       delete: async (cookieName: string) => {
@@ -223,7 +217,6 @@ export function createSiteObject(
     },
 
     // ── Interception ───────────────────────────────────────────────────────────
-
     intercept: {
       block: async (pattern: string) => {
         await client.addInterceptRule("block", pattern, {}, tabId);
@@ -244,7 +237,6 @@ export function createSiteObject(
     },
 
     // ── Network capture ────────────────────────────────────────────────────────
-
     capture: {
       start: async () => {
         await client.captureStart(tabId);
@@ -265,7 +257,6 @@ export function createSiteObject(
     },
 
     // ── Session ────────────────────────────────────────────────────────────────
-
     session: {
       export: async () => {
         const data = await client.sessionExport(tabId);
@@ -279,7 +270,6 @@ export function createSiteObject(
     },
 
     // ── Expose Function ─────────────────────────────────────────────────────────
-
     exposeFunction: async (fnName: string, handler: (data: any) => Promise<any> | any) => {
       await client.exposeFunction(fnName, handler, tabId);
       logger.success(`[${name}] exposed function: ${fnName}`);
@@ -298,29 +288,16 @@ export function createSiteObject(
       return site;
     },
 
-    exposeAndInject: async (
-      fnName: string,
-      handler: (data: any) => Promise<any> | any,
-      injectionJs: string | ((fnName: string) => string)
-    ) => {
+    exposeAndInject: async (fnName: string, handler: (data: any) => Promise<any> | any, injectionJs: string | ((fnName: string) => string)) => {
       await client.exposeFunction(fnName, handler, tabId);
-      
-      const js = typeof injectionJs === "function" 
-        ? injectionJs(fnName) 
-        : injectionJs;
-        
+      const js = typeof injectionJs === "function" ? injectionJs(fnName) : injectionJs;
       await client.evaluate(js, tabId);
       logger.success(`[${name}] exposed and injected: ${fnName}`);
       return site;
     },
 
     // ── Elysia API ─────────────────────────────────────────────────────────────
-
-    api: (
-      path: string,
-      handler: RouteHandler,
-      opts?: { ttl?: number; before?: BeforeMiddleware[]; method?: "GET" | "POST" | "PUT" | "DELETE" }
-    ) => {
+    api: (path: string, handler: RouteHandler, opts?: { ttl?: number; before?: BeforeMiddleware[]; method?: "GET" | "POST" | "PUT" | "DELETE" }) => {
       const key = `${name}:${path}`;
       if (routeRegistry.has(key)) {
         logger.warn(`[${name}] route ${path} already registered`);
@@ -351,21 +328,11 @@ export function createSiteObject(
   return site;
 }
 
-// ── Helper for creating structured APIs ───────────────────────────────────────
-
-export function createExposedAPI<T extends Record<string, (data: any) => any>>(
-  site: any,
-  apiName: string,
-  handlers: T
-): Promise<void> {
+export function createExposedAPI<T extends Record<string, (data: any) => any>>(site: any, apiName: string, handlers: T): Promise<void> {
   const wrappedHandler = async (call: any) => {
     const { method, args } = call;
     const handler = handlers[method as keyof T];
-    
-    if (!handler) {
-      throw new Error(`Unknown method: ${method}`);
-    }
-    
+    if (!handler) throw new Error(`Unknown method: ${method}`);
     try {
       return await handler(args);
     } catch (err) {
@@ -373,6 +340,5 @@ export function createExposedAPI<T extends Record<string, (data: any) => any>>(
       throw err;
     }
   };
-  
   return site.exposeFunction(apiName, wrappedHandler);
 }