chess-moments 0.10.0 → 0.11.0

package/functions/index.js

@@ -9,6 +9,7 @@ const prepare = require('./prepare');
 const prettify = require('./prettify');
 const shape = require('./shape');
 const split = require('./split');
+const train = require('./train');
 const tree = require('./tree');
 
 module.exports = {
@@ -23,5 +24,6 @@ module.exports = {
   prettify,
   shape,
   split,
+  train,
   tree,
 };

package/functions/prettify.js

@@ -1,9 +1,7 @@
 module.exports = (comment) => {
   try {
-    return comment
-      .split(']')
-      .map((it) => it.trim())
-      .filter((it) => it.indexOf('[') !== 0)[0];
+    // Remove any shape comments like [%cal Ya7,Ya6] or [%csl Ya7,Ya6]
+    return comment.replace(/\s*\[%c(?:al|sl) [^\]]+\]/g, '').trim();
   } catch (err) {
     return undefined;
   }

package/functions/train.js

@@ -0,0 +1,21 @@
+const flat = require('./flat');
+
+const train = (sloppyPgn, moveIndex) => {
+  const moments = flat(sloppyPgn);
+
+  // Remove sidelines (depth > 1) when no move index is specified
+  if (!moveIndex) {
+    return moments.filter((moment) => moment.depth === 1);
+  }
+
+  // Split the moments into two groups
+  const before = moments.slice(0, moveIndex);
+  const after = moments.slice(moveIndex);
+
+  // Remove sidelines after the specified move index with the move index depth
+  const filtered = after.filter((moment) => moment.depth <= moments[moveIndex].depth);
+
+  return before.concat(filtered);
+};
+
+module.exports = train;

package/index.js

@@ -1,8 +1,9 @@
-const { fen, pgn, flat, tree } = require('./functions');
+const { fen, pgn, flat, train, tree } = require('./functions');
 
 module.exports = {
   fen,
   pgn,
   flat,
+  train,
   tree,
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chess-moments",
-  "version": "0.10.0",
+  "version": "0.11.0",
   "description": "PGN reader is a javascript chess library that transforms PGN files into chess moments",
   "main": "dist/index.js",
   "files": [