use super::{

    client::HWClient,

    core::HWServer,

    coretypes::{ClientId, GameCfg, RoomId, VoteType},

    handlers,

    room::HWRoom,

    room::{GameInfo, RoomFlags},

};

use crate::{

    protocol::messages::{server_chat, HWProtocolMessage, HWServerMessage, HWServerMessage::*},

    utils::to_engine_msg,

};

use rand::{distributions::Uniform, thread_rng, Rng};

use std::{io, io::Write, iter::once, mem::replace};

#[cfg(feature = "official-server")]

use super::database;

pub enum Destination {

    ToId(ClientId),

    ToSelf,

    ToAll {

        room_id: Option<RoomId>,

        protocol: Option<u16>,

        skip_self: bool,

    },

}

pub struct PendingMessage {

    pub destination: Destination,

    pub message: HWServerMessage,

}

impl PendingMessage {

    pub fn send(message: HWServerMessage, client_id: ClientId) -> PendingMessage {

        PendingMessage {

            destination: Destination::ToId(client_id),

            message,

        }

    }

    pub fn send_self(message: HWServerMessage) -> PendingMessage {

        PendingMessage {

            destination: Destination::ToSelf,

            message,

        }

    }

    pub fn send_all(message: HWServerMessage) -> PendingMessage {

        let destination = Destination::ToAll {

            room_id: None,

            protocol: None,

            skip_self: false,

        };

        PendingMessage {

            destination,

            message,

        }

    }

    pub fn in_room(mut self, clients_room_id: RoomId) -> PendingMessage {

        if let Destination::ToAll {

            ref mut room_id, ..

        } = self.destination

        {

            *room_id = Some(clients_room_id)

        }

        self

    }

    pub fn with_protocol(mut self, protocol_number: u16) -> PendingMessage {

        if let Destination::ToAll {

            ref mut protocol, ..

        } = self.destination

        {

            *protocol = Some(protocol_number)

        }

        self

    }

    pub fn but_self(mut self) -> PendingMessage {

        if let Destination::ToAll {

            ref mut skip_self, ..

        } = self.destination

        {

            *skip_self = true

        }

        self

    }

}

impl HWServerMessage {

    pub fn send(self, client_id: ClientId) -> PendingMessage {

        PendingMessage::send(self, client_id)

    }

    pub fn send_self(self) -> PendingMessage {

        PendingMessage::send_self(self)

    }

    pub fn send_all(self) -> PendingMessage {

        PendingMessage::send_all(self)

    }

}

pub enum Action {

    ChangeMaster(RoomId, Option<ClientId>),

    SendRoomUpdate(Option<String>),

    StartRoomGame(RoomId),

    SendTeamRemovalMessage(String),

    FinishRoomGame(RoomId),

    SendRoomData {

        to: ClientId,

        teams: bool,

        config: bool,

        flags: bool,

    },

    AddVote {

        vote: bool,

        is_forced: bool,

    },

    ApplyVoting(VoteType, RoomId),

}

use self::Action::*;

