Capturing king now means checkmate

Author: Aversefun

index.ts

@@ -437,7 +437,6 @@ function get_allowed_moves(piece: [Piece, Color], from: Square, ignore_check: bo
       }
       return !in_check(piece[1], [piece[0], from, v]);
     })
-    .filter((v) => get_square(v) === null || get_square(v)![0] !== Piece.King);
 
   return allowed_movements;
 }
@@ -738,6 +737,10 @@ board.addEventListener('click', function (event) {
       }
     }
 
+    if (get_square(square) !== null && get_square(square)![0] === Piece.King && get_square(square)![1] === Color.opposite(turn)) {
+      checkmate = true;
+    }
+
     set_square(square, piece);
     set_square(squareToMove, null);