diff -r 7732013ce64c -r c5a6e8566425 rust/hedgewars-server/src/handlers/actions.rs --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/rust/hedgewars-server/src/handlers/actions.rs Tue May 28 19:04:18 2019 +0300 @@ -0,0 +1,115 @@ +use crate::{ + core::{ + client::HWClient, + server::HWServer, + types::{ClientId, GameCfg, RoomId, VoteType}, + room::HWRoom, + room::{GameInfo, RoomFlags} + }, + 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}; + +pub enum DestinationGroup { + All, + Lobby, + Room(RoomId), + Protocol(u16), +} + +pub enum Destination { + ToId(ClientId), + ToIds(Vec), + ToSelf, + ToAll { + group: DestinationGroup, + 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_many(message: HWServerMessage, client_ids: Vec) -> PendingMessage { + PendingMessage { + destination: Destination::ToIds(client_ids), + message, + } + } + + pub fn send_self(message: HWServerMessage) -> PendingMessage { + PendingMessage { + destination: Destination::ToSelf, + message, + } + } + + pub fn send_all(message: HWServerMessage) -> PendingMessage { + let destination = Destination::ToAll { + group: DestinationGroup::All, + skip_self: false, + }; + PendingMessage { + destination, + message, + } + } + + pub fn in_room(mut self, clients_room_id: RoomId) -> PendingMessage { + if let Destination::ToAll { ref mut group, .. } = self.destination { + *group = DestinationGroup::Room(clients_room_id) + } + self + } + + pub fn in_lobby(mut self) -> PendingMessage { + if let Destination::ToAll { ref mut group, .. } = self.destination { + *group = DestinationGroup::Lobby + } + self + } + + pub fn with_protocol(mut self, protocol_number: u16) -> PendingMessage { + if let Destination::ToAll { ref mut group, .. } = self.destination { + *group = DestinationGroup::Protocol(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_many(self, client_ids: Vec) -> PendingMessage { + PendingMessage::send_many(self, client_ids) + } + pub fn send_self(self) -> PendingMessage { + PendingMessage::send_self(self) + } + pub fn send_all(self) -> PendingMessage { + PendingMessage::send_all(self) + } +}