rust/hedgewars-server/src/handlers/actions.rs
changeset 15074 c5a6e8566425
parent 14788 6dea1ca64992
child 15075 e935b1ad23f3
--- /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<ClientId>),
+    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<ClientId>) -> 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<ClientId>) -> 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)
+    }
+}