llm-chess-mcp 0.4.9 → 0.4.10

package/dist/chess-copy.js

@@ -5,6 +5,42 @@ const ORIGINAL_PIECES = {
     r: 2,
     n: 2,
 };
+const CANONICAL_PGN_HEADERS = new Map([
+    "Event",
+    "Site",
+    "Date",
+    "Round",
+    "White",
+    "Black",
+    "Result",
+    "SetUp",
+    "FEN",
+].map((name) => [name.toLowerCase(), name]));
+function restoreHeaders(chess, sourceHeaders) {
+    const sourceNames = new Set(sourceHeaders.map(([key]) => key.toLowerCase()));
+    for (const key of Object.keys(chess.getHeaders())) {
+        if (!sourceNames.has(key.toLowerCase()))
+            chess.removeHeader(key);
+    }
+    const names = new Map();
+    for (const key of Object.keys(chess.getHeaders())) {
+        const existing = names.get(key.toLowerCase());
+        if (existing)
+            existing.push(key);
+        else
+            names.set(key.toLowerCase(), [key]);
+    }
+    for (const [key, value] of sourceHeaders) {
+        const lower = key.toLowerCase();
+        const canonical = CANONICAL_PGN_HEADERS.get(lower) ?? key;
+        for (const existing of names.get(lower) ?? []) {
+            if (existing !== canonical)
+                chess.removeHeader(existing);
+        }
+        chess.setHeader(canonical, value);
+        names.set(lower, [canonical]);
+    }
+}
 function squareColor(square) {
     return ((square.charCodeAt(0) - 97 + Number(square[1])) % 2);
 }
@@ -146,14 +182,30 @@ export function snapshotChess(chess) {
     assertSafeFenCounters(initialFen);
     const snapshot = new Chess(initialFen);
     assertLegalPosition(snapshot);
-    const comments = new Map(chess.getComments().map(({ fen, comment }) => [fen, comment]));
-    for (const [key, value] of Object.entries(chess.getHeaders())) {
-        snapshot.setHeader(key, value);
+    const comments = new Map(chess.getComments().map(({ fen, comment }) => [
+        fen,
+        /[{}]/.test(comment) ? comment.replace(/[\r\n]+/g, " ") : comment,
+    ]));
+    const sourceHeaders = Object.entries(chess.getHeaders());
+    const unsafeComments = [...comments.values()].some((comment) => /[{}]/.test(comment));
+    let markerPrefix = "\uE000";
+    if (unsafeComments) {
+        const occupied = [...sourceHeaders.flat(), ...comments.values()].join("\u0000");
+        while (occupied.includes(markerPrefix))
+            markerPrefix += "\uE001";
     }
+    const markerComments = [];
     const restoreComment = () => {
         const comment = comments.get(snapshot.fen());
-        if (comment !== undefined)
+        if (comment === undefined)
+            return;
+        if (!unsafeComments || !/[{}]/.test(comment)) {
             snapshot.setComment(comment);
+            return;
+        }
+        const marker = `${markerPrefix}${markerComments.length}${markerPrefix}`;
+        markerComments.push(comment);
+        snapshot.setComment(marker);
     };
     restoreComment();
     for (const move of history) {
@@ -161,5 +213,31 @@ export function snapshotChess(chess) {
         restoreComment();
     }
     assertSafeFenCounters(snapshot.fen());
-    return snapshot;
+    if (!unsafeComments) {
+        restoreHeaders(snapshot, sourceHeaders);
+        return snapshot;
+    }
+    const escapedPrefix = markerPrefix.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+    const marker = new RegExp(`\\{${escapedPrefix}(\\d+)${escapedPrefix}\\}`, "g");
+    const pgn = snapshot.pgn().replace(marker, (_match, index) => {
+        return `;${markerComments[Number(index)]}\n`;
+    });
+    const restored = new Chess();
+    restored.loadPgn(pgn);
+    const restoredHistory = restored.history({ verbose: true });
+    while (restored.undo()) { }
+    const restoreSafeComment = () => {
+        const comment = comments.get(restored.fen());
+        if (comment !== undefined && !/[{}]/.test(comment)) {
+            restored.setComment(comment);
+        }
+    };
+    restoreSafeComment();
+    for (const move of restoredHistory) {
+        restored.move(moveDescriptor(move));
+        restoreSafeComment();
+    }
+    restoreHeaders(restored, sourceHeaders);
+    assertSafeFenCounters(restored.fen());
+    return restored;
 }

package/dist/pgn.js

@@ -278,12 +278,14 @@ function pgnText(chess) {
     for (const { comment } of comments)
         assertPgnTokenSize(comment);
     const raw = chess.pgn();
-    const separator = raw.indexOf("\n\n");
-    let movetext = headers.length === 0
-        ? raw
-        : separator < 0
-            ? ""
-            : raw.slice(separator + 2);
+    let movetextStart = 0;
+    for (let index = 0; index < headers.length; index += 1) {
+        const lineEnd = raw.indexOf("\n", movetextStart);
+        movetextStart = lineEnd < 0 ? raw.length : lineEnd + 1;
+    }
+    if (headers.length > 0 && raw[movetextStart] === "\n")
+        movetextStart += 1;
+    let movetext = raw.slice(movetextStart);
     const unsafeComments = [
         ...new Set(comments
             .map(({ comment }) => comment)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "llm-chess-mcp",
-  "version": "0.4.9",
+  "version": "0.4.10",
   "description": "MCP server that lets an LLM analyze, judge, and choose chess moves (Stockfish + Maia3 + Lichess)",
   "type": "module",
   "license": "AGPL-3.0",