author | alfadur <mail@none> |
Sat, 21 Dec 2019 23:33:50 +0300 | |
changeset 15519 | b3157d218ae2 |
parent 15518 | e705d30e0f10 |
child 15520 | fd3a20e9d095 |
permissions | -rw-r--r-- |
13416 | 1 |
use super::{ |
15075 | 2 |
client::HwClient, |
14693 | 3 |
indexslab::IndexSlab, |
15075 | 4 |
room::HwRoom, |
15519
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
5 |
types::{ClientId, RoomId, ServerVar, TeamInfo}, |
13416 | 6 |
}; |
15075 | 7 |
use crate::{protocol::messages::HwProtocolMessage::Greeting, utils}; |
14693 | 8 |
|
15519
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
9 |
use bitflags::_core::hint::unreachable_unchecked; |
14786
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
10 |
use bitflags::*; |
15517 | 11 |
use chrono::{offset, DateTime}; |
13805 | 12 |
use log::*; |
15517 | 13 |
use slab::Slab; |
15516 | 14 |
use std::{borrow::BorrowMut, collections::HashSet, iter, mem::replace, num::NonZeroU16}; |
12126 | 15 |
|
15441 | 16 |
#[derive(Debug)] |
15439 | 17 |
pub enum CreateRoomError { |
18 |
InvalidName, |
|
19 |
AlreadyExists, |
|
20 |
} |
|
21 |
||
15441 | 22 |
#[derive(Debug)] |
15439 | 23 |
pub enum JoinRoomError { |
24 |
DoesntExist, |
|
25 |
WrongProtocol, |
|
26 |
Full, |
|
27 |
Restricted, |
|
28 |
} |
|
29 |
||
15518 | 30 |
#[derive(Debug)] |
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)] |
15487 | 48 |
pub struct ChangeMasterResult { |
49 |
pub old_master_id: Option<ClientId>, |
|
50 |
pub new_master_id: ClientId, |
|
51 |
} |
|
52 |
||
53 |
#[derive(Debug)] |
|
54 |
pub enum ChangeMasterError { |
|
55 |
NoAccess, |
|
56 |
AlreadyMaster, |
|
57 |
NoClient, |
|
58 |
ClientNotInRoom, |
|
59 |
} |
|
60 |
||
61 |
#[derive(Debug)] |
|
15519
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
62 |
pub enum AddTeamError { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
63 |
TooManyTeams, |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
64 |
TooManyHedgehogs, |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
65 |
TeamAlreadyExists, |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
66 |
GameInProgress, |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
67 |
Restricted, |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
68 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
69 |
|
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
70 |
#[derive(Debug)] |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
71 |
pub enum RemoveTeamError { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
72 |
NoTeam, |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
73 |
TeamNotOwned, |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
74 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
75 |
|
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
76 |
#[derive(Debug)] |
15518 | 77 |
pub enum ModifyTeamError { |
78 |
NoTeam, |
|
79 |
NotMaster, |
|
80 |
} |
|
81 |
||
82 |
#[derive(Debug)] |
|
83 |
pub enum ModifyRoomNameError { |
|
84 |
AccessDenied, |
|
85 |
InvalidName, |
|
86 |
DuplicateName, |
|
87 |
} |
|
88 |
||
89 |
#[derive(Debug)] |
|
15492 | 90 |
pub enum StartGameError { |
91 |
NotEnoughClans, |
|
92 |
NotEnoughTeams, |
|
93 |
NotReady, |
|
94 |
AlreadyInGame, |
|
95 |
} |
|
96 |
||
97 |
#[derive(Debug)] |
|
15516 | 98 |
pub struct EndGameResult { |
99 |
pub joined_mid_game_clients: Vec<ClientId>, |
|
100 |
pub left_teams: Vec<String>, |
|
101 |
pub unreadied_nicks: Vec<String>, |
|
102 |
} |
|
103 |
||
104 |
#[derive(Debug)] |
|
15441 | 105 |
pub struct UninitializedError(); |
106 |
#[derive(Debug)] |
|
15439 | 107 |
pub struct AccessError(); |
108 |
||
15075 | 109 |
pub struct HwAnteClient { |
14693 | 110 |
pub nick: Option<String>, |
111 |
pub protocol_number: Option<NonZeroU16>, |
|
112 |
pub server_salt: String, |
|
14781 | 113 |
pub is_checker: bool, |
15176 | 114 |
pub is_local_admin: bool, |
15441 | 115 |
pub is_registered: bool, |
116 |
pub is_admin: bool, |
|
117 |
pub is_contributor: bool, |
|
14693 | 118 |
} |
119 |
||
15517 | 120 |
struct Ipv4AddrRange { |
121 |
min: [u8; 4], |
|
122 |
max: [u8; 4], |
|
123 |
} |
|
124 |
||
125 |
impl Ipv4AddrRange { |
|
126 |
fn contains(&self, addr: [u8; 4]) -> bool { |
|
127 |
(0..4).all(|i| self.min[i] <= addr[i] && addr[i] <= self.max[i]) |
|
128 |
} |
|
129 |
} |
|
130 |
||
131 |
struct BanCollection { |
|
132 |
ban_ips: Vec<Ipv4AddrRange>, |
|
133 |
ban_timeouts: Vec<DateTime<offset::Utc>>, |
|
134 |
ban_reasons: Vec<String>, |
|
135 |
} |
|
136 |
||
137 |
impl BanCollection { |
|
138 |
fn new() -> Self { |
|
139 |
Self { |
|
140 |
ban_ips: vec![], |
|
141 |
ban_timeouts: vec![], |
|
142 |
ban_reasons: vec![], |
|
143 |
} |
|
144 |
} |
|
145 |
||
146 |
fn find(&self, addr: [u8; 4]) -> Option<String> { |
|
147 |
let time = offset::Utc::now(); |
|
148 |
self.ban_ips |
|
149 |
.iter() |
|
150 |
.enumerate() |
|
151 |
.find(|(i, r)| r.contains(addr) && time < self.ban_timeouts[*i]) |
|
152 |
.map(|(i, _)| self.ban_reasons[i].clone()) |
|
153 |
} |
|
154 |
} |
|
155 |
||
15075 | 156 |
pub struct HwAnteroom { |
157 |
pub clients: IndexSlab<HwAnteClient>, |
|
15517 | 158 |
bans: BanCollection, |
14693 | 159 |
} |
160 |
||
15075 | 161 |
impl HwAnteroom { |
14693 | 162 |
pub fn new(clients_limit: usize) -> Self { |
163 |
let clients = IndexSlab::with_capacity(clients_limit); |
|
15517 | 164 |
HwAnteroom { |
165 |
clients, |
|
166 |
bans: BanCollection::new(), |
|
167 |
} |
|
168 |
} |
|
169 |
||
170 |
pub fn find_ip_ban(&self, addr: [u8; 4]) -> Option<String> { |
|
171 |
self.bans.find(addr) |
|
14693 | 172 |
} |
173 |
||
15176 | 174 |
pub fn add_client(&mut self, client_id: ClientId, salt: String, is_local_admin: bool) { |
15075 | 175 |
let client = HwAnteClient { |
14693 | 176 |
nick: None, |
177 |
protocol_number: None, |
|
178 |
server_salt: salt, |
|
14781 | 179 |
is_checker: false, |
15176 | 180 |
is_local_admin, |
15441 | 181 |
is_registered: false, |
182 |
is_admin: false, |
|
183 |
is_contributor: false, |
|
14693 | 184 |
}; |
185 |
self.clients.insert(client_id, client); |
|
186 |
} |
|
187 |
||
15075 | 188 |
pub fn remove_client(&mut self, client_id: ClientId) -> Option<HwAnteClient> { |
15439 | 189 |
let client = self.clients.remove(client_id); |
14693 | 190 |
client |
191 |
} |
|
192 |
} |
|
193 |
||
14783 | 194 |
pub struct ServerGreetings { |
195 |
pub for_latest_protocol: String, |
|
196 |
pub for_old_protocols: String, |
|
197 |
} |
|
198 |
||
199 |
impl ServerGreetings { |
|
200 |
fn new() -> Self { |
|
201 |
Self { |
|
202 |
for_latest_protocol: "\u{1f994} is watching".to_string(), |
|
203 |
for_old_protocols: "\u{1f994} is watching".to_string(), |
|
204 |
} |
|
205 |
} |
|
206 |
} |
|
207 |
||
14786
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
208 |
bitflags! { |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
209 |
pub struct ServerFlags: u8 { |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
210 |
const REGISTERED_ONLY = 0b0000_1000; |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
211 |
} |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
212 |
} |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
213 |
|
15075 | 214 |
pub struct HwServer { |
215 |
pub clients: IndexSlab<HwClient>, |
|
216 |
pub rooms: Slab<HwRoom>, |
|
217 |
pub anteroom: HwAnteroom, |
|
14783 | 218 |
pub latest_protocol: u16, |
14786
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
219 |
pub flags: ServerFlags, |
14783 | 220 |
pub greetings: ServerGreetings, |
12126 | 221 |
} |
222 |
||
15075 | 223 |
impl HwServer { |
14780 | 224 |
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
|
225 |
let rooms = Slab::with_capacity(rooms_limit); |
14693 | 226 |
let clients = IndexSlab::with_capacity(clients_limit); |
14694 | 227 |
Self { |
14457 | 228 |
clients, |
229 |
rooms, |
|
15075 | 230 |
anteroom: HwAnteroom::new(clients_limit), |
14783 | 231 |
greetings: ServerGreetings::new(), |
232 |
latest_protocol: 58, |
|
14786
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
233 |
flags: ServerFlags::empty(), |
14694 | 234 |
} |
12126 | 235 |
} |
236 |
||
15441 | 237 |
#[inline] |
238 |
pub fn client(&self, client_id: ClientId) -> &HwClient { |
|
239 |
&self.clients[client_id] |
|
240 |
} |
|
241 |
||
242 |
#[inline] |
|
15518 | 243 |
fn client_mut(&mut self, client_id: ClientId) -> &mut HwClient { |
15441 | 244 |
&mut self.clients[client_id] |
245 |
} |
|
246 |
||
247 |
#[inline] |
|
15519
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
248 |
pub fn has_client(&self, client_id: ClientId) -> bool { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
249 |
self.clients.contains(client_id) |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
250 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
251 |
|
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
252 |
#[inline] |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
253 |
pub fn iter_clients(&self) -> impl Iterator<Item = &HwClient> { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
254 |
self.clients.iter().map(|(_, c)| c) |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
255 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
256 |
|
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
257 |
#[inline] |
15441 | 258 |
pub fn room(&self, room_id: RoomId) -> &HwRoom { |
259 |
&self.rooms[room_id] |
|
260 |
} |
|
261 |
||
262 |
#[inline] |
|
263 |
pub fn room_mut(&mut self, room_id: RoomId) -> &mut HwRoom { |
|
264 |
&mut self.rooms[room_id] |
|
265 |
} |
|
266 |
||
267 |
#[inline] |
|
15482 | 268 |
pub fn client_and_room(&self, client_id: ClientId, room_id: RoomId) -> (&HwClient, &HwRoom) { |
269 |
(&self.clients[client_id], &self.rooms[room_id]) |
|
270 |
} |
|
271 |
||
272 |
#[inline] |
|
273 |
pub fn client_and_room_mut( |
|
274 |
&mut self, |
|
275 |
client_id: ClientId, |
|
276 |
room_id: RoomId, |
|
15519
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
277 |
) -> (&HwClient, &mut HwRoom) { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
278 |
(&self.clients[client_id], &mut self.rooms[room_id]) |
15482 | 279 |
} |
280 |
||
281 |
#[inline] |
|
15441 | 282 |
pub fn is_admin(&self, client_id: ClientId) -> bool { |
283 |
self.clients |
|
284 |
.get(client_id) |
|
285 |
.map(|c| c.is_admin()) |
|
286 |
.unwrap_or(false) |
|
287 |
} |
|
288 |
||
15075 | 289 |
pub fn add_client(&mut self, client_id: ClientId, data: HwAnteClient) { |
14693 | 290 |
if let (Some(protocol), Some(nick)) = (data.protocol_number, data.nick) { |
15075 | 291 |
let mut client = HwClient::new(client_id, protocol.get(), nick); |
14781 | 292 |
client.set_is_checker(data.is_checker); |
15176 | 293 |
#[cfg(not(feature = "official-server"))] |
294 |
client.set_is_admin(data.is_local_admin); |
|
295 |
||
15441 | 296 |
#[cfg(feature = "official-server")] |
297 |
{ |
|
298 |
client.set_is_registered(info.is_registered); |
|
299 |
client.set_is_admin(info.is_admin); |
|
300 |
client.set_is_contributor(info.is_contributor); |
|
301 |
} |
|
302 |
||
14693 | 303 |
self.clients.insert(client_id, client); |
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
304 |
} |
12126 | 305 |
} |
12127 | 306 |
|
14673
08a8605bafaf
Server action refactoring part 4 of N
alfadur <mail@none>
parents:
14672
diff
changeset
|
307 |
pub fn remove_client(&mut self, client_id: ClientId) { |
14696 | 308 |
self.clients.remove(client_id); |
12127 | 309 |
} |
310 |
||
15441 | 311 |
pub fn get_greetings(&self, client: &HwClient) -> &str { |
312 |
if client.protocol_number < self.latest_protocol { |
|
14783 | 313 |
&self.greetings.for_old_protocols |
314 |
} else { |
|
315 |
&self.greetings.for_latest_protocol |
|
316 |
} |
|
317 |
} |
|
318 |
||
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
319 |
#[inline] |
15439 | 320 |
pub fn get_client_nick(&self, client_id: ClientId) -> &str { |
321 |
&self.clients[client_id].nick |
|
322 |
} |
|
323 |
||
324 |
#[inline] |
|
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
325 |
pub fn create_room( |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
326 |
&mut self, |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
327 |
creator_id: ClientId, |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
328 |
name: String, |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
329 |
password: Option<String>, |
15439 | 330 |
) -> Result<(&HwClient, &HwRoom), CreateRoomError> { |
331 |
use CreateRoomError::*; |
|
332 |
if utils::is_name_illegal(&name) { |
|
333 |
Err(InvalidName) |
|
334 |
} else if self.has_room(&name) { |
|
335 |
Err(AlreadyExists) |
|
336 |
} else { |
|
337 |
Ok(create_room( |
|
338 |
&mut self.clients[creator_id], |
|
339 |
&mut self.rooms, |
|
340 |
name, |
|
341 |
password, |
|
342 |
)) |
|
343 |
} |
|
344 |
} |
|
345 |
||
346 |
pub fn join_room( |
|
347 |
&mut self, |
|
348 |
client_id: ClientId, |
|
349 |
room_id: RoomId, |
|
350 |
) -> Result<(&HwClient, &HwRoom, impl Iterator<Item = &HwClient> + Clone), JoinRoomError> { |
|
351 |
use JoinRoomError::*; |
|
352 |
let room = &mut self.rooms[room_id]; |
|
353 |
let client = &mut self.clients[client_id]; |
|
354 |
||
355 |
if client.protocol_number != room.protocol_number { |
|
356 |
Err(WrongProtocol) |
|
357 |
} else if room.is_join_restricted() { |
|
358 |
Err(Restricted) |
|
359 |
} else if room.players_number == u8::max_value() { |
|
360 |
Err(Full) |
|
361 |
} else { |
|
362 |
move_to_room(client, room); |
|
363 |
let room_id = room.id; |
|
364 |
Ok(( |
|
365 |
&self.clients[client_id], |
|
366 |
&self.rooms[room_id], |
|
367 |
self.clients.iter().map(|(_, c)| c), |
|
368 |
)) |
|
369 |
} |
|
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
370 |
} |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
371 |
|
14671
455865ccd36c
Server action refactoring part 2 of N
alfadur <mail@none>
parents:
14504
diff
changeset
|
372 |
#[inline] |
15439 | 373 |
pub fn join_room_by_name( |
374 |
&mut self, |
|
375 |
client_id: ClientId, |
|
376 |
room_name: &str, |
|
377 |
) -> Result<(&HwClient, &HwRoom, impl Iterator<Item = &HwClient> + Clone), JoinRoomError> { |
|
378 |
use JoinRoomError::*; |
|
379 |
let room = self.rooms.iter().find(|(_, r)| r.name == room_name); |
|
380 |
if let Some((_, room)) = room { |
|
381 |
let room_id = room.id; |
|
382 |
self.join_room(client_id, room_id) |
|
383 |
} else { |
|
384 |
Err(DoesntExist) |
|
385 |
} |
|
386 |
} |
|
387 |
||
15482 | 388 |
pub fn leave_room(&mut self, client_id: ClientId) -> Result<LeaveRoomResult, LeaveRoomError> { |
389 |
let client = &mut self.clients[client_id]; |
|
390 |
if let Some(room_id) = client.room_id { |
|
391 |
let room = &mut self.rooms[room_id]; |
|
392 |
||
393 |
room.players_number -= 1; |
|
394 |
client.room_id = None; |
|
395 |
||
396 |
let is_empty = room.players_number == 0; |
|
397 |
let is_fixed = room.is_fixed(); |
|
398 |
let was_master = room.master_id == Some(client_id); |
|
399 |
let was_in_game = client.is_in_game(); |
|
400 |
let mut removed_teams = vec![]; |
|
401 |
||
402 |
if is_empty && !is_fixed { |
|
403 |
if client.is_ready() && room.ready_players_number > 0 { |
|
404 |
room.ready_players_number -= 1; |
|
405 |
} |
|
406 |
||
407 |
removed_teams = room |
|
408 |
.client_teams(client.id) |
|
409 |
.map(|t| t.name.clone()) |
|
410 |
.collect(); |
|
411 |
||
412 |
for team_name in &removed_teams { |
|
413 |
room.remove_team(team_name); |
|
414 |
} |
|
415 |
||
416 |
if client.is_master() && !is_fixed { |
|
417 |
client.set_is_master(false); |
|
418 |
room.master_id = None; |
|
419 |
} |
|
420 |
} |
|
421 |
||
422 |
client.set_is_ready(false); |
|
423 |
client.set_is_in_game(false); |
|
424 |
||
425 |
if !is_fixed { |
|
426 |
if room.players_number == 0 { |
|
427 |
self.rooms.remove(room_id); |
|
428 |
} else if room.master_id == None { |
|
429 |
let new_master_id = self.room_clients(room_id).next(); |
|
430 |
if let Some(new_master_id) = new_master_id { |
|
431 |
let room = &mut self.rooms[room_id]; |
|
432 |
room.master_id = Some(new_master_id); |
|
433 |
let new_master = &mut self.clients[new_master_id]; |
|
434 |
new_master.set_is_master(true); |
|
435 |
||
436 |
if room.protocol_number < 42 { |
|
437 |
room.name = new_master.nick.clone(); |
|
438 |
} |
|
439 |
||
440 |
room.set_join_restriction(false); |
|
441 |
room.set_team_add_restriction(false); |
|
442 |
room.set_unregistered_players_restriction(true); |
|
443 |
} |
|
444 |
} |
|
445 |
} |
|
446 |
||
447 |
if is_empty && !is_fixed { |
|
448 |
Ok(LeaveRoomResult::RoomRemoved) |
|
449 |
} else { |
|
450 |
Ok(LeaveRoomResult::RoomRemains { |
|
451 |
is_empty, |
|
452 |
was_master, |
|
453 |
was_in_game, |
|
454 |
new_master: self.rooms[room_id].master_id, |
|
455 |
removed_teams, |
|
456 |
}) |
|
457 |
} |
|
458 |
} else { |
|
459 |
Err(LeaveRoomError::NoRoom) |
|
460 |
} |
|
461 |
} |
|
462 |
||
15487 | 463 |
pub fn change_master( |
464 |
&mut self, |
|
465 |
client_id: ClientId, |
|
466 |
room_id: RoomId, |
|
467 |
new_master_nick: String, |
|
468 |
) -> Result<ChangeMasterResult, ChangeMasterError> { |
|
469 |
let client = &mut self.clients[client_id]; |
|
470 |
let room = &mut self.rooms[room_id]; |
|
471 |
||
472 |
if client.is_admin() || room.master_id == Some(client_id) { |
|
473 |
let new_master_id = self |
|
474 |
.clients |
|
475 |
.iter() |
|
476 |
.find(|(_, c)| c.nick == new_master_nick) |
|
477 |
.map(|(id, _)| id); |
|
478 |
||
479 |
match new_master_id { |
|
480 |
Some(new_master_id) if new_master_id == client_id => { |
|
481 |
Err(ChangeMasterError::AlreadyMaster) |
|
482 |
} |
|
483 |
Some(new_master_id) => { |
|
484 |
let new_master = &mut self.clients[new_master_id]; |
|
485 |
if new_master.room_id == Some(room_id) { |
|
486 |
self.clients[new_master_id].set_is_master(true); |
|
487 |
let old_master_id = room.master_id; |
|
488 |
if let Some(master_id) = old_master_id { |
|
489 |
self.clients[master_id].set_is_master(false); |
|
490 |
} |
|
491 |
room.master_id = Some(new_master_id); |
|
492 |
Ok(ChangeMasterResult { |
|
493 |
old_master_id, |
|
494 |
new_master_id, |
|
495 |
}) |
|
496 |
} else { |
|
497 |
Err(ChangeMasterError::ClientNotInRoom) |
|
498 |
} |
|
499 |
} |
|
500 |
None => Err(ChangeMasterError::NoClient), |
|
501 |
} |
|
502 |
} else { |
|
503 |
Err(ChangeMasterError::NoAccess) |
|
504 |
} |
|
505 |
} |
|
506 |
||
15492 | 507 |
pub fn start_game(&mut self, room_id: RoomId) -> Result<Vec<String>, StartGameError> { |
508 |
let (room_clients, room_nicks): (Vec<_>, Vec<_>) = self |
|
509 |
.clients |
|
510 |
.iter() |
|
511 |
.map(|(id, c)| (id, c.nick.clone())) |
|
512 |
.unzip(); |
|
513 |
||
514 |
let room = &mut self.rooms[room_id]; |
|
515 |
||
516 |
if !room.has_multiple_clans() { |
|
517 |
Err(StartGameError::NotEnoughClans) |
|
518 |
} else if room.protocol_number <= 43 && room.players_number != room.ready_players_number { |
|
519 |
Err(StartGameError::NotReady) |
|
520 |
} else if room.game_info.is_some() { |
|
521 |
Err(StartGameError::AlreadyInGame) |
|
522 |
} else { |
|
523 |
room.start_round(); |
|
524 |
for id in room_clients { |
|
525 |
let c = &mut self.clients[id]; |
|
526 |
c.set_is_in_game(true); |
|
527 |
c.team_indices = room.client_team_indices(c.id); |
|
528 |
} |
|
529 |
Ok(room_nicks) |
|
530 |
} |
|
531 |
} |
|
532 |
||
15519
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
533 |
pub fn leave_game(&mut self, client_id: ClientId) -> Option<Vec<String>> { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
534 |
let client = &mut self.clients[client_id]; |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
535 |
let client_left = client.is_in_game(); |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
536 |
if client_left { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
537 |
client.set_is_in_game(false); |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
538 |
let room = &mut self.rooms[client.room_id.expect("Client should've been in the game")]; |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
539 |
|
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
540 |
let team_names: Vec<_> = room |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
541 |
.client_teams(client_id) |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
542 |
.map(|t| t.name.clone()) |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
543 |
.collect(); |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
544 |
|
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
545 |
if let Some(ref mut info) = room.game_info { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
546 |
info.teams_in_game -= team_names.len() as u8; |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
547 |
|
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
548 |
for team_name in &team_names { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
549 |
let remove_msg = |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
550 |
utils::to_engine_msg(std::iter::once(b'F').chain(team_name.bytes())); |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
551 |
if let Some(m) = &info.sync_msg { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
552 |
info.msg_log.push(m.clone()); |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
553 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
554 |
if info.sync_msg.is_some() { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
555 |
info.sync_msg = None |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
556 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
557 |
info.msg_log.push(remove_msg); |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
558 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
559 |
Some(team_names) |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
560 |
} else { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
561 |
unreachable!(); |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
562 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
563 |
} else { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
564 |
None |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
565 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
566 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
567 |
|
15516 | 568 |
pub fn end_game(&mut self, room_id: RoomId) -> EndGameResult { |
569 |
let room = &mut self.rooms[room_id]; |
|
15519
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
570 |
room.ready_players_number = room.master_id.is_some() as u8; |
15516 | 571 |
|
572 |
if let Some(info) = replace(&mut room.game_info, None) { |
|
573 |
let joined_mid_game_clients = self |
|
574 |
.clients |
|
575 |
.iter() |
|
576 |
.filter(|(_, c)| c.room_id == Some(room_id) && c.is_joined_mid_game()) |
|
577 |
.map(|(_, c)| c.id) |
|
578 |
.collect(); |
|
579 |
||
580 |
let unreadied_nicks: Vec<_> = self |
|
581 |
.clients |
|
582 |
.iter_mut() |
|
583 |
.filter(|(_, c)| c.room_id == Some(room_id)) |
|
584 |
.map(|(_, c)| { |
|
585 |
c.set_is_ready(c.is_master()); |
|
586 |
c.set_is_joined_mid_game(false); |
|
587 |
c |
|
588 |
}) |
|
589 |
.filter_map(|c| { |
|
590 |
if !c.is_master() { |
|
591 |
Some(c.nick.clone()) |
|
592 |
} else { |
|
593 |
None |
|
594 |
} |
|
595 |
}) |
|
596 |
.collect(); |
|
597 |
||
598 |
EndGameResult { |
|
599 |
joined_mid_game_clients, |
|
600 |
left_teams: info.left_teams.clone(), |
|
601 |
unreadied_nicks, |
|
602 |
} |
|
603 |
} else { |
|
604 |
unreachable!() |
|
605 |
} |
|
606 |
} |
|
607 |
||
15518 | 608 |
pub fn enable_super_power(&mut self, client_id: ClientId) -> bool { |
609 |
let client = &mut self.clients[client_id]; |
|
610 |
if client.is_admin() { |
|
611 |
client.set_has_super_power(true); |
|
612 |
} |
|
613 |
client.is_admin() |
|
614 |
} |
|
615 |
||
616 |
pub fn set_room_name( |
|
617 |
&mut self, |
|
618 |
client_id: ClientId, |
|
619 |
room_id: RoomId, |
|
620 |
mut name: String, |
|
621 |
) -> Result<String, ModifyRoomNameError> { |
|
622 |
let room_exists = self.has_room(&name); |
|
623 |
let room = &mut self.rooms[room_id]; |
|
624 |
if room.is_fixed() || room.master_id != Some(client_id) { |
|
625 |
Err(ModifyRoomNameError::AccessDenied) |
|
626 |
} else if utils::is_name_illegal(&name) { |
|
627 |
Err(ModifyRoomNameError::InvalidName) |
|
628 |
} else if room_exists { |
|
629 |
Err(ModifyRoomNameError::DuplicateName) |
|
630 |
} else { |
|
631 |
std::mem::swap(&mut room.name, &mut name); |
|
632 |
Ok(name) |
|
633 |
} |
|
634 |
} |
|
635 |
||
15519
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
636 |
pub fn add_team( |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
637 |
&mut self, |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
638 |
client_id: ClientId, |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
639 |
mut info: Box<TeamInfo>, |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
640 |
) -> Result<&TeamInfo, AddTeamError> { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
641 |
let client = &mut self.clients[client_id]; |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
642 |
if let Some(room_id) = client.room_id { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
643 |
let room = &mut self.rooms[room_id]; |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
644 |
if room.teams.len() >= room.max_teams as usize { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
645 |
Err(AddTeamError::TooManyTeams) |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
646 |
} else if room.addable_hedgehogs() == 0 { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
647 |
Err(AddTeamError::TooManyHedgehogs) |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
648 |
} else if room.find_team(|t| t.name == info.name) != None { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
649 |
Err(AddTeamError::TeamAlreadyExists) |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
650 |
} else if room.game_info.is_some() { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
651 |
Err(AddTeamError::GameInProgress) |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
652 |
} else if room.is_team_add_restricted() { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
653 |
Err(AddTeamError::Restricted) |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
654 |
} else { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
655 |
info.owner = client.nick.clone(); |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
656 |
let team = room.add_team(client.id, *info, client.protocol_number < 42); |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
657 |
client.teams_in_game += 1; |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
658 |
client.clan = Some(team.color); |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
659 |
Ok(team) |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
660 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
661 |
} else { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
662 |
unreachable!() |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
663 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
664 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
665 |
|
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
666 |
pub fn remove_team( |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
667 |
&mut self, |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
668 |
client_id: ClientId, |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
669 |
team_name: &str, |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
670 |
) -> Result<(), RemoveTeamError> { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
671 |
let client = &mut self.clients[client_id]; |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
672 |
if let Some(room_id) = client.room_id { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
673 |
let room = &mut self.rooms[room_id]; |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
674 |
match room.find_team_owner(team_name) { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
675 |
None => Err(RemoveTeamError::NoTeam), |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
676 |
Some((id, _)) if id != client_id => Err(RemoveTeamError::TeamNotOwned), |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
677 |
Some(_) => { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
678 |
client.teams_in_game -= 1; |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
679 |
client.clan = room.find_team_color(client.id); |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
680 |
room.remove_team(team_name); |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
681 |
Ok(()) |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
682 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
683 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
684 |
} else { |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
685 |
unreachable!(); |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
686 |
} |
b3157d218ae2
disallow mutable clients to leave the server
alfadur <mail@none>
parents:
15518
diff
changeset
|
687 |
} |
15518 | 688 |
|
689 |
pub fn set_team_color( |
|
690 |
&mut self, |
|
691 |
client_id: ClientId, |
|
692 |
room_id: RoomId, |
|
693 |
team_name: &str, |
|
694 |
color: u8, |
|
695 |
) -> Result<(), ModifyTeamError> { |
|
696 |
let client = &self.clients[client_id]; |
|
697 |
let room = &mut self.rooms[room_id]; |
|
698 |
if let Some((owner, team)) = room.find_team_and_owner_mut(|t| t.name == team_name) { |
|
699 |
if !client.is_master() { |
|
700 |
Err(ModifyTeamError::NotMaster) |
|
701 |
} else { |
|
702 |
team.color = color; |
|
703 |
self.clients[owner].clan = Some(color); |
|
704 |
Ok(()) |
|
705 |
} |
|
706 |
} else { |
|
707 |
Err(ModifyTeamError::NoTeam) |
|
708 |
} |
|
709 |
} |
|
710 |
||
711 |
pub fn toggle_ready(&mut self, client_id: ClientId) -> bool { |
|
712 |
let client = &mut self.clients[client_id]; |
|
713 |
if let Some(room_id) = client.room_id { |
|
714 |
let room = &mut self.rooms[room_id]; |
|
715 |
||
716 |
client.set_is_ready(!client.is_ready()); |
|
717 |
if client.is_ready() { |
|
718 |
room.ready_players_number += 1; |
|
719 |
} else { |
|
720 |
room.ready_players_number -= 1; |
|
721 |
} |
|
722 |
} |
|
723 |
client.is_ready() |
|
724 |
} |
|
725 |
||
15439 | 726 |
#[inline] |
727 |
pub fn set_var(&mut self, client_id: ClientId, var: ServerVar) -> Result<(), AccessError> { |
|
728 |
if self.clients[client_id].is_admin() { |
|
729 |
match var { |
|
730 |
ServerVar::MOTDNew(msg) => self.greetings.for_latest_protocol = msg, |
|
731 |
ServerVar::MOTDOld(msg) => self.greetings.for_old_protocols = msg, |
|
732 |
ServerVar::LatestProto(n) => self.latest_protocol = n, |
|
733 |
} |
|
734 |
Ok(()) |
|
735 |
} else { |
|
736 |
Err(AccessError()) |
|
737 |
} |
|
14671
455865ccd36c
Server action refactoring part 2 of N
alfadur <mail@none>
parents:
14504
diff
changeset
|
738 |
} |
455865ccd36c
Server action refactoring part 2 of N
alfadur <mail@none>
parents:
14504
diff
changeset
|
739 |
|
15439 | 740 |
#[inline] |
741 |
pub fn get_vars(&self, client_id: ClientId) -> Result<[ServerVar; 3], AccessError> { |
|
742 |
if self.clients[client_id].is_admin() { |
|
743 |
Ok([ |
|
744 |
ServerVar::MOTDNew(self.greetings.for_latest_protocol.clone()), |
|
745 |
ServerVar::MOTDOld(self.greetings.for_old_protocols.clone()), |
|
746 |
ServerVar::LatestProto(self.latest_protocol), |
|
747 |
]) |
|
748 |
} else { |
|
749 |
Err(AccessError()) |
|
750 |
} |
|
751 |
} |
|
752 |
||
753 |
pub fn get_used_protocols(&self, client_id: ClientId) -> Result<Vec<u16>, AccessError> { |
|
754 |
if self.clients[client_id].is_admin() { |
|
755 |
let mut protocols: HashSet<_> = self |
|
756 |
.clients |
|
757 |
.iter() |
|
758 |
.map(|(_, c)| c.protocol_number) |
|
759 |
.chain(self.rooms.iter().map(|(_, r)| r.protocol_number)) |
|
760 |
.collect(); |
|
761 |
let mut protocols: Vec<_> = protocols.drain().collect(); |
|
762 |
protocols.sort(); |
|
763 |
Ok(protocols) |
|
764 |
} else { |
|
765 |
Err(AccessError()) |
|
766 |
} |
|
767 |
} |
|
768 |
||
769 |
#[inline] |
|
13416 | 770 |
pub fn has_room(&self, name: &str) -> bool { |
14697 | 771 |
self.find_room(name).is_some() |
13416 | 772 |
} |
773 |
||
15439 | 774 |
#[inline] |
15075 | 775 |
pub fn find_room(&self, name: &str) -> Option<&HwRoom> { |
14457 | 776 |
self.rooms |
777 |
.iter() |
|
778 |
.find_map(|(_, r)| Some(r).filter(|r| r.name == name)) |
|
13416 | 779 |
} |
780 |
||
15075 | 781 |
pub fn find_room_mut(&mut self, name: &str) -> Option<&mut HwRoom> { |
14457 | 782 |
self.rooms |
783 |
.iter_mut() |
|
784 |
.find_map(|(_, r)| Some(r).filter(|r| r.name == name)) |
|
13416 | 785 |
} |
786 |
||
15075 | 787 |
pub fn find_client(&self, nick: &str) -> Option<&HwClient> { |
14457 | 788 |
self.clients |
789 |
.iter() |
|
790 |
.find_map(|(_, c)| Some(c).filter(|c| c.nick == nick)) |
|
13478 | 791 |
} |
792 |
||
15075 | 793 |
pub fn find_client_mut(&mut self, nick: &str) -> Option<&mut HwClient> { |
14457 | 794 |
self.clients |
795 |
.iter_mut() |
|
796 |
.find_map(|(_, c)| Some(c).filter(|c| c.nick == nick)) |
|
13478 | 797 |
} |
798 |
||
14789 | 799 |
pub fn all_clients(&self) -> impl Iterator<Item = ClientId> + '_ { |
800 |
self.clients.iter().map(|(id, _)| id) |
|
801 |
} |
|
802 |
||
803 |
pub fn filter_clients<'a, F>(&'a self, f: F) -> impl Iterator<Item = ClientId> + 'a |
|
804 |
where |
|
15075 | 805 |
F: Fn(&(usize, &HwClient)) -> bool + 'a, |
14789 | 806 |
{ |
807 |
self.clients.iter().filter(f).map(|(_, c)| c.id) |
|
808 |
} |
|
809 |
||
810 |
pub fn filter_rooms<'a, F>(&'a self, f: F) -> impl Iterator<Item = RoomId> + 'a |
|
811 |
where |
|
15075 | 812 |
F: Fn(&(usize, &HwRoom)) -> bool + 'a, |
14789 | 813 |
{ |
814 |
self.rooms.iter().filter(f).map(|(_, c)| c.id) |
|
815 |
} |
|
816 |
||
14781 | 817 |
pub fn collect_clients<F>(&self, f: F) -> Vec<ClientId> |
14457 | 818 |
where |
15075 | 819 |
F: Fn(&(usize, &HwClient)) -> bool, |
14457 | 820 |
{ |
14789 | 821 |
self.filter_clients(f).collect() |
13416 | 822 |
} |
823 |
||
14781 | 824 |
pub fn collect_nicks<F>(&self, f: F) -> Vec<String> |
825 |
where |
|
15075 | 826 |
F: Fn(&(usize, &HwClient)) -> bool, |
14781 | 827 |
{ |
828 |
self.clients |
|
829 |
.iter() |
|
830 |
.filter(f) |
|
831 |
.map(|(_, c)| c.nick.clone()) |
|
832 |
.collect() |
|
14694 | 833 |
} |
834 |
||
14789 | 835 |
pub fn lobby_clients(&self) -> impl Iterator<Item = ClientId> + '_ { |
836 |
self.filter_clients(|(_, c)| c.room_id == None) |
|
14781 | 837 |
} |
838 |
||
14789 | 839 |
pub fn room_clients(&self, room_id: RoomId) -> impl Iterator<Item = ClientId> + '_ { |
840 |
self.filter_clients(move |(_, c)| c.room_id == Some(room_id)) |
|
13416 | 841 |
} |
842 |
||
14789 | 843 |
pub fn protocol_clients(&self, protocol: u16) -> impl Iterator<Item = ClientId> + '_ { |
844 |
self.filter_clients(move |(_, c)| c.protocol_number == protocol) |
|
845 |
} |
|
846 |
||
847 |
pub fn protocol_rooms(&self, protocol: u16) -> impl Iterator<Item = RoomId> + '_ { |
|
848 |
self.filter_rooms(move |(_, r)| r.protocol_number == protocol) |
|
13416 | 849 |
} |
850 |
||
851 |
pub fn other_clients_in_room(&self, self_id: ClientId) -> Vec<ClientId> { |
|
852 |
let room_id = self.clients[self_id].room_id; |
|
14781 | 853 |
self.collect_clients(|(id, c)| *id != self_id && c.room_id == room_id) |
13416 | 854 |
} |
14786
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
855 |
|
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
856 |
pub fn is_registered_only(&self) -> bool { |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
857 |
self.flags.contains(ServerFlags::REGISTERED_ONLY) |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
858 |
} |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
859 |
|
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
860 |
pub fn set_is_registered_only(&mut self, value: bool) { |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
861 |
self.flags.set(ServerFlags::REGISTERED_ONLY, value) |
8ecdb5c6bb2a
implement info, registered only & super power messages
alfadur
parents:
14783
diff
changeset
|
862 |
} |
13445
d3c86ade3d4d
Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents:
13442
diff
changeset
|
863 |
} |
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
864 |
|
15075 | 865 |
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
|
866 |
let entry = rooms.vacant_entry(); |
15075 | 867 |
let room = HwRoom::new(entry.key()); |
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
868 |
entry.insert(room) |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
869 |
} |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
870 |
|
15439 | 871 |
fn create_room<'a, 'b>( |
872 |
client: &'a mut HwClient, |
|
873 |
rooms: &'b mut Slab<HwRoom>, |
|
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
874 |
name: String, |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
875 |
password: Option<String>, |
15439 | 876 |
) -> (&'a HwClient, &'b HwRoom) { |
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
877 |
let room = allocate_room(rooms); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
878 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
879 |
room.master_id = Some(client.id); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
880 |
room.name = name; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
881 |
room.password = password; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
882 |
room.protocol_number = client.protocol_number; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
883 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
884 |
room.players_number = 1; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
885 |
room.ready_players_number = 1; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
886 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
887 |
client.room_id = Some(room.id); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
888 |
client.set_is_master(true); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
889 |
client.set_is_ready(true); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
890 |
client.set_is_joined_mid_game(false); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
891 |
|
15439 | 892 |
(client, room) |
14504
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
893 |
} |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
894 |
|
15075 | 895 |
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
|
896 |
debug_assert!(client.room_id != Some(room.id)); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
897 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
898 |
room.players_number += 1; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
899 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
900 |
client.room_id = Some(room.id); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
901 |
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
|
902 |
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
|
903 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
904 |
if let Some(ref mut info) = room.game_info { |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
905 |
let teams = info.client_teams(client.id); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
906 |
client.teams_in_game = teams.clone().count() as u8; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
907 |
client.clan = teams.clone().next().map(|t| t.color); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
908 |
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
|
909 |
|
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
910 |
if !team_names.is_empty() { |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
911 |
info.left_teams.retain(|name| !team_names.contains(&name)); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
912 |
info.teams_in_game += team_names.len() as u8; |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
913 |
room.teams = info |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
914 |
.teams_at_start |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
915 |
.iter() |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
916 |
.filter(|(_, t)| !team_names.contains(&t.name)) |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
917 |
.cloned() |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
918 |
.collect(); |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
919 |
} |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
920 |
} |
6cc0fce249f9
Server action refactoring part 1 of N
alfadur <mail@none>
parents:
14457
diff
changeset
|
921 |
} |