rust/hedgewars-server/src/core/server.rs
author sheepluva
Sat, 04 Jul 2020 03:11:41 +0200
changeset 15665 63e2b7b2ec47
parent 15569 4b2f3228f13b
child 15697 25371cae9fd7
permissions -rw-r--r--
Merge chat-scaling (= current state of branch "ui-scaling") into default branch
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
     1
use super::{
15520
fd3a20e9d095 move the anteroom out of the server
alfadur <mail@none>
parents: 15519
diff changeset
     2
    anteroom::HwAnteroomClient,
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
     3
    client::HwClient,
14693
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14686
diff changeset
     4
    indexslab::IndexSlab,
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
     5
    room::HwRoom,
15525
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
     6
    types::{ClientId, GameCfg, RoomId, ServerVar, TeamInfo, Vote, VoteType, Voting},
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
     7
};
15533
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15532
diff changeset
     8
use crate::utils;
14693
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14686
diff changeset
     9
14786
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
    10
use bitflags::*;
13805
0463a4221327 cleanup crate imports
alfadur
parents: 13798
diff changeset
    11
use log::*;
15517
abd5eb807166 add ip ban check
alfadur <mail@none>
parents: 15516
diff changeset
    12
use slab::Slab;
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    13
use std::{borrow::BorrowMut, cmp::min, collections::HashSet, iter, mem::replace};
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
    14
15441
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
    15
#[derive(Debug)]
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
    16
pub enum CreateRoomError {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
    17
    InvalidName,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
    18
    AlreadyExists,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
    19
}
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
    20
15441
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
    21
#[derive(Debug)]
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
    22
pub enum JoinRoomError {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
    23
    DoesntExist,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
    24
    WrongProtocol,
15533
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15532
diff changeset
    25
    WrongPassword,
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
    26
    Full,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
    27
    Restricted,
15534
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15533
diff changeset
    28
    RegistrationRequired,
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
    29
}
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
    30
15518
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
    31
#[derive(Debug)]
15482
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
    32
pub enum LeaveRoomResult {
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
    33
    RoomRemoved,
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
    34
    RoomRemains {
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
    35
        is_empty: bool,
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
    36
        was_master: bool,
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
    37
        was_in_game: bool,
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
    38
        new_master: Option<ClientId>,
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
    39
        removed_teams: Vec<String>,
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
    40
    },
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
    41
}
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
    42
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
    43
#[derive(Debug)]
15487
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    44
pub struct ChangeMasterResult {
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    45
    pub old_master_id: Option<ClientId>,
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    46
    pub new_master_id: ClientId,
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    47
}
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    48
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    49
#[derive(Debug)]
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    50
pub enum ChangeMasterError {
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    51
    NoAccess,
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    52
    AlreadyMaster,
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    53
    NoClient,
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    54
    ClientNotInRoom,
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    55
}
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    56
91f0c5ec37b5 move change_master into the server
alfadur
parents: 15482
diff changeset
    57
#[derive(Debug)]
15519
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    58
pub enum AddTeamError {
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    59
    TooManyTeams,
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    60
    TooManyHedgehogs,
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    61
    TeamAlreadyExists,
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    62
    GameInProgress,
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    63
    Restricted,
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    64
}
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    65
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    66
#[derive(Debug)]
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    67
pub enum RemoveTeamError {
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    68
    NoTeam,
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    69
    TeamNotOwned,
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    70
}
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    71
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
    72
#[derive(Debug)]
15518
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
    73
pub enum ModifyTeamError {
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
    74
    NoTeam,
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
    75
    NotMaster,
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
    76
}
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
    77
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
    78
#[derive(Debug)]
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    79
pub enum SetTeamCountError {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    80
    InvalidNumber,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    81
    NotMaster,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    82
}
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    83
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    84
#[derive(Debug)]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    85
pub enum SetHedgehogsError {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    86
    NoTeam,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    87
    InvalidNumber(u8),
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    88
    NotMaster,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    89
}
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    90
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    91
#[derive(Debug)]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    92
pub enum SetConfigError {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    93
    NotMaster,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    94
    RoomFixed,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    95
}
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    96
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
    97
#[derive(Debug)]
15518
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
    98
pub enum ModifyRoomNameError {
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
    99
    AccessDenied,
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
   100
    InvalidName,
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
   101
    DuplicateName,
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
   102
}
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
   103
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
   104
#[derive(Debug)]
15525
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   105
pub enum StartVoteError {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   106
    VotingInProgress,
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   107
}
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   108
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   109
#[derive(Debug)]
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   110
pub enum VoteResult {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   111
    Submitted,
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   112
    Succeeded(VoteType),
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   113
    Failed,
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   114
}
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   115
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   116
#[derive(Debug)]
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   117
pub enum VoteError {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   118
    NoVoting,
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   119
    AlreadyVoted,
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   120
}
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   121
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   122
#[derive(Debug)]
15492
395be40faa51 move start_game into the server
alfadur
parents: 15487
diff changeset
   123
pub enum StartGameError {
395be40faa51 move start_game into the server
alfadur
parents: 15487
diff changeset
   124
    NotEnoughClans,
395be40faa51 move start_game into the server
alfadur
parents: 15487
diff changeset
   125
    NotReady,
395be40faa51 move start_game into the server
alfadur
parents: 15487
diff changeset
   126
    AlreadyInGame,
395be40faa51 move start_game into the server
alfadur
parents: 15487
diff changeset
   127
}
395be40faa51 move start_game into the server
alfadur
parents: 15487
diff changeset
   128
395be40faa51 move start_game into the server
alfadur
parents: 15487
diff changeset
   129
#[derive(Debug)]
15516
b907b9071ec5 move end_game into server
alfadur
parents: 15492
diff changeset
   130
pub struct EndGameResult {
b907b9071ec5 move end_game into server
alfadur
parents: 15492
diff changeset
   131
    pub left_teams: Vec<String>,
b907b9071ec5 move end_game into server
alfadur
parents: 15492
diff changeset
   132
    pub unreadied_nicks: Vec<String>,
b907b9071ec5 move end_game into server
alfadur
parents: 15492
diff changeset
   133
}
b907b9071ec5 move end_game into server
alfadur
parents: 15492
diff changeset
   134
b907b9071ec5 move end_game into server
alfadur
parents: 15492
diff changeset
   135
