chess-moments 1.4.1 → 1.4.2

package/functions/extras/index.js

@@ -4,6 +4,7 @@ const getNextMoments = require('./get-next-moments');
 const getPrevMoment = require('./get-prev-moment');
 const momentsToPgn = require('./moments-to-pgn');
 const moveTrainer = require('./move-trainer');
+const promoteMainlineTree = require('./promote-mainline-tree');
 const promoteMainline = require('./promote-mainline');
 
 module.exports = {
@@ -13,5 +14,6 @@ module.exports = {
   getPrevMoment,
   momentsToPgn,
   moveTrainer,
+  promoteMainlineTree,
   promoteMainline,
 };

package/functions/extras/promote-mainline-tree.js

@@ -0,0 +1,24 @@
+const { groupBy, flatten } = require('lodash');
+const promoteMainline = require('./promote-mainline');
+
+const promoteMainlineTree = (moments, current) => {
+  try {
+    const isNested = moments.length > 0 && Array.isArray(moments[0]);
+    if (!isNested) {
+      return moments;
+    }
+
+    // Flatten nested array
+    const flatMoments = flatten(moments);
+    const newMoments = promoteMainline(flatMoments, current);
+
+    // Regroup by depth into a nested array
+    const grouped = groupBy(newMoments, 'depth');
+    return Object.values(grouped);
+  } catch {
+    // In case of any error, return the original moments array
+    return moments;
+  }
+};
+
+module.exports = promoteMainlineTree;

package/index.js

@@ -6,7 +6,8 @@ const {
   getPrevMoment,
   momentsToPgn,
   moveTrainer,
-  promoteMainline
+  promoteMainlineTree,
+  promoteMainline,
 } = require('./functions/extras');
 
 module.exports = {
@@ -21,5 +22,6 @@ module.exports = {
   getPrevMoment,
   momentsToPgn,
   moveTrainer,
-  promoteMainline
+  promoteMainlineTree,
+  promoteMainline,
 };

package/package.json

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