import { SpawnExecution } from "../src/core/execution/SpawnExecution";
import { Game, Player, PlayerInfo, PlayerType } from "../src/core/game/Game";
import { GameID } from "../src/core/Schemas";
import { setup } from "./util/Setup";

const gameID: GameID = "test_game";

function addPlayerWithGold(
  game: Game,
  id: string,
  type: PlayerType,
  gold: bigint,
): Player {
  game.addPlayer(new PlayerInfo(id, type, null, id));
  const player = game.player(id);
  player.addGold(gold);
  return player;
}

describe("DefaultConfig.conquerGoldAmount", () => {
  let game: Game;

  beforeEach(async () => {
    game = await setup("ocean_and_land");
  });

  test("returns full gold for Bot", () => {
    const bot = addPlayerWithGold(game, "bot", PlayerType.Bot, 1000n);
    expect(game.config().conquerGoldAmount(bot)).toBe(1000n);
  });

  test("returns full gold for Nation", () => {
    const nation = addPlayerWithGold(game, "nation", PlayerType.Nation, 2000n);
    expect(game.config().conquerGoldAmount(nation)).toBe(2000n);
  });

  test("returns half gold for Human", () => {
    const human = addPlayerWithGold(game, "human", PlayerType.Human, 1000n);
    expect(game.config().conquerGoldAmount(human)).toBe(500n);
  });
});

describe("Conquest gold transfer", () => {
  let game: Game;
  let conqueror: Player;

  beforeEach(async () => {
    game = await setup("ocean_and_land");
    const conquerorInfo = new PlayerInfo(
      "conqueror",
      PlayerType.Human,
      null,
      "conqueror",
    );
    game.addPlayer(conquerorInfo);
    game.addExecution(
      new SpawnExecution(gameID, conquerorInfo, game.ref(0, 10)),
    );
    while (game.inSpawnPhase()) {
      game.executeNextTick();
    }
    conqueror = game.player(conquerorInfo.id);
  });

  test("conqueror receives 100% of gold when conquering a Bot", () => {
    const bot = addPlayerWithGold(game, "bot", PlayerType.Bot, 1000n);
    const goldBefore = conqueror.gold();
    game.conquerPlayer(conqueror, bot);
    expect(conqueror.gold()).toBe(goldBefore + 1000n);
    expect(bot.gold()).toBe(0n);
  });

  test("conqueror receives 100% of gold when conquering a Nation", () => {
    const nation = addPlayerWithGold(game, "nation", PlayerType.Nation, 800n);
    const goldBefore = conqueror.gold();
    game.conquerPlayer(conqueror, nation);
    expect(conqueror.gold()).toBe(goldBefore + 800n);
    expect(nation.gold()).toBe(0n);
  });

  test("conqueror receives 50% of gold when conquering a Human who has attacked", () => {
    // clientID must be non-null for stats tracking to work
    game.addPlayer(
      new PlayerInfo("victim", PlayerType.Human, "victim_client", "victim"),
    );
    const victim = game.player("victim");
    victim.addGold(1000n);
    // Record an attack so the gold transfer is not skipped
    game.stats().attack(victim, game.terraNullius(), 100);
    const goldBefore = conqueror.gold();
    game.conquerPlayer(conqueror, victim);
    expect(conqueror.gold()).toBe(goldBefore + 500n);
    expect(victim.gold()).toBe(0n);
  });

  test("conqueror receives no gold when conquering a Human who never attacked", () => {
    const victim = addPlayerWithGold(game, "afk", PlayerType.Human, 1000n);
    const goldBefore = conqueror.gold();
    game.conquerPlayer(conqueror, victim);
    expect(conqueror.gold()).toBe(goldBefore);
    expect(victim.gold()).toBe(1000n);
  });
});