## Bullet economy test: SNNBot vs WallsBot.
## Prints per-round and summary stats for tuning. No hard pass/fail.
## Requires TR server JARs (set TR_SERVER_JAR / TR_BATTLE_RUNNER env vars).
## Set TR_SAMPLE_BOTS to point to sample-bots directory (e.g. sample-bots-nim-linux-1.0.7).

import std/[os, strformat]
import test_framework/battle_result

const
  DefaultServerJar = "/home/davide/Projects/tank-royale/server/build/libs/robocode-tankroyale-server-0.35.5-all.jar"
  DefaultRunnerJar = "/home/davide/Projects/tank-royale/runner/examples/lib/robocode-tankroyale-runner.jar"
  DefaultSampleBots = "/home/davide/Downloads/robocode-tankroyale/sample-bots-nim-linux-1.0.7"

let serverJar = getEnv("TR_SERVER_JAR",    DefaultServerJar)
let runnerJar = getEnv("TR_BATTLE_RUNNER", DefaultRunnerJar)
let sampleBotsDir = getEnv("TR_SAMPLE_BOTS", DefaultSampleBots)

if not fileExists(serverJar) or not fileExists(runnerJar):
  echo "Skipping: JAR files not found (set TR_SERVER_JAR / TR_BATTLE_RUNNER to override)"
  quit(0)

if not dirExists(sampleBotsDir):
  echo "Skipping: sample bots directory not found at ", sampleBotsDir
  echo "Set TR_SAMPLE_BOTS to override"
  quit(0)

import test_framework/test_framework

const
  snnbotDir = currentSourcePath().parentDir.parentDir

let wallsbotDir = sampleBotsDir / "Walls"

let r = runBattle(@[snnbotDir, wallsbotDir], rounds = 10)

# Per-round stats
echo "=== Per-round results ==="
var snnWins, wallsWins: int
for rnd in r.rounds:
  var snn, walls: BotRoundResult
  for br in rnd.results:
    if br.name == "SNNBot":      snn   = br
    if br.name == "Walls (Nim)": walls = br
  # rank 1 = winner; lower rank is better; default rank=0 means bot not found
  let winner = if snn.rank > 0 and (walls.rank == 0 or snn.rank < walls.rank): "SNNBot" else: "Walls"
  if winner == "SNNBot": inc snnWins else: inc wallsWins
  echo fmt"Round {rnd.round:2}: {winner} wins | SNNBot score={snn.score} survived={snn.survived} | Walls score={walls.score} survived={walls.survived}"

# Summary
echo ""
echo "=== Summary (10 rounds) ==="
var snnFinal, wallsFinal: BotResult
for b in r.results:
  if b.name == "SNNBot":      snnFinal   = b
  if b.name == "Walls (Nim)": wallsFinal = b

echo fmt"SNNBot : wins={snnWins}/10  totalScore={snnFinal.totalScore}  survivalRounds={snnFinal.survivalCount}  firstPlaces={snnFinal.firstPlaces}"
echo fmt"Walls  : wins={wallsWins}/10  totalScore={wallsFinal.totalScore}  survivalRounds={wallsFinal.survivalCount}  firstPlaces={wallsFinal.firstPlaces}"
echo fmt"Win rate : {snnWins * 100 div 10}%"