WebMCP Tool Generator & Browser AI Configurator

import { WebMcpClient, defineBrowserTool } from "@webmcp/sdk";

async function extract_page_content(args) {
  const selector = args.selector || "main, article, body";
  const container = document.querySelector(selector) || document.body;

  if (args.includeHeadingsOnly) {
    const headings = Array.from(container.querySelectorAll("h1, h2, h3, h4"))
      .map(h => ({ tag: h.tagName.toLowerCase(), text: h.textContent?.trim() || "" }))
      .filter(h => h.text.length > 0);
    return { title: document.title, headingsCount: headings.length, headings };
  }

  const text = container.innerText.slice(0, 5000);
  return {
    title: document.title,
    url: window.location.href,
    contentLength: text.length,
    textContent: text
  };
}

export const extract_page_contentWebMcpTool = defineBrowserTool({
  name: "extract_page_content",
  description: "Extracts structured headings, main paragraph text, title, and metadata from the currently active web page DOM.",
  inputSchema: {
  "type": "object",
  "properties": {
    "selector": {
      "type": "string",
      "description": "CSS selector for the target container element (defaults to main or body)",
      "default": "main, article, body"
    },
    "includeHeadingsOnly": {
      "type": "boolean",
      "description": "Set to true to return only H1, H2, and H3 elements",
      "default": false
    }
  }
},
  execute: extract_page_content
});

// Initialize in client browser session
export function initBrowserMcpAgent(client: WebMcpClient) {
  client.registerTool(extract_page_contentWebMcpTool);
  console.log("Registered WebMCP Tool:", "extract_page_content");
}