feat(domains): migrate page_info, wait, wait_for_load with typed timeout

1 files changed, 88 insertions, 0 deletions

diff --git a/src/domains/page.ts b/src/domains/page.ts
new file mode 100644
index 0000000..25b3c67
--- /dev/null
+++ b/src/domains/page.ts
@@ -0,0 +1,88 @@
+import { Type } from "typebox";
+import { type Result, err, ok } from "../util/result";
+import { defineBrowserTool, type ToolErr, type ToolOk } from "../util/tool";
+import { sleep } from "../util/time";
+import { safeJs } from "../util/js-template";
+
+export const pageInfoTool = defineBrowserTool({
+  name: "browser_page_info",
+  label: "Browser Page Info",
+  description:
+    "Get current page state: URL, title, viewport size, scroll position, page dimensions. If a JS dialog is open, returns dialog info instead.",
+  promptSnippet: "Get current page URL, title, viewport, and scroll position",
+  promptGuidelines: [
+    "Use browser_page_info to quickly check what page you're on and whether a JS dialog is blocking interaction.",
+    "If browser_page_info returns a dialog, use browser_handle_dialog before any other browser actions.",
+    "JS dialogs freeze the page's JS thread, so no other interaction works until the dialog is handled.",
+    "browser_page_info auto-detects alert, confirm, prompt, and beforeunload dialogs.",
+  ],
+  parameters: Type.Object({}),
+  async handler(_args, { client }): Promise<Result<ToolOk, ToolErr>> {
+    const r = await client.pageInfo();
+    if (!r.success) return err({ kind: "cdp_error", message: r.error.message });
+    if ("dialog" in r.data) {
+      const d = r.data.dialog;
+      return ok({
+        text: `⚠️  DIALOG OPEN: ${d.type}\nMessage: ${d.message}${d.defaultPrompt ? `\nDefault: ${d.defaultPrompt}` : ""}\n\nUse browser_handle_dialog to accept or dismiss.`,
+        details: { dialog: d },
+      });
+    }
+    const i = r.data;
+    return ok({
+      text: `URL: ${i.url}\nTitle: ${i.title}\nViewport: ${i.width}x${i.height}\nScroll: (${i.scrollX}, ${i.scrollY})\nPage size: ${i.pageWidth}x${i.pageHeight}`,
+      details: { ...i },
+    });
+  },
+});
+
+const WaitArgs = Type.Object({
+  seconds: Type.Number({ minimum: 0, maximum: 60, description: "Seconds to wait" }),
+});
+
+export const waitTool = defineBrowserTool({
+  name: "browser_wait",
+  label: "Browser Wait",
+  description: "Wait N seconds before continuing.",
+  promptSnippet: "Wait N seconds",
+  promptGuidelines: ["Use sparingly — prefer browser_wait_for_load when waiting for a page load."],
+  parameters: WaitArgs,
+  async handler(args, { signal }): Promise<Result<ToolOk, ToolErr>> {
+    try {
+      await sleep(Math.round(args.seconds * 1000), signal);
+      return ok({ text: `Waited ${args.seconds}s` });
+    } catch (e) {
+      return err({ kind: "internal", message: e instanceof Error ? e.message : String(e) });
+    }
+  },
+});
+
+const WaitForLoadArgs = Type.Object({
+  timeout: Type.Optional(Type.Number({ default: 15, minimum: 1, maximum: 120, description: "Max seconds to wait. Default: 15." })),
+});
+
+export const waitForLoadTool = defineBrowserTool({
+  name: "browser_wait_for_load",
+  label: "Browser Wait For Load",
+  description:
+    "Poll until the current page reports document.readyState === 'complete'. Returns a typed timeout error if the deadline elapses.",
+  promptSnippet: "Wait for the page to finish loading",
+  promptGuidelines: [
+    "Call after browser_navigate / browser_open_urls before extracting data.",
+    "Returns when readyState becomes 'complete' OR the timeout elapses.",
+  ],
+  parameters: WaitForLoadArgs,
+  async handler(args, { client, signal }): Promise<Result<ToolOk, ToolErr>> {
+    const timeoutMs = (args.timeout ?? 15) * 1000;
+    const start = Date.now();
+    while (Date.now() - start < timeoutMs) {
+      if (signal?.aborted) return err({ kind: "internal", message: "aborted" });
+      const r = await client.evaluateJs(safeJs`document.readyState`);
+      if (r.success && r.data === "complete") {
+        const ms = Date.now() - start;
+        return ok({ text: `Page loaded in ${Math.round(ms / 100) / 10}s`, details: { ms } });
+      }
+      await new Promise((res) => setTimeout(res, 50));
+    }
+    return err({ kind: "timeout", message: `Page did not finish loading in ${args.timeout ?? 15}s` });
+  },
+});