Clorange: almost functional draft (promotions, capture non-violent...)

[xogo.git] / variants / Clorange / class.js

import ChessRules from "/base_rules.js";

export default class ClorangeRules extends ChessRules {

  static get Options() {
    return {
      select: C.Options.select,
      styles:
        C.Options.styles.filter(s => !["crazyhouse","recycle"].includes(s))
    };
  }

  get hasReserve() {
    return true;
  }

  getReserveFen(o) {
    if (o.init)
      return "00000000000000000000";
    return (
      ["w","b"].map(c => Object.values(this.reserve[c]).join("")).join("")
    );
  }

  pieces(color, x, y) {
    let res = super.pieces(color, x, y);
    res['s'] = {"class": "nv-pawn", moveas: "p"};
    res['u'] = {"class": "nv-rook", moveas: "r"};
    res['o'] = {"class": "nv-knight", moveas: "n"};
    res['c'] = {"class": "nv-bishop", moveas: "b"};
    res['t'] = {"class": "nv-queen", moveas: "q"};
    return res;
  }

  static get V_PIECES() {
    return ['p', 'r', 'n', 'b', 'q'];
  }
  static get NV_PIECES() {
    return ['s', 'u', 'o', 'c', 't'];
  }

  setOtherVariables(fen) {
    super.setOtherVariables(fen, V.V_PIECES.concat(V.NV_PIECES));
  }

  // Forbid non-violent pieces to capture
  canTake([x1, y1], [x2, y2]) {
    return (
      this.getColor(x1, y1) !== this.getColor(x2, y2) &&
      (['k'].concat(V.V_PIECES)).includes(this.getPiece(x1, y1))
    );
  }

  prePlay(move) {
    super.prePlay(move);
    // No crazyhouse or recycle, so the last call didn't update reserve:
    if (
      (move.vanish.length == 2 && move.appear.length == 1) ||
      move.vanish.length == 0 //drop
    ) {
      const trPiece =
        (move.vanish.length > 0 ? move.vanish[1].p : move.appear[0].p);
      const normal = V.V_PIECES.includes(trPiece);
      const pIdx = (normal ? V.V_PIECES : V.NV_PIECES).indexOf(trPiece);
      const resPiece = (normal ? V.NV_PIECES : V.V_PIECES)[pIdx];
      if (move.vanish.length > 0) {
        super.updateReserve(C.GetOppTurn(this.turn), resPiece,
          this.reserve[C.GetOppTurn(this.turn)][resPiece] + 1);
      }
      else {
        super.updateReserve(this.turn, resPiece,
          this.reserve[this.turn][resPiece] - 1);
      }
    }
  }

};