chess-moments 0.14.5 → 0.14.7

package/functions/extras/get-next-moments.js

@@ -7,35 +7,44 @@ const getNextMoments = (moments, current) => {
 
     // If it's the first moment without move, add the next moment
     if (current.index === 0 && !current.move) {
-      return moments[1];
+      return [moments[1]];
     }
 
-    // Split moments after the current index
-    const currentIndex = moments.indexOf(current);
-    let moves = moments.slice(currentIndex + 1).filter((m) => m.move);
+    // If there is no next moment, return empty array
+    const nextIndex = moments.indexOf(current) + 1;
+    const nextMoment = moments[nextIndex];
+    if (!nextMoment) {
+      return [];
+    }
+
+    // Add the next moment if it has a move and is at the same depth
+    if (nextMoment.move && current.depth === nextMoment.depth) {
+      next.push(nextMoment);
+    }
 
-    // If the immediate next moment has increased depth, filter out all moments until depth returns to current
-    if (moves.length > 0 && moves[0].depth > current.depth) {
-      let foundReturnToCurrentDepth = false;
-      moves = moves.filter((moment) => {
-        if (foundReturnToCurrentDepth) {
-          return true; // Keep all moments after we return to current depth
-        }
-        if (moment.depth === current.depth) {
-          foundReturnToCurrentDepth = true;
-          return true; // Keep this moment and all following
-        }
-        return false; // Skip moments with increased depth
-      });
+    // If the next next moment has a move, we don't have any other sidelines
+    const nextNextMoment = moments[nextIndex + 1];
+    if (nextMoment?.move && nextNextMoment?.move) {
+      return next;
     }
 
     // Add fullmove number to the current moment
     current.fullmove = Number(current.fen.split(' ')[5]);
 
+    // Keep only slim moments
+    const slim = moments.filter((moment) => {
+      const fullmove = Number(moment.fen.split(' ')[5]);
+      return (
+        moment.index > current.index + 1 &&
+        fullmove <= current.fullmove + 1 &&
+        moment.move // Also filter out moments without a move
+      );
+    });
+
     // Active color after current move (who moves next)
     const activeColorAfterCurrent = current.fen.split(' ')[1];
 
-    for (const moment of moves) {
+    for (const moment of slim) {
       // Only process main line and immediate variations
       if (moment.depth > current.depth + 1) {
         continue;

package/functions/extras/get-prev-moment.js

@@ -29,7 +29,8 @@ const getPrevMoment = (moments, current) => {
         const fullmove = Number(moment.fen.split(' ')[5]);
 
         if (
-          moment.depth === current.depth - 1 &&
+          (moment.depth === current.depth - 1 ||
+            moment.depth === current.depth) &&
           activeColor === 'b' &&
           fullmove === current.fullmove - 1
         ) {
@@ -47,7 +48,8 @@ const getPrevMoment = (moments, current) => {
         const fullmove = Number(moment.fen.split(' ')[5]);
 
         if (
-          moment.depth === current.depth - 1 &&
+          (moment.depth === current.depth - 1 ||
+            moment.depth === current.depth) &&
           activeColor === 'w' &&
           fullmove === current.fullmove
         ) {

package/package.json

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