From c04d4a449bd147c632c0b6ceae04f0514803b66f Mon Sep 17 00:00:00 2001
From: Ilion Beyst <ilion.beyst@gmail.com>
Date: Sun, 26 Dec 2021 21:06:52 +0100
Subject: rename to planetwars-cli

---
 planetwars-cli/src/lib.rs | 156 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 156 insertions(+)
 create mode 100644 planetwars-cli/src/lib.rs

(limited to 'planetwars-cli/src/lib.rs')

diff --git a/planetwars-cli/src/lib.rs b/planetwars-cli/src/lib.rs
new file mode 100644
index 0000000..7eea75a
--- /dev/null
+++ b/planetwars-cli/src/lib.rs
@@ -0,0 +1,156 @@
+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(())
+}
-- 
cgit v1.2.3