chess-moments 0.11.1 → 0.11.2

package/functions/{split-by-move-index.js → split-by-move.js}

@@ -1,10 +1,10 @@
-function splitByMoveIndex(inputData, moveIndex) {
+function splitByMoveIndex(inputData, current) {
   const before = [];
   const after = [];
 
   inputData.forEach((innerArray) => {
     // Check if every element in the inner array has an index less than moveIndex
-    if (innerArray.every((item) => item.index < moveIndex)) {
+    if (innerArray.every((item) => item.index < current.index)) {
       if (innerArray.length) {
         before.push(innerArray);
       }

package/functions/train.js

@@ -1,8 +1,8 @@
 const { flatten } = require('lodash');
-const splitByMoveIndex = require('./split-by-move-index');
+const splitByMoveIndex = require('./split-by-move');
 const tree = require('./tree');
 
-const train = (sloppyPgn, moveIndex) => {
+const train = (sloppyPgn, fen) => {
   // Tree is an array of array depths
   const moments = tree(sloppyPgn);
 
@@ -10,7 +10,7 @@ const train = (sloppyPgn, moveIndex) => {
   const training = [];
 
   // Remove sidelines (depth > 1) when no move index is specified
-  if (!moveIndex) {
+  if (!fen) {
     for (const innerArray of moments) {
       const hasDepth = innerArray.filter((moment) => moment.depth === 1);
       if (hasDepth.length) {
@@ -20,15 +20,15 @@ const train = (sloppyPgn, moveIndex) => {
     return training;
   }
 
+  // Get the current move from the move index
+  const current = flatten(moments).find((moment) => moment.fen === fen);
+
   // Split the moments into two groups
-  const { before, after } = splitByMoveIndex(moments, moveIndex);
+  const { before, after } = splitByMoveIndex(moments, current);
   for (const innerArray of before) {
     training.push(innerArray);
   }
 
-  // Get the current move from the move index
-  const current = flatten(moments).find((moment) => moment.index === moveIndex);
-
   // Only future moves remaining
   for (const innerArray of after) {
     // Remove every inner array with a depth not equal than the depth of the move index

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chess-moments",
-  "version": "0.11.1",
+  "version": "0.11.2",
   "description": "PGN parser that transforms PGN files into chess \"moments\"",
   "main": "dist/index.js",
   "files": [