/*
    Hurry Curry! - a game about cooking
    Copyright 2024 metamuffin

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, version 3 of the License only.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.

*/
use crate::{
    entity::{bot::BotDriver, tutorial::Tutorial},
    server::Server,
    trm,
};
use anyhow::{anyhow, bail, Result};
use clap::{Parser, ValueEnum};
use hurrycurry_bot::algos::ALGO_CONSTRUCTORS;
use hurrycurry_protocol::{Menu, Message, PacketC, PlayerClass, PlayerID};
use std::{fmt::Write, time::Duration};

#[derive(Parser)]
#[clap(multicall = true)]
enum Command {
    /// Start a new game
    Start {
        /// Gamedata specification
        #[arg(default_value = "junior")]
        spec: String,
        /// Duration in seconds
        timer: Option<u64>,
    },
    /// Shows the best entries of the scoreboard for this map.
    #[clap(alias = "top5")]
    Top {
        /// Name of the map, default: current
        map: Option<String>,
    },
    #[clap(alias = "mapinfo")]
    Info {
        /// Name of the map, default: current
        map: Option<String>,
    },
    /// Abort the current game
    End,
    /// Download recipe/map's source declaration
    Download {
        /// Resource kind
        #[arg(value_enum)]
        r#type: DownloadType,
        /// Name
        name: String,
    },
    /// List all recipes and maps
    List,
    /// Send an effect
    Effect {
        name: String,
    },
    /// Send an item
    Item {
        name: String,
    },
    /// Reload the resource index
    ReloadIndex,
    #[clap(alias = "summon", alias = "bot")]
    CreateBot {
        algo: String,
        name: Option<String>,
    },
    /// Reload the current map
    #[clap(alias = "r")]
    Reload,
    /// Shows the recipe book
    Book,
    #[clap(alias = "tutorial")]
    StartTutorial {
        item: String,
    },
    EndTutorial,
    #[clap(alias = "tr")]
    TranslateMessage {
        message_id: String,
        arguments: Vec<String>,
    },
}

#[derive(ValueEnum, Clone)]
enum DownloadType {
    Map,
    Recipes,
}

