diff --git a/src/index.js b/src/index.js index f3bb390..38c16c3 100644 --- a/src/index.js +++ b/src/index.js @@ -188,16 +188,17 @@ class Board extends React.Component { let shift = pieceIsBlack ? -1 : 1; let startLine = pieceIsBlack ? 6 : 1; - if (y === startLine) { - return [ - [x, y + shift], - [x, y + (shift * 2)], - ]; - } else { - return [ - [x, y + shift], - ]; + if (this.squareAt(x, y + shift) == null) { + moves.push([x, y + shift]); + if (y === startLine && this.squareAt(x, y + (shift * 2)) == null) { + moves.push([x, y + (shift * 2)]); + } } + [x + 1, x - 1].forEach((x) => { + if (this.isEnemyOf(piece, x, y + shift)) { + moves.push([x, y + shift]); + } + }); } else if (isRook(piece)) { // Down for (i = y + 1; i < 8; i++) {