chess-moments 0.13.0 → 0.14.1

package/functions/extras/index.js

@@ -0,0 +1,9 @@
+const getNextMoments = require('./get-next-moments');
+const getPrevMoment = require('./get-prev-moment');
+const momentsToPgn = require('./moments-to-pgn');
+
+module.exports = {
+  getNextMoments,
+  getPrevMoment,
+  momentsToPgn,
+};

package/functions/extras/moments-to-pgn.js

@@ -0,0 +1,158 @@
+const fen = require('../fen');
+const { getBrushCode } = require('../helpers');
+
+/**
+ * Converts chess moments back to PGN format
+ * @param {Array} moments - Array of chess moment objects
+ * @returns {String} PGN string
+ */
+const momentsToPgn = (moments) => {
+  if (!Array.isArray(moments) || moments.length === 0) {
+    return '';
+  }
+
+  let pgn = '';
+  let currentDepth = 1;
+  let variationStack = [];
+
+  // Check if we need to add FEN headers for non-standard starting position
+  const initialMoment = moments[0];
+  if (initialMoment && initialMoment.fen && initialMoment.fen !== fen.initial) {
+    pgn += `[SetUp "1"]\n`;
+    pgn += `[FEN "${initialMoment.fen}"]\n\n`;
+  }
+
+  for (let i = 0; i < moments.length; i++) {
+    const moment = moments[i];
+    const depth = moment.depth || 1;
+
+    // Handle moments without moves (initial position or variation breaks)
+    if (!moment.move) {
+      // If this is a variation marker (depth > 1) and not the initial position
+      if (depth > 1 && i > 0) {
+        // Check if we need to start a new variation at the same depth
+        if (depth === currentDepth) {
+          // Close current variation and start a new one at the same level
+          pgn += ') (';
+        } else if (depth > currentDepth) {
+          // Starting a new variation at a deeper level
+          variationStack.push(currentDepth);
+          pgn += ' (';
+          currentDepth = depth;
+        }
+      }
+
+      // Add initial comment if present
+      if (moment.comment) {
+        pgn += `{${moment.comment}} `;
+      }
+      continue;
+    }
+
+    // Determine who made this move and what move number it is from the FEN
+    const fenParts = moment.fen.split(' ');
+    const activeColorAfterMove = fenParts[1]; // 'w' or 'b' - who moves next
+    const fullmoveNumber = parseInt(fenParts[5]);
+
+    // If black is to move after this move, then white made this move
+    const moveWasByWhite = activeColorAfterMove === 'b';
+    // For white moves, the move number is the fullmove number
+    // For black moves, the move number is fullmove number - 1 (since fullmove increments after black's move)
+    const moveNumber = moveWasByWhite ? fullmoveNumber : fullmoveNumber - 1;
+
+    // Handle depth changes (variations)
+    if (depth > currentDepth) {
+      // Starting a new variation
+      variationStack.push(currentDepth);
+      pgn += ' (';
+      currentDepth = depth;
+    } else if (depth < currentDepth) {
+      // Ending variation(s) - close as many as needed to get to the right depth
+      while (currentDepth > depth) {
+        variationStack.pop();
+        pgn += ')';
+        currentDepth--;
+      }
+      if (pgn.endsWith(')')) {
+        pgn += ' ';
+      }
+    }
+
+    // Add move number for white moves or when starting a variation
+    if (moveWasByWhite || (depth > 1 && pgn.endsWith('('))) {
+      pgn += `${moveNumber}.`;
+      if (!moveWasByWhite) {
+        pgn += '..';
+      }
+      pgn += ' ';
+    }
+
+    // Add the move
+    pgn += moment.move;
+
+    // Add comment if present
+    if (moment.comment) {
+      pgn += ` {${moment.comment}}`;
+    }
+
+    // Add shapes if present
+    if (moment.shapes && moment.shapes.length > 0) {
+      let shapesComment = '';
+      const squareHighlights = moment.shapes.filter(
+        (s) => !s.dest || s.dest === ']'
+      );
+      const arrows = moment.shapes.filter((s) => s.dest && s.dest !== ']');
+
+      if (squareHighlights.length > 0) {
+        const coloredSquares = squareHighlights
+          .map((s) => {
+            const colorCode = getBrushCode(s.brush);
+            return `${colorCode}${s.orig}`;
+          })
+          .join(',');
+        shapesComment += `[%csl ${coloredSquares}]`;
+      }
+
+      if (arrows.length > 0) {
+        const coloredArrows = arrows
+          .map((s) => {
+            const colorCode = getBrushCode(s.brush);
+            return `${colorCode}${s.orig}${s.dest}`;
+          })
+          .join(',');
+        if (shapesComment) shapesComment += ' ';
+        shapesComment += `[%cal ${coloredArrows}]`;
+      }
+
+      if (shapesComment) {
+        if (moment.comment) {
+          // If there's already a comment, add shapes to it
+          pgn = pgn.replace(/\{([^}]*)\}$/, `{$1 ${shapesComment}}`);
+        } else {
+          pgn += ` {${shapesComment}}`;
+        }
+      }
+    }
+
+    // Add space after move if not the last move
+    if (i < moments.length - 1) {
+      const nextMoment = moments[i + 1];
+      if (nextMoment && nextMoment.move) {
+        pgn += ' ';
+      }
+    }
+  }
+
+  // Close any remaining variations
+  while (variationStack.length > 0) {
+    variationStack.pop();
+    pgn += ')';
+  }
+
+  // Add default result
+  pgn += ' *';
+
+  return pgn.trim();
+};
+
+module.exports = momentsToPgn;

