rust/hedgewars-server/src/core/room.rs
author Wuzzy <Wuzzy@disroot.org>
Wed, 28 Jun 2023 18:57:00 +0200 (18 months ago)
changeset 15967 f362e82cddcf
parent 15804 747278149393
child 16000 d9f1b239b6d7
permissions -rw-r--r--
CS: Disable game flags incompatible with this script
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
15074
c5a6e8566425 shuffle server files
alfadur
parents: 14788
diff changeset
     1
use super::{
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
     2
    client::HwClient,
15804
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15569
diff changeset
     3
    types::{ClientId, RoomId, Voting},
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
     4
};
13805
0463a4221327 cleanup crate imports
alfadur
parents: 13801
diff changeset
     5
use bitflags::*;
15804
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15569
diff changeset
     6
use hedgewars_network_protocol::types::{
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15569
diff changeset
     7
    GameCfg, GameCfg::*, RoomConfig, TeamInfo, MAX_HEDGEHOGS_PER_TEAM,
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15569
diff changeset
     8
};
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     9
use serde::{Deserialize, Serialize};
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    10
use serde_derive::{Deserialize, Serialize};
13529
662f7df89d06 Implement room config export
alfadur
parents: 13528
diff changeset
    11
use serde_yaml;
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    12
use std::{collections::HashMap, iter};
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
    13
14788
6dea1ca64992 implement maxteams and teamchat messages
alfadur
parents: 14785
diff changeset
    14
pub const MAX_TEAMS_IN_ROOM: u8 = 8;
15525
16d3c9acd715 restore voting
alfadur <mail@none>
parents: 15482
diff changeset
    15
pub const MAX_HEDGEHOGS_IN_ROOM: u8 = MAX_TEAMS_IN_ROOM * MAX_HEDGEHOGS_PER_TEAM;
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
    16
15540
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
    17
fn client_teams_impl(
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
    18
    teams: &[(ClientId, TeamInfo)],
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
    19
    client_id: ClientId,
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
    20
) -> impl Iterator<Item = &TeamInfo> + Clone {
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
    21
    teams
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
    22
        .iter()
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
    23
        .filter(move |(id, _)| *id == client_id)
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
    24
        .map(|(_, t)| t)
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
    25
}
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
    26
13423
87a6cad20c90 Implement game start & engine messages
alfadur
parents: 13422
diff changeset
    27
pub struct GameInfo {
15541
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
    28
    pub original_teams: Vec<(ClientId, TeamInfo)>,
13427
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13426
diff changeset
    29
    pub left_teams: Vec<String>,
13428
c8425fbcf1d9 Fix spectators joining midgame
alfadur
parents: 13427
diff changeset
    30
    pub msg_log: Vec<String>,
13443
2501428303a2 Fix team remove synchronization
alfadur
parents: 13428
diff changeset
    31
    pub sync_msg: Option<String>,
13427
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13426
diff changeset
    32
    pub is_paused: bool,
15541
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
    33
    original_config: RoomConfig,
13426
f091f69d59e4 Additional round cleanup
alfadur
parents: 13423
diff changeset
    34
}
f091f69d59e4 Additional round cleanup
alfadur
parents: 13423
diff changeset
    35
f091f69d59e4 Additional round cleanup
alfadur
parents: 13423
diff changeset
    36
