Chess.js: In move_from_san(), only bother generating relevant candidate moves.

[remoteglot] / www / js / chess.js

diff --git a/www/js/chess.js b/www/js/chess.js
index 91f6c1dda7af308f342e08e05095ca26238566c7..3c97b646e03c07565556ed54209c78b7dc99ee85 100644 (file)
--- a/www/js/chess.js
+++ b/www/js/chess.js
@@ -1176,10 +1176,10 @@ var Chess = function(fen) {
     return '';
   }
 
-  // Find all pseudolegal moves featuring the given piece attacking
+  // Find all pseudolegal moves featuring the given piece moving to
   // the given square (using symmetry of all non-pawn-or-castle moves,
-  // we simply generate // moves backwards). Does not support kings
-  // or pawns. Assumes there's // not already a piece of our own color
+  // we simply generate moves backwards). Does not support pawns.
+  // Assumes there's not already a piece of our own color
   // on the destination square.
   function find_attacking_moves(to, piece, us) {
     let moves = [];
@@ -1195,7 +1195,7 @@ var Chess = function(fen) {
         if (square & 0x88) break;
 
         if (board[square] != null) {
-          if (board[square].color !== us) break;
+          if (board[square].color !== us || board[square].type !== piece) break;
           if (board[to] == null) {
             add_move(board, moves, square, to, BITS.NORMAL);
           } else {
@@ -1204,8 +1204,8 @@ var Chess = function(fen) {
           break;
         }
 
-        /* break if knight */
-        if (piece === 'n') break;
+        /* break if knight or king */
+        if (piece === 'n' || piece === 'k') break;
       }
     }
 
@@ -1259,7 +1259,23 @@ var Chess = function(fen) {
       }
     }
 
-    var moves = generate_moves();
+    let moves;
+    let piece_matches = clean_move.match(/^([NBRQK])x?([a-h][1-8])$/);
+    if (piece_matches) {
+      // Only look for moves by the given piece to the given square.
+      let to = SQUARES[piece_matches[2]];
+      if (board[to] != null && board[to].color === turn) {
+        // Cannot capture our own piece.
+        return null;
+      }
+      moves = find_attacking_moves(to, piece_matches[1].toLowerCase(), turn);
+      // Legal moves only.
+      moves = possibly_filter_moves(moves, turn, true);
+    } else {
+      // Fallback (also used for pawns): Any (legal) moves.
+      moves = generate_moves();
+    }
+
     for (var i = 0, len = moves.length; i < len; i++) {
       // try the strict parser first, then the sloppy parser if requested
       // by the user