use server::coretypes::{ServerVar, GameCfg, TeamInfo, HedgehogInfo};

use std;

use std::ops;

use std::convert::From;

#[derive(PartialEq, Eq, Clone, Debug)]

pub enum HWProtocolMessage {

    // core

    Ping,

    Pong,

    Quit(Option<String>),

    //Cmd(String, Vec<String>),

    Global(String),

    Watch(String),

    ToggleServerRegisteredOnly,

    SuperPower,

    Info(String),

    // not entered state

    Nick(String),

    Proto(u32),

    Password(String, String),

    Checker(u32, String, String),

    // lobby

    List,

    Chat(String),

    CreateRoom(String, Option<String>),

    JoinRoom(String, Option<String>),

    Follow(String),

    Rnd(Vec<String>),

    Kick(String),

    Ban(String, String, u32),

    BanIP(String, String, u32),

    BanNick(String, String, u32),

    BanList,

    Unban(String),

    SetServerVar(ServerVar),

    GetServerVar,

    RestartServer,

    Stats,

    // in room

    Part(Option<String>),

    Cfg(GameCfg),

    AddTeam(TeamInfo),

    RemoveTeam(String),

    SetHedgehogsNumber(String, u8),

    SetTeamColor(String, u8),

    ToggleReady,

    StartGame,

    EngineMessage(String),

    RoundFinished,

    ToggleRestrictJoin,

    ToggleRestrictTeams,

    ToggleRegisteredOnly,

    RoomName(String),

    Delegate(String),

    TeamChat(String),

    MaxTeams(u8),

    Fix,

    Unfix,

    Greeting(String),

    CallVote(Option<(String, Option<String>)>),

    Vote(String),

    ForceVote(String),

    Save(String, String),

    Delete(String),

    SaveRoom(String),

    LoadRoom(String),

    Malformed,

    Empty,

}

#[derive(Debug)]

pub enum HWServerMessage {

    Ping,

    Pong,

    Bye(String),

    Nick(String),

    Proto(u32),

    LobbyLeft(String, String),

    LobbyJoined(Vec<String>),

    ChatMsg(String, String),

    ClientFlags(String, Vec<String>),

    Rooms(Vec<String>),

    RoomAdd(Vec<String>),

    RoomJoined(Vec<String>),

    RoomLeft(String, String),

    RoomRemove(String),

    RoomUpdated(String, Vec<String>),

    TeamAdd(Vec<String>),

    TeamRemove(String),

    TeamAccepted(String),

    TeamColor(String, u8),

    HedgehogsNumber(String, u8),

    ConfigEntry(String, Vec<String>),

    RunGame,

    ForwardEngineMessage(Vec<String>),

    RoundFinished,

    ServerMessage(String),

    Warning(String),

    Error(String),

    Connected(u32),

    Unreachable,

}

impl GameCfg {

    pub fn into_server_msg(self) -> HWServerMessage {

        use self::HWServerMessage::ConfigEntry;

        use server::coretypes::GameCfg::*;

        match self {

            FeatureSize(s) => ConfigEntry("FEATURE_SIZE".to_string(), vec![s.to_string()]),

            MapType(t) => ConfigEntry("MAP".to_string(), vec![t.to_string()]),

            MapGenerator(g) => ConfigEntry("MAPGEN".to_string(), vec![g.to_string()]),

            MazeSize(s) => ConfigEntry("MAZE_SIZE".to_string(), vec![s.to_string()]),

            Seed(s) => ConfigEntry("SEED".to_string(), vec![s.to_string()]),

            Template(t) => ConfigEntry("TEMPLATE".to_string(), vec![t.to_string()]),

            Ammo(n, None) => ConfigEntry("AMMO".to_string(), vec![n.to_string()]),

            Ammo(n, Some(s)) => ConfigEntry("AMMO".to_string(), vec![n.to_string(), s.to_string()]),

            Scheme(n, None) => ConfigEntry("SCHEME".to_string(), vec![n.to_string()]),

            Scheme(n, Some(s)) => ConfigEntry("SCHEME".to_string(), {

                let mut v = vec![n.to_string()];

                v.extend(s.into_iter());

                v

            }),

            Script(s) => ConfigEntry("SCRIPT".to_string(), vec![s.to_string()]),

            Theme(t) => ConfigEntry("THEME".to_string(), vec![t.to_string()]),

            DrawnMap(m) => ConfigEntry("DRAWNMAP".to_string(), vec![m.to_string()])

        }

    }

}

macro_rules! const_braces {

    ($e: expr) => { "{}\n" }

}

macro_rules! msg {

    [$($part: expr),*] => {

        format!(concat!($(const_braces!($part)),*, "\n"), $($part),*);

    };

}

fn construct_message(header: &[&str], msg: &Vec<String>) -> String {

    let mut v: Vec<_> = header.iter().map(|s| *s).collect();

    v.extend(msg.iter().map(|s| &s[..]));

    v.push("\n");

    v.join("\n")

}

impl HWServerMessage {

    pub fn to_raw_protocol(&self) -> String {

        use self::HWServerMessage::*;

        match self {

            Ping => msg!["PING"],

            Pong => msg!["PONG"],

            Connected(protocol_version) => msg![

                "CONNECTED",

                "Hedgewars server http://www.hedgewars.org/",

                protocol_version],

            Bye(msg) => msg!["BYE", msg],

            Nick(nick) => msg!["NICK", nick],

            Proto(proto) => msg!["PROTO", proto],

            LobbyLeft(nick, msg) => msg!["LOBBY:LEFT", nick, msg],

            LobbyJoined(nicks) =>

                construct_message(&["LOBBY:JOINED"], &nicks),

            ClientFlags(flags, nicks) =>

                construct_message(&["CLIENT_FLAGS", flags], &nicks),

            Rooms(info) =>

                construct_message(&["ROOMS"], &info),

            RoomAdd(info) =>

                construct_message(&["ROOM", "ADD"], &info),

            RoomJoined(nicks) =>

                construct_message(&["JOINED"], &nicks),

            RoomLeft(nick, msg) => msg!["LEFT", nick, msg],

            RoomRemove(name) => msg!["ROOM", "DEL", name],

            RoomUpdated(name, info) =>

                construct_message(&["ROOM", "UPD", name], &info),

            TeamAdd(info) =>

                construct_message(&["ADD_TEAM"], &info),

            TeamRemove(name) => msg!["REMOVE_TEAM", name],

            TeamAccepted(name) => msg!["TEAM_ACCEPTED", name],

            TeamColor(name, color) => msg!["TEAM_COLOR", name, color],

            HedgehogsNumber(name, number) => msg!["HH_NUM", name, number],

            ConfigEntry(name, values) =>

                construct_message(&["CFG", name], &values),

            RunGame => msg!["RUN_GAME"],

            ForwardEngineMessage(em) =>

                construct_message(&["EM"], &em),

            RoundFinished => msg!["ROUND_FINISHED"],

            ChatMsg(nick, msg) => msg!["CHAT", nick, msg],

            ServerMessage(msg) => msg!["SERVER_MESSAGE", msg],

            Warning(msg) => msg!["WARNING", msg],

            Error(msg) => msg!["ERROR", msg],

            _ => msg!["ERROR", "UNIMPLEMENTED"],

        }

    }

}