package/functions/helpers/get-brush-code.js

@@ -0,0 +1,14 @@
+/**
+ * Convert brush color name to PGN color code
+ */
+const getBrushCode = (brush) => {
+  const brushCodes = {
+    green: 'G',
+    red: 'R',
+    blue: 'B',
+    yellow: 'Y',
+  };
+  return brushCodes[brush] || 'G';
+};
+
+module.exports = getBrushCode;

package/functions/helpers/get-brush-color.js

@@ -0,0 +1,15 @@
+/**
+ * Convert PGN color code to brush color name
+ */
+const getBrushColor = (letter) => {
+  const brushes = {
+    R: 'red',
+    Y: 'yellow',
+    G: 'green',
+    B: 'blue',
+  };
+
+  return brushes[letter] || 'green';
+};
+
+module.exports = getBrushColor;

package/functions/helpers/index.js

@@ -0,0 +1,7 @@
+const getBrushCode = require('./get-brush-code');
+const getBrushColor = require('./get-brush-color');
+
+module.exports = {
+  getBrushCode,
+  getBrushColor,
+};

package/functions/index.js

@@ -1,8 +1,5 @@
-const brush = require('./brush');
 const fen = require('./fen');
 const flat = require('./flat');
-const getNextMoments = require('./get-next-moments');
-const getPrevMoment = require('./get-prev-moment');
 const lines = require('./lines');
 const moment = require('./moment');
 const parser = require('./parser');
@@ -15,11 +12,8 @@ const train = require('./train');
 const tree = require('./tree');
 
 module.exports = {
-  brush,
   fen,
   flat,
-  getNextMoments,
-  getPrevMoment,
   lines,
   moment,
   parser,

package/functions/shape.js

@@ -1,4 +1,4 @@
-const brush = require('./brush');
+const { getBrushColor } = require('./helpers');
 
 module.exports = (comment) => {
   try {
@@ -16,7 +16,7 @@ module.exports = (comment) => {
 
       for (const target of shapes.join('').split(',')) {
         draw.push({
-          brush: brush(target.substr(0, 1)),
+          brush: getBrushColor(target.substr(0, 1)),
           orig: target.substr(1, 2),
           dest: target.substr(3, 2),
         });

package/index.js

@@ -1,12 +1,9 @@
+const { fen, pgn, flat, train, tree } = require('./functions');
 const {
-  fen,
-  pgn,
-  flat,
-  train,
-  tree,
   getNextMoments,
   getPrevMoment,
-} = require('./functions');
+  momentsToPgn,
+} = require('./functions/extras');
 
 module.exports = {
   fen,
@@ -16,4 +13,5 @@ module.exports = {
   tree,
   getNextMoments,
   getPrevMoment,
+  momentsToPgn,
 };

package/package.json

@@ -1,12 +1,12 @@
 {
   "name": "chess-moments",
-  "version": "0.13.0",
+  "version": "0.14.1",
   "description": "PGN parser that transforms PGN files into chess \"moments\"",
   "main": "dist/index.js",
   "files": [
     "index.js",
     "chess.js",
-    "functions/*.js"
+    "functions/**/*.js"
   ],
   "scripts": {
     "test": "mocha",

package/functions/brush.js

@@ -1,9 +0,0 @@
-module.exports = (letter) => {
-  const brushes = {
-    R: 'red',
-    Y: 'yellow',
-    G: 'green',
-  };
-
-  return brushes[letter] || 'green';
-};