2021-12-25 14:45:05 +01:00
|
|
|
use match_runner::{MatchBot, MatchConfig};
|
|
|
|
use serde::Deserialize;
|
|
|
|
|
|
|
|
mod match_runner;
|
|
|
|
|
|
|
|
use serde::Serialize;
|
|
|
|
use std::collections::HashMap;
|
|
|
|
use std::env;
|
|
|
|
use std::io;
|
|
|
|
use std::path::{Path, PathBuf};
|
|
|
|
use toml;
|
|
|
|
|
|
|
|
use clap::{Parser, Subcommand};
|
|
|
|
|
|
|
|
#[derive(Parser)]
|
|
|
|
#[clap(name = "pwcli")]
|
|
|
|
#[clap(author, version, about)]
|
|
|
|
struct Cli {
|
|
|
|
#[clap(subcommand)]
|
|
|
|
command: Commands,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Subcommand)]
|
|
|
|
enum Commands {
|
2021-12-25 15:36:23 +01:00
|
|
|
/// Initialize a new project
|
|
|
|
InitProject(InitProjectCommand),
|
2021-12-25 14:45:05 +01:00
|
|
|
/// Run a match
|
|
|
|
RunMatch(RunMatchCommand),
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Parser)]
|
|
|
|
struct RunMatchCommand {
|
|
|
|
/// map name
|
|
|
|
map: String,
|
|
|
|
/// bot names
|
|
|
|
bots: Vec<String>,
|
|
|
|
}
|
|
|
|
|
2021-12-25 15:36:23 +01:00
|
|
|
#[derive(Parser)]
|
|
|
|
struct InitProjectCommand {
|
|
|
|
/// project root directory
|
|
|
|
path: String,
|
|
|
|
}
|
|
|
|
|
2021-12-25 14:45:05 +01:00
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
|
|
struct ProjectConfig {
|
|
|
|
bots: HashMap<String, BotConfig>,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
|
|
|
pub struct BotConfig {
|
|
|
|
path: String,
|
|
|
|
argv: Vec<String>,
|
|
|
|
}
|
|
|
|
|
|
|
|
pub async fn run() {
|
|
|
|
let matches = Cli::parse();
|
|
|
|
let res = match matches.command {
|
|
|
|
Commands::RunMatch(command) => run_match(command).await,
|
2021-12-25 15:36:23 +01:00
|
|
|
Commands::InitProject(command) => init_project(command),
|
2021-12-25 14:45:05 +01:00
|
|
|
};
|
|
|
|
if let Err(err) = res {
|
|
|
|
eprintln!("{}", err);
|
|
|
|
std::process::exit(1);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
async fn run_match(command: RunMatchCommand) -> io::Result<()> {
|
|
|
|
let project_dir = env::current_dir().unwrap();
|
|
|
|
|
|
|
|
let config_path = project_dir.join("pw_project.toml");
|
|
|
|
|
|
|
|
let map_path = project_dir.join(format!("maps/{}.json", command.map));
|
|
|
|
|
|
|
|
let timestamp = chrono::Local::now().format("%Y-%m-%d-%H-%M-%S");
|
|
|
|
let log_path = project_dir.join(format!("matches/{}.log", timestamp));
|
|
|
|
|
|
|
|
let config_str = std::fs::read_to_string(config_path).unwrap();
|
|
|
|
let project_config: ProjectConfig = toml::from_str(&config_str).unwrap();
|
|
|
|
|
|
|
|
let players = command
|
|
|
|
.bots
|
|
|
|
.into_iter()
|
|
|
|
.map(|bot_name| {
|
|
|
|
let bot_config = project_config.bots.get(&bot_name).unwrap().clone();
|
|
|
|
let resolved_config = resolve_bot_config(&project_dir, bot_config);
|
|
|
|
MatchBot {
|
|
|
|
name: bot_name,
|
|
|
|
bot_config: resolved_config,
|
|
|
|
}
|
|
|
|
})
|
|
|
|
.collect();
|
|
|
|
|
|
|
|
let match_config = MatchConfig {
|
|
|
|
map_path,
|
|
|
|
log_path,
|
|
|
|
players,
|
|
|
|
};
|
|
|
|
|
|
|
|
match_runner::run_match(match_config).await;
|
2021-12-25 15:36:23 +01:00
|
|
|
println!("match completed successfully");
|
|
|
|
// TODO: don't hardcode match path.
|
|
|
|
// maybe print the match result as well?
|
|
|
|
println!("wrote match log to matches/{}.log", timestamp);
|
2021-12-25 14:45:05 +01:00
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn resolve_bot_config(project_dir: &Path, config: BotConfig) -> BotConfig {
|
|
|
|
let mut path = PathBuf::from(project_dir);
|
|
|
|
path.push(&config.path);
|
|
|
|
BotConfig {
|
|
|
|
path: path.to_str().unwrap().to_string(),
|
|
|
|
argv: config.argv,
|
|
|
|
}
|
|
|
|
}
|
2021-12-25 15:36:23 +01:00
|
|
|
|
|
|
|
macro_rules! copy_asset {
|
|
|
|
($path:expr, $file_name:literal) => {
|
|
|
|
::std::fs::write(
|
|
|
|
$path.join($file_name),
|
|
|
|
include_bytes!(concat!("../assets/", $file_name)),
|
|
|
|
)?;
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
fn init_project(command: InitProjectCommand) -> io::Result<()> {
|
|
|
|
let path = PathBuf::from(&command.path);
|
|
|
|
|
|
|
|
// create directories
|
|
|
|
std::fs::create_dir_all(&path)?;
|
|
|
|
std::fs::create_dir(path.join("maps"))?;
|
|
|
|
std::fs::create_dir(path.join("matches"))?;
|
|
|
|
std::fs::create_dir_all(path.join("bots/simplebot"))?;
|
|
|
|
|
|
|
|
// create files
|
|
|
|
copy_asset!(path, "pw_project.toml");
|
|
|
|
copy_asset!(path.join("maps"), "hex.json");
|
|
|
|
copy_asset!(path.join("bots/simplebot"), "simplebot.py");
|
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|