rust/hedgewars-server/src/core/server.rs
author alfadur
Tue, 28 May 2019 21:28:32 +0300
changeset 15096 e935b1ad23f3
parent 15095 c5a6e8566425
child 15197 f6115638aa92
permissions -rw-r--r--
normalize type names
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
     1
use super::{
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
     2
    client::HwClient,
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
     3
    indexslab::IndexSlab,
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
     4
    room::HwRoom,
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
     5
    types::{ClientId, RoomId},
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
     6
};
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
     7
use crate::{protocol::messages::HwProtocolMessage::Greeting, utils};
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
     8
14807
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
     9
use bitflags::*;
13810
0463a4221327 cleanup crate imports
alfadur
parents: 13771
diff changeset
    10
use log::*;
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
    11
use slab;
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    12
use std::{borrow::BorrowMut, iter, num::NonZeroU16};
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    13
12852
bd35cb2302b3 Quick dirty fix for building
unc0rr
parents: 12148
diff changeset
    14
type Slab<T> = slab::Slab<T>;
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    15
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    16
pub struct HwAnteClient {
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    17
    pub nick: Option<String>,
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    18
    pub protocol_number: Option<NonZeroU16>,
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    19
    pub server_salt: String,
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
    20
    pub is_checker: bool,
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    21
}
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    22
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    23
pub struct HwAnteroom {
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    24
    pub clients: IndexSlab<HwAnteClient>,
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    25
}
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    26
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    27
impl HwAnteroom {
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    28
    pub fn new(clients_limit: usize) -> Self {
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    29
        let clients = IndexSlab::with_capacity(clients_limit);
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    30
        HwAnteroom { clients }
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    31
    }
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    32
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    33
    pub fn add_client(&mut self, client_id: ClientId, salt: String) {
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    34
        let client = HwAnteClient {
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    35
            nick: None,
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    36
            protocol_number: None,
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    37
            server_salt: salt,
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
    38
            is_checker: false,
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    39
        };
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    40
        self.clients.insert(client_id, client);
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    41
    }
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    42
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    43
    pub fn remove_client(&mut self, client_id: ClientId) -> Option<HwAnteClient> {
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    44
        let mut client = self.clients.remove(client_id);
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    45
        client
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    46
    }
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    47
}
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    48
14804
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    49
pub struct ServerGreetings {
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    50
    pub for_latest_protocol: String,
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    51
    pub for_old_protocols: String,
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    52
}
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    53
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    54
impl ServerGreetings {
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    55
    fn new() -> Self {
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    56
        Self {
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    57
            for_latest_protocol: "\u{1f994} is watching".to_string(),
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    58
            for_old_protocols: "\u{1f994} is watching".to_string(),
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    59
        }
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    60
    }
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    61
}
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    62
14807
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
    63
