OpenFrontIO/src/core/execution/NukeExecution.ts

import {
  Cell,
  Execution,
  Game,
  Player,
  PlayerID,
  Unit,
  UnitType,
  TerraNullius,
  MessageType,
  NukeType,
} from "../game/Game";
import { PseudoRandom } from "../PseudoRandom";
import { consolex } from "../Consolex";
import { TileRef } from "../game/GameMap";

export class NukeExecution implements Execution {
  private player: Player;
  private active = true;
  private mg: Game;
  private nuke: Unit;

  private random: PseudoRandom;

  constructor(
    private type: NukeType,
    private senderID: PlayerID,
    private dst: TileRef,
    private src?: TileRef,
    private speed: number = 4,
    private waitTicks = 0,
  ) {}

  init(mg: Game, ticks: number): void {
    if (!mg.hasPlayer(this.senderID)) {
      console.warn(`NukeExecution: sender ${this.senderID} not found`);
      this.active = false;
      return;
    }

    this.mg = mg;
    this.player = mg.player(this.senderID);
    this.random = new PseudoRandom(ticks);
  }

  public target(): Player | TerraNullius {
    return this.mg.owner(this.dst);
  }

  tick(ticks: number): void {
    if (this.nuke == null) {
      const spawn = this.src ?? this.player.canBuild(this.type, this.dst);
      if (spawn == false) {
        consolex.warn(`cannot build Nuke`);
        this.active = false;
        return;
      }
      this.nuke = this.player.buildUnit(this.type, 0, spawn, {
        detonationDst: this.dst,
      });
      if (this.mg.hasOwner(this.dst)) {
        const target = this.mg.owner(this.dst) as Player;
        if (this.type == UnitType.AtomBomb) {
          this.mg.displayMessage(
            `${this.player.name()} - atom bomb inbound`,
            MessageType.ERROR,
            target.id(),
          );
        }
        if (this.type == UnitType.HydrogenBomb) {
          this.mg.displayMessage(
            `${this.player.name()} - hydrogen bomb inbound`,
            MessageType.ERROR,
            target.id(),
          );
        }

        this.mg
          .stats()
          .increaseNukeCount(
            this.senderID,
            target.id(),
            this.nuke.type() as NukeType,
          );
      }

      // after sending an nuke set the missilesilo on cooldown
      const silo = this.player
        .units(UnitType.MissileSilo)
        .find((silo) => silo.tile() === spawn);
      if (silo) {
        silo.setCooldown(true);
      }
    }

    // make the nuke unactive if it was intercepted
    if (!this.nuke.isActive()) {
      consolex.log(`Nuke destroyed before reaching target`);
      this.active = false;
      return;
    }

    if (this.waitTicks > 0) {
      this.waitTicks--;
      return;
    }

    const r = (this.mg.y(this.dst) * this.mg.x(this.dst)) % 10;
    const s = this.speed + (this.mg.ticks() % r);

    for (let i = 0; i < this.speed; i++) {
      const x = this.mg.x(this.nuke.tile());
      const y = this.mg.y(this.nuke.tile());
      const dstX = this.mg.x(this.dst);
      const dstY = this.mg.y(this.dst);

      // If we've reached the destination, detonate
      if (x === dstX && y === dstY) {
        this.detonate();
        return;
      }

      // Calculate next position
      let nextX = x;
      let nextY = y;

      const ratio = Math.floor(
        1 + Math.abs(dstY - y) / (Math.abs(dstX - x) + 1),
      );

      if (this.random.chance(ratio) && x != dstX) {
        if (x < dstX) nextX++;
        else if (x > dstX) nextX--;
      } else {
        if (y < dstY) nextY++;
        else if (y > dstY) nextY--;
      }

      // Move to next tile
      const nextTile = this.mg.ref(nextX, nextY);
      if (nextTile !== undefined) {
        this.nuke.move(nextTile);
      } else {
        consolex.warn(`invalid tile position ${nextX},${nextY}`);
        this.active = false;
        return;
      }
    }
  }

  private detonate() {
    let magnitude;
    switch (this.type) {
      case UnitType.MIRVWarhead:
        magnitude = { inner: 25, outer: 30 };
        break;
      case UnitType.AtomBomb:
        magnitude = { inner: 12, outer: 30 };
        break;
      case UnitType.HydrogenBomb:
        magnitude = { inner: 80, outer: 100 };
        break;
    }

    const rand = new PseudoRandom(this.mg.ticks());
    const toDestroy = this.mg.bfs(this.dst, (_, n: TileRef) => {
      const d = this.mg.euclideanDist(this.dst, n);
      return (d <= magnitude.inner || rand.chance(2)) && d <= magnitude.outer;
    });

    const attacked = new Map<Player, number>();
    for (const tile of toDestroy) {
      const owner = this.mg.owner(tile);
      if (owner.isPlayer()) {
        const mp = this.mg.player(owner.id());
        mp.relinquish(tile);
        mp.removeTroops((5 * mp.population()) / mp.numTilesOwned());
        if (!attacked.has(mp)) {
          attacked.set(mp, 0);
        }
        const prev = attacked.get(mp);
        attacked.set(mp, prev + 1);
      }

      if (this.mg.isLand(tile)) {
        this.mg.setFallout(tile, true);
      }
    }
    for (const [other, tilesDestroyed] of attacked) {
      if (tilesDestroyed > 100 && this.nuke.type() != UnitType.MIRVWarhead) {
        // Mirv warheads shouldn't break alliances
        const alliance = this.player.allianceWith(other);
        if (alliance != null) {
          this.player.breakAlliance(alliance);
        }
        if (other != this.player) {
          other.updateRelation(this.player, -100);
        }
      }
    }

    for (const unit of this.mg.units()) {
      if (
        unit.type() != UnitType.AtomBomb &&
        unit.type() != UnitType.HydrogenBomb &&
        unit.type() != UnitType.MIRVWarhead &&
        unit.type() != UnitType.MIRV
      ) {
        if (this.mg.euclideanDist(this.dst, unit.tile()) < magnitude.outer) {
          unit.delete();
        }
      }
    }
    this.active = false;
    this.nuke.delete(false);
  }

  owner(): Player {
    return this.player;
  }

  isActive(): boolean {
    return this.active;
  }

  activeDuringSpawnPhase(): boolean {
    return false;
  }
}