@lineadraw/mcp 0.3.0 → 0.5.0

package/dist/server.js

@@ -2,9 +2,8 @@
 
 // src/server.ts
 import { readFileSync } from "fs";
+import http from "http";
 import https from "https";
-import cors from "cors";
-import express from "express";
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
@@ -13,21 +12,18 @@ import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/
 import { newDrawing, openDrawing, drawingFromJson } from "@lineadraw/sdk";
 var DocumentRegistry = class {
   docs = /* @__PURE__ */ new Map();
-  revisions = /* @__PURE__ */ new Map();
   activeId = null;
   create(name) {
     const doc = newDrawing(name);
     const id = doc.document.id;
-    this.docs.set(id, doc);
-    this.revisions.set(id, 0);
+    this.docs.set(id, { doc, revision: 0 });
     this.activeId = id;
     return { id, doc };
   }
   async open(path2) {
     const doc = await openDrawing(path2);
     const id = doc.document.id;
-    this.docs.set(id, doc);
-    this.revisions.set(id, 0);
+    this.docs.set(id, { doc, revision: 0 });
     this.activeId = id;
     return { id, doc };
   }
@@ -35,32 +31,33 @@ var DocumentRegistry = class {
   get(id) {
     const key = id ?? this.activeId;
     if (!key) throw new Error("No document \u2014 call linea_new_document first");
-    const doc = this.docs.get(key);
-    if (!doc) throw new Error(`Unknown document "${key}"`);
-    return { id: key, doc };
+    const entry = this.docs.get(key);
+    if (!entry) throw new Error(`Unknown document "${key}"`);
+    return { id: key, doc: entry.doc };
   }
   /**
    * Replace a document's full state (validated + migrated) — how in-chat
    * editor edits land. The id must match; conversation undo history resets.
    */
   replace(id, raw) {
-    if (!this.docs.has(id)) throw new Error(`Unknown document "${id}"`);
+    const entry = this.docs.get(id);
+    if (!entry) throw new Error(`Unknown document "${id}"`);
     const doc = drawingFromJson(JSON.stringify(raw));
     if (doc.document.id !== id)
       throw new Error(
         `Document id mismatch: payload has "${doc.document.id}", expected "${id}"`
       );
-    this.docs.set(id, doc);
+    entry.doc = doc;
     return { doc, revision: this.bump(id) };
   }
   /** Mark a document changed; the viewer compares revisions to refresh. */
   bump(id) {
-    const next = (this.revisions.get(id) ?? 0) + 1;
-    this.revisions.set(id, next);
-    return next;
+    const entry = this.docs.get(id);
+    if (!entry) return 0;
+    return ++entry.revision;
   }
   revision(id) {
-    return this.revisions.get(id) ?? 0;
+    return this.docs.get(id)?.revision ?? 0;
   }
 };
 
@@ -94,12 +91,6 @@ var toolResult = (data) => ({
   structuredContent: data
 });
 var isRawToolResult = (v) => typeof v === "object" && v !== null && Array.isArray(v.content);
-var bytesToBase64 = (bytes) => {
-  if (typeof Buffer !== "undefined") return Buffer.from(bytes).toString("base64");
-  let bin = "";
-  for (const b of bytes) bin += String.fromCharCode(b);
-  return btoa(bin);
-};
 var toolError = (e) => ({
   content: [
     { type: "text", text: String(e instanceof Error ? e.message : e) }
@@ -115,11 +106,6 @@ var runTool = async (def, ctx, args) => {
     return toolError(e);
   }
 };
-var docPayload = (s) => ({
-  documentId: s.id,
-  revision: s.revision(),
-  document: s.getDocument()
-});
 var coreLineaTools = [
   {
     name: "linea_add_objects",
@@ -273,7 +259,7 @@ var coreLineaTools = [
   {
     name: "linea_set_dim_style",
     title: "Set dimension style",
-    description: "Patch a document dimension style (default: the first): textHeight, textOffset, arrowType (filled|open|tick|dot|none), arrowSize, extensionType (long|short), extensionOffset, extensionOvershoot, precision, font, widthFactor. NOTE: annotation scale is NOT a style key \u2014 set the top-level `scale` field on each dimension object.",
+    description: "Patch a document dimension style (default: the first): textHeight, textOffset, arrowType (filled|open|tick|dot|none), arrowSize, extensionType (long|short), extensionOffset, extensionOvershoot, precision, font (arial|osifont|jetbrains-mono|source-serif), widthFactor. NOTE: annotation scale is NOT a style key \u2014 set the top-level `scale` field on each dimension object.",
     inputSchema: {
       patch: z.record(z.string(), z.unknown()),
       styleId: z.string().optional(),
@@ -289,7 +275,7 @@ var coreLineaTools = [
   {
     name: "linea_set_text_style",
     title: "Set text style",
-    description: "Patch a document text style (default: the first): font, widthFactor, textHeight (default height for new text), arrowType, arrowSize (leader arrows). Per-object styleOverride still wins over the style.",
+    description: "Patch a document text style (default: the first): font (arial|osifont|jetbrains-mono|source-serif), widthFactor, textHeight (default height for new text), arrowType, arrowSize (leader arrows). Per-object styleOverride still wins over the style.",
     inputSchema: {
       patch: z.record(z.string(), z.unknown()),
       styleId: z.string().optional(),
@@ -401,7 +387,8 @@ var coreLineaTools = [
           content: [
             {
               type: "image",
-              data: bytesToBase64(bytes),
+              // renderPng exists on the Node adapter only, so Buffer is here.
+              data: Buffer.from(bytes).toString("base64"),
               mimeType: "image/png"
             }
           ]
@@ -448,7 +435,8 @@ var coreLineaTools = [
     description: "The full document (validated .linea JSON) + revision. The in-chat viewer calls this to refresh; agents usually want linea_summary instead.",
     inputSchema: { documentId },
     run(ctx, args) {
-      return docPayload(ctx.get(args.documentId));
+      const s = ctx.get(args.documentId);
+      return { documentId: s.id, revision: s.revision(), document: s.getDocument() };
     }
   },
   {
@@ -469,135 +457,67 @@ var coreLineaTools = [
 
 // src/tools.ts
 var EDITOR_RESOURCE_URI = "ui://linea/editor.html";
-var docPayload2 = (registry2, id, doc) => ({
-  documentId: id,
-  revision: registry2.revision(id),
-  document: doc.document
-});
-var DrawingSession = class {
-  constructor(registry2, id, doc) {
-    this.registry = registry2;
-    this.id = id;
-    this.doc = doc;
-  }
-  registry;
-  id;
-  doc;
-  revision() {
-    return this.registry.revision(this.id);
-  }
-  bump() {
-    return this.registry.bump(this.id);
-  }
-  add(objects, opts) {
-    return this.doc.add(objects, {
-      layout: opts?.layout
-    });
-  }
-  update(entries) {
-    this.doc.update(entries);
-  }
-  remove(ids) {
-    const removed = ids.filter((id) => this.doc.getObject(id) !== void 0);
-    this.doc.remove(ids);
+var drawingSession = (registry2, id, doc) => ({
+  id,
+  revision: () => registry2.revision(id),
+  bump: () => registry2.bump(id),
+  add: (objects, opts) => doc.add(objects, { layout: opts?.layout }),
+  update: (entries) => doc.update(entries),
+  remove: (ids) => {
+    const removed = ids.filter((oid) => doc.getObject(oid) !== void 0);
+    doc.remove(ids);
     return removed.length;
-  }
-  query(filter) {
+  },
+  query: (filter) => {
     const { ids, ...rest } = filter;
-    let objects = this.doc.query(rest);
+    let objects = doc.query(rest);
     if (ids) {
       const wanted = new Set(ids);
       objects = objects.filter((o) => wanted.has(o.id));
     }
     return objects;
-  }
-  measure(id) {
-    return {
-      bbox: this.doc.bboxOf([id]),
-      length: this.doc.length(id),
-      area: this.doc.area(id)
-    };
-  }
-  listLayers() {
-    const current = this.doc.document.currentLayerId;
-    return this.doc.layers.map((l) => ({ ...l, current: l.id === current }));
-  }
-  addLayer(props) {
-    return this.doc.addLayer(props);
-  }
-  updateLayer(nameOrId, patch) {
-    this.doc.updateLayer(nameOrId, patch);
-  }
-  removeLayer(nameOrId) {
-    this.doc.removeLayer(nameOrId);
-  }
-  setCurrentLayer(nameOrId) {
-    this.doc.setCurrentLayer(nameOrId);
-  }
-  renderSvg(opts) {
-    return this.doc.toSvg(opts);
-  }
-  renderPng(opts) {
-    return this.doc.toPng(opts);
-  }
-  undo() {
-    return this.doc.undo();
-  }
-  redo() {
-    return this.doc.redo();
-  }
-  move(ids, delta) {
-    this.doc.move(ids, delta);
-  }
-  rotate(ids, angleRad, center) {
-    this.doc.rotate(ids, angleRad, center);
-  }
-  scale(ids, factor, center) {
-    this.doc.scale(ids, factor, center);
-  }
-  mirror(ids, axisA, axisB) {
-    this.doc.mirror(ids, axisA, axisB);
-  }
-  copy(ids, delta) {
-    return this.doc.copy(ids, delta);
-  }
-  array(ids, count, delta) {
-    return this.doc.array(ids, count, delta);
-  }
-  createLayout(name, opts) {
-    return this.doc.createLayout(name, {
-      sheet: opts.sheet,
-      orientation: opts.orientation,
-      viewport: opts.scale !== void 0 ? { scale: opts.scale } : void 0
-    });
-  }
-  updateLayout(nameOrId, patch) {
-    this.doc.updateLayoutMeta(nameOrId, patch);
-  }
-  removeLayout(nameOrId) {
-    this.doc.removeLayout(nameOrId);
-  }
-  setDimStyle(patch, styleId) {
-    this.doc.setDimStyle(patch, styleId);
-  }
-  setTextStyle(patch, styleId) {
-    this.doc.setTextStyle(patch, styleId);
-  }
-  summary() {
-    return this.doc.summary();
-  }
-  getDocument() {
-    return this.doc.document;
-  }
-  replace(raw) {
-    return this.registry.replace(this.id, raw).revision;
-  }
-};
+  },
+  measure: (oid) => ({
+    bbox: doc.bboxOf([oid]),
+    length: doc.length(oid),
+    area: doc.area(oid)
+  }),
+  listLayers: () => {
+    const current = doc.document.currentLayerId;
+    return doc.layers.map((l) => ({ ...l, current: l.id === current }));
+  },
+  addLayer: (props) => doc.addLayer(props),
+  updateLayer: doc.updateLayer.bind(doc),
+  removeLayer: doc.removeLayer.bind(doc),
+  setCurrentLayer: doc.setCurrentLayer.bind(doc),
+  renderSvg: doc.toSvg.bind(doc),
+  renderPng: doc.toPng.bind(doc),
+  undo: doc.undo.bind(doc),
+  redo: doc.redo.bind(doc),
+  move: doc.move.bind(doc),
+  rotate: doc.rotate.bind(doc),
+  scale: doc.scale.bind(doc),
+  mirror: doc.mirror.bind(doc),
+  copy: doc.copy.bind(doc),
+  array: doc.array.bind(doc),
+  createLayout: (name, opts) => doc.createLayout(name, {
+    sheet: opts.sheet,
+    orientation: opts.orientation,
+    viewport: opts.scale !== void 0 ? { scale: opts.scale } : void 0
+  }),
+  updateLayout: doc.updateLayoutMeta.bind(doc),
+  removeLayout: doc.removeLayout.bind(doc),
+  setDimStyle: (patch, styleId) => doc.setDimStyle(patch, styleId),
+  setTextStyle: (patch, styleId) => doc.setTextStyle(patch, styleId),
+  summary: () => doc.summary(),
+  getDocument: () => doc.document,
+  replace: (raw) => registry2.replace(id, raw).revision
+});
 function registerLineaTools(server, registry2, opts = {}) {
   const ctx = {
     get(idArg) {
       const { id, doc } = registry2.get(idArg);
-      return new DrawingSession(registry2, id, doc);
+      return drawingSession(registry2, id, doc);
     }
   };
   for (const def of coreLineaTools) {
@@ -698,7 +618,11 @@ function registerLineaTools(server, registry2, opts = {}) {
               text: `Showing "${doc.document.name}" in the in-chat editor (${doc.objects.length} model objects)`
             }
           ],
-          structuredContent: docPayload2(registry2, id, doc)
+          structuredContent: {
+            documentId: id,
+            revision: registry2.revision(id),
+            document: doc.document
+          }
         };
       } catch (e) {
         return toolError(e);
@@ -736,37 +660,46 @@ var buildServer = () => {
   return server;
 };
 function startHttpServer() {
-  const app = express();
-  app.use(cors());
-  app.use(express.json({ limit: "16mb" }));
-  app.use((req, _res, next) => {
-    console.log(`${(/* @__PURE__ */ new Date()).toISOString()} ${req.method} ${req.path}`);
-    next();
-  });
-  app.all(["/", "/mcp"], async (req, res) => {
+  const handle = async (req, res) => {
+    const path2 = (req.url ?? "/").split("?")[0];
+    console.log(`${(/* @__PURE__ */ new Date()).toISOString()} ${req.method} ${path2}`);
+    res.setHeader("Access-Control-Allow-Origin", "*");
+    res.setHeader("Access-Control-Allow-Methods", "GET,POST,DELETE,OPTIONS");
+    res.setHeader("Access-Control-Allow-Headers", "*");
+    if (req.method === "OPTIONS") {
+      res.writeHead(204).end();
+      return;
+    }
+    if (path2 !== "/" && path2 !== "/mcp") {
+      res.writeHead(404).end();
+      return;
+    }
     const transport = new StreamableHTTPServerTransport({
       sessionIdGenerator: void 0,
       enableJsonResponse: true
     });
     res.on("close", () => transport.close());
     await buildServer().connect(transport);
-    await transport.handleRequest(req, res, req.body);
-  });
-  const { TLS_CERT, TLS_KEY } = process.env;
-  if (TLS_CERT && TLS_KEY) {
-    https.createServer(
-      { cert: readFileSync(TLS_CERT), key: readFileSync(TLS_KEY) },
-      app
-    ).listen(PORT, () => {
-      console.log(`Linea MCP server on https://localhost:${PORT}/mcp`);
-    });
-  } else {
-    app.listen(PORT, () => {
-      console.log(
-        `Linea MCP server on http://localhost:${PORT}/mcp (set TLS_CERT/TLS_KEY for https)`
-      );
+    await transport.handleRequest(req, res);
+  };
+  const listener = (req, res) => {
+    void handle(req, res).catch((e) => {
+      console.error(e);
+      if (!res.headersSent) res.writeHead(500);
+      res.end();
     });
-  }
+  };
+  const { TLS_CERT, TLS_KEY } = process.env;
+  const tls = Boolean(TLS_CERT && TLS_KEY);
+  const server = tls ? https.createServer(
+    { cert: readFileSync(TLS_CERT), key: readFileSync(TLS_KEY) },
+    listener
+  ) : http.createServer(listener);
+  server.listen(PORT, () => {
+    console.log(
+      tls ? `Linea MCP server on https://localhost:${PORT}/mcp` : `Linea MCP server on http://localhost:${PORT}/mcp (set TLS_CERT/TLS_KEY for https)`
+    );
+  });
 }
 if (process.argv.includes("--stdio")) {
   await buildServer().connect(new StdioServerTransport());