impl Server {
    pub async fn handle_command_parse(
        &mut self,
        player: PlayerID,
        command: &str,
    ) -> Result<Vec<PacketC>> {
        let mut replies = Vec::new();
        for line in command.split("\n") {
            self.handle_command(
                player,
                Command::try_parse_from(
                    shlex::split(line)
                        .ok_or(anyhow!("quoting invalid"))?
                        .into_iter(),
                )?,
                &mut replies,
            )
            .await?;
        }
        Ok(replies)
    }
    async fn handle_command(
        &mut self,
        player: PlayerID,
        command: Command,
        replies: &mut Vec<PacketC>,
    ) -> Result<()> {
        match command {
            Command::Start { spec, timer } => {
                if !self.game.lobby {
                    self.tx
                        .send(PacketC::ServerMessage {
                            message: trm!(
                                "s.state.game_aborted",
                                s = self
                                    .game
                                    .players
                                    .get(&player)
                                    .ok_or(anyhow!("player missing"))?
                                    .name
                                    .clone()
                            ),
                            error: false,
                        })
                        .ok();
                }
                let data = self.index.generate(&spec).await?;
                self.load(data, timer.map(Duration::from_secs));
            }
            Command::End => {
                self.tx
                    .send(PacketC::ServerMessage {
                        message: trm!(
                            "s.state.game_aborted",
                            s = self
                                .game
                                .players
                                .get(&player)
                                .ok_or(anyhow!("player missing"))?
                                .name
                                .clone()
                        ),
                        error: false,
                    })
                    .ok();
                self.load(self.index.generate("lobby").await?, None);
            }
            Command::Reload => {
                if self.count_chefs() > 1 {
                    bail!("must be at most one player to reload");
                }
                self.load(
                    self.index.generate(&self.game.data.current_map).await?,
                    None,
                );
            }
            Command::ReloadIndex => {
                self.index.reload().await?;
            }
            Command::Book => replies.push(PacketC::Menu(Menu::Book)),
            Command::Download { r#type, name } => {
                let source = match r#type {
                    DownloadType::Map => self.index.read_map(&name).await,
                    DownloadType::Recipes => self.index.read_recipes(&name).await,
                }?;
                bail!("{source}");
            }
            Command::List => {
                bail!(
                    "Maps: {:?}\nRecipes: {:?}",
                    self.index.maps.keys().collect::<Vec<_>>(),
                    self.index.recipes
                )
            }
            Command::Effect { name } => {
                self.tx.send(PacketC::Effect { name, player }).ok();
            }
            Command::Item { name } => {
                let item = self
                    .game
                    .data
                    .get_item_by_name(&name)
                    .ok_or(anyhow!("unknown item"))?;
                self.tx
                    .send(PacketC::Communicate {
                        player,
                        message: Some(Message::Item(item)),
                        timeout: None,
                    })
                    .ok();
            }
            Command::CreateBot { algo, name } => {
                let (aname, cons) = ALGO_CONSTRUCTORS
                    .iter()
                    .find(|(name, _)| *name == algo.as_str())
                    .ok_or(anyhow!("algo name unknown"))?;
                let algo = cons();
                self.entities.push(Box::new(BotDriver::new(
                    format!("{}-bot", name.unwrap_or((*aname).to_owned())),
                    51,
                    PlayerClass::Bot,
                    algo,
                )));
            }
            Command::Top { map } => {
                let mapname = map.as_ref().unwrap_or(&self.game.data.current_map);
                if let Some(board) = self.scoreboard.get(mapname) {
                    let mut o = format!("Scoreboard for {}:\n", mapname);
                    for (i, entry) in board.best.iter().take(5).enumerate() {
                        writeln!(
                            o,
                            "  {i}. {} points by {}",
                            entry.score.points,
                            entry.players.clone().join(", ")
                        )
                        .unwrap();
                    }
                    bail!("{o}");
                }
            }
            Command::Info { map } => {
                let mapname = map.as_ref().unwrap_or(&self.game.data.current_map);
                let info = self
                    .index
                    .maps
                    .get(mapname)
                    .ok_or(anyhow!("no information available"))?;
                bail!(
                    "{:?}\nRecommended player count: {}\nDifficulty: {}",
                    info.name,
                    info.difficulty,
                    info.players
                )
            }
            Command::StartTutorial { item } => {
                let item = self
                    .game
                    .data
                    .get_item_by_name(&item)
                    .ok_or(anyhow!("unknown item"))?;
                #[cfg(not(test))] // TODO rust-analyser does not undestand trait upcasting
                if self
                    .entities
                    .iter()
                    .find(|e| {
                        <dyn std::any::Any>::downcast_ref::<Tutorial>(e.as_ref())
                            .map_or(false, |t| t.player == player)
                    })
                    .is_some()
                {
                    bail!("Tutorial already running")
                }
                self.entities.push(Box::new(Tutorial::new(player, item)));
            }
            Command::EndTutorial => {
                #[cfg(not(test))] // TODO rust-analyser does not undestand trait upcasting
                if let Some(tutorial) = self
                    .entities
                    .iter_mut()
                    .find_map(|e| <dyn std::any::Any>::downcast_mut::<Tutorial>(e.as_mut()))
                {
                    tutorial.ended = true;
                } else {
                    bail!("No tutorial running")
                }
            }
            Command::TranslateMessage {
                message_id,
                arguments,
            } => {
                self.packet_out.push_back(PacketC::Communicate {
                    player,
                    message: Some(Message::Translation {
                        id: message_id,
                        params: arguments.into_iter().map(|c| Message::Text(c)).collect(),
                    }),
                    timeout: None,
                });
            }
        }
        Ok(())
    }
}