bitflags! {
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
    64
    pub struct ServerFlags: u8 {
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
    65
        const REGISTERED_ONLY = 0b0000_1000;
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
    66
    }
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
    67
}
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
    68
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    69
pub struct HwServer {
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    70
    pub clients: IndexSlab<HwClient>,
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    71
    pub rooms: Slab<HwRoom>,
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    72
    pub anteroom: HwAnteroom,
14804
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    73
    pub latest_protocol: u16,
14807
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
    74
    pub flags: ServerFlags,
14804
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    75
    pub greetings: ServerGreetings,
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    76
}
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    77
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    78
impl HwServer {
14801
65861ba8b4e8 move room saves to IO thread
alfadur
parents: 14800
diff changeset
    79
    pub fn new(clients_limit: usize, rooms_limit: usize) -> Self {
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    80
        let rooms = Slab::with_capacity(rooms_limit);
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    81
        let clients = IndexSlab::with_capacity(clients_limit);
14715
25c564f77b7d remove lobby room
alfadur <mail@none>
parents: 14714
diff changeset
    82
        Self {
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
    83
            clients,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
    84
            rooms,
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    85
            anteroom: HwAnteroom::new(clients_limit),
14804
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    86
            greetings: ServerGreetings::new(),
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
    87
            latest_protocol: 58,
14807
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
    88
            flags: ServerFlags::empty(),
14715
25c564f77b7d remove lobby room
alfadur <mail@none>
parents: 14714
diff changeset
    89
        }
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    90
    }
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    91
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    92
    pub fn add_client(&mut self, client_id: ClientId, data: HwAnteClient) {
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    93
        if let (Some(protocol), Some(nick)) = (data.protocol_number, data.nick) {
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    94
            let mut client = HwClient::new(client_id, protocol.get(), nick);
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
    95
            client.set_is_checker(data.is_checker);
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14707
diff changeset
    96
            self.clients.insert(client_id, client);
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    97
        }
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    98
    }
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    99
14694
08a8605bafaf Server action refactoring part 4 of N
alfadur <mail@none>
parents: 14693
diff changeset
   100
    pub fn remove_client(&mut self, client_id: ClientId) {
14717
8a45c90f4580 fix client removal
alfadur
parents: 14716
diff changeset
   101
        self.clients.remove(client_id);
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
   102
    }
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
   103
14804
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
   104
    pub fn get_greetings(&self, client_id: ClientId) -> &str {
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
   105
        if self.clients[client_id].protocol_number < self.latest_protocol {
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
   106
            &self.greetings.for_old_protocols
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
   107
        } else {
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
   108
            &self.greetings.for_latest_protocol
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
   109
        }
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
   110
    }
b3adc030104b implement server vars
alfadur
parents: 14802
diff changeset
   111
14525
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   112
    #[inline]
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   113
    pub fn create_room(
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   114
        &mut self,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   115
        creator_id: ClientId,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   116
        name: String,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   117
        password: Option<String>,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   118
    ) -> RoomId {
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   119
        create_room(
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   120
            &mut self.clients[creator_id],
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   121
            &mut self.rooms,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   122
            name,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   123
            password,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   124
        )
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   125
    }
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   126
14692
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14525
diff changeset
   127
    #[inline]
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14525
diff changeset
   128
    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: 14525
diff changeset
   129
        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: 14525
diff changeset
   130
    }
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14525
diff changeset
   131
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   132
    pub fn has_room(&self, name: &str) -> bool {
14718
f64e21f164a5 fix some warnings
alfadur
parents: 14717
diff changeset
   133
        self.find_room(name).is_some()
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   134
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   135
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   136
    pub fn find_room(&self, name: &str) -> Option<&HwRoom> {
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   137
        self.rooms
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   138
            .iter()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   139
            .find_map(|(_, r)| Some(r).filter(|r| r.name == name))
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   140
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   141
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   142
    pub fn find_room_mut(&mut self, name: &str) -> Option<&mut HwRoom> {
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   143
        self.rooms
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   144
            .iter_mut()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   145
            .find_map(|(_, r)| Some(r).filter(|r| r.name == name))
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   146
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   147
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   148
    pub fn find_client(&self, nick: &str) -> Option<&HwClient> {
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   149
        self.clients
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   150
            .iter()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   151
            .find_map(|(_, c)| Some(c).filter(|c| c.nick == nick))
13450
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   152
    }
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   153
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   154
    pub fn find_client_mut(&mut self, nick: &str) -> Option<&mut HwClient> {
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   155
        self.clients
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   156
            .iter_mut()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   157
            .find_map(|(_, c)| Some(c).filter(|c| c.nick == nick))
13450
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   158
    }
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   159
14810
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   160
    pub fn all_clients(&self) -> impl Iterator<Item = ClientId> + '_ {
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   161
        self.clients.iter().map(|(id, _)| id)
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   162
    }
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   163
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   164
    pub fn filter_clients<'a, F>(&'a self, f: F) -> impl Iterator<Item = ClientId> + 'a
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   165
    where
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   166
        F: Fn(&(usize, &HwClient)) -> bool + 'a,
14810
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   167
    {
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   168
        self.clients.iter().filter(f).map(|(_, c)| c.id)
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   169
    }
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   170
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   171
    pub fn filter_rooms<'a, F>(&'a self, f: F) -> impl Iterator<Item = RoomId> + 'a
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   172
    where
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   173
        F: Fn(&(usize, &HwRoom)) -> bool + 'a,
14810
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   174
    {
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   175
        self.rooms.iter().filter(f).map(|(_, c)| c.id)
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   176
    }
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   177
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
   178
    pub fn collect_clients<F>(&self, f: F) -> Vec<ClientId>
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   179
    where
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   180
        F: Fn(&(usize, &HwClient)) -> bool,
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
   181
    {
14810
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   182
        self.filter_clients(f).collect()
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   183
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   184
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
   185
    pub fn collect_nicks<F>(&self, f: F) -> Vec<String>
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
   186
    where
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   187
        F: Fn(&(usize, &HwClient)) -> bool,
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
   188
    {
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
   189
        self.clients
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
   190
            .iter()
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
   191
            .filter(f)
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
   192
            .map(|(_, c)| c.nick.clone())
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
   193
            .collect()
14715
25c564f77b7d remove lobby room
alfadur <mail@none>
parents: 14714
diff changeset
   194
    }
25c564f77b7d remove lobby room
alfadur <mail@none>
parents: 14714
diff changeset
   195
14810
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   196
    pub fn lobby_clients(&self) -> impl Iterator<Item = ClientId> + '_ {
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   197
        self.filter_clients(|(_, c)| c.room_id == None)
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
   198
    }
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
   199
