use mio;

use super::common::rnd_reply;

use crate::{

    protocol::messages::{

        add_flags, remove_flags, HWProtocolMessage, HWServerMessage::*, ProtocolFlags as Flags,

    },

    server::{

        core::HWServer,

        coretypes::{ClientId, ServerVar},

    },

    utils::is_name_illegal,

};

use log::*;

pub fn handle(

    server: &mut HWServer,

    client_id: ClientId,

    response: &mut super::Response,

    message: HWProtocolMessage,

) {

    use crate::protocol::messages::HWProtocolMessage::*;

    match message {

        CreateRoom(name, password) => {

            if is_name_illegal(&name) {

                response.add(Warning("Illegal room name! A room name must be between 1-40 characters long, must not have a trailing or leading space and must not have any of these characters: $()*+?[]^{|}".to_string()).send_self());

            } else if server.has_room(&name) {

                response.add(

                    Warning("A room with the same name already exists.".to_string()).send_self(),

                );

            } else {

                let flags_msg = ClientFlags(

                    add_flags(&[Flags::RoomMaster, Flags::Ready]),

                    vec![server.clients[client_id].nick.clone()],

                );

                let room_id = server.create_room(client_id, name, password);

                let room = &server.rooms[room_id];

                let client = &server.clients[client_id];

                response.add(

                    RoomAdd(room.info(Some(&client)))

                        .send_all()

                        .with_protocol(room.protocol_number),

                );

                response.add(flags_msg.send_self());

                response.add(

                    ClientFlags(add_flags(&[Flags::InRoom]), vec![client.nick.clone()]).send_self(),

                );

            };

        }

        Chat(msg) => {

            response.add(

                ChatMsg {

                    nick: server.clients[client_id].nick.clone(),

                    msg,

                }

                .send_all()

                .in_lobby()

                .but_self(),

            );

        }

        JoinRoom(name, _password) => {

            let room = server.rooms.iter().find(|(_, r)| r.name == name);

            let room_id = room.map(|(_, r)| r.id);

            let client = &mut server.clients[client_id];

            if let Some((_, room)) = room {

                if client.protocol_number != room.protocol_number {

                    response.add(

                        Warning("Room version incompatible to your Hedgewars version!".to_string())

                            .send_self(),

                    );

                } else if room.is_join_restricted() {

                    response.add(

                        Warning(

                            "Access denied. This room currently doesn't allow joining.".to_string(),

                        )

                        .send_self(),

                    );

                } else if room.players_number == u8::max_value() {

                    response.add(Warning("This room is already full".to_string()).send_self());

                } else if let Some(room_id) = room_id {

                    let nick = client.nick.clone();

                    server.move_to_room(client_id, room_id);

                    response.add(RoomJoined(vec![nick.clone()]).send_all().in_room(room_id));

                    response.add(ClientFlags(add_flags(&[Flags::InRoom]), vec![nick]).send_all());

                    response.add(RoomJoined(nicks).send_self());

                    let room = &server.rooms[room_id];

                    if !room.greeting.is_empty() {

                        response.add(

                            ChatMsg {

                                nick: "[greeting]".to_string(),

                                msg: room.greeting.clone(),

                            }

                            .send_self(),

                        );

                    }

                }

            } else {

                response.add(Warning("No such room.".to_string()).send_self());

            }

        }

        SetServerVar(var) => {

            if !server.clients[client_id].is_admin() {

                response.add(Warning("Access denied.".to_string()).send_self());

            } else {

                match var {

                    ServerVar::MOTDNew(msg) => server.greetings.for_latest_protocol = msg,

                    ServerVar::MOTDOld(msg) => server.greetings.for_old_protocols = msg,

                    ServerVar::LatestProto(n) => server.latest_protocol = n,

                }

            }

        }

        GetServerVar => {

            if !server.clients[client_id].is_admin() {

                response.add(Warning("Access denied.".to_string()).send_self());

            } else {

                let vars: Vec<_> = [

                    ServerVar::MOTDNew(server.greetings.for_latest_protocol.clone()),

                    ServerVar::MOTDOld(server.greetings.for_old_protocols.clone()),

                    ServerVar::LatestProto(server.latest_protocol),

                ]

                .iter()

                .flat_map(|v| v.to_protocol())

                .collect();

                response.add(ServerVars(vars).send_self());

            }

        }

        Rnd(v) => {

            response.add(rnd_reply(&v).send_self());

        }

        List => warn!("Deprecated LIST message received"),

        _ => warn!("Incorrect command in lobby state"),

    }

}