nothing-browser 0.0.17 → 0.0.19

package/piggy.ts

@@ -1,5 +1,5 @@
 // piggy.ts
-import { detectBinary, type BinaryMode } from "./piggy/launch/detect";
+import { detectBinary } from "./piggy/launch/detect";
 import { spawnBrowser, killBrowser, spawnBrowserOnSocket } from "./piggy/launch/spawn";
 import { PiggyClient } from "./piggy/client";
 import { setClient, setHumanMode, createSiteObject, type SiteObject } from "./piggy/register";
@@ -9,14 +9,21 @@ import logger from "./piggy/logger";
 
 type TabMode = "tab" | "process";
 
+// "headless" and "headfull" are the built-in modes.
+// Any other string is treated as a raw path to a binary —
+// e.g. "../my-browser/browser", "C:\\browsers\\mybrowser.exe",
+// "/opt/weird-browser/bin/wb", etc.
+// The (string & {}) trick keeps autocomplete for the two literals
+// while still accepting any arbitrary string.
+export type BinaryMode = "headless" | "headfull" | (string & {});
+
 let _client: PiggyClient | null = null;
 let _tabMode: TabMode = "tab";
 const _extraProcs: { socket: string; client: PiggyClient }[] = [];
 const _sites: Record<string, SiteObject> = [];
 
 const piggy: any = {
-  // ── Lifecycle ─────────────────────────────────────────────────────────────
-
+  // ── Local launch (socket) ─────────────────────────────────────────────────
   launch: async (opts?: { mode?: TabMode; binary?: BinaryMode }) => {
     _tabMode = opts?.mode ?? "tab";
     const binaryMode: BinaryMode = opts?.binary ?? "headless";
@@ -29,12 +36,23 @@ const piggy: any = {
     return piggy;
   },
 
+  // ── Remote connect (HTTP) ─────────────────────────────────────────────────
+  connect: async (opts: { host: string; key: string }) => {
+    _tabMode = "tab";
+    _client = new PiggyClient({ host: opts.host, key: opts.key });
+    await _client.connect();
+    setClient(_client);
+    logger.info(`[piggy] connected (HTTP) → ${opts.host}`);
+    return piggy;
+  },
+
+  // ── Register ──────────────────────────────────────────────────────────────
   register: async (
     name: string,
     url: string,
     opts?: {
       binary?: BinaryMode;
-      pool?: number;          // number of tabs to pool — enables concurrent requests
+      pool?: number;
     }
   ) => {
     if (!url?.trim()) throw new Error(`No URL for site "${name}"`);
@@ -42,21 +60,16 @@ const piggy: any = {
     const poolSize = opts?.pool ?? 0;
 
     if (_tabMode === "tab") {
-      if (!_client) throw new Error("No client. Call piggy.launch() first.");
+      if (!_client) throw new Error("No client. Call piggy.launch() or piggy.connect() first.");
 
       if (poolSize > 1) {
-        // Pool mode — create N tabs, wrap in TabPool
         const pool = new TabPool(_client, poolSize, url, name);
         await pool.init();
-
-        // Use first tab as the "default" tabId for event subscriptions
-        const firstTabId = "default";
-        const siteObj = createSiteObject(name, url, _client, firstTabId, pool);
+        const siteObj = createSiteObject(name, url, _client, "default", pool);
         _sites[name] = siteObj;
         piggy[name] = siteObj;
         logger.success(`[${name}] registered with pool of ${poolSize} tabs`);
       } else {
-        // Single tab mode
         const tabId = await _client.newTab();
         const siteObj = createSiteObject(name, url, _client, tabId);
         _sites[name] = siteObj;
@@ -80,7 +93,6 @@ const piggy: any = {
   },
 
   // ── Global controls ───────────────────────────────────────────────────────
-
   actHuman: (enable: boolean) => {
     setHumanMode(enable);
     logger.info(`[piggy] actHuman: ${enable}`);
@@ -90,23 +102,41 @@ const piggy: any = {
   mode: (m: TabMode) => { _tabMode = m; return piggy; },
 
   // ── Expose Function ───────────────────────────────────────────────────────
-
   expose: async (name: string, handler: (data: any) => Promise<any> | any, tabId = "default") => {
-    if (!_client) throw new Error("No client. Call piggy.launch() first.");
+    if (!_client) throw new Error("No client. Call piggy.launch() or piggy.connect() first.");
     await _client.exposeFunction(name, handler, tabId);
     logger.success(`[piggy] exposed global function: ${name}`);
     return piggy;
   },
 
   unexpose: async (name: string, tabId = "default") => {
-    if (!_client) throw new Error("No client. Call piggy.launch() first.");
+    if (!_client) throw new Error("No client. Call piggy.launch() or piggy.connect() first.");
     await _client.unexposeFunction(name, tabId);
     logger.info(`[piggy] unexposed function: ${name}`);
     return piggy;
   },
 
-  // ── Elysia server ─────────────────────────────────────────────────────────
+  // ── Proxy ─────────────────────────────────────────────────────────────────
+  proxy: {
+    load:     (path: string)                                               => { if (!_client) throw new Error("No client"); return _client.proxyLoad(path); },
+    fetch:    (url: string)                                                => { if (!_client) throw new Error("No client"); return _client.proxyFetch(url); },
+    ovpn:     (path: string)                                               => { if (!_client) throw new Error("No client"); return _client.proxyOvpn(path); },
+    set:      (opts: Parameters<PiggyClient["proxySet"]>[0])               => { if (!_client) throw new Error("No client"); return _client.proxySet(opts); },
+    test:     ()                                                           => { if (!_client) throw new Error("No client"); return _client.proxyTest(); },
+    testStop: ()                                                           => { if (!_client) throw new Error("No client"); return _client.proxyTestStop(); },
+    next:     ()                                                           => { if (!_client) throw new Error("No client"); return _client.proxyNext(); },
+    disable:  ()                                                           => { if (!_client) throw new Error("No client"); return _client.proxyDisable(); },
+    enable:   ()                                                           => { if (!_client) throw new Error("No client"); return _client.proxyEnable(); },
+    current:  ()                                                           => { if (!_client) throw new Error("No client"); return _client.proxyCurrent(); },
+    stats:    ()                                                           => { if (!_client) throw new Error("No client"); return _client.proxyStats(); },
+    list:     (limit?: number)                                             => { if (!_client) throw new Error("No client"); return _client.proxyList(limit); },
+    rotation: (mode: "none" | "timed" | "perrequest", interval?: number)  => { if (!_client) throw new Error("No client"); return _client.proxyRotation(mode, interval); },
+    config:   (opts: { skipDead?: boolean; autoCheck?: boolean })          => { if (!_client) throw new Error("No client"); return _client.proxyConfig(opts); },
+    save:     (path: string, filter?: "alive" | "dead" | "all")           => { if (!_client) throw new Error("No client"); return _client.proxySave(path, filter); },
+    on:       (event: string, handler: (data: any) => void)               => { if (!_client) throw new Error("No client"); return _client.onProxyEvent(event, handler); },
+  },
 
+  // ── Elysia server ─────────────────────────────────────────────────────────
   serve: (
     port: number,
     opts?: {
@@ -121,7 +151,6 @@ const piggy: any = {
   stopServer,
 
   // ── Route listing ─────────────────────────────────────────────────────────
-
   routes: () =>
     Array.from(routeRegistry.entries()).map(([key, cfg]) => {
       const [site] = key.split(":");
@@ -135,7 +164,6 @@ const piggy: any = {
     }),
 
   // ── Multi-site ────────────────────────────────────────────────────────────
-
   all: (sites: SiteObject[]) =>
     new Proxy({} as any, {
       get: (_, method: string) =>
@@ -152,7 +180,6 @@ const piggy: any = {
     }),
 
   // ── Shutdown ──────────────────────────────────────────────────────────────
-
   close: async (opts?: { force?: boolean }) => {
     stopServer();
     if (opts?.force) {
@@ -182,9 +209,6 @@ const piggy: any = {
 };
 
 // ── usePiggy ──────────────────────────────────────────────────────────────────
-// Typed accessor — call AFTER register() so sites exist on piggy.
-// const { amazon, ebay } = usePiggy<"amazon" | "ebay">()
-
 type TypedPiggy<Sites extends string> = typeof piggy & {
   [K in Sites]: SiteObject;
 };