rust/hedgewars-server/src/server/actions.rs
author alfadur <mail@none>
Sat, 02 Feb 2019 15:06:39 +0300
changeset 14692 455865ccd36c
parent 14525 6cc0fce249f9
child 14694 08a8605bafaf
permissions -rw-r--r--
Server action refactoring part 2 of N

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
    }

    pub fn action(self) -> Action {
        Send(self)
    }
}

impl Into<Action> for PendingMessage {
    fn into(self) -> Action {
        self.action()
    }
}

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 {
    Send(PendingMessage),
    RemoveClient,
    ByeClient(String),
    CheckRegistered,
    JoinLobby,
    RemoveRoom(RoomId),
    MoveToRoom(RoomId),
    MoveToLobby(String),
    ChangeMaster(RoomId, Option<ClientId>),
    RemoveTeam(String),
    RemoveClientTeams,
    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),
    Warn(String),
    ProtocolError(String),
}

use self::Action::*;

pub fn run_action(server: &mut HWServer, client_id: usize, action: Action) {
    match action {
        Send(msg) => server.send(client_id, &msg.destination, msg.message),
        ByeClient(msg) => {
            let c = &server.clients[client_id];
            let nick = c.nick.clone();

            if let Some(id) = c.room_id {
                if id != server.lobby_id {
                    server.react(
                        client_id,
                        vec![MoveToLobby(format!("quit: {}", msg.clone()))],
                    );
                }
            }

            server.react(
                client_id,
                vec![
                    LobbyLeft(nick, msg.clone()).send_all().action(),
                    Bye(msg).send_self().action(),
                    RemoveClient,
                ],
            );
        }
        RemoveClient => {
            server.removed_clients.push(client_id);
            if server.clients.contains(client_id) {
                server.clients.remove(client_id);
            }
        }
        CheckRegistered => {
            let client = &server.clients[client_id];
            if client.protocol_number > 0 && client.nick != "" {
                let has_nick_clash = server
                    .clients
                    .iter()
                    .any(|(id, c)| id != client_id && c.nick == client.nick);

                let actions = if !client.is_checker() && has_nick_clash {
                    if client.protocol_number < 38 {
                        vec![ByeClient("Nickname is already in use".to_string())]
                    } else {
                        server.clients[client_id].nick.clear();
                        vec![Notice("NickAlreadyInUse".to_string()).send_self().action()]
                    }
                } else {
                    vec![JoinLobby]
                };
                server.react(client_id, actions);
            }
        }
        JoinLobby => {
            server.clients[client_id].room_id = Some(server.lobby_id);

            let mut lobby_nicks = Vec::new();
            for (_, c) in server.clients.iter() {
                if c.room_id.is_some() {
                    lobby_nicks.push(c.nick.clone());
                }
            }
            let joined_msg = LobbyJoined(lobby_nicks);

            let everyone_msg = LobbyJoined(vec![server.clients[client_id].nick.clone()]);
            let flags_msg = ClientFlags(
                "+i".to_string(),
                server
                    .clients
                    .iter()
                    .filter(|(_, c)| c.room_id.is_some())
                    .map(|(_, c)| c.nick.clone())
                    .collect(),
            );
            let server_msg = ServerMessage("\u{1f994} is watching".to_string());
            let rooms_msg = Rooms(
                server
                    .rooms
                    .iter()
                    .filter(|(id, _)| *id != server.lobby_id)
                    .flat_map(|(_, r)| r.info(r.master_id.map(|id| &server.clients[id])))
                    .collect(),
            );
            server.react(
                client_id,
                vec![
                    everyone_msg.send_all().but_self().action(),
                    joined_msg.send_self().action(),
                    flags_msg.send_self().action(),
                    server_msg.send_self().action(),
                    rooms_msg.send_self().action(),
                ],
            );
        }
        RemoveRoom(room_id) => {
            let r = &mut server.rooms[room_id];
            let actions = vec![RoomRemove(r.name.clone())
                .send_all()
                .with_protocol(r.protocol_number)
                .action()];
            server.rooms.remove(room_id);
            server.react(client_id, actions);
        }
        MoveToRoom(room_id) => {
            let r = &mut server.rooms[room_id];
            let c = &mut server.clients[client_id];
            r.players_number += 1;
            c.room_id = Some(room_id);

            let is_master = r.master_id == Some(c.id);
            c.set_is_master(is_master);
            c.set_is_ready(is_master);
            c.set_is_joined_mid_game(false);

            if is_master {
                r.ready_players_number += 1;
            }

            let mut v = vec![
                RoomJoined(vec![c.nick.clone()])
                    .send_all()
                    .in_room(room_id)
                    .action(),
                ClientFlags("+i".to_string(), vec![c.nick.clone()])
                    .send_all()
                    .action(),
                SendRoomUpdate(None),
            ];

            if !r.greeting.is_empty() {
                v.push(
                    ChatMsg {
                        nick: "[greeting]".to_string(),
                        msg: r.greeting.clone(),
                    }
                    .send_self()
                    .action(),
                );
            }

            if !c.is_master() {
                let team_names: Vec<_>;
                if let Some(ref mut info) = r.game_info {
                    c.set_is_in_game(true);
                    c.set_is_joined_mid_game(true);

                    {
                        let teams = info.client_teams(c.id);
                        c.teams_in_game = teams.clone().count() as u8;
                        c.clan = teams.clone().next().map(|t| t.color);
                        team_names = teams.map(|t| t.name.clone()).collect();
                    }

                    if !team_names.is_empty() {
                        info.left_teams.retain(|name| !team_names.contains(&name));
                        info.teams_in_game += team_names.len() as u8;
                        r.teams = info
                            .teams_at_start
                            .iter()
                            .filter(|(_, t)| !team_names.contains(&t.name))
                            .cloned()
                            .collect();
                    }
                } else {
                    team_names = Vec::new();
                }

                v.push(SendRoomData {
                    to: client_id,
                    teams: true,
                    config: true,
                    flags: true,
                });

                if let Some(ref info) = r.game_info {
                    v.push(RunGame.send_self().action());
                    v.push(
                        ClientFlags("+g".to_string(), vec![c.nick.clone()])
                            .send_all()
                            .in_room(r.id)
                            .action(),
                    );
                    v.push(
                        ForwardEngineMessage(vec![to_engine_msg("e$spectate 1".bytes())])
                            .send_self()
                            .action(),
                    );
                    v.push(
                        ForwardEngineMessage(info.msg_log.clone())
                            .send_self()
                            .action(),
                    );

                    for name in &team_names {
                        v.push(
                            ForwardEngineMessage(vec![to_engine_msg(
                                once(b'G').chain(name.bytes()),
                            )])
                            .send_all()
                            .in_room(r.id)
                            .action(),
                        );
                    }
                    if info.is_paused {
                        v.push(
                            ForwardEngineMessage(vec![to_engine_msg(once(b'I'))])
                                .send_all()
                                .in_room(r.id)
                                .action(),
                        )
                    }
                }
            }
            server.react(client_id, v);
        }
        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 {
                    actions.push(
                        ConfigEntry("FULLMAPCONFIG".to_string(), r.map_config())
                            .send(to)
                            .action(),
                    );
                    for cfg in r.game_config() {
                        actions.push(cfg.to_server_msg().send(to).action());
                    }
                }
                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() {
                        actions.push(
                            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 {
                        actions.push(
                            ClientFlags("+h".to_string(), vec![server.clients[id].nick.clone()])
                                .send(to)
                                .action(),
                        );
                    }
                    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() {
                        actions.push(ClientFlags("+r".to_string(), nicks).send(to).action());
                    }
                }
            }
            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) {
                        actions.push(
                            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 {
                        actions.push(
                            server_chat("You already have voted.".to_string())
                                .send_self()
                                .action(),
                        );
                    }
                } else {
                    actions.push(
                        server_chat("There's no voting going on.".to_string())
                            .send_self()
                            .action(),
                    );
                }

                if let Some(res) = result {
                    actions.push(
                        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(Kicked.send_self().action());
                            actions.push(MoveToLobby("kicked".to_string()));
                        }
                    }
                }
                VoteType::Map(None) => (),
                VoteType::Map(Some(name)) => {
                    if let Some(location) = server.rooms[room_id].load_config(&name) {
                        actions.push(
                            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;
                        actions.push(
                            server_chat("Pause toggled.".to_string())
                                .send_all()
                                .in_room(room_id)
                                .action(),
                        );
                        actions.push(
                            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);
                    actions.push(cfg.to_server_msg().send_all().in_room(room_id).action());
                    server.rooms[room_id].set_config(cfg);
                }
                VoteType::HedgehogsPerTeam(number) => {
                    let r = &mut server.rooms[room_id];
                    let nicks = r.set_hedgehogs_number(number);
                    actions.extend(nicks.into_iter().map(|n| {
                        HedgehogsNumber(n, number)
                            .send_all()
                            .in_room(room_id)
                            .action()
                    }));
                }
            }
            server.react(id, actions);
        }
        MoveToLobby(msg) => {
            let mut actions = Vec::new();
            let lobby_id = server.lobby_id;
            if let (c, Some(r)) = server.client_and_room(client_id) {
                r.players_number -= 1;
                if c.is_ready() && r.ready_players_number > 0 {
                    r.ready_players_number -= 1;
                }
                if c.is_master() && (r.players_number > 0 || r.is_fixed()) {
                    actions.push(ChangeMaster(r.id, None));
                }
                actions.push(
                    ClientFlags("-i".to_string(), vec![c.nick.clone()])
                        .send_all()
                        .action(),
                );
            }
            server.react(client_id, actions);
            actions = Vec::new();

            if let (c, Some(r)) = server.client_and_room(client_id) {
                c.room_id = Some(lobby_id);
                if r.players_number == 0 && !r.is_fixed() {
                    actions.push(RemoveRoom(r.id));
                } else {
                    actions.push(RemoveClientTeams);
                    actions.push(
                        RoomLeft(c.nick.clone(), msg)
                            .send_all()
                            .in_room(r.id)
                            .but_self()
                            .action(),
                    );
                    actions.push(SendRoomUpdate(Some(r.name.clone())));
                }
            }
            server.react(client_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;
                        actions.push(
                            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 {
                    actions.push(
                        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);
        }
        RemoveTeam(name) => {
            let mut actions = Vec::new();
            if let (c, Some(r)) = server.client_and_room(client_id) {
                r.remove_team(&name);
                if let Some(ref mut info) = r.game_info {
                    info.left_teams.push(name.clone());
                }
                actions.push(TeamRemove(name.clone()).send_all().in_room(r.id).action());
                actions.push(SendRoomUpdate(None));
                if r.game_info.is_some() && c.is_in_game() {
                    actions.push(SendTeamRemovalMessage(name));
                }
            }
            server.react(client_id, actions);
        }
        RemoveClientTeams => {
            if let (c, Some(r)) = server.client_and_room(client_id) {
                let actions = r
                    .client_teams(c.id)
                    .map(|t| RemoveTeam(t.name.clone()))
                    .collect();
                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());
                let actions = vec![RoomUpdated(name, r.info(Some(&c)))
                    .send_all()
                    .with_protocol(r.protocol_number)
                    .action()];
                server.react(client_id, actions);
            }
        }
        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() {
                    vec![Warn(
                        "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
                {
                    vec![Warn("Not all players are ready".to_string())]
                } else if room.game_info.is_some() {
                    vec![Warn("The game is already in progress".to_string())]
                } 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![
                        RunGame.send_all().in_room(room.id).action(),
                        SendRoomUpdate(None),
                        ClientFlags("+g".to_string(), room_nicks)
                            .send_all()
                            .in_room(room.id)
                            .action(),
                    ]
                }
            };
            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());
                    actions.push(
                        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());
                    actions.push(
                        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));
            actions.push(RoundFinished.send_all().in_room(r.id).action());

            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 {
                            actions.push(TeamRemove(name.clone()).send(c.id).action());
                        }
                    }
                }
            }

            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);
                    c
                })
                .filter_map(|c| {
                    if !c.is_master() {
                        Some(c.nick.clone())
                    } else {
                        None
                    }
                })
                .collect();

            if !nicks.is_empty() {
                let msg = if r.protocol_number < 38 {
                    LegacyReady(false, nicks)
                } else {
                    ClientFlags("-r".to_string(), nicks)
                };
                actions.push(msg.send_all().in_room(room_id).action());
            }
            server.react(client_id, actions);
        }
        Warn(msg) => {
            run_action(server, client_id, Warning(msg).send_self().action());
        }
        ProtocolError(msg) => run_action(server, client_id, Error(msg).send_self().action()),
    }
}