#[derive(Debug)]
15441
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   136
pub struct UninitializedError();
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   137
#[derive(Debug)]
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   138
pub struct AccessError();
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   139
14783
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   140
pub struct ServerGreetings {
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   141
    pub for_latest_protocol: String,
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   142
    pub for_old_protocols: String,
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   143
}
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   144
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   145
impl ServerGreetings {
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   146
    fn new() -> Self {
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   147
        Self {
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   148
            for_latest_protocol: "\u{1f994} is watching".to_string(),
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   149
            for_old_protocols: "\u{1f994} is watching".to_string(),
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   150
        }
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   151
    }
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   152
}
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   153
14786
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   154
bitflags! {
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   155
    pub struct ServerFlags: u8 {
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   156
        const REGISTERED_ONLY = 0b0000_1000;
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   157
    }
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   158
}
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   159
15532
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   160
struct HwChecker {
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   161
    pub id: ClientId,
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   162
    pub is_ready: bool,
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   163
}
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   164
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   165
impl HwChecker {
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   166
    pub fn new(id: ClientId) -> Self {
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   167
        Self {
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   168
            id,
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   169
            is_ready: false,
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   170
        }
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   171
    }
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   172
}
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   173
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
   174
pub struct HwServer {
15520
fd3a20e9d095 move the anteroom out of the server
alfadur <mail@none>
parents: 15519
diff changeset
   175
    clients: IndexSlab<HwClient>,
15522
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   176
    rooms: Slab<HwRoom>,
15532
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   177
    checkers: IndexSlab<HwChecker>,
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   178
    latest_protocol: u16,
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   179
    flags: ServerFlags,
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   180
    greetings: ServerGreetings,
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
   181
}
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
   182
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
   183
impl HwServer {
14780
65861ba8b4e8 move room saves to IO thread
alfadur
parents: 14779
diff changeset
   184
    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
   185
        let rooms = Slab::with_capacity(rooms_limit);
14693
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14686
diff changeset
   186
        let clients = IndexSlab::with_capacity(clients_limit);
15532
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   187
        let checkers = IndexSlab::new();
14694
25c564f77b7d remove lobby room
alfadur <mail@none>
parents: 14693
diff changeset
   188
        Self {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   189
            clients,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   190
            rooms,
15532
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   191
            checkers,
14783
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   192
            greetings: ServerGreetings::new(),
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   193
            latest_protocol: 58,
14786
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   194
            flags: ServerFlags::empty(),
14694
25c564f77b7d remove lobby room
alfadur <mail@none>
parents: 14693
diff changeset
   195
        }
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
   196
    }
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
   197
15441
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   198
    #[inline]
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   199
    pub fn client(&self, client_id: ClientId) -> &HwClient {
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   200
        &self.clients[client_id]
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   201
    }
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   202
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   203
    #[inline]
15519
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
   204
    pub fn has_client(&self, client_id: ClientId) -> bool {
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
   205
        self.clients.contains(client_id)
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
   206
    }
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
   207
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
   208
    #[inline]
15534
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15533
diff changeset
   209
    pub fn iter_clients(&self) -> impl Iterator<Item = &HwClient> + Clone {
15519
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
   210
        self.clients.iter().map(|(_, c)| c)
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
   211
    }
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
   212
b3157d218ae2 disallow mutable clients to leave the server
alfadur <mail@none>
parents: 15518
diff changeset
   213
    #[inline]
15441
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   214
    pub fn room(&self, room_id: RoomId) -> &HwRoom {
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   215
        &self.rooms[room_id]
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   216
    }
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   217
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   218
    #[inline]
15522
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   219
    pub fn get_room(&self, room_id: RoomId) -> Option<&HwRoom> {
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   220
        self.rooms.get(room_id)
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   221
    }
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   222
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   223
    #[inline]
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   224
    fn get_room_mut(&mut self, room_id: RoomId) -> Option<&mut HwRoom> {
15522
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   225
        self.rooms.get_mut(room_id)
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   226
    }
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   227
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   228
    #[inline]
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   229
    pub fn iter_rooms(&self) -> impl Iterator<Item = &HwRoom> {
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   230
        self.rooms.iter().map(|(_, r)| r)
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   231
    }
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   232
4a0b06b03199 make rooms field private
alfadur <mail@none>
parents: 15520
diff changeset
   233
    #[inline]
15482
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
   234
    pub fn client_and_room(&self, client_id: ClientId, room_id: RoomId) -> (&HwClient, &HwRoom) {
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
   235
        (&self.clients[client_id], &self.rooms[room_id])
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
   236
    }
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
   237
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
   238
    #[inline]
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   239
    fn client_and_room_mut(&mut self, client_id: ClientId) -> Option<(&mut HwClient, &mut HwRoom)> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   240
        let client = &mut self.clients[client_id];
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   241
        if let Some(room_id) = client.room_id {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   242
            Some((client, &mut self.rooms[room_id]))
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   243
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   244
            None
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   245
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   246
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   247
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   248
    #[inline]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   249
    pub fn get_room_control(&mut self, client_id: ClientId) -> Option<HwRoomControl> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   250
        HwRoomControl::new(self, client_id)
15482
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
   251
    }
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
   252
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15441
diff changeset
   253
    #[inline]
15441
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   254
    pub fn is_admin(&self, client_id: ClientId) -> bool {
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   255
        self.clients
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   256
            .get(client_id)
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   257
            .map(|c| c.is_admin())
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   258
            .unwrap_or(false)
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   259
    }
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   260
15520
fd3a20e9d095 move the anteroom out of the server
alfadur <mail@none>
parents: 15519
diff changeset
   261
    pub fn add_client(&mut self, client_id: ClientId, data: HwAnteroomClient) {
15532
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   262
        if data.is_checker {
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   263
            self.checkers.insert(client_id, HwChecker::new(client_id));
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15531
diff changeset
   264
        } else if let (Some(protocol), Some(nick)) = (data.protocol_number, data.nick) {
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
   265
            let mut client = HwClient::new(client_id, protocol.get(), nick);
15176
f6115638aa92 enable local admins for non-official servers
alfadur
parents: 15075
diff changeset
   266
            #[cfg(not(feature = "official-server"))]
f6115638aa92 enable local admins for non-official servers
alfadur
parents: 15075
diff changeset
   267
            client.set_is_admin(data.is_local_admin);
f6115638aa92 enable local admins for non-official servers
alfadur
parents: 15075
diff changeset
   268
15441
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   269
            #[cfg(feature = "official-server")]
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   270
            {
15531
ede5f4ec48f3 fix official server
alfadur <mail@none>
parents: 15529
diff changeset
   271
                client.set_is_registered(data.is_registered);
ede5f4ec48f3 fix official server
alfadur <mail@none>
parents: 15529
diff changeset
   272
                client.set_is_admin(data.is_admin);
ede5f4ec48f3 fix official server
alfadur <mail@none>
parents: 15529
diff changeset
   273
                client.set_is_contributor(data.is_contributor);
15441
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   274
            }
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   275
14693
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14686
diff changeset
   276
            self.clients.insert(client_id, client);
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
   277
        }
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents:
diff changeset
   278
    }
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
   279
