use std::path::PathBuf;

use diesel::{PgConnection, QueryResult};
use planetwars_matchrunner::{self as runner, docker_runner::DockerBotSpec, BotSpec, MatchConfig};
use runner::MatchOutcome;
use tokio::task::JoinHandle;

use crate::{
    db::{
        self,
        matches::{MatchData, MatchResult},
    },
    util::gen_alphanumeric,
    ConnectionPool, BOTS_DIR, MAPS_DIR, MATCHES_DIR,
};

const PYTHON_IMAGE: &str = "python:3.10-slim-buster";

pub struct RunMatch {
    log_file_name: String,
    players: Vec<MatchPlayer>,
    match_id: Option<i32>,
}

pub struct MatchPlayer {
    bot_spec: Box<dyn BotSpec>,
    // metadata that will be passed on to database
    code_bundle_id: Option<i32>,
}

impl MatchPlayer {
    pub fn from_bot_version(bot: &db::bots::Bot, version: &db::bots::BotVersion) -> Self {
        MatchPlayer {
            bot_spec: bot_version_to_botspec(bot, version),
            code_bundle_id: Some(version.id),
        }
    }

    /// Construct a MatchPlayer from a BotVersion that certainly contains a code bundle path.
    /// Will panic when this is not the case.
    pub fn from_code_bundle_version(version: &db::bots::BotVersion) -> Self {
        let code_bundle_path = version
            .code_bundle_path
            .as_ref()
            .expect("no code_bundle_path found");
        MatchPlayer {
            bot_spec: python_docker_bot_spec(code_bundle_path),
            code_bundle_id: Some(version.id),
        }
    }

    pub fn from_bot_spec(bot_spec: Box<dyn BotSpec>) -> Self {
        MatchPlayer {
            bot_spec,
            code_bundle_id: None,
        }
    }
}

impl RunMatch {
    pub fn from_players(players: Vec<MatchPlayer>) -> Self {
        let log_file_name = format!("{}.log", gen_alphanumeric(16));
        RunMatch {
            log_file_name,
            players,
            match_id: None,
        }
    }

    pub fn into_runner_config(self) -> runner::MatchConfig {
        runner::MatchConfig {
            map_path: PathBuf::from(MAPS_DIR).join("hex.json"),
            map_name: "hex".to_string(),
            log_path: PathBuf::from(MATCHES_DIR).join(&self.log_file_name),
            players: self
                .players
                .into_iter()
                .map(|player| runner::MatchPlayer {
                    bot_spec: player.bot_spec,
                })
                .collect(),
        }
    }

    pub fn store_in_database(&mut self, db_conn: &PgConnection) -> QueryResult<MatchData> {
        // don't store the same match twice
        assert!(self.match_id.is_none());

        let new_match_data = db::matches::NewMatch {
            state: db::matches::MatchState::Playing,
            log_path: &self.log_file_name,
        };
        let new_match_players = self
            .players
            .iter()
            .map(|p| db::matches::MatchPlayerData {
                code_bundle_id: p.code_bundle_id,
            })
            .collect::<Vec<_>>();

        let match_data = db::matches::create_match(&new_match_data, &new_match_players, &db_conn)?;
        self.match_id = Some(match_data.base.id);
        Ok(match_data)
    }

    pub fn spawn(self, pool: ConnectionPool) -> JoinHandle<MatchOutcome> {
        let match_id = self.match_id.expect("match must be saved before running");
        let runner_config = self.into_runner_config();
        tokio::spawn(run_match_task(pool, runner_config, match_id))
    }
}

pub fn bot_version_to_botspec(
    bot: &db::bots::Bot,
    bot_version: &db::bots::BotVersion,
) -> Box<dyn BotSpec> {
    if let Some(code_bundle_path) = &bot_version.code_bundle_path {
        python_docker_bot_spec(code_bundle_path)
    } else if let Some(container_digest) = &bot_version.container_digest {
        // TODO: put this in config
        let registry_url = "localhost:9001";
        Box::new(DockerBotSpec {
            image: format!("{}/{}@{}", registry_url, bot.name, container_digest),
            binds: None,
            argv: None,
            working_dir: None,
        })
    } else {
        panic!("bad bot version")
    }
}

fn python_docker_bot_spec(code_bundle_path: &str) -> Box<dyn BotSpec> {
    let code_bundle_rel_path = PathBuf::from(BOTS_DIR).join(code_bundle_path);
    let code_bundle_abs_path = std::fs::canonicalize(&code_bundle_rel_path).unwrap();
    let code_bundle_path_str = code_bundle_abs_path.as_os_str().to_str().unwrap();

    // TODO: it would be good to simplify this configuration
    Box::new(DockerBotSpec {
        image: PYTHON_IMAGE.to_string(),
        binds: Some(vec![format!("{}:{}", code_bundle_path_str, "/workdir")]),
        argv: Some(vec!["python".to_string(), "bot.py".to_string()]),
        working_dir: Some("/workdir".to_string()),
    })
}

async fn run_match_task(
    connection_pool: ConnectionPool,
    match_config: MatchConfig,
    match_id: i32,
) -> MatchOutcome {
    let outcome = runner::run_match(match_config).await;

    // update match state in database
    let conn = connection_pool
        .get()
        .await
        .expect("could not get database connection");

    let result = MatchResult::Finished {
        winner: outcome.winner.map(|w| (w - 1) as i32), // player numbers in matchrunner start at 1
    };

    db::matches::save_match_result(match_id, result, &conn).expect("could not save match result");

    outcome
}