chess-moments 0.13.0 → 0.14.0

package/functions/index.js

@@ -1,8 +1,5 @@
-const brush = require('./brush');
 const fen = require('./fen');
 const flat = require('./flat');
-const getNextMoments = require('./get-next-moments');
-const getPrevMoment = require('./get-prev-moment');
 const lines = require('./lines');
 const moment = require('./moment');
 const parser = require('./parser');
@@ -15,11 +12,8 @@ const train = require('./train');
 const tree = require('./tree');
 
 module.exports = {
-  brush,
   fen,
   flat,
-  getNextMoments,
-  getPrevMoment,
   lines,
   moment,
   parser,

package/functions/shape.js

@@ -1,4 +1,4 @@
-const brush = require('./brush');
+const { getBrushColor } = require('./helpers');
 
 module.exports = (comment) => {
   try {
@@ -16,7 +16,7 @@ module.exports = (comment) => {
 
       for (const target of shapes.join('').split(',')) {
         draw.push({
-          brush: brush(target.substr(0, 1)),
+          brush: getBrushColor(target.substr(0, 1)),
           orig: target.substr(1, 2),
           dest: target.substr(3, 2),
         });

package/index.js

@@ -1,12 +1,9 @@
+const { fen, pgn, flat, train, tree } = require('./functions');
 const {
-  fen,
-  pgn,
-  flat,
-  train,
-  tree,
   getNextMoments,
   getPrevMoment,
-} = require('./functions');
+  momentsToPgn,
+} = require('./functions/extras');
 
 module.exports = {
   fen,
@@ -16,4 +13,5 @@ module.exports = {
   tree,
   getNextMoments,
   getPrevMoment,
+  momentsToPgn,
 };

package/package.json

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

package/functions/brush.js

@@ -1,9 +0,0 @@
-module.exports = (letter) => {
-  const brushes = {
-    R: 'red',
-    Y: 'yellow',
-    G: 'green',
-  };
-
-  return brushes[letter] || 'green';
-};

package/functions/get-next-moments.js

@@ -1,65 +0,0 @@
-/**
- * Always returns an array because there can be more than one next moment
- */
-const getNextMoments = (moments, current) => {
-  try {
-    const next = [];
-    const sameDepth = moments[moments.indexOf(current) + 1];
-    if (sameDepth.move && sameDepth.depth === current.depth) {
-      next.push(sameDepth);
-    }
-
-    // If it's the first moment without move, add the next moment
-    if (current.index === 0 && !current.move) {
-      next.push(moments[1]);
-    }
-
-    // Split moments after the current index
-    const currentIndex = moments.indexOf(current);
-    const moves = moments.slice(currentIndex + 1).filter((m) => m.move);
-
-    // Add fullmove number to the current moment
-    current.fullmove = Number(current.fen.split(' ')[5]);
-
-    // Active color is white
-    if (current.fen.split(' ')[1] === 'w') {
-      for (const moment of moves) {
-        const activeColor = moment.fen.split(' ')[1];
-        const fullmove = Number(moment.fen.split(' ')[5]);
-
-        // Check if the next moment is a valid move for white
-        if (
-          moment.depth === current.depth + 1 &&
-          activeColor === 'b' &&
-          fullmove === current.fullmove
-        ) {
-          next.push(moment);
-        }
-      }
-    }
-
-    // Active color is black
-    if (current.fen.split(' ')[1] === 'b') {
-      for (const moment of moves) {
-        const activeColor = moment.fen.split(' ')[1];
-        const fullmove = Number(moment.fen.split(' ')[5]);
-
-        // Check if the next moment is a valid move for black
-        if (
-          moment.depth === current.depth + 1 &&
-          activeColor === 'w' &&
-          fullmove === current.fullmove + 1
-        ) {
-          next.push(moment);
-        }
-      }
-    }
-
-    // If no next moment is found, return an empty array
-    return next;
-  } catch {
-    return [];
-  }
-};
-
-module.exports = getNextMoments;

package/functions/get-prev-moment.js

@@ -1,66 +0,0 @@
-/**
- * Always returns an object because there can be only one previous moment
- */
-const getPrevMoment = (moments, current) => {
-  try {
-    const sameDepth = moments[moments.indexOf(current) - 1];
-    if (sameDepth.move && sameDepth.depth === current.depth) {
-      return sameDepth;
-    }
-
-    // If it's the first moment with move, return the previous moment (starting position)
-    if (current.index === 1 && current.move) {
-      return moments[0];
-    }
-
-    // Split moments until the current index
-    const currentIndex = moments.indexOf(current);
-    const moves = moments.slice(0, currentIndex).filter((m) => m.move);
-
-    // Add fullmove number to the current moment
-    current.fullmove = Number(current.fen.split(' ')[5]);
-
-    // Active color is white
-    if (current.fen.split(' ')[1] === 'w') {
-      // Traverse moves in reverse order to optimize search
-      for (let i = moves.length - 1; i >= 0; i--) {
-        const moment = moves[i];
-        const activeColor = moment.fen.split(' ')[1];
-        const fullmove = Number(moment.fen.split(' ')[5]);
-
-        if (
-          moment.depth === current.depth - 1 &&
-          activeColor === 'b' &&
-          fullmove === current.fullmove - 1
-        ) {
-          return moment;
-        }
-      }
-    }
-
-    // Active color is black
-    if (current.fen.split(' ')[1] === 'b') {
-      // Traverse moves in reverse order to optimize search
-      for (let i = moves.length - 1; i >= 0; i--) {
-        const moment = moves[i];
-        const activeColor = moment.fen.split(' ')[1];
-        const fullmove = Number(moment.fen.split(' ')[5]);
-
-        if (
-          moment.depth === current.depth - 1 &&
-          activeColor === 'w' &&
-          fullmove === current.fullmove
-        ) {
-          return moment;
-        }
-      }
-    }
-
-    // If no previous moment is found, return an empty object
-    return {};
-  } catch {
-    return {};
-  }
-};
-
-module.exports = getPrevMoment;