14673
08a8605bafaf Server action refactoring part 4 of N
alfadur <mail@none>
parents: 14672
diff changeset
   280
    pub fn remove_client(&mut self, client_id: ClientId) {
14696
8a45c90f4580 fix client removal
alfadur
parents: 14695
diff changeset
   281
        self.clients.remove(client_id);
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
   282
    }
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
   283
15441
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   284
    pub fn get_greetings(&self, client: &HwClient) -> &str {
61a0bd0bb021 refactor root handler
alfadur
parents: 15439
diff changeset
   285
        if client.protocol_number < self.latest_protocol {
14783
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   286
            &self.greetings.for_old_protocols
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   287
        } else {
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   288
            &self.greetings.for_latest_protocol
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   289
        }
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   290
    }
b3adc030104b implement server vars
alfadur
parents: 14781
diff changeset
   291
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   292
    #[inline]
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   293
    pub fn create_room(
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   294
        &mut self,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   295
        creator_id: ClientId,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   296
        name: String,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   297
        password: Option<String>,
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   298
    ) -> Result<(&HwClient, &HwRoom), CreateRoomError> {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   299
        use CreateRoomError::*;
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   300
        if utils::is_name_illegal(&name) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   301
            Err(InvalidName)
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   302
        } else if self.has_room(&name) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   303
            Err(AlreadyExists)
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   304
        } else {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   305
            Ok(create_room(
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   306
                &mut self.clients[creator_id],
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   307
                &mut self.rooms,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   308
                name,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   309
                password,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   310
            ))
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   311
        }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   312
    }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   313
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   314
    pub fn join_room(
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   315
        &mut self,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   316
        client_id: ClientId,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   317
        room_id: RoomId,
15533
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15532
diff changeset
   318
        room_password: Option<&str>,
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   319
    ) -> Result<(&HwClient, &HwRoom, impl Iterator<Item = &HwClient> + Clone), JoinRoomError> {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   320
        use JoinRoomError::*;
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   321
        let room = &mut self.rooms[room_id];
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   322
        let client = &mut self.clients[client_id];
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   323
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   324
        if client.protocol_number != room.protocol_number {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   325
            Err(WrongProtocol)
15534
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15533
diff changeset
   326
        } else if room.password.is_some()
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15533
diff changeset
   327
            && room_password != room.password.as_deref()
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15533
diff changeset
   328
            && !client.has_super_power()
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15533
diff changeset
   329
        {
15533
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15532
diff changeset
   330
            Err(WrongPassword)
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   331
        } else if room.is_join_restricted() {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   332
            Err(Restricted)
15534
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15533
diff changeset
   333
        } else if room.is_registration_required() {
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15533
diff changeset
   334
            Err(RegistrationRequired)
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   335
        } else if room.players_number == u8::max_value() {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   336
            Err(Full)
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   337
        } else {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   338
            move_to_room(client, room);
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   339
            let room_id = room.id;
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   340
            Ok((
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   341
                &self.clients[client_id],
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   342
                &self.rooms[room_id],
15534
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15533
diff changeset
   343
                self.iter_clients()
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15533
diff changeset
   344
                    .filter(move |c| c.room_id == Some(room_id)),
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   345
            ))
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   346
        }
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   347
    }
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   348
14671
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
   349
    #[inline]
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   350
    pub fn join_room_by_name(
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   351
        &mut self,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   352
        client_id: ClientId,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   353
        room_name: &str,
15533
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15532
diff changeset
   354
        room_password: Option<&str>,
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   355
    ) -> Result<(&HwClient, &HwRoom, impl Iterator<Item = &HwClient> + Clone), JoinRoomError> {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   356
        use JoinRoomError::*;
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   357
        let room = self.rooms.iter().find(|(_, r)| r.name == room_name);
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   358
        if let Some((_, room)) = room {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   359
            let room_id = room.id;
15533
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15532
diff changeset
   360
            self.join_room(client_id, room_id, room_password)
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   361
        } else {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   362
            Err(DoesntExist)
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   363
        }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   364
    }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   365
15518
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
   366
    pub fn enable_super_power(&mut self, client_id: ClientId) -> bool {
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
   367
        let client = &mut self.clients[client_id];
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
   368
        if client.is_admin() {
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
   369
            client.set_has_super_power(true);
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
   370
        }
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
   371
        client.is_admin()
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
   372
    }
