rust/hedgewars-server/src/server/core.rs
author alfadur <mail@none>
Sat, 02 Feb 2019 15:06:39 +0300
changeset 14671 455865ccd36c
parent 14504 6cc0fce249f9
child 14672 6e6632068a33
permissions -rw-r--r--
Server action refactoring part 2 of N
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
     1
use super::{
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     2
    actions,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     3
    actions::{Destination, PendingMessage},
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     4
    client::HWClient,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     5
    coretypes::{ClientId, RoomId},
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     6
    handlers,
14392
e335b3120f59 pull file io out of server handler
alfadur
parents: 14375
diff changeset
     7
    io::HWServerIO,
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     8
    room::HWRoom,
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
     9
};
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13524
diff changeset
    10
use crate::protocol::messages::*;
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    11
use crate::utils;
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    12
use base64::encode;
13805
0463a4221327 cleanup crate imports
alfadur
parents: 13798
diff changeset
    13
use log::*;
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    14
use rand::{thread_rng, RngCore};
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    15
use slab;
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    16
use std::borrow::BorrowMut;
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    17
12852
bd35cb2302b3 Quick dirty fix for building
unc0rr
parents: 12148
diff changeset
    18
type Slab<T> = slab::Slab<T>;
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    19
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    20
pub struct HWServer {
12143
7e874846afe3 Toss code around
unc0rr
parents: 12142
diff changeset
    21
    pub clients: Slab<HWClient>,
7e874846afe3 Toss code around
unc0rr
parents: 12142
diff changeset
    22
    pub rooms: Slab<HWRoom>,
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    23
    pub lobby_id: RoomId,
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    24
    pub output: Vec<(Vec<ClientId>, HWServerMessage)>,
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    25
    pub removed_clients: Vec<ClientId>,
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    26
    pub io: Box<dyn HWServerIO>,
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    27
}
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    28
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    29
impl HWServer {
14392
e335b3120f59 pull file io out of server handler
alfadur
parents: 14375
diff changeset
    30
    pub fn new(clients_limit: usize, rooms_limit: usize, io: Box<dyn HWServerIO>) -> HWServer {
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    31
        let rooms = Slab::with_capacity(rooms_limit);
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    32
        let clients = Slab::with_capacity(clients_limit);
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    33
        let mut server = HWServer {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    34
            clients,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    35
            rooms,
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    36
            lobby_id: 0,
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    37
            output: vec![],
14392
e335b3120f59 pull file io out of server handler
alfadur
parents: 14375
diff changeset
    38
            removed_clients: vec![],
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    39
            io,
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    40
        };
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    41
        server.lobby_id = server.add_room().id;
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    42
        server
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    43
    }
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    44
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    45
    pub fn add_client(&mut self) -> ClientId {
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    46
        let key: ClientId;
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    47
        {
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    48
            let entry = self.clients.vacant_entry();
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    49
            key = entry.key();
13798
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
    50
            let mut salt = [0u8; 18];
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
    51
            thread_rng().fill_bytes(&mut salt);
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
    52
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
    53
            let client = HWClient::new(entry.key(), encode(&salt));
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    54
            entry.insert(client);
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    55
        }
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    56
        self.send(
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    57
            key,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    58
            &Destination::ToSelf,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    59
            HWServerMessage::Connected(utils::PROTOCOL_VERSION),
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    60
        );
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    61
        key
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    62
    }
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    63
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    64
    pub fn client_lost(&mut self, client_id: ClientId) {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    65
        actions::run_action(
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    66
            self,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    67
            client_id,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    68
            actions::Action::ByeClient("Connection reset".to_string()),
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    69
        );
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    70
    }
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    71
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    72
    pub fn add_room(&mut self) -> &mut HWRoom {
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    73
        allocate_room(&mut self.rooms)
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    74
    }
12137
193dfdcb0620 - Use logging facilities instead of plain println!
unc0rr
parents: 12136
diff changeset
    75
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    76
    #[inline]
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    77
    pub fn create_room(
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    78
        &mut self,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    79
        creator_id: ClientId,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    80
        name: String,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    81
        password: Option<String>,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    82
    ) -> RoomId {
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    83
        create_room(
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    84
            &mut self.clients[creator_id],
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    85
            &mut self.rooms,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    86
            name,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    87
            password,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    88
        )
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    89
    }
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    90
14671
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    91
    #[inline]
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    92
    pub fn move_to_room(&mut self, client_id: ClientId, room_id: RoomId) {
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    93
        move_to_room(&mut self.clients[client_id], &mut self.rooms[room_id])
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    94
    }
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    95
13524
5359ff75da3a indulge clippy
alfadur
parents: 13521
diff changeset
    96
    fn get_recipients(&self, client_id: ClientId, destination: &Destination) -> Vec<ClientId> {
5359ff75da3a indulge clippy
alfadur
parents: 13521
diff changeset
    97
        let mut ids = match *destination {
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    98
            Destination::ToSelf => vec![client_id],
13426
f091f69d59e4 Additional round cleanup
alfadur
parents: 13422
diff changeset
    99
            Destination::ToId(id) => vec![id],
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   100
            Destination::ToAll {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   101
                room_id: Some(id), ..
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   102
            } => self.room_clients(id),
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   103
            Destination::ToAll {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   104
                protocol: Some(proto),
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   105
                ..
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   106
            } => self.protocol_clients(proto),
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   107
            Destination::ToAll { .. } => self.clients.iter().map(|(id, _)| id).collect::<Vec<_>>(),
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   108
        };
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   109
        if let Destination::ToAll {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   110
            skip_self: true, ..
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   111
        } = destination
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   112
        {
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   113
            if let Some(index) = ids.iter().position(|id| *id == client_id) {
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   114
                ids.remove(index);
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   115
            }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   116
        }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   117
        ids
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   118
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   119
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   120
    pub fn send(
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   121
        &mut self,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   122
        client_id: ClientId,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   123
        destination: &Destination,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   124
        message: HWServerMessage,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   125
    ) {
13524
5359ff75da3a indulge clippy
alfadur
parents: 13521
diff changeset
   126
        let ids = self.get_recipients(client_id, &destination);
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   127
        self.output.push((ids, message));
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   128
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   129
14671
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
   130
    pub fn send_msg(&mut self, client_id: ClientId, message: PendingMessage) {
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
   131
        self.send(client_id, &message.destination, message.message)
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
   132
    }
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
   133
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
   134
    pub fn react(&mut self, client_id: ClientId, actions: Vec<actions::Action>) {
12139
f3121d7dedec - Handle errors
unc0rr
parents: 12138
diff changeset
   135
        for action in actions {
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
   136
            actions::run_action(self, client_id, action);
12138
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12137
diff changeset
   137
        }
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12137
diff changeset
   138
    }
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   139
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   140
    pub fn lobby(&self) -> &HWRoom {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   141
        &self.rooms[self.lobby_id]
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   142
    }
13521
ba5211dddb21 Assorted chat fixes
alfadur
parents: 13520
diff changeset
   143
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   144
    pub fn has_room(&self, name: &str) -> bool {
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   145
        self.rooms.iter().any(|(_, r)| r.name == name)
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   146
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   147
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   148
    pub fn find_room(&self, name: &str) -> Option<&HWRoom> {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   149
        self.rooms
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   150
            .iter()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   151
            .find_map(|(_, r)| Some(r).filter(|r| r.name == name))
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   152
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   153
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   154
    pub fn find_room_mut(&mut self, name: &str) -> Option<&mut HWRoom> {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   155
        self.rooms
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   156
            .iter_mut()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   157
            .find_map(|(_, r)| Some(r).filter(|r| r.name == name))
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   158
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   159
13478
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   160
    pub fn find_client(&self, nick: &str) -> Option<&HWClient> {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   161
        self.clients
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   162
            .iter()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   163
            .find_map(|(_, c)| Some(c).filter(|c| c.nick == nick))
13478
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   164
    }
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   165
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   166
    pub fn find_client_mut(&mut self, nick: &str) -> Option<&mut HWClient> {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   167
        self.clients
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   168
            .iter_mut()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   169
            .find_map(|(_, c)| Some(c).filter(|c| c.nick == nick))
13478
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   170
    }
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   171
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   172
    pub fn select_clients<F>(&self, f: F) -> Vec<ClientId>
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   173
    where
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   174
        F: Fn(&(usize, &HWClient)) -> bool,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   175
    {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   176
        self.clients.iter().filter(f).map(|(_, c)| c.id).collect()
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   177
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   178
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   179
    pub fn room_clients(&self, room_id: RoomId) -> Vec<ClientId> {
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   180
        self.select_clients(|(_, c)| c.room_id == Some(room_id))
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   181
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   182
13520
1ee192f13456 Better packing for clients
alfadur
parents: 13478
diff changeset
   183
    pub fn protocol_clients(&self, protocol: u16) -> Vec<ClientId> {
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   184
        self.select_clients(|(_, c)| c.protocol_number == protocol)
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   185
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   186
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   187
    pub fn other_clients_in_room(&self, self_id: ClientId) -> Vec<ClientId> {
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   188
        let room_id = self.clients[self_id].room_id;
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   189
        self.select_clients(|(id, c)| *id != self_id && c.room_id == room_id)
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   190
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   191
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   192
    pub fn client_and_room(&mut self, client_id: ClientId) -> (&mut HWClient, Option<&mut HWRoom>) {
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   193
        let c = &mut self.clients[client_id];
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   194
        if let Some(room_id) = c.room_id {
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   195
            (c, Some(&mut self.rooms[room_id]))
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   196
        } else {
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   197
            (c, None)
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   198
        }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   199
    }
13445
d3c86ade3d4d Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents: 13442
diff changeset
   200
d3c86ade3d4d Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents: 13442
diff changeset
   201
    pub fn room(&mut self, client_id: ClientId) -> Option<&mut HWRoom> {
d3c86ade3d4d Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents: 13442
diff changeset
   202
        self.client_and_room(client_id).1
d3c86ade3d4d Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents: 13442
diff changeset
   203
    }
d3c86ade3d4d Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents: 13442
diff changeset
   204
}
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   205
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   206
fn allocate_room(rooms: &mut Slab<HWRoom>) -> &mut HWRoom {
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   207
    let entry = rooms.vacant_entry();
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   208
    let key = entry.key();
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   209
    let room = HWRoom::new(entry.key());
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   210
    entry.insert(room)
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   211
}
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   212
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   213
fn create_room(
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   214
    client: &mut HWClient,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   215
    rooms: &mut Slab<HWRoom>,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   216
    name: String,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   217
    password: Option<String>,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   218
) -> RoomId {
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   219
    let room = allocate_room(rooms);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   220
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   221
    room.master_id = Some(client.id);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   222
    room.name = name;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   223
    room.password = password;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   224
    room.protocol_number = client.protocol_number;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   225
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   226
    room.players_number = 1;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   227
    room.ready_players_number = 1;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   228
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   229
    client.room_id = Some(room.id);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   230
    client.set_is_master(true);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   231
    client.set_is_ready(true);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   232
    client.set_is_joined_mid_game(false);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   233
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   234
    room.id
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   235
}
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   236
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   237
fn move_to_room(client: &mut HWClient, room: &mut HWRoom) {
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   238
    debug_assert!(client.room_id != Some(room.id));
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   239
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   240
    room.players_number += 1;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   241
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   242
    client.room_id = Some(room.id);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   243
    client.set_is_joined_mid_game(room.game_info.is_some());
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   244
    client.set_is_in_game(room.game_info.is_some());
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   245
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   246
    if let Some(ref mut info) = room.game_info {
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   247
        let teams = info.client_teams(client.id);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   248
        client.teams_in_game = teams.clone().count() as u8;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   249
        client.clan = teams.clone().next().map(|t| t.color);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   250
        let team_names: Vec<_> = teams.map(|t| t.name.clone()).collect();
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   251
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   252
        if !team_names.is_empty() {
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   253
            info.left_teams.retain(|name| !team_names.contains(&name));
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   254
            info.teams_in_game += team_names.len() as u8;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   255
            room.teams = info
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   256
                .teams_at_start
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   257
                .iter()
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   258
                .filter(|(_, t)| !team_names.contains(&t.name))
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   259
                .cloned()
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   260
                .collect();
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   261
        }
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   262
    }
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   263
}