14810
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   200
    pub fn room_clients(&self, room_id: RoomId) -> impl Iterator<Item = ClientId> + '_ {
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   201
        self.filter_clients(move |(_, c)| c.room_id == Some(room_id))
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   202
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   203
14810
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   204
    pub fn protocol_clients(&self, protocol: u16) -> impl Iterator<Item = ClientId> + '_ {
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   205
        self.filter_clients(move |(_, c)| c.protocol_number == protocol)
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   206
    }
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   207
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   208
    pub fn protocol_rooms(&self, protocol: u16) -> impl Iterator<Item = RoomId> + '_ {
18240b308505 implement stats message
alfadur
parents: 14807
diff changeset
   209
        self.filter_rooms(move |(_, r)| r.protocol_number == protocol)
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   210
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   211
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   212
    pub fn other_clients_in_room(&self, self_id: ClientId) -> Vec<ClientId> {
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   213
        let room_id = self.clients[self_id].room_id;
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14801
diff changeset
   214
        self.collect_clients(|(id, c)| *id != self_id && c.room_id == room_id)
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   215
    }
14807
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
   216
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
   217
    pub fn is_registered_only(&self) -> bool {
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
   218
        self.flags.contains(ServerFlags::REGISTERED_ONLY)
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
   219
    }
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
   220
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
   221
    pub fn set_is_registered_only(&mut self, value: bool) {
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
   222
        self.flags.set(ServerFlags::REGISTERED_ONLY, value)
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14804
diff changeset
   223
    }
13445
d3c86ade3d4d Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents: 13442
diff changeset
   224
}
14525
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   225
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   226
fn allocate_room(rooms: &mut Slab<HwRoom>) -> &mut HwRoom {
14525
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   227
    let entry = rooms.vacant_entry();
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   228
    let room = HwRoom::new(entry.key());
14525
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   229
    entry.insert(room)
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   230
}
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   231
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   232
fn create_room(
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   233
    client: &mut HwClient,
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   234
    rooms: &mut Slab<HwRoom>,
14525
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   235
    name: String,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   236
    password: Option<String>,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   237
) -> RoomId {
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   238
    let room = allocate_room(rooms);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   239
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   240
    room.master_id = Some(client.id);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   241
    room.name = name;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   242
    room.password = password;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   243
    room.protocol_number = client.protocol_number;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   244
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   245
    room.players_number = 1;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   246
    room.ready_players_number = 1;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   247
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   248
    client.room_id = Some(room.id);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   249
    client.set_is_master(true);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   250
    client.set_is_ready(true);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   251
    client.set_is_joined_mid_game(false);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   252
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   253
    room.id
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   254
}
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   255
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   256
fn move_to_room(client: &mut HwClient, room: &mut HwRoom) {
14525
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   257
    debug_assert!(client.room_id != Some(room.id));
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   258
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   259
    room.players_number += 1;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   260
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   261
    client.room_id = Some(room.id);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   262
    client.set_is_joined_mid_game(room.game_info.is_some());
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   263
    client.set_is_in_game(room.game_info.is_some());
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   264
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   265
    if let Some(ref mut info) = room.game_info {
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   266
        let teams = info.client_teams(client.id);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   267
        client.teams_in_game = teams.clone().count() as u8;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   268
        client.clan = teams.clone().next().map(|t| t.color);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   269
        let team_names: Vec<_> = teams.map(|t| t.name.clone()).collect();
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   270
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   271
        if !team_names.is_empty() {
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   272
            info.left_teams.retain(|name| !team_names.contains(&name));
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   273
            info.teams_in_game += team_names.len() as u8;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   274
            room.teams = info
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   275
                .teams_at_start
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   276
                .iter()
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   277
                .filter(|(_, t)| !team_names.contains(&t.name))
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   278
                .cloned()
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   279
                .collect();
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   280
        }
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   281
    }
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
   282
}