e705d30e0f10 some more room handler cleanup
alfadur <mail@none>
parents: 15517
diff changeset
   373
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   374
    #[inline]
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   375
    pub fn set_var(&mut self, client_id: ClientId, var: ServerVar) -> Result<(), AccessError> {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   376
        if self.clients[client_id].is_admin() {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   377
            match var {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   378
                ServerVar::MOTDNew(msg) => self.greetings.for_latest_protocol = msg,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   379
                ServerVar::MOTDOld(msg) => self.greetings.for_old_protocols = msg,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   380
                ServerVar::LatestProto(n) => self.latest_protocol = n,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   381
            }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   382
            Ok(())
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   383
        } else {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   384
            Err(AccessError())
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   385
        }
14671
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
   386
    }
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
   387
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   388
    #[inline]
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   389
    pub fn get_vars(&self, client_id: ClientId) -> Result<[ServerVar; 3], AccessError> {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   390
        if self.clients[client_id].is_admin() {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   391
            Ok([
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   392
                ServerVar::MOTDNew(self.greetings.for_latest_protocol.clone()),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   393
                ServerVar::MOTDOld(self.greetings.for_old_protocols.clone()),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   394
                ServerVar::LatestProto(self.latest_protocol),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   395
            ])
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   396
        } else {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   397
            Err(AccessError())
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   398
        }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   399
    }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   400
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   401
    pub fn get_used_protocols(&self, client_id: ClientId) -> Result<Vec<u16>, AccessError> {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   402
        if self.clients[client_id].is_admin() {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   403
            let mut protocols: HashSet<_> = self
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   404
                .clients
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   405
                .iter()
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   406
                .map(|(_, c)| c.protocol_number)
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   407
                .chain(self.rooms.iter().map(|(_, r)| r.protocol_number))
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   408
                .collect();
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   409
            let mut protocols: Vec<_> = protocols.drain().collect();
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   410
            protocols.sort();
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   411
            Ok(protocols)
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   412
        } else {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   413
            Err(AccessError())
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   414
        }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   415
    }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   416
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   417
    #[inline]
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   418
    pub fn has_room(&self, name: &str) -> bool {
14697
f64e21f164a5 fix some warnings
alfadur
parents: 14696
diff changeset
   419
        self.find_room(name).is_some()
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   420
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   421
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
   422
    #[inline]
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
   423
    pub fn find_room(&self, name: &str) -> Option<&HwRoom> {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   424
        self.rooms
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   425
            .iter()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   426
            .find_map(|(_, r)| Some(r).filter(|r| r.name == name))
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   427
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   428
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   429
    fn find_room_mut(&mut self, name: &str) -> Option<&mut HwRoom> {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   430
        self.rooms
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   431
            .iter_mut()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   432
            .find_map(|(_, r)| Some(r).filter(|r| r.name == name))
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   433
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   434
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
   435
    pub fn find_client(&self, nick: &str) -> Option<&HwClient> {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   436
        self.clients
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   437
            .iter()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   438
            .find_map(|(_, c)| Some(c).filter(|c| c.nick == nick))
13478
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   439
    }
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   440
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   441
    fn find_client_mut(&mut self, nick: &str) -> Option<&mut HwClient> {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   442
        self.clients
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   443
            .iter_mut()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   444
            .find_map(|(_, c)| Some(c).filter(|c| c.nick == nick))
13478
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   445
    }
d79795acaa73 Mostly implement voting
alfadur
parents: 13445
diff changeset
   446
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   447
    pub fn iter_client_ids(&self) -> impl Iterator<Item = ClientId> + '_ {
14789
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   448
        self.clients.iter().map(|(id, _)| id)
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   449
    }
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   450
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   451
    pub fn filter_clients<'a, F>(&'a self, f: F) -> impl Iterator<Item = ClientId> + 'a
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   452
    where
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
   453
        F: Fn(&(usize, &HwClient)) -> bool + 'a,
14789
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   454
    {
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   455
        self.clients.iter().filter(f).map(|(_, c)| c.id)
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   456
    }
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   457
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   458
    pub fn filter_rooms<'a, F>(&'a self, f: F) -> impl Iterator<Item = RoomId> + 'a
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   459
    where
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
   460
        F: Fn(&(usize, &HwRoom)) -> bool + 'a,
14789
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   461
    {
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   462
        self.rooms.iter().filter(f).map(|(_, c)| c.id)
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   463
    }
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   464
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   465
    pub fn collect_client_ids<F>(&self, f: F) -> Vec<ClientId>
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   466
    where
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
   467
        F: Fn(&(usize, &HwClient)) -> bool,
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   468
    {
14789
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   469
        self.filter_clients(f).collect()
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   470
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   471
14781
01f8ab45f806 fix lobby joining
alfadur
parents: 14780
diff changeset
   472
    pub fn collect_nicks<F>(&self, f: F) -> Vec<String>
01f8ab45f806 fix lobby joining
alfadur
parents: 14780
diff changeset
   473
    where
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
   474
        F: Fn(&(usize, &HwClient)) -> bool,
14781
01f8ab45f806 fix lobby joining
alfadur
parents: 14780
diff changeset
   475
    {
01f8ab45f806 fix lobby joining
alfadur
parents: 14780
diff changeset
   476
        self.clients
01f8ab45f806 fix lobby joining
alfadur
parents: 14780
diff changeset
   477
            .iter()
01f8ab45f806 fix lobby joining
alfadur
parents: 14780
diff changeset
   478
            .filter(f)
01f8ab45f806 fix lobby joining
alfadur
parents: 14780
diff changeset
   479
            .map(|(_, c)| c.nick.clone())
01f8ab45f806 fix lobby joining
alfadur
parents: 14780
diff changeset
   480
            .collect()
14694
25c564f77b7d remove lobby room
alfadur <mail@none>
parents: 14693
diff changeset
   481
    }
25c564f77b7d remove lobby room
alfadur <mail@none>
parents: 14693
diff changeset
   482
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   483
    pub fn lobby_client_ids(&self) -> impl Iterator<Item = ClientId> + '_ {
14789
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   484
        self.filter_clients(|(_, c)| c.room_id == None)
14781
01f8ab45f806 fix lobby joining
alfadur
parents: 14780
diff changeset
   485
    }
01f8ab45f806 fix lobby joining
alfadur
parents: 14780
diff changeset
   486
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   487
    pub fn room_client_ids(&self, room_id: RoomId) -> impl Iterator<Item = ClientId> + '_ {
14789
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   488
        self.filter_clients(move |(_, c)| c.room_id == Some(room_id))
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   489
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   490
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   491
    pub fn protocol_client_ids(&self, protocol: u16) -> impl Iterator<Item = ClientId> + '_ {
14789
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   492
        self.filter_clients(move |(_, c)| c.protocol_number == protocol)
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   493
    }
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   494
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   495
    pub fn protocol_room_ids(&self, protocol: u16) -> impl Iterator<Item = RoomId> + '_ {
14789
18240b308505 implement stats message
alfadur
parents: 14786
diff changeset
   496
        self.filter_rooms(move |(_, r)| r.protocol_number == protocol)
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   497
    }
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   498
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   499
    pub fn other_client_ids_in_room(&self, self_id: ClientId) -> Vec<ClientId> {
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   500
        let room_id = self.clients[self_id].room_id;
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   501
        self.collect_client_ids(|(id, c)| *id != self_id && c.room_id == room_id)
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   502
    }