impl GameInfo {
13427
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13426
diff changeset
    37
    fn new(teams: Vec<(ClientId, TeamInfo)>, config: RoomConfig) -> GameInfo {
13426
f091f69d59e4 Additional round cleanup
alfadur
parents: 13423
diff changeset
    38
        GameInfo {
13427
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13426
diff changeset
    39
            left_teams: Vec::new(),
13428
c8425fbcf1d9 Fix spectators joining midgame
alfadur
parents: 13427
diff changeset
    40
            msg_log: Vec::new(),
13443
2501428303a2 Fix team remove synchronization
alfadur
parents: 13428
diff changeset
    41
            sync_msg: None,
13427
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13426
diff changeset
    42
            is_paused: false,
15541
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
    43
            original_teams: teams,
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
    44
            original_config: config,
13426
f091f69d59e4 Additional round cleanup
alfadur
parents: 13423
diff changeset
    45
        }
f091f69d59e4 Additional round cleanup
alfadur
parents: 13423
diff changeset
    46
    }
15540
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
    47
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
    48
    pub fn client_teams(&self, client_id: ClientId) -> impl Iterator<Item = &TeamInfo> + Clone {
15541
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
    49
        client_teams_impl(&self.original_teams, client_id)
15540
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
    50
    }
15569
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    51
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    52
    pub fn mark_left_teams<'a, I>(&mut self, team_names: I)
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    53
    where
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    54
        I: Iterator<Item = &'a String>,
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    55
    {
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    56
        if let Some(m) = &self.sync_msg {
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    57
            self.msg_log.push(m.clone());
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    58
            self.sync_msg = None
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    59
        }
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    60
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    61
        for team_name in team_names {
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    62
            self.left_teams.push(team_name.clone());
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    63
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    64
            let remove_msg = crate::utils::to_engine_msg(iter::once(b'F').chain(team_name.bytes()));
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    65
            self.msg_log.push(remove_msg);
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    66
        }
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
    67
    }
13423
87a6cad20c90 Implement game start & engine messages
alfadur
parents: 13422
diff changeset
    68
}
87a6cad20c90 Implement game start & engine messages
alfadur
parents: 13422
diff changeset
    69
13529
662f7df89d06 Implement room config export
alfadur
parents: 13528
diff changeset
    70
#[derive(Serialize, Deserialize)]
13527
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
    71
pub struct RoomSave {
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
    72
    pub location: String,
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    73
    config: RoomConfig,
13527
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
    74
}
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
    75
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    76
bitflags! {
13523
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
    77
    pub struct RoomFlags: u8 {
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
    78
        const FIXED = 0b0000_0001;
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
    79
        const RESTRICTED_JOIN = 0b0000_0010;
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
    80
        const RESTRICTED_TEAM_ADD = 0b0000_0100;
15534
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15525
diff changeset
    81
        const REGISTRATION_REQUIRED = 0b0000_1000;
13523
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
    82
    }
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
    83
}
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
    84
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
    85
pub struct HwRoom {
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
    86
    pub id: RoomId,
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
    87
    pub master_id: Option<ClientId>,
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
    88
    pub name: String,
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
    89
    pub password: Option<String>,
13523
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
    90
    pub greeting: String,
13520
1ee192f13456 Better packing for clients
alfadur
parents: 13478
diff changeset
    91
    pub protocol_number: u16,
13523
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
    92
    pub flags: RoomFlags,
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
    93
13523
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
    94
    pub players_number: u8,
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    95
    pub default_hedgehog_number: u8,
14788
6dea1ca64992 implement maxteams and teamchat messages
alfadur
parents: 14785
diff changeset
    96
    pub max_teams: u8,
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
    97
    pub ready_players_number: u8,
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    98
    pub teams: Vec<(ClientId, TeamInfo)>,
13422
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
    99
    config: RoomConfig,
13478
d79795acaa73 Mostly implement voting
alfadur
parents: 13477
diff changeset
   100
    pub voting: Option<Voting>,
13527
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
   101
    pub saves: HashMap<String, RoomSave>,
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   102
    pub game_info: Option<GameInfo>,
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
   103
}
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
   104
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
   105
