npm - @adeu/mcp-server - Versions diffs - 1.8.0 → 1.9.0 - Mend

@adeu/mcp-server 1.8.0 → 1.9.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (18) hide show

package/dist/assets/adeu.svg +3 -0
package/dist/assets/logo.png +0 -0
package/dist/assets/marked.min.js +69 -0
package/dist/index.js +523 -395
package/dist/index.js.map +1 -1
package/dist/templates/email_ui.html +667 -0
package/dist/templates/markdown_ui.html +745 -0
package/package.json +4 -2
package/src/assets/adeu.svg +3 -0
package/src/assets/logo.png +0 -0
package/src/assets/marked.min.js +69 -0
package/src/index.ts +510 -407
package/src/mcp.cloud.test.ts +13 -0
package/src/response-builders.ts +111 -50
package/src/templates/email_ui.html +667 -0
package/src/templates/markdown_ui.html +745 -0
package/src/tools/email.ts +61 -2
package/tsup.config.ts +35 -11

package/src/index.ts CHANGED Viewed

@@ -1,11 +1,15 @@
-import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+// FILE: node/packages/mcp-server/src/index.ts
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { readFileSync, existsSync } from "node:fs";
+import { basename, resolve, extname, dirname, join } from "node:path";
+import { z } from "zod";
 import {
-  CallToolRequestSchema,
-  ListToolsRequestSchema,
-} from "@modelcontextprotocol/sdk/types.js";
-import { readFileSync } from "node:fs";
-import { basename, resolve, extname, dirname } from "node:path";
+  registerAppTool,
+  registerAppResource,
+  RESOURCE_MIME_TYPE,
+} from "@modelcontextprotocol/ext-apps/server";
+import fs from "node:fs";
 import {
   identifyEngine,
   extractTextFromBuffer,
@@ -15,13 +19,21 @@ import {
   create_word_patch_diff,
   finalize_document,
 } from "@adeu/core";
 import {
   build_paginated_response,
   build_outline_response,
   build_appendix_response,
 } from "./response-builders.js";
 import { login_to_adeu_cloud, logout_of_adeu_cloud } from "./tools/auth.js";
-import { search_and_fetch_emails, create_email_draft } from "./tools/email.js";
+import {
+  search_and_fetch_emails,
+  create_email_draft,
+  list_available_mailboxes,
+} from "./tools/email.js";
+import { MARKDOWN_UI_URI, EMAIL_UI_URI } from "./shared.js";
 function readFileBytesOrThrow(filePath: string): Buffer {
   try {
     return readFileSync(filePath);
@@ -32,7 +44,23 @@ function readFileBytesOrThrow(filePath: string): Buffer {
     throw err;
   }
 }
-// --- Tool Description Constants (Parity with Python) ---
+// --- Asset Loaders for UI ---
+const DIST_DIR = import.meta.dirname;
+function getAssetContent(
+  folder: "templates" | "assets",
+  filename: string,
+  fallbackMessage: string,
+): string {
+  const filePath = join(DIST_DIR, folder, filename);
+  if (existsSync(filePath)) {
+    return readFileSync(filePath, "utf-8");
+  }
+  return fallbackMessage;
+}
+// --- Tool Description Constants ---
 const READ_DOCX_COMMON_DESC =
   "Reads a DOCX file. Returns text with inline CriticMarkup for Tracked Changes and Comments: {++inserted++}, {--deleted--}, {==highlighted==}{>>comment<<}. Set clean_view=True for the finalized 'Accepted' text without markup.\n\n";
 const READ_DOCX_TAIL =
@@ -47,451 +75,526 @@ const DIFF_DOCX_DESC =
   "Compares two DOCX files and returns a unified diff of their text content. Useful for analyzing differences between versions before editing.";
 // --- Server Setup ---
-const server = new Server(
-  {
-    name: "adeu-redlining-service",
-    version: "1.0.0",
-  },
-  {
-    capabilities: {
-      tools: {},
-    },
+const server = new McpServer({
+  name: "adeu-redlining-service",
+  version: "1.0.0",
+});
+// Common CSP allowing Google Fonts used by Adeu UI templates
+const UI_CSP = {
+  connectDomains: ["https://fonts.googleapis.com", "https://fonts.gstatic.com"],
+  resourceDomains: [
+    "https://fonts.googleapis.com",
+    "https://fonts.gstatic.com",
+  ],
+};
+// ==========================================
+// 1. UI RESOURCES
+// ==========================================
+registerAppResource(
+  server,
+  MARKDOWN_UI_URI,
+  MARKDOWN_UI_URI,
+  { mimeType: RESOURCE_MIME_TYPE, description: "Adeu Markdown Viewer UI" },
+  async () => {
+    let html = getAssetContent(
+      "templates",
+      "markdown_ui.html",
+      "<html><body>UI Template Not Found</body></html>",
+    );
+    const markedJs = getAssetContent(
+      "assets",
+      "marked.min.js",
+      "window.__MARKED_ERROR = 'marked.min.js not found';",
+    );
+    const svg = getAssetContent("assets", "adeu.svg", "");
+    html = html
+      .replace("[[marked_js_code | safe]]", markedJs)
+      .replace("[[ adeu_svg_code ]]", svg);
+    return {
+      contents: [
+        {
+          uri: MARKDOWN_UI_URI,
+          mimeType: RESOURCE_MIME_TYPE,
+          text: html,
+          _meta: { ui: { csp: UI_CSP } },
+        },
+      ],
+    };
   },
 );
-// --- Tool Registration ---
-server.setRequestHandler(ListToolsRequestSchema, async () => {
-  return {
-    tools: [
-      {
-        name: "read_docx",
-        description: READ_DOCX_COMMON_DESC + READ_DOCX_TAIL,
-        inputSchema: {
-          type: "object",
-          properties: {
-            file_path: {
-              type: "string",
-              description: "Absolute path to the DOCX file.",
-            },
-            clean_view: {
-              type: "boolean",
-              description:
-                "If False (default), returns the 'Raw' text with inline CriticMarkup. If True, returns 'Accepted' text.",
-              default: false,
-            },
-            mode: {
-              type: "string",
-              enum: ["full", "outline", "appendix"],
-              description:
-                "'full' returns body content. 'outline' returns a structural heading map. 'appendix' returns defined terms.",
-              default: "full",
-            },
-            page: {
-              type: "number",
-              description:
-                "Page number (1-indexed) for mode='full'. Defaults to 1.",
-              default: 1,
-            },
-            outline_max_level: {
-              type: "number",
-              description: "For mode='outline' only: cap on heading depth.",
-              default: 2,
-            },
-            outline_verbose: {
-              type: "boolean",
-              description: "For mode='outline' only: includes metadata.",
-              default: false,
-            },
-          },
-          required: ["file_path"],
-        },
-      },
-      {
-        name: "process_document_batch",
-        description: PROCESS_BATCH_COMMON_DESC + PROCESS_BATCH_OPERATIONS_DESC,
-        inputSchema: {
-          type: "object",
-          properties: {
-            original_docx_path: {
-              type: "string",
-              description: "Absolute path to the source file.",
-            },
-            author_name: {
-              type: "string",
-              description:
-                "Name to appear in Track Changes (e.g., 'Reviewer AI').",
-            },
-            changes: {
-              type: "array",
-              description:
-                "List of changes to apply. Each change must specify 'type'.",
-              items: { type: "object" },
-            },
-            output_path: {
-              type: "string",
-              description: "Optional output path.",
-            },
-          },
-          required: ["original_docx_path", "author_name", "changes"],
-        },
-      },
-      {
-        name: "accept_all_changes",
-        description:
-          "Accepts all tracked changes and removes all comments in a single operation, producing a finalized clean document. Use this when a document review is entirely complete and you want to clear all redlines.",
-        inputSchema: {
-          type: "object",
-          properties: {
-            docx_path: {
-              type: "string",
-              description: "Absolute path to the DOCX file.",
-            },
-            output_path: {
-              type: "string",
-              description: "Optional output path.",
-            },
-          },
-          required: ["docx_path"],
-        },
-      },
-      {
-        name: "diff_docx_files",
-        description: DIFF_DOCX_DESC,
-        inputSchema: {
-          type: "object",
-          properties: {
-            original_path: {
-              type: "string",
-              description: "Absolute path to the baseline DOCX file.",
-            },
-            modified_path: {
-              type: "string",
-              description: "Absolute path to the modified DOCX file.",
-            },
-            compare_clean: {
-              type: "boolean",
-              description:
-                "If True, compares 'Accepted' state. If False, compares raw text.",
-              default: true,
-            },
-          },
-          required: ["original_path", "modified_path"],
+registerAppResource(
+  server,
+  EMAIL_UI_URI,
+  EMAIL_UI_URI,
+  { mimeType: RESOURCE_MIME_TYPE, description: "Adeu Email Viewer UI" },
+  async () => {
+    let html = getAssetContent(
+      "templates",
+      "email_ui.html",
+      "<html><body>UI Template Not Found</body></html>",
+    );
+    const svg = getAssetContent("assets", "adeu.svg", "");
+    html = html.replace("[[ adeu_svg_code ]]", svg);
+    return {
+      contents: [
+        {
+          uri: EMAIL_UI_URI,
+          mimeType: RESOURCE_MIME_TYPE,
+          text: html,
+          _meta: { ui: { csp: UI_CSP } },
         },
-      },
-      {
-        name: "finalize_document",
-        description:
-          "Prepares a document for external distribution or e-signature. This tool combines metadata sanitization, document locking (protection), and markup resolution into a single step. NOTE: PDF export and AES encryption are disabled in this environment.",
-        inputSchema: {
-          type: "object",
-          properties: {
-            file_path: {
-              type: "string",
-              description: "Absolute path to the DOCX file.",
-            },
-            output_path: {
-              type: "string",
-              description: "Optional output path.",
-            },
-            sanitize_mode: {
-              type: "string",
-              enum: ["full", "keep-markup"],
-              description:
-                "full removes all markup, keep-markup redacts metadata but keeps comments/redlines.",
-            },
-            accept_all: {
-              type: "boolean",
-              description:
-                "If true, auto-accepts all unresolved track changes before finalizing.",
-            },
-            protection_mode: {
-              type: "string",
-              enum: ["read_only", "encrypt"],
-              description:
-                "Native OOXML document locking. encrypt falls back to read_only in this environment.",
-            },
-            password: {
-              type: "string",
-              description: "Ignored in this environment.",
-            },
-            author: {
-              type: "string",
-              description:
-                "Replace all remaining markup authorship with this name.",
-            },
-            export_pdf: {
-              type: "boolean",
-              description: "Ignored in this environment.",
-            },
-          },
-          required: ["file_path"],
-        },
-      },
-      {
-        name: "login_to_adeu_cloud",
-        description:
-          "Logs the user into the Adeu Cloud backend. Securely opens a browser window for authentication.",
-        inputSchema: { type: "object", properties: {} },
-      },
-      {
-        name: "logout_of_adeu_cloud",
-        description:
-          "Logs out of the Adeu Cloud backend by clearing the local API key.",
-        inputSchema: { type: "object", properties: {} },
-      },
-      {
-        name: "search_and_fetch_emails",
-        description:
-          "Searches the user's live email inbox. By default, searches only the Inbox folder. Returns a list of lightweight previews. Call again with `email_id` to fetch the full body and download attachments.",
-        inputSchema: {
-          type: "object",
-          properties: {
-            sender: { type: "string" },
-            subject: { type: "string" },
-            has_attachments: { type: "boolean" },
-            attachment_name: { type: "string" },
-            is_unread: { type: "boolean" },
-            days_ago: { type: "number" },
-            folder: { type: "string", enum: ["inbox", "sent", "all"] },
-            limit: { type: "number", default: 10 },
-            offset: { type: "number", default: 0 },
-            email_id: { type: "string" },
-            working_directory: { type: "string" },
+      ],
+    };
+  },
+);
+// ==========================================
+// 2. UI-ENABLED TOOLS
+// ==========================================
+registerAppTool(
+  server,
+  "read_docx",
+  {
+    title: "Read DOCX",
+    description: READ_DOCX_COMMON_DESC + READ_DOCX_TAIL,
+    inputSchema: z.object({
+      file_path: z.string().describe("Absolute path to the DOCX file."),
+      clean_view: z
+        .boolean()
+        .default(false)
+        .describe(
+          "If False (default), returns the 'Raw' text with inline CriticMarkup. If True, returns 'Accepted' text.",
+        ),
+      mode: z
+        .enum(["full", "outline", "appendix"])
+        .default("full")
+        .describe(
+          "'full' returns body content. 'outline' returns a structural heading map. 'appendix' returns defined terms.",
+        ),
+      page: z
+        .number()
+        .default(1)
+        .describe("Page number (1-indexed) for mode='full'. Defaults to 1."),
+      outline_max_level: z
+        .number()
+        .default(2)
+        .describe("For mode='outline' only: cap on heading depth."),
+      outline_verbose: z
+        .boolean()
+        .default(false)
+        .describe("For mode='outline' only: includes metadata."),
+    }),
+    _meta: { ui: { resourceUri: MARKDOWN_UI_URI } },
+  },
+  async ({
+    file_path,
+    clean_view,
+    mode,
+    page,
+    outline_max_level,
+    outline_verbose,
+  }) => {
+    try {
+      const buf = readFileBytesOrThrow(file_path);
+      const text = await extractTextFromBuffer(buf, clean_view);
+      if (mode === "outline") {
+        const doc = await DocumentObject.load(buf);
+        return build_outline_response(
+          doc,
+          text,
+          file_path,
+          outline_max_level,
+          outline_verbose,
+        ) as any;
+      }
+      if (mode === "appendix") {
+        return build_appendix_response(text, page, file_path) as any;
+      }
+      return build_paginated_response(text, page, file_path) as any;
+    } catch (e: any) {
+      return {
+        isError: true,
+        content: [
+          {
+            type: "text",
+            text: `Error executing tool read_docx: ${e.message}`,
           },
-        },
-      },
-      {
-        name: "create_email_draft",
-        description:
-          "Creates an email draft in the user's native draft box. Provide `reply_to_email_id` to reply, or `subject` and `to_recipients` for a new email.",
-        inputSchema: {
-          type: "object",
-          properties: {
-            body_markdown: { type: "string" },
-            reply_to_email_id: { type: "string" },
-            subject: { type: "string" },
-            to_recipients: { type: "array", items: { type: "string" } },
-            attachment_paths: { type: "array", items: { type: "string" } },
+        ],
+      };
+    }
+  },
+);
+registerAppTool(
+  server,
+  "search_and_fetch_emails",
+  {
+    title: "Search & Fetch Emails",
+    description:
+      "Searches the user's live email inbox. Returns previews. Call again with `email_id` to fetch the full body.",
+    inputSchema: z.object({
+      sender: z.string().optional(),
+      subject: z.string().optional(),
+      has_attachments: z.boolean().optional(),
+      attachment_name: z.string().optional(),
+      is_unread: z.boolean().optional(),
+      days_ago: z.number().optional(),
+      folder: z.enum(["inbox", "sent", "all"]).optional(),
+      limit: z.number().default(10),
+      offset: z.number().default(0),
+      email_id: z.string().optional(),
+      working_directory: z.string().optional(),
+      mailbox_address: z
+        .string()
+        .optional()
+        .describe("Optional target mailbox email address to search within."),
+    }),
+    _meta: { ui: { resourceUri: EMAIL_UI_URI } },
+  },
+  async (args) => {
+    try {
+      return (await search_and_fetch_emails(args)) as any;
+    } catch (e: any) {
+      return {
+        isError: true,
+        content: [
+          {
+            type: "text",
+            text: `Error executing tool search_and_fetch_emails: ${e.message}`,
           },
-          required: ["body_markdown"],
-        },
-      },
-    ],
-  };
-});
+        ],
+      };
+    }
+  },
+);
-// --- Tool Execution ---
-server.setRequestHandler(
-  CallToolRequestSchema,
-  async (request): Promise<any> => {
-    const { name, arguments: args } = request.params;
+// ==========================================
+// 3. HEADLESS TOOLS (No UI)
+// ==========================================
+server.registerTool(
+  "process_document_batch",
+  {
+    description: PROCESS_BATCH_COMMON_DESC + PROCESS_BATCH_OPERATIONS_DESC,
+    inputSchema: {
+      original_docx_path: z
+        .string()
+        .describe("Absolute path to the source file."),
+      author_name: z
+        .string()
+        .describe("Name to appear in Track Changes (e.g., 'Reviewer AI')."),
+      changes: z
+        .array(z.any())
+        .describe("List of changes to apply. Each change must specify 'type'."),
+      output_path: z.string().optional().describe("Optional output path."),
+    },
+  },
+  async ({ original_docx_path, author_name, changes, output_path }) => {
     try {
-      if (name === "read_docx") {
-        const filePath = args?.file_path as string;
-        const cleanView = (args?.clean_view as boolean) ?? false;
-        const mode = (args?.mode as string) ?? "full";
-        const page = (args?.page as number) ?? 1;
-        const outline_max_level = (args?.outline_max_level as number) ?? 2;
-        const outline_verbose = (args?.outline_verbose as boolean) ?? false;
-        const buf = readFileBytesOrThrow(filePath);
-        const text = await extractTextFromBuffer(buf, cleanView);
-        if (mode === "outline") {
-          const doc = await DocumentObject.load(buf);
-          return build_outline_response(
-            doc,
-            text,
-            filePath,
-            outline_max_level,
-            outline_verbose,
-          );
-        }
-        if (mode === "appendix") {
-          return build_appendix_response(text, page, filePath);
-        }
-        return build_paginated_response(text, page, filePath);
+      if (!author_name || !author_name.trim())
+        return {
+          content: [
+            { type: "text", text: "Error: author_name cannot be empty." },
+          ],
+        };
+      if (!changes || changes.length === 0)
+        return {
+          content: [{ type: "text", text: "Error: No changes provided." }],
+        };
+      let outPath = output_path;
+      if (!outPath) {
+        const ext = extname(original_docx_path);
+        const base = basename(original_docx_path, ext);
+        const dir = dirname(original_docx_path);
+        outPath = resolve(dir, `${base}_processed${ext}`);
       }
-      if (name === "process_document_batch") {
-        const origPath = args?.original_docx_path as string;
-        const authorName = args?.author_name as string;
-        const changes = args?.changes as any[];
-        let outPath = args?.output_path as string;
-        if (!authorName || !authorName.trim()) {
+      const buf = readFileBytesOrThrow(original_docx_path);
+      const doc = await DocumentObject.load(buf);
+      const engine = new RedlineEngine(doc, author_name);
+      let stats;
+      try {
+        stats = engine.process_batch(changes);
+      } catch (e: any) {
+        if (e instanceof BatchValidationError) {
           return {
+            isError: true,
             content: [
-              { type: "text", text: "Error: author_name cannot be empty." },
+              {
+                type: "text",
+                text: `Batch rejected. Some edits failed validation:\n\n${e.errors.join("\n\n")}`,
+              },
             ],
           };
         }
+        throw e;
+      }
-        if (!changes || changes.length === 0) {
-          return {
-            content: [{ type: "text", text: "Error: No changes provided." }],
-          };
-        }
-        if (!outPath) {
-          const ext = extname(origPath);
-          const base = basename(origPath, ext);
-          const dir = dirname(origPath);
-          outPath = resolve(dir, `${base}_processed${ext}`);
-        }
-        const buf = readFileBytesOrThrow(origPath);
-        const doc = await DocumentObject.load(buf);
-        const engine = new RedlineEngine(doc, authorName);
-        let stats;
-        try {
-          stats = engine.process_batch(changes);
-        } catch (e) {
-          if (e instanceof BatchValidationError) {
-            return {
-              content: [
-                {
-                  type: "text",
-                  text: `Batch rejected. Some edits failed validation:\n\n${(e as BatchValidationError).errors.join("\n\n")}`,
-                },
-              ],
-              isError: true,
-            };
-          }
-          throw e;
-        }
+      const outBuf = await doc.save();
-        const outBuf = await doc.save();
-        // Using dynamic import of fs/promises or just sync write
-        const fs = await import("node:fs");
-        fs.writeFileSync(outPath, outBuf);
+      fs.writeFileSync(outPath, outBuf);
-        let res = `Batch complete. Saved to: ${outPath}\nActions: ${stats.actions_applied} applied, ${stats.actions_skipped} skipped.\nEdits: ${stats.edits_applied} applied, ${stats.edits_skipped} skipped.`;
-        if (stats.skipped_details?.length > 0) {
-          res += `\n\nSkipped Details:\n${stats.skipped_details.join("\n")}`;
-        }
+      let res = `Batch complete. Saved to: ${outPath}\nActions: ${stats.actions_applied} applied, ${stats.actions_skipped} skipped.\nEdits: ${stats.edits_applied} applied, ${stats.edits_skipped} skipped.`;
+      if (stats.skipped_details?.length > 0) {
+        res += `\n\nSkipped Details:\n${stats.skipped_details.join("\n")}`;
+      }
+      return { content: [{ type: "text", text: res }] };
+    } catch (e: any) {
+      return {
+        isError: true,
+        content: [{ type: "text", text: `Error: ${e.message}` }],
+      };
+    }
+  },
+);
-        return {
-          content: [{ type: "text", text: res }],
-        };
+server.registerTool(
+  "accept_all_changes",
+  {
+    description:
+      "Accepts all tracked changes and removes all comments in a single operation.",
+    inputSchema: {
+      docx_path: z.string().describe("Absolute path to the DOCX file."),
+      output_path: z.string().optional().describe("Optional output path."),
+    },
+  },
+  async ({ docx_path, output_path }) => {
+    try {
+      let outPath = output_path;
+      if (!outPath) {
+        const ext = extname(docx_path);
+        const base = basename(docx_path, ext);
+        const dir = dirname(docx_path);
+        outPath = resolve(dir, `${base}_clean${ext}`);
       }
-      if (name === "accept_all_changes") {
-        const docxPath = args?.docx_path as string;
-        let outPath = args?.output_path as string;
+      const buf = readFileBytesOrThrow(docx_path);
+      const doc = await DocumentObject.load(buf);
+      const engine = new RedlineEngine(doc);
-        if (!outPath) {
-          const ext = extname(docxPath);
-          const base = basename(docxPath, ext);
-          const dir = dirname(docxPath);
-          outPath = resolve(dir, `${base}_clean${ext}`);
-        }
+      engine.accept_all_revisions();
-        const buf = readFileBytesOrThrow(docxPath);
-        const doc = await DocumentObject.load(buf);
-        const engine = new RedlineEngine(doc);
+      const outBuf = await doc.save();
-        // We implement the public facing accept_all wrapper from python
-        engine.accept_all_revisions();
+      fs.writeFileSync(outPath, outBuf);
-        const outBuf = await doc.save();
-        const fs = await import("node:fs");
-        fs.writeFileSync(outPath, outBuf);
+      return {
+        content: [
+          { type: "text", text: `Accepted all changes. Saved to: ${outPath}` },
+        ],
+      };
+    } catch (e: any) {
+      return {
+        isError: true,
+        content: [{ type: "text", text: `Error: ${e.message}` }],
+      };
+    }
+  },
+);
-        return {
-          content: [
-            {
-              type: "text",
-              text: `Accepted all changes. Saved to: ${outPath}`,
-            },
-          ],
-        };
-      }
-      if (name === "diff_docx_files") {
-        const origPath = args?.original_path as string;
-        const modPath = args?.modified_path as string;
-        const compareClean = (args?.compare_clean as boolean) ?? true;
-        const origBuf = readFileBytesOrThrow(origPath);
-        const modBuf = readFileBytesOrThrow(modPath);
-        // Pass compareClean flag into extraction
-        const origText = await extractTextFromBuffer(origBuf, compareClean);
-        const modText = await extractTextFromBuffer(modBuf, compareClean);
-        const diff = create_word_patch_diff(
-          origText,
-          modText,
-          basename(origPath),
-          basename(modPath),
-        );
+server.registerTool(
+  "diff_docx_files",
+  {
+    description: DIFF_DOCX_DESC,
+    inputSchema: {
+      original_path: z
+        .string()
+        .describe("Absolute path to the baseline DOCX file."),
+      modified_path: z
+        .string()
+        .describe("Absolute path to the modified DOCX file."),
+      compare_clean: z
+        .boolean()
+        .default(true)
+        .describe(
+          "If True, compares 'Accepted' state. If False, compares raw text.",
+        ),
+    },
+  },
+  async ({ original_path, modified_path, compare_clean }) => {
+    try {
+      const origBuf = readFileBytesOrThrow(original_path);
+      const modBuf = readFileBytesOrThrow(modified_path);
-        return {
-          content: [{ type: "text", text: diff || "No differences found." }],
-        };
-      }
+      const origText = await extractTextFromBuffer(origBuf, compare_clean);
+      const modText = await extractTextFromBuffer(modBuf, compare_clean);
-      if (name === "finalize_document") {
-        const filePath = args?.file_path as string;
-        let outPath = args?.output_path as string;
+      const diff = create_word_patch_diff(
+        origText,
+        modText,
+        basename(original_path),
+        basename(modified_path),
+      );
-        if (!outPath) {
-          const ext = extname(filePath);
-          const base = basename(filePath, ext);
-          const dir = dirname(filePath);
-          outPath = resolve(dir, `${base}_final${ext}`);
-        }
+      return {
+        content: [{ type: "text", text: diff || "No differences found." }],
+      };
+    } catch (e: any) {
+      return {
+        isError: true,
+        content: [{ type: "text", text: `Error: ${e.message}` }],
+      };
+    }
+  },
+);
-        const buf = readFileBytesOrThrow(filePath);
-        const doc = await DocumentObject.load(buf);
+server.registerTool(
+  "finalize_document",
+  {
+    description:
+      "Prepares a document for external distribution or e-signature.",
+    inputSchema: {
+      file_path: z.string().describe("Absolute path to the DOCX file."),
+      output_path: z.string().optional().describe("Optional output path."),
+      sanitize_mode: z
+        .enum(["full", "keep-markup"])
+        .optional()
+        .describe("full removes all markup, keep-markup redacts metadata."),
+      accept_all: z
+        .boolean()
+        .optional()
+        .describe(
+          "If true, auto-accepts all unresolved track changes before finalizing.",
+        ),
+      protection_mode: z
+        .enum(["read_only", "encrypt"])
+        .optional()
+        .describe("Native OOXML document locking."),
+      password: z.string().optional().describe("Ignored in this environment."),
+      author: z
+        .string()
+        .optional()
+        .describe("Replace all remaining markup authorship with this name."),
+      export_pdf: z
+        .boolean()
+        .optional()
+        .describe("Ignored in this environment."),
+    },
+  },
+  async ({
+    file_path,
+    output_path,
+    sanitize_mode,
+    accept_all,
+    protection_mode,
+    author,
+    export_pdf,
+  }) => {
+    try {
+      let outPath = output_path;
+      if (!outPath) {
+        const ext = extname(file_path);
+        const base = basename(file_path, ext);
+        const dir = dirname(file_path);
+        outPath = resolve(dir, `${base}_final${ext}`);
+      }
-        const result = await finalize_document(doc, {
-          filename: basename(filePath),
-          sanitize_mode: (args?.sanitize_mode as any) || "full",
-          accept_all: args?.accept_all as boolean,
-          protection_mode: args?.protection_mode as any,
-          author: args?.author as string,
-          export_pdf: args?.export_pdf as boolean,
-        });
+      const buf = readFileBytesOrThrow(file_path);
+      const doc = await DocumentObject.load(buf);
-        const fs = await import("node:fs");
-        fs.writeFileSync(outPath, result.outBuffer!);
+      const result = await finalize_document(doc, {
+        filename: basename(file_path),
+        sanitize_mode: (sanitize_mode as any) || "full",
+        accept_all: accept_all as boolean,
+        protection_mode: protection_mode as any,
+        author: author as string,
+        export_pdf: export_pdf as boolean,
+      });
+      fs.writeFileSync(outPath, result.outBuffer!);
-        return {
-          content: [
-            {
-              type: "text",
-              text: `Saved to: ${outPath}\n\n${result.reportText}`,
-            },
-          ],
-        };
-      }
-      if (name === "login_to_adeu_cloud") {
-        return await login_to_adeu_cloud();
-      }
-      if (name === "logout_of_adeu_cloud") {
-        return await logout_of_adeu_cloud();
-      }
-      if (name === "search_and_fetch_emails") {
-        return await search_and_fetch_emails(args || {});
-      }
-      if (name === "create_email_draft") {
-        return await create_email_draft(args || {});
-      }
-      throw new Error(`Unknown tool: ${name}`);
-    } catch (error: any) {
       return {
         content: [
           {
             type: "text",
-            text: `Error executing tool ${name}: ${error.message}`,
+            text: `Saved to: ${outPath}\n\n${result.reportText}`,
           },
         ],
+      };
+    } catch (e: any) {
+      return {
         isError: true,
+        content: [{ type: "text", text: `Error: ${e.message}` }],
       };
     }
   },
 );
+server.registerTool(
+  "login_to_adeu_cloud",
+  { description: "Logs the user into the Adeu Cloud backend." },
+  async () => {
+    try {
+      return (await login_to_adeu_cloud()) as any;
+    } catch (e: any) {
+      return { isError: true, content: [{ type: "text", text: e.message }] };
+    }
+  },
+);
+server.registerTool(
+  "logout_of_adeu_cloud",
+  { description: "Logs out of the Adeu Cloud backend." },
+  async () => {
+    try {
+      return (await logout_of_adeu_cloud()) as any;
+    } catch (e: any) {
+      return { isError: true, content: [{ type: "text", text: e.message }] };
+    }
+  },
+);
+server.registerTool(
+  "create_email_draft",
+  {
+    description: "Creates an email draft in the user's native draft box.",
+    inputSchema: {
+      body_markdown: z.string(),
+      reply_to_email_id: z.string().optional(),
+      subject: z.string().optional(),
+      to_recipients: z.array(z.string()).optional(),
+      attachment_paths: z.array(z.string()).optional(),
+      mailbox_address: z
+        .string()
+        .optional()
+        .describe(
+          "Optional target mailbox email address to create the draft in.",
+        ),
+    },
+  },
+  async (args) => {
+    try {
+      return (await create_email_draft(args)) as any;
+    } catch (e: any) {
+      return { isError: true, content: [{ type: "text", text: e.message }] };
+    }
+  },
+);
+server.registerTool(
+  "list_available_mailboxes",
+  {
+    description:
+      "Lists all personal and shared delegated mailboxes configured for the authenticated profile. Use this to discover valid email addresses to scope search and draft operations.",
+    inputSchema: {},
+  },
+  async () => {
+    try {
+      return (await list_available_mailboxes()) as any;
+    } catch (e: any) {
+      return { isError: true, content: [{ type: "text", text: e.message }] };
+    }
+  },
+);
 // --- Startup ---
 async function main() {
   const transport = new StdioServerTransport();