diff options
Diffstat (limited to 'planetwars-localdev/src/lib.rs')
-rw-r--r-- | planetwars-localdev/src/lib.rs | 156 |
1 files changed, 0 insertions, 156 deletions
diff --git a/planetwars-localdev/src/lib.rs b/planetwars-localdev/src/lib.rs deleted file mode 100644 index 7eea75a..0000000 --- a/planetwars-localdev/src/lib.rs +++ /dev/null @@ -1,156 +0,0 @@ -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 { - /// Initialize a new project - InitProject(InitProjectCommand), - /// Run a match - RunMatch(RunMatchCommand), - /// Host local webserver - Serve(ServeCommand), -} - -#[derive(Parser)] -struct RunMatchCommand { - /// map name - map: String, - /// bot names - bots: Vec<String>, -} - -#[derive(Parser)] -struct InitProjectCommand { - /// project root directory - path: String, -} - -#[derive(Parser)] -struct ServeCommand; - -#[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, - Commands::InitProject(command) => init_project(command), - Commands::Serve(_) => run_webserver().await, - }; - 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_name: command.map, - map_path, - log_path, - players, - }; - - match_runner::run_match(match_config).await; - 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); - 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, - } -} - -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(()) -} - -mod web; -async fn run_webserver() -> io::Result<()> { - let project_dir = env::current_dir().unwrap(); - - web::run(project_dir).await; - Ok(()) -} |