impl HwRoom {
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
   106
    pub fn new(id: RoomId) -> HwRoom {
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
   107
        HwRoom {
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
   108
            id,
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   109
            master_id: None,
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
   110
            name: String::new(),
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
   111
            password: None,
13477
f748a72432f2 Implement greetings & fixed rooms
alfadur
parents: 13443
diff changeset
   112
            greeting: "".to_string(),
13523
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   113
            flags: RoomFlags::empty(),
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
   114
            protocol_number: 0,
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   115
            players_number: 0,
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   116
            default_hedgehog_number: 4,
14788
6dea1ca64992 implement maxteams and teamchat messages
alfadur
parents: 14785
diff changeset
   117
            max_teams: MAX_TEAMS_IN_ROOM,
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
   118
            ready_players_number: 0,
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   119
            teams: Vec::new(),
13422
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   120
            config: RoomConfig::new(),
13478
d79795acaa73 Mostly implement voting
alfadur
parents: 13477
diff changeset
   121
            voting: None,
13527
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
   122
            saves: HashMap::new(),
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   123
            game_info: None,
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
   124
        }
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents:
diff changeset
   125
    }
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   126
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   127
    pub fn hedgehogs_number(&self) -> u8 {
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   128
        self.teams.iter().map(|(_, t)| t.hedgehogs_number).sum()
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   129
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   130
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   131
    pub fn addable_hedgehogs(&self) -> u8 {
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   132
        MAX_HEDGEHOGS_IN_ROOM - self.hedgehogs_number()
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   133
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   134
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   135
    pub fn add_team(
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   136
        &mut self,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   137
        owner_id: ClientId,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   138
        mut team: TeamInfo,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   139
        preserve_color: bool,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   140
    ) -> &TeamInfo {
13801
5fb40c8e5542 port some legacy protocol support
alfadur
parents: 13795
diff changeset
   141
        if !preserve_color {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   142
            team.color = iter::repeat(())
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   143
                .enumerate()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   144
                .map(|(i, _)| i as u8)
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   145
                .take(u8::max_value() as usize + 1)
13801
5fb40c8e5542 port some legacy protocol support
alfadur
parents: 13795
diff changeset
   146
                .find(|i| self.teams.iter().all(|(_, t)| t.color != *i))
5fb40c8e5542 port some legacy protocol support
alfadur
parents: 13795
diff changeset
   147
                .unwrap_or(0u8)
5fb40c8e5542 port some legacy protocol support
alfadur
parents: 13795
diff changeset
   148
        };
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   149
        team.hedgehogs_number = if self.teams.is_empty() {
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   150
            self.default_hedgehog_number
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   151
        } else {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   152
            self.teams[0]
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   153
                .1
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   154
                .hedgehogs_number
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   155
                .min(self.addable_hedgehogs())
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   156
        };
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   157
        self.teams.push((owner_id, team));
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   158
        &self.teams.last().unwrap().1
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   159
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   160
15482
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15075
diff changeset
   161
    pub fn remove_team(&mut self, team_name: &str) {
4cc9ec732392 refactor some of the room handler
alfadur
parents: 15075
diff changeset
   162
        if let Some(index) = self.teams.iter().position(|(_, t)| t.name == team_name) {
15540
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
   163
            self.teams.remove(index);
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   164
        }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   165
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   166
13478
d79795acaa73 Mostly implement voting
alfadur
parents: 13477
diff changeset
   167
    pub fn set_hedgehogs_number(&mut self, n: u8) -> Vec<String> {
d79795acaa73 Mostly implement voting
alfadur
parents: 13477
diff changeset
   168
        let mut names = Vec::new();
15552
0031683bfa76 don't apply voted 🦔 to the in-game teams
alfadur
parents: 15541
diff changeset
   169
        let teams = &mut self.teams;
15540
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
   170
13478
d79795acaa73 Mostly implement voting
alfadur
parents: 13477
diff changeset
   171
        if teams.len() as u8 * n <= MAX_HEDGEHOGS_IN_ROOM {
d79795acaa73 Mostly implement voting
alfadur
parents: 13477
diff changeset
   172
            for (_, team) in teams.iter_mut() {
d79795acaa73 Mostly implement voting
alfadur
parents: 13477
diff changeset
   173
                team.hedgehogs_number = n;
d79795acaa73 Mostly implement voting
alfadur
parents: 13477
diff changeset
   174
                names.push(team.name.clone())
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   175
            }
13478
d79795acaa73 Mostly implement voting
alfadur
parents: 13477
diff changeset
   176
            self.default_hedgehog_number = n;
d79795acaa73 Mostly implement voting
alfadur
parents: 13477
diff changeset
   177
        }
d79795acaa73 Mostly implement voting
alfadur
parents: 13477
diff changeset
   178
        names
d79795acaa73 Mostly implement voting
alfadur
parents: 13477
diff changeset
   179
    }
d79795acaa73 Mostly implement voting
alfadur
parents: 13477
diff changeset
   180
15541
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
   181
    pub fn teams_in_game(&self) -> Option<u8> {
15569
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   182
        self.game_info
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   183
            .as_ref()
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   184
            .map(|info| (info.original_teams.len() - info.left_teams.len()) as u8)
15541
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
   185
    }
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
   186
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   187
    pub fn find_team_and_owner_mut<F>(&mut self, f: F) -> Option<(ClientId, &mut TeamInfo)>
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   188
    where
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   189
        F: Fn(&TeamInfo) -> bool,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   190
    {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   191
        self.teams
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   192
            .iter_mut()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   193
            .find(|(_, t)| f(t))
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   194
            .map(|(id, t)| (*id, t))
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   195
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   196
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   197
    pub fn find_team<F>(&self, f: F) -> Option<&TeamInfo>
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   198
    where
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   199
        F: Fn(&TeamInfo) -> bool,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   200
    {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   201
        self.teams
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   202
            .iter()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   203
            .find_map(|(_, t)| Some(t).filter(|t| f(&t)))
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   204
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   205
15540
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
   206
    pub fn client_teams(&self, client_id: ClientId) -> impl Iterator<Item = &TeamInfo> {
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15536
diff changeset
   207
        client_teams_impl(&self.teams, client_id)
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   208
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   209
13423
87a6cad20c90 Implement game start & engine messages
alfadur
parents: 13422
diff changeset
   210
    pub fn client_team_indices(&self, client_id: ClientId) -> Vec<u8> {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   211
        self.teams
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   212
            .iter()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   213
            .enumerate()
13423
87a6cad20c90 Implement game start & engine messages
alfadur
parents: 13422
diff changeset
   214
            .filter(move |(_, (id, _))| *id == client_id)
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   215
            .map(|(i, _)| i as u8)
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   216
            .collect()
13423
87a6cad20c90 Implement game start & engine messages
alfadur
parents: 13422
diff changeset
   217
    }
87a6cad20c90 Implement game start & engine messages
alfadur
parents: 13422
diff changeset
   218
14788
6dea1ca64992 implement maxteams and teamchat messages
alfadur
parents: 14785
diff changeset
   219
    pub fn clan_team_owners(&self, color: u8) -> impl Iterator<Item = ClientId> + '_ {
6dea1ca64992 implement maxteams and teamchat messages
alfadur
parents: 14785
diff changeset
   220
        self.teams
6dea1ca64992 implement maxteams and teamchat messages
alfadur
parents: 14785
diff changeset
   221
            .iter()
6dea1ca64992 implement maxteams and teamchat messages
alfadur
parents: 14785
diff changeset
   222
            .filter(move |(_, t)| t.color == color)
6dea1ca64992 implement maxteams and teamchat messages
alfadur
parents: 14785
diff changeset
   223
            .map(|(id, _)| *id)
6dea1ca64992 implement maxteams and teamchat messages
alfadur
parents: 14785
diff changeset
   224
    }
6dea1ca64992 implement maxteams and teamchat messages
alfadur
parents: 14785
diff changeset
   225
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   226
    pub fn find_team_owner(&self, team_name: &str) -> Option<(ClientId, &str)> {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   227
        self.teams
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   228
            .iter()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   229
            .find(|(_, t)| t.name == team_name)
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   230
            .map(|(id, t)| (*id, &t.name[..]))
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   231
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   232
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   233
    pub fn find_team_color(&self, owner_id: ClientId) -> Option<u8> {
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   234
        self.client_teams(owner_id).nth(0).map(|t| t.color)
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   235
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   236
13423
87a6cad20c90 Implement game start & engine messages
alfadur
parents: 13422
diff changeset
   237
    pub fn has_multiple_clans(&self) -> bool {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   238
        self.teams.iter().min_by_key(|(_, t)| t.color)
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   239
            != self.teams.iter().max_by_key(|(_, t)| t.color)
13423
87a6cad20c90 Implement game start & engine messages
alfadur
parents: 13422
diff changeset
   240
    }
87a6cad20c90 Implement game start & engine messages
alfadur
parents: 13422
diff changeset
   241
13422
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   242
    pub fn set_config(&mut self, cfg: GameCfg) {
14785
a1077e8d26f4 implement watch message apart from replay deserializing
alfadur
parents: 14457
diff changeset
   243
        self.config.set_config(cfg);
13422
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   244
    }
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   245
13427
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13426
diff changeset
   246
    pub fn start_round(&mut self) {
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13426
diff changeset
   247
        if self.game_info.is_none() {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   248
            self.game_info = Some(GameInfo::new(self.teams.clone(), self.config.clone()));
13427
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13426
diff changeset
   249
        }
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13426
diff changeset
   250
    }
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13426
diff changeset
   251
13523
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   252
    pub fn is_fixed(&self) -> bool {
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   253
        self.flags.contains(RoomFlags::FIXED)
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   254
    }
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   255
    pub fn is_join_restricted(&self) -> bool {
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   256
        self.flags.contains(RoomFlags::RESTRICTED_JOIN)
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   257
    }
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   258
    pub fn is_team_add_restricted(&self) -> bool {
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   259
        self.flags.contains(RoomFlags::RESTRICTED_TEAM_ADD)
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   260
    }
15534
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15525
diff changeset
   261
    pub fn is_registration_required(&self) -> bool {
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15525
diff changeset
   262
        self.flags.contains(RoomFlags::REGISTRATION_REQUIRED)
13523
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   263
    }
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   264
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   265
    pub fn set_is_fixed(&mut self, value: bool) {
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   266
        self.flags.set(RoomFlags::FIXED, value)
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   267
    }
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   268
    pub fn set_join_restriction(&mut self, value: bool) {
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   269
        self.flags.set(RoomFlags::RESTRICTED_JOIN, value)
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   270
    }
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   271
    pub fn set_team_add_restriction(&mut self, value: bool) {
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   272
        self.flags.set(RoomFlags::RESTRICTED_TEAM_ADD, value)
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   273
    }
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   274
    pub fn set_unregistered_players_restriction(&mut self, value: bool) {
15534
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15525
diff changeset
   275
        self.flags.set(RoomFlags::REGISTRATION_REQUIRED, value)
13523
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   276
    }
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   277
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   278
    fn flags_string(&self) -> String {
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   279
        let mut result = "-".to_string();
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   280
        if self.game_info.is_some() {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   281
            result += "g"
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   282
        }
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   283
        if self.password.is_some() {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   284
            result += "p"
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   285
        }
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   286
        if self.is_join_restricted() {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   287
            result += "j"
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   288
        }
15534
bb93e9642b5b send cleared flags on room join
alfadur <mail@none>
parents: 15525
diff changeset
   289
        if self.is_registration_required() {
13523
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   290
            result += "r"
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   291
        }
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   292
        result
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   293
    }
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   294
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
   295
    pub fn info(&self, master: Option<&HwClient>) -> Vec<String> {
13422
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   296
        let c = &self.config;
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   297
        vec![
13523
8c5dd562c9f7 Add room flags
alfadur
parents: 13520
diff changeset
   298
            self.flags_string(),
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   299
            self.name.clone(),
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   300
            self.players_number.to_string(),
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   301
            self.teams.len().to_string(),
13422
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   302
            master.map_or("[]", |c| &c.nick).to_string(),
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   303
            c.map_type.to_string(),
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   304
            c.script.to_string(),
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   305
            c.scheme.name.to_string(),
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   306
            c.ammo.name.to_string(),
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   307
        ]
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
   308
    }
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   309
15569
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   310
    pub fn config(&self) -> &RoomConfig {
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   311
        &self.config
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   312
    }
4b2f3228f13b fix editing teams while a game is in progress
alfadur
parents: 15553
diff changeset
   313
14785
a1077e8d26f4 implement watch message apart from replay deserializing
alfadur
parents: 14457
diff changeset
   314
    pub fn active_config(&self) -> &RoomConfig {
a1077e8d26f4 implement watch message apart from replay deserializing
alfadur
parents: 14457
diff changeset
   315
        match self.game_info {
15541
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
   316
            Some(ref info) => &info.original_config,
14785
a1077e8d26f4 implement watch message apart from replay deserializing
alfadur
parents: 14457
diff changeset
   317
            None => &self.config,
a1077e8d26f4 implement watch message apart from replay deserializing
alfadur
parents: 14457
diff changeset
   318
        }
a1077e8d26f4 implement watch message apart from replay deserializing
alfadur
parents: 14457
diff changeset
   319
    }
a1077e8d26f4 implement watch message apart from replay deserializing
alfadur
parents: 14457
diff changeset
   320
13422
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   321
    pub fn map_config(&self) -> Vec<String> {
13427
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13426
diff changeset
   322
        match self.game_info {
15541
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
   323
            Some(ref info) => info.original_config.to_map_config(),
14785
a1077e8d26f4 implement watch message apart from replay deserializing
alfadur
parents: 14457
diff changeset
   324
            None => self.config.to_map_config(),
13427
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13426
diff changeset
   325
        }
13422
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   326
    }
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   327
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   328
    pub fn game_config(&self) -> Vec<GameCfg> {
13427
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13426
diff changeset
   329
        match self.game_info {
15541
d122b65bdf6f reapply some simplifications
alfadur <mail@none>
parents: 15540
diff changeset
   330
            Some(ref info) => info.original_config.to_game_config(),
14785
a1077e8d26f4 implement watch message apart from replay deserializing
alfadur
parents: 14457
diff changeset
   331
            None => self.config.to_game_config(),
13422
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   332
        }
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   333
    }
5fb27f94fc3b Implement game config messages
alfadur
parents: 13419
diff changeset
   334
13528
c8b626b0a3ad Implement save/delete commands
alfadur
parents: 13527
diff changeset
   335
    pub fn save_config(&mut self, name: String, location: String) {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   336
        self.saves.insert(
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   337
            name,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   338
            RoomSave {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   339
                location,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   340
                config: self.config.clone(),
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   341
            },
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   342
        );
13528
c8b626b0a3ad Implement save/delete commands
alfadur
parents: 13527
diff changeset
   343
    }
c8b626b0a3ad Implement save/delete commands
alfadur
parents: 13527
diff changeset
   344
13527
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
   345
    pub fn load_config(&mut self, name: &str) -> Option<&str> {
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
   346
        if let Some(save) = self.saves.get(name) {
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
   347
            self.config = save.config.clone();
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
   348
            Some(&save.location[..])
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
   349
        } else {
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
   350
            None
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
   351
        }
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
   352
    }
e3ae9eea0689 Implement map voting
alfadur
parents: 13525
diff changeset
   353
13528
c8b626b0a3ad Implement save/delete commands
alfadur
parents: 13527
diff changeset
   354
    pub fn delete_config(&mut self, name: &str) -> bool {
c8b626b0a3ad Implement save/delete commands
alfadur
parents: 13527
diff changeset
   355
        self.saves.remove(name).is_some()
c8b626b0a3ad Implement save/delete commands
alfadur
parents: 13527
diff changeset
   356
    }
c8b626b0a3ad Implement save/delete commands
alfadur
parents: 13527
diff changeset
   357
13529
662f7df89d06 Implement room config export
alfadur
parents: 13528
diff changeset
   358
    pub fn get_saves(&self) -> Result<String, serde_yaml::Error> {
662f7df89d06 Implement room config export
alfadur
parents: 13528
diff changeset
   359
        serde_yaml::to_string(&(&self.greeting, &self.saves))
662f7df89d06 Implement room config export
alfadur
parents: 13528
diff changeset
   360
    }
662f7df89d06 Implement room config export
alfadur
parents: 13528
diff changeset
   361
662f7df89d06 Implement room config export
alfadur
parents: 13528
diff changeset
   362
    pub fn set_saves(&mut self, text: &str) -> Result<(), serde_yaml::Error> {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   363
        serde_yaml::from_str::<(String, HashMap<String, RoomSave>)>(text).map(
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   364
            |(greeting, saves)| {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   365
                self.greeting = greeting;
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   366
                self.saves = saves;
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   367
            },
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   368
        )
13529
662f7df89d06 Implement room config export
alfadur
parents: 13528
diff changeset
   369
    }
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   370
}