pub fn run_action(server: &mut HWServer, client_id: usize, action: Action) {

    match action {

        SendRoomData {

            to,

            teams,

            config,

            flags,

        } => {

            let mut actions = Vec::new();

            let room_id = server.clients[client_id].room_id;

            if let Some(r) = room_id.and_then(|id| server.rooms.get(id)) {

                if config {

                        ConfigEntry("FULLMAPCONFIG".to_string(), r.map_config())

                            .send(to)

                            .action(),

                    for cfg in r.game_config() {

                    }

                }

                if teams {

                    let current_teams = match r.game_info {

                        Some(ref info) => &info.teams_at_start,

                        None => &r.teams,

                    };

                    for (owner_id, team) in current_teams.iter() {

                            TeamAdd(HWRoom::team_info(&server.clients[*owner_id], &team))

                                .send(to)

                                .action(),

                        );

                        actions.push(TeamColor(team.name.clone(), team.color).send(to).action());

                        actions.push(

                            HedgehogsNumber(team.name.clone(), team.hedgehogs_number)

                                .send(to)

                                .action(),

                    }

                }

                if flags {

                    if let Some(id) = r.master_id {

                    }

                    let nicks: Vec<_> = server

                        .clients

                        .iter()

                        .filter(|(_, c)| c.room_id == Some(r.id) && c.is_ready())

                        .map(|(_, c)| c.nick.clone())

                        .collect();

                    if !nicks.is_empty() {

                    }

                }

            }

            server.react(client_id, actions);

        }

        AddVote { vote, is_forced } => {

            let mut actions = Vec::new();

            if let Some(r) = server.room(client_id) {

                let mut result = None;

                if let Some(ref mut voting) = r.voting {

                    if is_forced || voting.votes.iter().all(|(id, _)| client_id != *id) {

                            server_chat("Your vote has been counted.".to_string())

                                .send_self()

                                .action(),

                        voting.votes.push((client_id, vote));

                        let i = voting.votes.iter();

                        let pro = i.clone().filter(|(_, v)| *v).count();

                        let contra = i.filter(|(_, v)| !*v).count();

                        let success_quota = voting.voters.len() / 2 + 1;

                        if is_forced && vote || pro >= success_quota {

                            result = Some(true);

                        } else if is_forced && !vote || contra > voting.voters.len() - success_quota

                        {

                            result = Some(false);

                        }

                    } else {

                            server_chat("You already have voted.".to_string())

                                .send_self()

                                .action(),

                    }

                } else {

                        server_chat("There's no voting going on.".to_string())

                            .send_self()

                            .action(),

                }

                if let Some(res) = result {

                        server_chat("Voting closed.".to_string())

                            .send_all()

                            .in_room(r.id)

                            .action(),

                    let voting = replace(&mut r.voting, None).unwrap();

                    if res {

                        actions.push(ApplyVoting(voting.kind, r.id));

                    }

                }

            }

            server.react(client_id, actions);

        }

        ApplyVoting(kind, room_id) => {

            let mut actions = Vec::new();

            let mut id = client_id;

            match kind {

                VoteType::Kick(nick) => {

                    if let Some(c) = server.find_client(&nick) {

                        if c.room_id == Some(room_id) {

                            id = c.id;

                            //actions.push(MoveToLobby("kicked".to_string()));

                        }

                    }

                }

                VoteType::Map(None) => (),

                VoteType::Map(Some(name)) => {

                    if let Some(location) = server.rooms[room_id].load_config(&name) {

                            server_chat(location.to_string())

                                .send_all()

                                .in_room(room_id)

                                .action(),

                        actions.push(SendRoomUpdate(None));

                        for (_, c) in server.clients.iter() {

                            if c.room_id == Some(room_id) {

                                actions.push(SendRoomData {

                                    to: c.id,

                                    teams: false,

                                    config: true,

                                    flags: false,

                                })

                            }

                        }

                    }

                }

                VoteType::Pause => {

                    if let Some(ref mut info) = server.rooms[room_id].game_info {

                        info.is_paused = !info.is_paused;

                            server_chat("Pause toggled.".to_string())

                                .send_all()

                                .in_room(room_id)

                                .action(),

                            ForwardEngineMessage(vec![to_engine_msg(once(b'I'))])

                                .send_all()

                                .in_room(room_id)

                                .action(),

                    }

                }

                VoteType::NewSeed => {

                    let seed = thread_rng().gen_range(0, 1_000_000_000).to_string();

                    let cfg = GameCfg::Seed(seed);

                    server.rooms[room_id].set_config(cfg);

                }

                VoteType::HedgehogsPerTeam(number) => {

                    let r = &mut server.rooms[room_id];

                    let nicks = r.set_hedgehogs_number(number);

                        HedgehogsNumber(n, number)

                            .send_all()

                            .in_room(room_id)

                            .action()

                }

            }

            server.react(id, actions);

        }

        ChangeMaster(room_id, new_id) => {

            let mut actions = Vec::new();

            let room_client_ids = server.room_clients(room_id);

            let new_id = if server

                .room(client_id)

                .map(|r| r.is_fixed())

                .unwrap_or(false)

            {

                new_id

            } else {

                new_id.or_else(|| room_client_ids.iter().find(|id| **id != client_id).cloned())

            };

            let new_nick = new_id.map(|id| server.clients[id].nick.clone());

            if let (c, Some(r)) = server.client_and_room(client_id) {

                match r.master_id {

                    Some(id) if id == c.id => {

                        c.set_is_master(false);

                        r.master_id = None;

                            ClientFlags("-h".to_string(), vec![c.nick.clone()])

                                .send_all()

                                .in_room(r.id)

                                .action(),

                    }

                    Some(_) => unreachable!(),

                    None => {}

                }

                r.master_id = new_id;

                if !r.is_fixed() && c.protocol_number < 42 {

                    r.name

                        .replace_range(.., new_nick.as_ref().map_or("[]", String::as_str));

                }

                r.set_join_restriction(false);

                r.set_team_add_restriction(false);

                let is_fixed = r.is_fixed();

                r.set_unregistered_players_restriction(is_fixed);

                if let Some(nick) = new_nick {

                        ClientFlags("+h".to_string(), vec![nick])

                            .send_all()

                            .in_room(r.id)

                            .action(),

                }

            }

            if let Some(id) = new_id {

                server.clients[id].set_is_master(true)

            }

            server.react(client_id, actions);

        }

        SendRoomUpdate(old_name) => {

            if let (c, Some(r)) = server.client_and_room(client_id) {

                let name = old_name.unwrap_or_else(|| r.name.clone());

                    .send_all()

                    .with_protocol(r.protocol_number)

                    .action()];

            }

        }

        StartRoomGame(room_id) => {

            let actions = {

                let (room_clients, room_nicks): (Vec<_>, Vec<_>) = server

                    .clients

                    .iter()

                    .map(|(id, c)| (id, c.nick.clone()))

                    .unzip();

                let room = &mut server.rooms[room_id];

                if !room.has_multiple_clans() {

                        "The game can't be started with less than two clans!".to_string(),

                } else if room.protocol_number <= 43

                    && room.players_number != room.ready_players_number

                {

                } else if room.game_info.is_some() {

                } else {

                    room.start_round();

                    for id in room_clients {

                        let c = &mut server.clients[id];

                        c.set_is_in_game(false);

                        c.team_indices = room.client_team_indices(c.id);

                    }

                    vec![

                        SendRoomUpdate(None),

                    ]

                }

            };

            server.react(client_id, actions);

        }

        SendTeamRemovalMessage(team_name) => {

            let mut actions = Vec::new();

            if let Some(r) = server.room(client_id) {

                if let Some(ref mut info) = r.game_info {

                    let msg = once(b'F').chain(team_name.bytes());

                        ForwardEngineMessage(vec![to_engine_msg(msg)])

                            .send_all()

                            .in_room(r.id)

                            .but_self()

                            .action(),

                    info.teams_in_game -= 1;

                    if info.teams_in_game == 0 {

                        actions.push(FinishRoomGame(r.id));

                    }

                    let remove_msg = to_engine_msg(once(b'F').chain(team_name.bytes()));

                    if let Some(m) = &info.sync_msg {

                        info.msg_log.push(m.clone());

                    }

                    if info.sync_msg.is_some() {

                        info.sync_msg = None

                    }

                    info.msg_log.push(remove_msg.clone());

                        ForwardEngineMessage(vec![remove_msg])

                            .send_all()

                            .in_room(r.id)

                            .but_self()

                            .action(),

                }

            }

            server.react(client_id, actions);

        }

        FinishRoomGame(room_id) => {

            let mut actions = Vec::new();

            let r = &mut server.rooms[room_id];

            r.ready_players_number = 1;

            actions.push(SendRoomUpdate(None));

            if let Some(info) = replace(&mut r.game_info, None) {

                for (_, c) in server.clients.iter() {

                    if c.room_id == Some(room_id) && c.is_joined_mid_game() {

                        actions.push(SendRoomData {

                            to: c.id,

                            teams: false,

                            config: true,

                            flags: false,

                        });

                        for name in &info.left_teams {

                        }

                    }

                }

            }

            let nicks: Vec<_> = server

                .clients

                .iter_mut()

                .filter(|(_, c)| c.room_id == Some(room_id))

                .map(|(_, c)| {

                    c.set_is_ready(c.is_master());

                    c.set_is_joined_mid_game(false);