Fix games rendering
[vchess.git] / client / src / variants / Rifle.js
1 import { ChessRules, PiPo, Move } from "@/base_rules";
2
3 export const VariantRules = class RifleRules extends ChessRules {
4 getEpSquare(moveOrSquare) {
5 if (typeof moveOrSquare !== "object" || moveOrSquare.appear.length > 0)
6 return super.getEpSquare(moveOrSquare);
7 // Capturing move: no en-passant
8 return undefined;
9 }
10
11 getBasicMove([sx, sy], [ex, ey], tr) {
12 let mv = new Move({
13 appear: [],
14 vanish: [],
15 start: {x:sx, y:sy},
16 end: {x:ex, y:ey}
17 });
18 if (this.board[ex][ey] != V.EMPTY) {
19 // No movement: just vanishing enemy piece
20 mv.vanish = [
21 new PiPo({
22 x: ex,
23 y: ey,
24 c: this.getColor(ex, ey),
25 p: this.getPiece(ex, ey)
26 })
27 ];
28 }
29 else {
30 // Normal move
31 mv.appear = [
32 new PiPo({
33 x: ex,
34 y: ey,
35 c: tr ? tr.c : this.getColor(sx, sy),
36 p: tr ? tr.p : this.getPiece(sx, sy)
37 })
38 ];
39 mv.vanish = [
40 new PiPo({
41 x: sx,
42 y: sy,
43 c: this.getColor(sx, sy),
44 p: this.getPiece(sx, sy)
45 })
46 ];
47 }
48
49 return mv;
50 }
51
52 getPotentialPawnMoves([x, y]) {
53 const color = this.turn;
54 let moves = [];
55 const [sizeX, sizeY] = [V.size.x, V.size.y];
56 const shiftX = color == "w" ? -1 : 1;
57 const startRank = color == "w" ? sizeX - 2 : 1;
58 const lastRank = color == "w" ? 0 : sizeX - 1;
59
60 const finalPieces =
61 x + shiftX == lastRank
62 ? [V.ROOK, V.KNIGHT, V.BISHOP, V.QUEEN]
63 : [V.PAWN];
64 if (this.board[x + shiftX][y] == V.EMPTY) {
65 for (let piece of finalPieces) {
66 moves.push(
67 this.getBasicMove([x, y], [x + shiftX, y], {
68 c: color,
69 p: piece
70 })
71 );
72 }
73 if (
74 x == startRank &&
75 this.board[x + 2 * shiftX][y] == V.EMPTY
76 ) {
77 moves.push(this.getBasicMove([x, y], [x + 2 * shiftX, y]));
78 }
79 }
80 // Captures
81 for (let shiftY of [-1, 1]) {
82 if (
83 y + shiftY >= 0 &&
84 y + shiftY < sizeY &&
85 this.board[x + shiftX][y + shiftY] != V.EMPTY &&
86 this.canTake([x, y], [x + shiftX, y + shiftY])
87 ) {
88 for (let piece of finalPieces) {
89 moves.push(
90 this.getBasicMove([x, y], [x + shiftX, y + shiftY], {
91 c: color,
92 p: piece
93 })
94 );
95 }
96 }
97 }
98
99 // En passant
100 const Lep = this.epSquares.length;
101 const epSquare = this.epSquares[Lep - 1]; //always at least one element
102 if (
103 !!epSquare &&
104 epSquare.x == x + shiftX &&
105 Math.abs(epSquare.y - y) == 1
106 ) {
107 let enpassantMove = new Move({
108 appear: [],
109 vanish: [],
110 start: {x:x, y:y},
111 end: {x:x+shiftX, y:epSquare.y}
112 });
113 enpassantMove.vanish.push({
114 x: x,
115 y: epSquare.y,
116 p: "p",
117 c: this.getColor(x, epSquare.y)
118 });
119 moves.push(enpassantMove);
120 }
121
122 return moves;
123 }
124 };