author | alfadur |
Sat, 26 Oct 2019 02:36:08 +0300 | |
changeset 15482 | 4cc9ec732392 |
parent 15441 | 61a0bd0bb021 |
child 15487 | 91f0c5ec37b5 |
permissions | -rw-r--r-- |
13416 | 1 |
use super::{ |
15075 | 2 |
client::HwClient, |
14693 | 3 |
indexslab::IndexSlab, |
15075 | 4 |
room::HwRoom, |
15439 | 5 |
types::{ClientId, RoomId, ServerVar}, |
13416 | 6 |
}; |
15075 | 7 |
use crate::{protocol::messages::HwProtocolMessage::Greeting, utils}; |
14693 | 8 |
|
15439 | 9 |
use crate::core::server::JoinRoomError::WrongProtocol; |
14786
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
10 |
use bitflags::*; |
13805 | 11 |
use log::*; |
14457 | 12 |
use slab; |
15439 | 13 |
use std::{borrow::BorrowMut, collections::HashSet, iter, num::NonZeroU16}; |
12126 | 14 |
|
12852 | 15 |
type Slab<T> = slab::Slab<T>; |
12127 | 16 |
|
15441 | 17 |
#[derive(Debug)] |
15439 | 18 |
pub enum CreateRoomError { |
19 |
InvalidName, |
|
20 |
AlreadyExists, |
|
21 |
} |
|
22 |
||
15441 | 23 |
#[derive(Debug)] |
15439 | 24 |
pub enum JoinRoomError { |
25 |
DoesntExist, |
|
26 |
WrongProtocol, |
|
27 |
Full, |
|
28 |
Restricted, |
|
29 |
} |
|
30 |
||
15482 | 31 |
pub enum LeaveRoomResult { |
32 |
RoomRemoved, |
|
33 |
RoomRemains { |
|
34 |
is_empty: bool, |
|
35 |
was_master: bool, |
|
36 |
was_in_game: bool, |
|
37 |
new_master: Option<ClientId>, |
|
38 |
removed_teams: Vec<String>, |
|
39 |
}, |
|
40 |
} |
|
41 |
||
42 |
#[derive(Debug)] |
|
43 |
pub enum LeaveRoomError { |
|
44 |
NoRoom, |
|
45 |
} |
|
46 |
||
15441 | 47 |
#[derive(Debug)] |
48 |
pub struct UninitializedError(); |
|
49 |
#[derive(Debug)] |
|
15439 | 50 |
pub struct AccessError(); |
51 |
||
15075 | 52 |
pub struct HwAnteClient { |
14693 | 53 |
pub nick: Option<String>, |
54 |
pub protocol_number: Option<NonZeroU16>, |
|
55 |
pub server_salt: String, |
|
14781 | 56 |
pub is_checker: bool, |
15176 | 57 |
pub is_local_admin: bool, |
15441 | 58 |
pub is_registered: bool, |
59 |
pub is_admin: bool, |
|
60 |
pub is_contributor: bool, |
|
14693 | 61 |
} |
62 |
||
15075 | 63 |
pub struct HwAnteroom { |
64 |
pub clients: IndexSlab<HwAnteClient>, |
|
14693 | 65 |
} |
66 |
||
15075 | 67 |
impl HwAnteroom { |
14693 | 68 |
pub fn new(clients_limit: usize) -> Self { |
69 |
let clients = IndexSlab::with_capacity(clients_limit); |
|
15075 | 70 |
HwAnteroom { clients } |
14693 | 71 |
} |
72 |
||
15176 | 73 |
pub fn add_client(&mut self, client_id: ClientId, salt: String, is_local_admin: bool) { |
15075 | 74 |
let client = HwAnteClient { |
14693 | 75 |
nick: None, |
76 |
protocol_number: None, |
|
77 |
server_salt: salt, |
|
14781 | 78 |
is_checker: false, |
15176 | 79 |
is_local_admin, |
15441 | 80 |
is_registered: false, |
81 |
is_admin: false, |
|
82 |
is_contributor: false, |
|
14693 | 83 |
}; |
84 |
self.clients.insert(client_id, client); |
|
85 |
} |
|
86 |
||
15075 | 87 |
pub fn remove_client(&mut self, client_id: ClientId) -> Option<HwAnteClient> { |
15439 | 88 |
let client = self.clients.remove(client_id); |
14693 | 89 |
client |
90 |
} |
|
91 |
} |
|
92 |
||
14783 | 93 |
pub struct ServerGreetings { |
94 |
pub for_latest_protocol: String, |
|
95 |
pub for_old_protocols: String, |
|
96 |
} |
|
97 |
||
98 |
impl ServerGreetings { |
|
99 |
fn new() -> Self { |
|
100 |
Self { |
|
101 |
for_latest_protocol: "\u{1f994} is watching".to_string(), |
|
102 |
for_old_protocols: "\u{1f994} is watching".to_string(), |
|
103 |
} |
|
104 |
} |
|
105 |
} |
|
106 |
||
14786
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
107 |
bitflags! { |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
108 |
pub struct ServerFlags: u8 { |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
109 |
const REGISTERED_ONLY = 0b0000_1000; |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
110 |
} |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
111 |
} |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
112 |
|
15075 | 113 |
pub struct HwServer { |
114 |
pub clients: IndexSlab<HwClient>, |
|
115 |
pub rooms: Slab<HwRoom>, |
|
116 |
pub anteroom: HwAnteroom, |
|
14783 | 117 |
pub latest_protocol: u16, |
14786
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
118 |
pub flags: ServerFlags, |
14783 | 119 |
pub greetings: ServerGreetings, |
12126 | 120 |
} |
121 |
||
15075 | 122 |
impl HwServer { |
14780 | 123 |
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
|
124 |
let rooms = Slab::with_capacity(rooms_limit); |
14693 | 125 |
let clients = IndexSlab::with_capacity(clients_limit); |
14694 | 126 |
Self { |
14457 | 127 |
clients, |
128 |
rooms, |
|
15075 | 129 |
anteroom: HwAnteroom::new(clients_limit), |
14783 | 130 |
greetings: ServerGreetings::new(), |
131 |
latest_protocol: 58, |
|
14786
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
132 |
flags: ServerFlags::empty(), |
14694 | 133 |
} |
12126 | 134 |
} |
135 |
||
15441 | 136 |
#[inline] |
137 |
pub fn client(&self, client_id: ClientId) -> &HwClient { |
|
138 |
&self.clients[client_id] |
|
139 |
} |
|
140 |
||
141 |
#[inline] |
|
142 |
pub fn client_mut(&mut self, client_id: ClientId) -> &mut HwClient { |
|
143 |
&mut self.clients[client_id] |
|
144 |
} |
|
145 |
||
146 |
#[inline] |
|
147 |
pub fn room(&self, room_id: RoomId) -> &HwRoom { |
|
148 |
&self.rooms[room_id] |
|
149 |
} |
|
150 |
||
151 |
#[inline] |
|
152 |
pub fn room_mut(&mut self, room_id: RoomId) -> &mut HwRoom { |
|
153 |
&mut self.rooms[room_id] |
|
154 |
} |
|
155 |
||
156 |
#[inline] |
|
15482 | 157 |
pub fn client_and_room(&self, client_id: ClientId, room_id: RoomId) -> (&HwClient, &HwRoom) { |
158 |
(&self.clients[client_id], &self.rooms[room_id]) |
|
159 |
} |
|
160 |
||
161 |
#[inline] |
|
162 |
pub fn client_and_room_mut( |
|
163 |
&mut self, |
|
164 |
client_id: ClientId, |
|
165 |
room_id: RoomId, |
|
166 |
) -> (&mut HwClient, &mut HwRoom) { |
|
167 |
(&mut self.clients[client_id], &mut self.rooms[room_id]) |
|
168 |
} |
|
169 |
||
170 |
#[inline] |
|
15441 | 171 |
pub fn is_admin(&self, client_id: ClientId) -> bool { |
172 |
self.clients |
|
173 |
.get(client_id) |
|
174 |
.map(|c| c.is_admin()) |
|
175 |
.unwrap_or(false) |
|
176 |
} |
|
177 |
||
15075 | 178 |
pub fn add_client(&mut self, client_id: ClientId, data: HwAnteClient) { |
14693 | 179 |
if let (Some(protocol), Some(nick)) = (data.protocol_number, data.nick) { |
15075 | 180 |
let mut client = HwClient::new(client_id, protocol.get(), nick); |
14781 | 181 |
client.set_is_checker(data.is_checker); |
15176 | 182 |
#[cfg(not(feature = "official-server"))] |
183 |
client.set_is_admin(data.is_local_admin); |
|
184 |
||
15441 | 185 |
#[cfg(feature = "official-server")] |
186 |
{ |
|
187 |
client.set_is_registered(info.is_registered); |
|
188 |
client.set_is_admin(info.is_admin); |
|
189 |
client.set_is_contributor(info.is_contributor); |
|
190 |
} |
|
191 |
||
14693 | 192 |
self.clients.insert(client_id, client); |
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
193 |
} |
12126 | 194 |
} |
12127 | 195 |
|
14673
08a8605bafaf
Server action refactoring part 4 of N
alfadur <mail@none>
parents:
14672
diff
changeset
|
196 |
pub fn remove_client(&mut self, client_id: ClientId) { |
14696 | 197 |
self.clients.remove(client_id); |
12127 | 198 |
} |
199 |
||
15441 | 200 |
pub fn get_greetings(&self, client: &HwClient) -> &str { |
201 |
if client.protocol_number < self.latest_protocol { |
|
14783 | 202 |
&self.greetings.for_old_protocols |
203 |
} else { |
|
204 |
&self.greetings.for_latest_protocol |
|
205 |
} |
|
206 |
} |
|
207 |
||
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
208 |
#[inline] |
15439 | 209 |
pub fn get_client_nick(&self, client_id: ClientId) -> &str { |
210 |
&self.clients[client_id].nick |
|
211 |
} |
|
212 |
||
213 |
#[inline] |
|
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
214 |
pub fn create_room( |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
215 |
&mut self, |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
216 |
creator_id: ClientId, |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
217 |
name: String, |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
218 |
password: Option<String>, |
15439 | 219 |
) -> Result<(&HwClient, &HwRoom), CreateRoomError> { |
220 |
use CreateRoomError::*; |
|
221 |
if utils::is_name_illegal(&name) { |
|
222 |
Err(InvalidName) |
|
223 |
} else if self.has_room(&name) { |
|
224 |
Err(AlreadyExists) |
|
225 |
} else { |
|
226 |
Ok(create_room( |
|
227 |
&mut self.clients[creator_id], |
|
228 |
&mut self.rooms, |
|
229 |
name, |
|
230 |
password, |
|
231 |
)) |
|
232 |
} |
|
233 |
} |
|
234 |
||
235 |
pub fn join_room( |
|
236 |
&mut self, |
|
237 |
client_id: ClientId, |
|
238 |
room_id: RoomId, |
|
239 |
) -> Result<(&HwClient, &HwRoom, impl Iterator<Item = &HwClient> + Clone), JoinRoomError> { |
|
240 |
use JoinRoomError::*; |
|
241 |
let room = &mut self.rooms[room_id]; |
|
242 |
let client = &mut self.clients[client_id]; |
|
243 |
||
244 |
if client.protocol_number != room.protocol_number { |
|
245 |
Err(WrongProtocol) |
|
246 |
} else if room.is_join_restricted() { |
|
247 |
Err(Restricted) |
|
248 |
} else if room.players_number == u8::max_value() { |
|
249 |
Err(Full) |
|
250 |
} else { |
|
251 |
move_to_room(client, room); |
|
252 |
let room_id = room.id; |
|
253 |
Ok(( |
|
254 |
&self.clients[client_id], |
|
255 |
&self.rooms[room_id], |
|
256 |
self.clients.iter().map(|(_, c)| c), |
|
257 |
)) |
|
258 |
} |
|
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
259 |
} |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
260 |
|
14671
455865ccd36c
Server action refactoring part 2 of N
alfadur <mail@none>
parents:
14504
diff
changeset
|
261 |
#[inline] |
15439 | 262 |
pub fn join_room_by_name( |
263 |
&mut self, |
|
264 |
client_id: ClientId, |
|
265 |
room_name: &str, |
|
266 |
) -> Result<(&HwClient, &HwRoom, impl Iterator<Item = &HwClient> + Clone), JoinRoomError> { |
|
267 |
use JoinRoomError::*; |
|
268 |
let room = self.rooms.iter().find(|(_, r)| r.name == room_name); |
|
269 |
if let Some((_, room)) = room { |
|
270 |
let room_id = room.id; |
|
271 |
self.join_room(client_id, room_id) |
|
272 |
} else { |
|
273 |
Err(DoesntExist) |
|
274 |
} |
|
275 |
} |
|
276 |
||
15482 | 277 |
pub fn leave_room(&mut self, client_id: ClientId) -> Result<LeaveRoomResult, LeaveRoomError> { |
278 |
let client = &mut self.clients[client_id]; |
|
279 |
if let Some(room_id) = client.room_id { |
|
280 |
let room = &mut self.rooms[room_id]; |
|
281 |
||
282 |
room.players_number -= 1; |
|
283 |
client.room_id = None; |
|
284 |
||
285 |
let is_empty = room.players_number == 0; |
|
286 |
let is_fixed = room.is_fixed(); |
|
287 |
let was_master = room.master_id == Some(client_id); |
|
288 |
let was_in_game = client.is_in_game(); |
|
289 |
let mut removed_teams = vec![]; |
|
290 |
||
291 |
if is_empty && !is_fixed { |
|
292 |
if client.is_ready() && room.ready_players_number > 0 { |
|
293 |
room.ready_players_number -= 1; |
|
294 |
} |
|
295 |
||
296 |
removed_teams = room |
|
297 |
.client_teams(client.id) |
|
298 |
.map(|t| t.name.clone()) |
|
299 |
.collect(); |
|
300 |
||
301 |
for team_name in &removed_teams { |
|
302 |
room.remove_team(team_name); |
|
303 |
} |
|
304 |
||
305 |
if client.is_master() && !is_fixed { |
|
306 |
client.set_is_master(false); |
|
307 |
room.master_id = None; |
|
308 |
} |
|
309 |
} |
|
310 |
||
311 |
client.set_is_ready(false); |
|
312 |
client.set_is_in_game(false); |
|
313 |
||
314 |
if !is_fixed { |
|
315 |
if room.players_number == 0 { |
|
316 |
self.rooms.remove(room_id); |
|
317 |
} else if room.master_id == None { |
|
318 |
let new_master_id = self.room_clients(room_id).next(); |
|
319 |
if let Some(new_master_id) = new_master_id { |
|
320 |
let room = &mut self.rooms[room_id]; |
|
321 |
room.master_id = Some(new_master_id); |
|
322 |
let new_master = &mut self.clients[new_master_id]; |
|
323 |
new_master.set_is_master(true); |
|
324 |
||
325 |
if room.protocol_number < 42 { |
|
326 |
room.name = new_master.nick.clone(); |
|
327 |
} |
|
328 |
||
329 |
room.set_join_restriction(false); |
|
330 |
room.set_team_add_restriction(false); |
|
331 |
room.set_unregistered_players_restriction(true); |
|
332 |
} |
|
333 |
} |
|
334 |
} |
|
335 |
||
336 |
if is_empty && !is_fixed { |
|
337 |
Ok(LeaveRoomResult::RoomRemoved) |
|
338 |
} else { |
|
339 |
Ok(LeaveRoomResult::RoomRemains { |
|
340 |
is_empty, |
|
341 |
was_master, |
|
342 |
was_in_game, |
|
343 |
new_master: self.rooms[room_id].master_id, |
|
344 |
removed_teams, |
|
345 |
}) |
|
346 |
} |
|
347 |
} else { |
|
348 |
Err(LeaveRoomError::NoRoom) |
|
349 |
} |
|
350 |
} |
|
351 |
||
15439 | 352 |
#[inline] |
353 |
pub fn set_var(&mut self, client_id: ClientId, var: ServerVar) -> Result<(), AccessError> { |
|
354 |
if self.clients[client_id].is_admin() { |
|
355 |
match var { |
|
356 |
ServerVar::MOTDNew(msg) => self.greetings.for_latest_protocol = msg, |
|
357 |
ServerVar::MOTDOld(msg) => self.greetings.for_old_protocols = msg, |
|
358 |
ServerVar::LatestProto(n) => self.latest_protocol = n, |
|
359 |
} |
|
360 |
Ok(()) |
|
361 |
} else { |
|
362 |
Err(AccessError()) |
|
363 |
} |
|
14671
455865ccd36c
Server action refactoring part 2 of N
alfadur <mail@none>
parents:
14504
diff
changeset
|
364 |
} |
455865ccd36c
Server action refactoring part 2 of N
alfadur <mail@none>
parents:
14504
diff
changeset
|
365 |
|
15439 | 366 |
#[inline] |
367 |
pub fn get_vars(&self, client_id: ClientId) -> Result<[ServerVar; 3], AccessError> { |
|
368 |
if self.clients[client_id].is_admin() { |
|
369 |
Ok([ |
|
370 |
ServerVar::MOTDNew(self.greetings.for_latest_protocol.clone()), |
|
371 |
ServerVar::MOTDOld(self.greetings.for_old_protocols.clone()), |
|
372 |
ServerVar::LatestProto(self.latest_protocol), |
|
373 |
]) |
|
374 |
} else { |
|
375 |
Err(AccessError()) |
|
376 |
} |
|
377 |
} |
|
378 |
||
379 |
pub fn get_used_protocols(&self, client_id: ClientId) -> Result<Vec<u16>, AccessError> { |
|
380 |
if self.clients[client_id].is_admin() { |
|
381 |
let mut protocols: HashSet<_> = self |
|
382 |
.clients |
|
383 |
.iter() |
|
384 |
.map(|(_, c)| c.protocol_number) |
|
385 |
.chain(self.rooms.iter().map(|(_, r)| r.protocol_number)) |
|
386 |
.collect(); |
|
387 |
let mut protocols: Vec<_> = protocols.drain().collect(); |
|
388 |
protocols.sort(); |
|
389 |
Ok(protocols) |
|
390 |
} else { |
|
391 |
Err(AccessError()) |
|
392 |
} |
|
393 |
} |
|
394 |
||
395 |
#[inline] |
|
13416 | 396 |
pub fn has_room(&self, name: &str) -> bool { |
14697 | 397 |
self.find_room(name).is_some() |
13416 | 398 |
} |
399 |
||
15439 | 400 |
#[inline] |
15075 | 401 |
pub fn find_room(&self, name: &str) -> Option<&HwRoom> { |
14457 | 402 |
self.rooms |
403 |
.iter() |
|
404 |
.find_map(|(_, r)| Some(r).filter(|r| r.name == name)) |
|
13416 | 405 |
} |
406 |
||
15075 | 407 |
pub fn find_room_mut(&mut self, name: &str) -> Option<&mut HwRoom> { |
14457 | 408 |
self.rooms |
409 |
.iter_mut() |
|
410 |
.find_map(|(_, r)| Some(r).filter(|r| r.name == name)) |
|
13416 | 411 |
} |
412 |
||
15075 | 413 |
pub fn find_client(&self, nick: &str) -> Option<&HwClient> { |
14457 | 414 |
self.clients |
415 |
.iter() |
|
416 |
.find_map(|(_, c)| Some(c).filter(|c| c.nick == nick)) |
|
13478 | 417 |
} |
418 |
||
15075 | 419 |
pub fn find_client_mut(&mut self, nick: &str) -> Option<&mut HwClient> { |
14457 | 420 |
self.clients |
421 |
.iter_mut() |
|
422 |
.find_map(|(_, c)| Some(c).filter(|c| c.nick == nick)) |
|
13478 | 423 |
} |
424 |
||
14789 | 425 |
pub fn all_clients(&self) -> impl Iterator<Item = ClientId> + '_ { |
426 |
self.clients.iter().map(|(id, _)| id) |
|
427 |
} |
|
428 |
||
429 |
pub fn filter_clients<'a, F>(&'a self, f: F) -> impl Iterator<Item = ClientId> + 'a |
|
430 |
where |
|
15075 | 431 |
F: Fn(&(usize, &HwClient)) -> bool + 'a, |
14789 | 432 |
{ |
433 |
self.clients.iter().filter(f).map(|(_, c)| c.id) |
|
434 |
} |
|
435 |
||
436 |
pub fn filter_rooms<'a, F>(&'a self, f: F) -> impl Iterator<Item = RoomId> + 'a |
|
437 |
where |
|
15075 | 438 |
F: Fn(&(usize, &HwRoom)) -> bool + 'a, |
14789 | 439 |
{ |
440 |
self.rooms.iter().filter(f).map(|(_, c)| c.id) |
|
441 |
} |
|
442 |
||
14781 | 443 |
pub fn collect_clients<F>(&self, f: F) -> Vec<ClientId> |
14457 | 444 |
where |
15075 | 445 |
F: Fn(&(usize, &HwClient)) -> bool, |
14457 | 446 |
{ |
14789 | 447 |
self.filter_clients(f).collect() |
13416 | 448 |
} |
449 |
||
14781 | 450 |
pub fn collect_nicks<F>(&self, f: F) -> Vec<String> |
451 |
where |
|
15075 | 452 |
F: Fn(&(usize, &HwClient)) -> bool, |
14781 | 453 |
{ |
454 |
self.clients |
|
455 |
.iter() |
|
456 |
.filter(f) |
|
457 |
.map(|(_, c)| c.nick.clone()) |
|
458 |
.collect() |
|
14694 | 459 |
} |
460 |
||
14789 | 461 |
pub fn lobby_clients(&self) -> impl Iterator<Item = ClientId> + '_ { |
462 |
self.filter_clients(|(_, c)| c.room_id == None) |
|
14781 | 463 |
} |
464 |
||
14789 | 465 |
pub fn room_clients(&self, room_id: RoomId) -> impl Iterator<Item = ClientId> + '_ { |
466 |
self.filter_clients(move |(_, c)| c.room_id == Some(room_id)) |
|
13416 | 467 |
} |
468 |
||
14789 | 469 |
pub fn protocol_clients(&self, protocol: u16) -> impl Iterator<Item = ClientId> + '_ { |
470 |
self.filter_clients(move |(_, c)| c.protocol_number == protocol) |
|
471 |
} |
|
472 |
||
473 |
pub fn protocol_rooms(&self, protocol: u16) -> impl Iterator<Item = RoomId> + '_ { |
|
474 |
self.filter_rooms(move |(_, r)| r.protocol_number == protocol) |
|
13416 | 475 |
} |
476 |
||
477 |
pub fn other_clients_in_room(&self, self_id: ClientId) -> Vec<ClientId> { |
|
478 |
let room_id = self.clients[self_id].room_id; |
|
14781 | 479 |
self.collect_clients(|(id, c)| *id != self_id && c.room_id == room_id) |
13416 | 480 |
} |
14786
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
481 |
|
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
482 |
pub fn is_registered_only(&self) -> bool { |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
483 |
self.flags.contains(ServerFlags::REGISTERED_ONLY) |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
484 |
} |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
485 |
|
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
486 |
pub fn set_is_registered_only(&mut self, value: bool) { |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
487 |
self.flags.set(ServerFlags::REGISTERED_ONLY, value) |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
488 |
} |
13445
d3c86ade3d4d
Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents:
13442
diff
changeset
|
489 |
} |
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
490 |
|
15075 | 491 |
fn allocate_room(rooms: &mut Slab<HwRoom>) -> &mut HwRoom { |
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
492 |
let entry = rooms.vacant_entry(); |
15075 | 493 |
let room = HwRoom::new(entry.key()); |
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
494 |
entry.insert(room) |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
495 |
} |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
496 |
|
15439 | 497 |
fn create_room<'a, 'b>( |
498 |
client: &'a mut HwClient, |
|
499 |
rooms: &'b mut Slab<HwRoom>, |
|
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
500 |
name: String, |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
501 |
password: Option<String>, |
15439 | 502 |
) -> (&'a HwClient, &'b HwRoom) { |
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
503 |
let room = allocate_room(rooms); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
504 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
505 |
room.master_id = Some(client.id); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
506 |
room.name = name; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
507 |
room.password = password; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
508 |
room.protocol_number = client.protocol_number; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
509 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
510 |
room.players_number = 1; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
511 |
room.ready_players_number = 1; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
512 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
513 |
client.room_id = Some(room.id); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
514 |
client.set_is_master(true); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
515 |
client.set_is_ready(true); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
516 |
client.set_is_joined_mid_game(false); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
517 |
|
15439 | 518 |
(client, room) |
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
519 |
} |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
520 |
|
15075 | 521 |
fn move_to_room(client: &mut HwClient, room: &mut HwRoom) { |
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
522 |
debug_assert!(client.room_id != Some(room.id)); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
523 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
524 |
room.players_number += 1; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
525 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
526 |
client.room_id = Some(room.id); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
527 |
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
|
528 |
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
|
529 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
530 |
if let Some(ref mut info) = room.game_info { |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
531 |
let teams = info.client_teams(client.id); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
532 |
client.teams_in_game = teams.clone().count() as u8; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
533 |
client.clan = teams.clone().next().map(|t| t.color); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
534 |
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
|
535 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
536 |
if !team_names.is_empty() { |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
537 |
info.left_teams.retain(|name| !team_names.contains(&name)); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
538 |
info.teams_in_game += team_names.len() as u8; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
539 |
room.teams = info |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
540 |
.teams_at_start |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
541 |
.iter() |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
542 |
.filter(|(_, t)| !team_names.contains(&t.name)) |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
543 |
.cloned() |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
544 |
.collect(); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
545 |
} |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
546 |
} |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
547 |
} |