chess-moments 1.4.8 → 1.6.0

package/constants/glyphs.js

@@ -0,0 +1,22 @@
+const glyphs = {
+  $7: '□', // Only move
+  $22: '⨀', // Zugzwang
+  $10: '=', // Equal position
+  $13: '∞', // Unclear position
+  $14: '⩲', // White is slightly better
+  $15: '⩱', // Black is slightly better
+  $16: '±', // White is better
+  $17: '∓', // Black is better
+  $18: '+−', // White is winning
+  $19: '-+', // Black is winning
+  $146: 'N', // Novelty
+  $32: '↑↑', // Development
+  $36: '↑', // Initiative
+  $40: '→', // Attack
+  $132: '⇆', // Counterplay
+  $138: '⊕', // Time trouble
+  $44: '=∞', // With compensation
+  $140: '∆', // With the idea
+};
+
+module.exports = glyphs;

package/functions/build-moment.js

@@ -1,7 +1,17 @@
 const prettify = require('./prettify');
 const shape = require('./shape');
 
-module.exports = ({ depth, move, fen, comment, suffix, from, to }) => {
+module.exports = ({
+  depth,
+  move,
+  fen,
+  comment,
+  suffix,
+  glyph,
+  from,
+  to,
+  headers,
+}) => {
   const moment = { depth, fen };
   if (move) {
     moment.move = move;
@@ -23,6 +33,12 @@ module.exports = ({ depth, move, fen, comment, suffix, from, to }) => {
   if (suffix) {
     moment.suffix = suffix;
   }
+  if (glyph) {
+    moment.glyph = glyph;
+  }
+  if (headers) {
+    moment.headers = headers;
+  }
 
   return moment;
 };

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

@@ -21,9 +21,46 @@ const momentsToPgn = (moments) => {
   let currentDepth = 1;
   let variationStack = [];
 
-  // Add FEN headers
-  pgn += `[SetUp "1"]\n`;
-  pgn += `[FEN "${moments[0].fen}"]\n\n`;
+  const headers = moments[0].headers;
+  const standardHeaders = [
+    'Event',
+    'Site',
+    'Date',
+    'Round',
+    'White',
+    'Black',
+    'Result',
+  ];
+
+  if (headers) {
+    // Add standard headers (excluding placeholder values)
+    for (const key of standardHeaders) {
+      if (
+        headers[key] &&
+        headers[key] !== '?' &&
+        headers[key] !== '????.??.??'
+      ) {
+        pgn += `[${key} "${headers[key]}"]\n`;
+      }
+    }
+
+    // Add custom headers
+    for (const key in headers) {
+      if (!standardHeaders.includes(key) && key !== 'FEN' && key !== 'SetUp') {
+        pgn += `[${key} "${headers[key]}"]\n`;
+      }
+    }
+  }
+
+  const fenString = headers?.FEN || moments[0].fen;
+
+  if (fenString !== fen.initial) {
+    const setUp = headers?.SetUp || '1';
+    pgn += `[SetUp "${setUp}"]\n`;
+    pgn += `[FEN "${fenString}"]\n`;
+  }
+
+  pgn += '\n';
 
   for (let i = 0; i < moments.length; i++) {
     const moment = moments[i];
@@ -105,6 +142,11 @@ const momentsToPgn = (moments) => {
       pgn += moment.suffix;
     }
 
+    // Add glyph if present (keep as NAG code from original PGN)
+    if (moment.glyph) {
+      pgn += ` ${moment.glyph}`;
+    }
+
     // Add comment if present
     if (moment.comment) {
       pgn += ` {${moment.comment}}`;
@@ -164,8 +206,8 @@ const momentsToPgn = (moments) => {
     pgn += ')';
   }
 
-  // Add default result
-  pgn += ' *';
+  // Add result from headers if available, otherwise default to *
+  pgn += ` ${headers?.Result || '*'}`;
 
   return pgn.trim();
 };

package/functions/make-moments.js

@@ -5,7 +5,11 @@ const pgn = require('./pgn');
 const split = require('./split');
 
 const make = (sloppyPgn) => {
-  const normalizedPgn = pgn.normalize(sloppyPgn);
+  let normalizedPgn = pgn.normalize(sloppyPgn);
+  // when PGN is empty, create a default one with initial FEN
+  if (!normalizedPgn) {
+    normalizedPgn = pgn.build([], fen.initial);
+  }
 
   // load PGN and check headers for existing FEN
   const chess = new Chess(); // can throw if PGN is invalid
@@ -18,7 +22,7 @@ const make = (sloppyPgn) => {
   };
 
   const history = new Map();
-  const variations = split(normalizedPgn).map(({ moves, depth }) => {
+  const variations = split(normalizedPgn).map(({ moves, depth }, index) => {
     // find previous FEN
     if (history.get(store.depth)) {
       store.fen = fen.previous(history, moves, depth, store.depth);
@@ -26,7 +30,13 @@ const make = (sloppyPgn) => {
     store.depth = depth;
 
     // parse current moves with the computed FEN
-    const moments = parser(moves, store.fen, depth);
+    // only pass headers to the first move
+    const moments = parser(
+      moves,
+      store.fen,
+      depth,
+      index === 0 ? headers : null
+    );
 
     // set history for the current depth
     history.set(depth, fen.history(moments, history, depth));

package/functions/parser.js

@@ -1,9 +1,11 @@
 const { Chess } = require('chess.js');
+const { invert } = require('lodash');
 const { initial } = require('./fen');
 const pgn = require('./pgn');
 const moment = require('./moment');
+const glyphs = require('../constants/glyphs');
 
-module.exports = (moves, fen = initial, depth = 1) => {
+module.exports = (moves, fen = initial, depth = 1, headers = null) => {
   const chess = new Chess();
   chess.loadPgn(pgn.build(moves, fen)); // can throw if PGN is invalid
 
@@ -17,6 +19,7 @@ module.exports = (moves, fen = initial, depth = 1) => {
     depth,
     comment: chess.getComment(),
     fen: chess.fen(),
+    headers,
   });
 
   const moments = history.map((item) => {
@@ -29,13 +32,14 @@ module.exports = (moves, fen = initial, depth = 1) => {
       to: item.to,
       comment: chess.getComment(),
       suffix: chess.getSuffixAnnotation(),
+      glyph: invert(glyphs)[chess.getGlyph()],
       fen: chess.fen(),
     });
   });
 
   // finally, add the first chess "moment" when needed
-  const headers = chess.getHeaders();
-  if (headers.FEN || fen === initial || first.comment || first.shapes) {
+  const pgnHeaders = chess.getHeaders();
+  if (pgnHeaders.FEN || fen === initial || first.comment || first.shapes) {
     moments.unshift(first);
   }
 

package/package.json

@@ -1,12 +1,13 @@
 {
   "name": "chess-moments",
-  "version": "1.4.8",
+  "version": "1.6.0",
   "description": "PGN parser that transforms PGN files into chess \"moments\"",
   "main": "index.js",
   "files": [
     "index.js",
     "chess.js",
-    "functions/**/*.js"
+    "functions/**/*.js",
+    "constants/**/*.js"
   ],
   "scripts": {
     "test": "mocha",
@@ -24,7 +25,7 @@
   ],
   "license": "MIT",
   "dependencies": {
-    "chess.js": "github:jhlywa/chess.js#master"
+    "chess.js": "github:brobert1/chess.js#master"
   },
   "devDependencies": {
     "chai": "^4.3.7",