14786
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   503
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   504
    pub fn is_registered_only(&self) -> bool {
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   505
        self.flags.contains(ServerFlags::REGISTERED_ONLY)
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   506
    }
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   507
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   508
    pub fn set_is_registered_only(&mut self, value: bool) {
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   509
        self.flags.set(ServerFlags::REGISTERED_ONLY, value)
8ecdb5c6bb2a implement info, registered only & super power messages
alfadur
parents: 14783
diff changeset
   510
    }
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   511
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   512
    pub fn set_room_saves(&mut self, room_id: RoomId, text: &str) -> Result<(), serde_yaml::Error> {
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   513
        if let Some(room) = self.rooms.get_mut(room_id) {
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   514
            room.set_saves(text)
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   515
        } else {
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   516
            Ok(())
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   517
        }
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   518
    }
13445
d3c86ade3d4d Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents: 13442
diff changeset
   519
}
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
   520
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   521
pub struct HwRoomControl<'a> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   522
    server: &'a mut HwServer,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   523
    client_id: ClientId,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   524
    room_id: RoomId,
15552
0031683bfa76 don't apply voted 🦔 to the in-game teams
alfadur
parents: 15551
diff changeset
   525
    is_room_removed: bool,
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   526
}
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   527
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   528
impl<'a> HwRoomControl<'a> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   529
    #[inline]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   530
    pub fn new(server: &'a mut HwServer, client_id: ClientId) -> Option<Self> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   531
        if let Some(room_id) = server.clients[client_id].room_id {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   532
            Some(Self {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   533
                server,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   534
                client_id,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   535
                room_id,
15552
0031683bfa76 don't apply voted 🦔 to the in-game teams
alfadur
parents: 15551
diff changeset
   536
                is_room_removed: false,
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   537
            })
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   538
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   539
            None
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   540
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   541
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   542
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   543
    #[inline]
15551
7478568cffbe fix phantom rooms
alfadur
parents: 15541
diff changeset
   544
    pub fn cleanup_room(self) {
7478568cffbe fix phantom rooms
alfadur
parents: 15541
diff changeset
   545
        if self.is_room_removed {
7478568cffbe fix phantom rooms
alfadur
parents: 15541
diff changeset
   546
            self.server.rooms.remove(self.room_id);
7478568cffbe fix phantom rooms
alfadur
parents: 15541
diff changeset
   547
        }
7478568cffbe fix phantom rooms
alfadur
parents: 15541
diff changeset
   548
    }
7478568cffbe fix phantom rooms
alfadur
parents: 15541
diff changeset
   549
7478568cffbe fix phantom rooms
alfadur
parents: 15541
diff changeset
   550
    #[inline]
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   551
    pub fn server(&self) -> &HwServer {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   552
        self.server
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   553
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   554
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   555
    #[inline]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   556
    pub fn client(&self) -> &HwClient {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   557
        &self.server.clients[self.client_id]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   558
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   559
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   560
    #[inline]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   561
    fn client_mut(&mut self) -> &mut HwClient {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   562
        &mut self.server.clients[self.client_id]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   563
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   564
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   565
    #[inline]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   566
    pub fn room(&self) -> &HwRoom {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   567
        &self.server.rooms[self.room_id]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   568
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   569
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   570
    #[inline]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   571
    fn room_mut(&mut self) -> &mut HwRoom {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   572
        &mut self.server.rooms[self.room_id]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   573
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   574
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   575
    #[inline]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   576
    pub fn get(&self) -> (&HwClient, &HwRoom) {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   577
        (self.client(), self.room())
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   578
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   579
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   580
    #[inline]
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   581
    fn get_mut(&mut self) -> (&mut HwClient, &mut HwRoom) {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   582
        (
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   583
            &mut self.server.clients[self.client_id],
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   584
            &mut self.server.rooms[self.room_id],
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   585
        )
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   586
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   587
15525
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   588
    pub fn change_client<'b: 'a>(self, client_id: ClientId) -> Option<HwRoomControl<'a>> {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   589
        let room_id = self.room_id;
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   590
        HwRoomControl::new(self.server, client_id).filter(|c| c.room_id == room_id)
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   591
    }
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   592
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   593
    pub fn leave_room(&mut self) -> LeaveRoomResult {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   594
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   595
        room.players_number -= 1;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   596
        client.room_id = None;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   597
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   598
        let is_empty = room.players_number == 0;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   599
        let is_fixed = room.is_fixed();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   600
        let was_master = room.master_id == Some(client.id);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   601
        let was_in_game = client.is_in_game();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   602
        let mut removed_teams = vec![];
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   603
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   604
        if is_empty && !is_fixed {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   605
            if client.is_ready() && room.ready_players_number > 0 {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   606
                room.ready_players_number -= 1;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   607
            }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   608
15569
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   609
            if let Some(ref mut info) = room.game_info {
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   610
                removed_teams = info
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   611
                    .client_teams(client.id)
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   612
                    .map(|t| t.name.clone())
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   613
                    .collect();
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   614
                info.mark_left_teams(removed_teams.iter());
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   615
            } else {
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   616
                removed_teams = room
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   617
                    .client_teams(client.id)
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   618
                    .map(|t| t.name.clone())
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   619
                    .collect();
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   620
                for team_name in &removed_teams {
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   621
                    room.remove_team(team_name);
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   622
                }
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   623
            }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   624
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   625
            if client.is_master() && !is_fixed {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   626
                client.set_is_master(false);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   627
                room.master_id = None;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   628
            }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   629
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   630
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   631
        client.set_is_ready(false);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   632
        client.set_is_in_game(false);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   633
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   634
        if !is_fixed {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   635
            if room.players_number == 0 {
15551
7478568cffbe fix phantom rooms
alfadur
parents: 15541
diff changeset
   636
                self.is_room_removed = true
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   637
            } else if room.master_id == None {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   638
                let protocol_number = room.protocol_number;
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   639
                let new_master_id = self.server.room_client_ids(self.room_id).next();
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   640
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   641
                if let Some(new_master_id) = new_master_id {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   642
                    let room = self.room_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   643
                    room.master_id = Some(new_master_id);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   644
                    let new_master = &mut self.server.clients[new_master_id];
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   645
                    new_master.set_is_master(true);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   646
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   647
                    if protocol_number < 42 {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   648
                        let nick = new_master.nick.clone();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   649
                        self.room_mut().name = nick;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   650
                    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   651
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   652
                    let room = self.room_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   653
                    room.set_join_restriction(false);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   654
                    room.set_team_add_restriction(false);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   655
                    room.set_unregistered_players_restriction(true);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   656
                }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   657
            }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   658
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   659
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   660
        if is_empty && !is_fixed {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   661
            LeaveRoomResult::RoomRemoved
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   662
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   663
            LeaveRoomResult::RoomRemains {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   664
                is_empty,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   665
                was_master,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   666
                was_in_game,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   667
                new_master: self.room().master_id,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   668
                removed_teams,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   669
            }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   670
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   671
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   672
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   673
    pub fn change_master(
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   674
        &mut self,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   675
        new_master_nick: String,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   676
    ) -> Result<ChangeMasterResult, ChangeMasterError> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   677
        use ChangeMasterError::*;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   678
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   679
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   680
        if client.is_admin() || room.master_id == Some(client.id) {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   681
            let new_master_id = self
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   682
                .server
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   683
                .clients
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   684
                .iter()
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   685
                .find(|(_, c)| c.nick == new_master_nick)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   686
                .map(|(id, _)| id);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   687
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   688
            match new_master_id {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   689
                Some(new_master_id) if new_master_id == self.client_id => Err(AlreadyMaster),
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   690
                Some(new_master_id) => {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   691
                    let new_master = &mut self.server.clients[new_master_id];
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   692
                    if new_master.room_id == Some(self.room_id) {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   693
                        self.server.clients[new_master_id].set_is_master(true);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   694
                        let old_master_id = self.room().master_id;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   695
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   696
                        if let Some(master_id) = old_master_id {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   697
                            self.server.clients[master_id].set_is_master(false);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   698
                        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   699
                        self.room_mut().master_id = Some(new_master_id);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   700
                        Ok(ChangeMasterResult {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   701
                            old_master_id,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   702
                            new_master_id,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   703
                        })
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   704
                    } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   705
                        Err(ClientNotInRoom)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   706
                    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   707
                }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   708
                None => Err(NoClient),
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   709
            }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   710
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   711
            Err(NoAccess)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   712
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   713
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   714
15525
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   715
    pub fn start_vote(&mut self, kind: VoteType) -> Result<(), StartVoteError> {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   716
        use StartVoteError::*;
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   717
        match self.room().voting {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   718
            Some(_) => Err(VotingInProgress),
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   719
            None => {
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15525
diff changeset
   720
                let voting = Voting::new(kind, self.server.room_client_ids(self.room_id).collect());
15525
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   721
                self.room_mut().voting = Some(voting);
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   722
                Ok(())
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   723
            }
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   724
        }
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   725
    }
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   726
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   727
    pub fn vote(&mut self, vote: Vote) -> Result<VoteResult, VoteError> {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   728
        use self::{VoteError::*, VoteResult::*};
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   729
        let client_id = self.client_id;
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   730
        if let Some(ref mut voting) = self.room_mut().voting {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   731
            if vote.is_forced || voting.votes.iter().all(|(id, _)| client_id != *id) {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   732
                voting.votes.push((client_id, vote.is_pro));
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   733
                let i = voting.votes.iter();
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   734
                let pro = i.clone().filter(|(_, v)| *v).count();
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   735
                let contra = i.filter(|(_, v)| !*v).count();
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   736
                let success_quota = voting.voters.len() / 2 + 1;
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   737
                if vote.is_forced && vote.is_pro || pro >= success_quota {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   738
                    let voting = self.room_mut().voting.take().unwrap();
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   739
                    Ok(Succeeded(voting.kind))
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   740
                } else if vote.is_forced && !vote.is_pro
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   741
                    || contra > voting.voters.len() - success_quota
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   742
                {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   743
                    Ok(Failed)
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   744
                } else {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   745
                    Ok(Submitted)
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   746
                }
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   747
            } else {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   748
                Err(AlreadyVoted)
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   749
            }
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   750
        } else {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   751
            Err(NoVoting)
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   752
        }
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   753
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   754
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   755
    pub fn toggle_flag(&mut self, flags: super::room::RoomFlags) -> bool {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   756
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   757
        if client.is_master() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   758
            room.flags.toggle(flags);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   759
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   760
        client.is_master()
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   761
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   762
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   763
    pub fn fix_room(&mut self) -> Result<(), AccessError> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   764
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   765
        if client.is_admin() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   766
            room.set_is_fixed(true);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   767
            room.set_join_restriction(false);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   768
            room.set_team_add_restriction(false);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   769
            room.set_unregistered_players_restriction(true);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   770
            Ok(())
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   771
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   772
            Err(AccessError())
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   773
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   774
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   775
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   776
    pub fn unfix_room(&mut self) -> Result<(), AccessError> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   777
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   778
        if client.is_admin() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   779
            room.set_is_fixed(false);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   780
            Ok(())
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   781
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   782
            Err(AccessError())
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   783
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   784
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   785
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   786
    pub fn set_room_name(&mut self, mut name: String) -> Result<String, ModifyRoomNameError> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   787
        use ModifyRoomNameError::*;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   788
        let room_exists = self.server.has_room(&name);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   789
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   790
        if room.is_fixed() || room.master_id != Some(client.id) {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   791
            Err(AccessDenied)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   792
        } else if utils::is_name_illegal(&name) {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   793
            Err(InvalidName)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   794
        } else if room_exists {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   795
            Err(DuplicateName)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   796
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   797
            std::mem::swap(&mut room.name, &mut name);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   798
            Ok(name)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   799
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   800
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   801
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   802
    pub fn set_room_greeting(&mut self, greeting: Option<String>) -> Result<(), AccessError> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   803
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   804
        if client.is_admin() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   805
            room.greeting = greeting.unwrap_or(String::new());
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   806
            Ok(())
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   807
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   808
            Err(AccessError())
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   809
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   810
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   811
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   812
    pub fn set_room_max_teams(&mut self, count: u8) -> Result<(), SetTeamCountError> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   813
        use SetTeamCountError::*;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   814
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   815
        if !client.is_master() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   816
            Err(NotMaster)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   817
        } else if !(2..=super::room::MAX_TEAMS_IN_ROOM).contains(&count) {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   818
            Err(InvalidNumber)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   819
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   820
            room.max_teams = count;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   821
            Ok(())
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   822
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   823
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   824
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   825
    pub fn set_team_hedgehogs_number(
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   826
        &mut self,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   827
        team_name: &str,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   828
        number: u8,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   829
    ) -> Result<(), SetHedgehogsError> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   830
        use SetHedgehogsError::*;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   831
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   832
        let addable_hedgehogs = room.addable_hedgehogs();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   833
        if let Some((_, team)) = room.find_team_and_owner_mut(|t| t.name == team_name) {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   834
            let max_hedgehogs = min(
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   835
                super::room::MAX_HEDGEHOGS_IN_ROOM,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   836
                addable_hedgehogs + team.hedgehogs_number,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   837
            );
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   838
            if !client.is_master() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   839
                Err(NotMaster)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   840
            } else if !(1..=max_hedgehogs).contains(&number) {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   841
                Err(InvalidNumber(team.hedgehogs_number))
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   842
            } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   843
                team.hedgehogs_number = number;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   844
                Ok(())
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   845
            }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   846
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   847
            Err(NoTeam)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   848
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   849
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   850
15525
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   851
    pub fn set_hedgehogs_number(&mut self, number: u8) -> Vec<String> {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   852
        self.room_mut().set_hedgehogs_number(number)
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   853
    }
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   854
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   855
    pub fn add_team(&mut self, mut info: Box<TeamInfo>) -> Result<&TeamInfo, AddTeamError> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   856
        use AddTeamError::*;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   857
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   858
        if room.teams.len() >= room.max_teams as usize {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   859
            Err(TooManyTeams)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   860
        } else if room.addable_hedgehogs() == 0 {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   861
            Err(TooManyHedgehogs)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   862
        } else if room.find_team(|t| t.name == info.name) != None {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   863
            Err(TeamAlreadyExists)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   864
        } else if room.game_info.is_some() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   865
            Err(GameInProgress)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   866
        } else if room.is_team_add_restricted() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   867
            Err(Restricted)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   868
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   869
            info.owner = client.nick.clone();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   870
            let team = room.add_team(client.id, *info, client.protocol_number < 42);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   871
            client.teams_in_game += 1;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   872
            client.clan = Some(team.color);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   873
            Ok(team)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   874
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   875
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   876
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   877
    pub fn remove_team(&mut self, team_name: &str) -> Result<(), RemoveTeamError> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   878
        use RemoveTeamError::*;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   879
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   880
        match room.find_team_owner(team_name) {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   881
            None => Err(NoTeam),
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   882
            Some((id, _)) if id != client.id => Err(RemoveTeamError::TeamNotOwned),
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   883
            Some(_) => {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   884
                client.teams_in_game -= 1;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   885
                client.clan = room.find_team_color(client.id);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   886
                room.remove_team(team_name);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   887
                Ok(())
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   888
            }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   889
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   890
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   891
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   892
    pub fn set_team_color(&mut self, team_name: &str, color: u8) -> Result<(), ModifyTeamError> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   893
        use ModifyTeamError::*;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   894
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   895
        if let Some((owner, team)) = room.find_team_and_owner_mut(|t| t.name == team_name) {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   896
            if !client.is_master() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   897
                Err(NotMaster)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   898
            } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   899
                team.color = color;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   900
                self.server.clients[owner].clan = Some(color);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   901
                Ok(())
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   902
            }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   903
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   904
            Err(NoTeam)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   905
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   906
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   907
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   908
    pub fn set_config(&mut self, cfg: GameCfg) -> Result<(), SetConfigError> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   909
        use SetConfigError::*;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   910
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   911
        if room.is_fixed() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   912
            Err(RoomFixed)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   913
        } else if !client.is_master() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   914
            Err(NotMaster)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   915
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   916
            let cfg = match cfg {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   917
                GameCfg::Scheme(name, mut values) => {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   918
                    if client.protocol_number == 49 && values.len() >= 2 {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   919
                        let mut s = "X".repeat(50);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   920
                        s.push_str(&values.pop().unwrap());
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   921
                        values.push(s);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   922
                    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   923
                    GameCfg::Scheme(name, values)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   924
                }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   925
                cfg => cfg,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   926
            };
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   927
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   928
            room.set_config(cfg);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   929
            Ok(())
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   930
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   931
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   932
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   933
    pub fn save_config(&mut self, name: String, location: String) {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   934
        self.room_mut().save_config(name, location);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   935
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   936
15525
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   937
    pub fn load_config(&mut self, name: &str) -> Option<&str> {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   938
        self.room_mut().load_config(name)
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   939
    }
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   940
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   941
    pub fn delete_config(&mut self, name: &str) -> bool {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   942
        self.room_mut().delete_config(name)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   943
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   944
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   945
    pub fn toggle_ready(&mut self) -> bool {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   946
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   947
        client.set_is_ready(!client.is_ready());
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   948
        if client.is_ready() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   949
            room.ready_players_number += 1;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   950
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   951
            room.ready_players_number -= 1;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   952
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   953
        client.is_ready()
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   954
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   955
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   956
    pub fn start_game(&mut self) -> Result<Vec<String>, StartGameError> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   957
        use StartGameError::*;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   958
        let (room_clients, room_nicks): (Vec<_>, Vec<_>) = self
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   959
            .server
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   960
            .clients
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   961
            .iter()
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   962
            .map(|(id, c)| (id, c.nick.clone()))
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   963
            .unzip();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   964
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   965
        let room = self.room_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   966
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   967
        if !room.has_multiple_clans() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   968
            Err(NotEnoughClans)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   969
        } else if room.protocol_number <= 43 && room.players_number != room.ready_players_number {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   970
            Err(NotReady)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   971
        } else if room.game_info.is_some() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   972
            Err(AlreadyInGame)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   973
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   974
            room.start_round();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   975
            for id in room_clients {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   976
                let team_indices = self.room().client_team_indices(id);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   977
                let c = &mut self.server.clients[id];
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   978
                c.set_is_in_game(true);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   979
                c.team_indices = team_indices;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   980
            }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   981
            Ok(room_nicks)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   982
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   983
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   984
15525
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   985
    pub fn toggle_pause(&mut self) -> bool {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   986
        if let Some(ref mut info) = self.room_mut().game_info {
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   987
            info.is_paused = !info.is_paused;
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   988
        }
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   989
        self.room_mut().game_info.is_some()
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   990
    }
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15523
diff changeset
   991
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   992
    pub fn leave_game(&mut self) -> Option<Vec<String>> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   993
        let (client, room) = self.get_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   994
        let client_left = client.is_in_game();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   995
        if client_left {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   996
            client.set_is_in_game(false);
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
   997
15553
852682c2003f fix in-game team handling
alfadur
parents: 15552
diff changeset
   998
            if let Some(ref mut info) = room.game_info {
852682c2003f fix in-game team handling
alfadur
parents: 15552
diff changeset
   999
                let team_names: Vec<_> = info
852682c2003f fix in-game team handling
alfadur
parents: 15552
diff changeset
  1000
                    .client_teams(client.id)
852682c2003f fix in-game team handling
alfadur
parents: 15552
diff changeset
  1001
                    .map(|t| t.name.clone())
852682c2003f fix in-game team handling
alfadur
parents: 15552
diff changeset
  1002
                    .collect();
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1003
15569
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
  1004
                info.mark_left_teams(team_names.iter());
15553
852682c2003f fix in-game team handling
alfadur
parents: 15552
diff changeset
  1005
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1006
                Some(team_names)
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1007
            } else {
15553
852682c2003f fix in-game team handling
alfadur
parents: 15552
diff changeset
  1008
                None
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1009
            }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1010
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1011
            None
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1012
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1013
    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1014
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1015
    pub fn end_game(&mut self) -> Option<EndGameResult> {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1016
        let room = self.room_mut();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1017
        room.ready_players_number = room.master_id.is_some() as u8;
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1018
15541
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
  1019
        if let Some(mut info) = replace(&mut room.game_info, None) {
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1020
            let room_id = room.id;
15541
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
  1021
            for team_name in &info.left_teams {
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
  1022
                room.remove_team(team_name);
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
  1023
            }
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
  1024
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1025
            let unreadied_nicks: Vec<_> = self
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1026
                .server
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1027
                .clients
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1028
                .iter_mut()
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1029
                .filter(|(_, c)| c.room_id == Some(room_id))
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1030
                .map(|(_, c)| {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1031
                    c.set_is_ready(c.is_master());
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1032
                    c
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1033
                })
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1034
                .filter_map(|c| {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1035
                    if !c.is_master() {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1036
                        Some(c.nick.clone())
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1037
                    } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1038
                        None
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1039
                    }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1040
                })
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1041
                .collect();
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1042
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1043
            Some(EndGameResult {
15541
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
  1044
                left_teams: replace(&mut info.left_teams, vec![]),
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1045
                unreadied_nicks,
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1046
            })
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1047
        } else {
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1048
            None
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1049
        }
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1050
    }
15527
428a0e7da27b restore engine message handler
alfadur <mail@none>
parents: 15526
diff changeset
  1051
428a0e7da27b restore engine message handler
alfadur <mail@none>
parents: 15526
diff changeset
  1052
    pub fn log_engine_msg(&mut self, log_msg: String, sync_msg: Option<Option<String>>) {
428a0e7da27b restore engine message handler
alfadur <mail@none>
parents: 15526
diff changeset
  1053
        if let Some(ref mut info) = self.room_mut().game_info {
428a0e7da27b restore engine message handler
alfadur <mail@none>
parents: 15526
diff changeset
  1054
            if !log_msg.is_empty() {
428a0e7da27b restore engine message handler
alfadur <mail@none>
parents: 15526
diff changeset
  1055
                info.msg_log.push(log_msg);
428a0e7da27b restore engine message handler
alfadur <mail@none>
parents: 15526
diff changeset
  1056
            }
428a0e7da27b restore engine message handler
alfadur <mail@none>
parents: 15526
diff changeset
  1057
            if let Some(msg) = sync_msg {
428a0e7da27b restore engine message handler
alfadur <mail@none>
parents: 15526
diff changeset
  1058
                info.sync_msg = msg;
428a0e7da27b restore engine message handler
alfadur <mail@none>
parents: 15526
diff changeset
  1059
            }
428a0e7da27b restore engine message handler
alfadur <mail@none>
parents: 15526
diff changeset
  1060
        }
428a0e7da27b restore engine message handler
alfadur <mail@none>
parents: 15526
diff changeset
  1061
    }
15523
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1062
}
f4f6060b536c add a separate interface for modifying room state
alfadur <mail@none>
parents: 15522
diff changeset
  1063
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
  1064
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
  1065
    let entry = rooms.vacant_entry();
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
  1066
    let room = HwRoom::new(entry.key());
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1067
    entry.insert(room)
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1068
}
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1069
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
  1070
fn create_room<'a, 'b>(
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
  1071
    client: &'a mut HwClient,
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
  1072
    rooms: &'b mut Slab<HwRoom>,
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1073
    name: String,
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1074
    password: Option<String>,
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
  1075
) -> (&'a HwClient, &'b HwRoom) {
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1076
    let room = allocate_room(rooms);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1077
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1078
    room.master_id = Some(client.id);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1079
    room.name = name;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1080
    room.password = password;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1081
    room.protocol_number = client.protocol_number;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1082
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1083
    room.players_number = 1;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1084
    room.ready_players_number = 1;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1085
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1086
    client.room_id = Some(room.id);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1087
    client.set_is_master(true);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1088
    client.set_is_ready(true);
15533
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15532
diff changeset
  1089
    client.set_is_in_game(false);
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15532
diff changeset
  1090
    client.clan = None;
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15532
diff changeset
  1091
    client.teams_in_game = 0;
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15532
diff changeset
  1092
    client.team_indices = vec![];
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1093
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15176
diff changeset
  1094
    (client, room)
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1095
}
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1096
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
  1097
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
  1098
    debug_assert!(client.room_id != Some(room.id));
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1099
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1100
    room.players_number += 1;
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1101
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1102
    client.room_id = Some(room.id);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1103
    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
  1104
15540
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
  1105
    if let Some(ref mut info) = room.game_info {
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
  1106
        let teams = info.client_teams(client.id);
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
  1107
        client.teams_in_game = teams.clone().count() as u8;
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
  1108
        client.clan = teams.clone().next().map(|t| t.color);
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
  1109
        let team_names: Vec<_> = teams.map(|t| t.name.clone()).collect();
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1110
15540
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
  1111
        if !team_names.is_empty() {
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
  1112
            info.left_teams.retain(|name| !team_names.contains(&name));
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1113
        }
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1114
    }
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
  1115
}