nothing-browser 0.0.16 → 0.0.17

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "nothing-browser",
-  "version": "0.0.16",
+  "version": "0.0.17",
   "description": "Browser automation library powered by Nothing Browser",
   "homepage": "https://github.com/ernest-tech-house-co-operation/nothing-browser#readme",
   "repository": {
@@ -49,6 +49,8 @@
     "prepublishOnly": "bun run build"
   },
   "dependencies": {
+    "@elysiajs/openapi": "^1.4.15",
+    "better-sqlite3": "^12.9.0",
     "elysia": "^1.0.0"
   },
   "peerDependencies": {

package/piggy/pool/index.d.ts

@@ -0,0 +1,12 @@
+// piggy/pool/index.d.ts
+import { PiggyClient } from "../client";
+
+export declare class TabPool {
+  constructor(client: PiggyClient, size: number, seedUrl: string, name: string);
+  init(): Promise<void>;
+  acquire(): Promise<string>;
+  release(tabId: string): void;
+  withTab<T>(fn: (tabId: string) => Promise<T>): Promise<T>;
+  close(): Promise<void>;
+  get stats(): { idle: number; busy: number; queued: number; total: number };
+}

package/piggy/pool/index.ts

@@ -0,0 +1,75 @@
+// piggy/pool/index.ts
+import { PiggyClient } from "../client";
+import logger from "../logger";
+
+export class TabPool {
+  private idle: string[] = [];
+  private busy = new Set<string>();
+  private queue: ((tabId: string) => void)[] = [];
+
+  constructor(
+    private client: PiggyClient,
+    private size: number,
+    private seedUrl: string,
+    private name: string
+  ) {}
+
+  async init() {
+    for (let i = 0; i < this.size; i++) {
+      const tabId = await this.client.newTab();
+      await this.client.navigate(this.seedUrl, tabId);
+      this.idle.push(tabId);
+      logger.success(`[${this.name}] pool tab ${i + 1}/${this.size} ready: ${tabId}`);
+    }
+  }
+
+  acquire(): Promise<string> {
+    return new Promise(resolve => {
+      const tabId = this.idle.pop();
+      if (tabId) {
+        this.busy.add(tabId);
+        resolve(tabId);
+      } else {
+        this.queue.push(resolve);
+      }
+    });
+  }
+
+  release(tabId: string) {
+    this.busy.delete(tabId);
+    const next = this.queue.shift();
+    if (next) {
+      this.busy.add(tabId);
+      next(tabId);
+    } else {
+      this.idle.push(tabId);
+    }
+  }
+
+  async withTab<T>(fn: (tabId: string) => Promise<T>): Promise<T> {
+    const tabId = await this.acquire();
+    try {
+      return await fn(tabId);
+    } finally {
+      this.release(tabId);
+    }
+  }
+
+  async close() {
+    for (const tabId of [...this.idle, ...this.busy]) {
+      try { await this.client.closeTab(tabId); } catch {}
+    }
+    this.idle = [];
+    this.busy.clear();
+    this.queue = [];
+  }
+
+  get stats() {
+    return {
+      idle:   this.idle.length,
+      busy:   this.busy.size,
+      queued: this.queue.length,
+      total:  this.size,
+    };
+  }
+}