rust/hedgewars-server/src/server/handlers/lobby.rs
author alfadur
Thu, 11 Apr 2019 21:20:41 +0300
changeset 14789 18240b308505
parent 14787 0e64acbc3f8b
child 14797 ce2268ae261f
permissions -rw-r--r--
implement stats message
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
     1
use mio;
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
     2
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     3
use super::common::rnd_reply;
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13523
diff changeset
     4
use crate::{
14782
50fcef24003f add typed protocol flags
alfadur
parents: 14694
diff changeset
     5
    protocol::messages::{
14789
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
     6
        add_flags, remove_flags, server_chat, HWProtocolMessage, HWServerMessage::*,
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
     7
        ProtocolFlags as Flags,
14782
50fcef24003f add typed protocol flags
alfadur
parents: 14694
diff changeset
     8
    },
14783
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
     9
    server::{
14787
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
    10
        client::HWClient,
14783
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
    11
        core::HWServer,
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
    12
        coretypes::{ClientId, ServerVar},
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
    13
    },
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    14
    utils::is_name_illegal,
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
    15
};
13805
0463a4221327 cleanup crate imports
alfadur
parents: 13666
diff changeset
    16
use log::*;
14789
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
    17
use std::{collections::HashSet, convert::identity};
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
    18
14671
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    19
pub fn handle(
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    20
    server: &mut HWServer,
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    21
    client_id: ClientId,
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    22
    response: &mut super::Response,
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    23
    message: HWProtocolMessage,
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    24
) {
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13523
diff changeset
    25
    use crate::protocol::messages::HWProtocolMessage::*;
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
    26
    match message {
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
    27
        CreateRoom(name, password) => {
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    28
            if is_name_illegal(&name) {
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    29
                response.add(Warning("Illegal room name! A room name must be between 1-40 characters long, must not have a trailing or leading space and must not have any of these characters: $()*+?[]^{|}".to_string()).send_self());
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    30
            } else if server.has_room(&name) {
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    31
                response.add(
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    32
                    Warning("A room with the same name already exists.".to_string()).send_self(),
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    33
                );
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    34
            } else {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    35
                let flags_msg = ClientFlags(
14782
50fcef24003f add typed protocol flags
alfadur
parents: 14694
diff changeset
    36
                    add_flags(&[Flags::RoomMaster, Flags::Ready]),
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    37
                    vec![server.clients[client_id].nick.clone()],
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    38
                );
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    39
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    40
                let room_id = server.create_room(client_id, name, password);
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    41
                let room = &server.rooms[room_id];
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    42
                let client = &server.clients[client_id];
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    43
14671
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    44
                response.add(
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    45
                    RoomAdd(room.info(Some(&client)))
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    46
                        .send_all()
14671
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    47
                        .with_protocol(room.protocol_number),
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    48
                );
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    49
                response.add(flags_msg.send_self());
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    50
14782
50fcef24003f add typed protocol flags
alfadur
parents: 14694
diff changeset
    51
                response.add(
50fcef24003f add typed protocol flags
alfadur
parents: 14694
diff changeset
    52
                    ClientFlags(add_flags(&[Flags::InRoom]), vec![client.nick.clone()]).send_self(),
50fcef24003f add typed protocol flags
alfadur
parents: 14694
diff changeset
    53
                );
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    54
            };
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    55
        }
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    56
        Chat(msg) => {
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    57
            response.add(
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    58
                ChatMsg {
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    59
                    nick: server.clients[client_id].nick.clone(),
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    60
                    msg,
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    61
                }
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    62
                .send_all()
14694
25c564f77b7d remove lobby room
alfadur <mail@none>
parents: 14690
diff changeset
    63
                .in_lobby()
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    64
                .but_self(),
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    65
            );
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    66
        }
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13523
diff changeset
    67
        JoinRoom(name, _password) => {
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13523
diff changeset
    68
            let room = server.rooms.iter().find(|(_, r)| r.name == name);
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13523
diff changeset
    69
            let room_id = room.map(|(_, r)| r.id);
14785
a1077e8d26f4 implement watch message apart from replay deserializing
alfadur
parents: 14783
diff changeset
    70
14686
9f98086de1b6 Server action refactoring part 9 of N
alfadur <mail@none>
parents: 14672
diff changeset
    71
            let client = &mut server.clients[client_id];
13427
6f6a866c86a2 Send more data on room joining
alfadur
parents: 13419
diff changeset
    72
14686
9f98086de1b6 Server action refactoring part 9 of N
alfadur <mail@none>
parents: 14672
diff changeset
    73
            if let Some((_, room)) = room {
9f98086de1b6 Server action refactoring part 9 of N
alfadur <mail@none>
parents: 14672
diff changeset
    74
                if client.protocol_number != room.protocol_number {
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    75
                    response.add(
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    76
                        Warning("Room version incompatible to your Hedgewars version!".to_string())
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    77
                            .send_self(),
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    78
                    );
14686
9f98086de1b6 Server action refactoring part 9 of N
alfadur <mail@none>
parents: 14672
diff changeset
    79
                } else if room.is_join_restricted() {
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    80
                    response.add(
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    81
                        Warning(
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    82
                            "Access denied. This room currently doesn't allow joining.".to_string(),
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    83
                        )
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    84
                        .send_self(),
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    85
                    );
14686
9f98086de1b6 Server action refactoring part 9 of N
alfadur <mail@none>
parents: 14672
diff changeset
    86
                } else if room.players_number == u8::max_value() {
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    87
                    response.add(Warning("This room is already full".to_string()).send_self());
14671
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    88
                } else if let Some(room_id) = room_id {
14787
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
    89
                    super::common::enter_room(server, client_id, room_id, response);
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13523
diff changeset
    90
                }
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13523
diff changeset
    91
            } else {
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    92
                response.add(Warning("No such room.".to_string()).send_self());
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    93
            }
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    94
        }
14787
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
    95
        Follow(nick) => {
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
    96
            if let Some(HWClient {
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
    97
                room_id: Some(room_id),
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
    98
                ..
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
    99
            }) = server.find_client(&nick)
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
   100
            {
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
   101
                let room = &server.rooms[*room_id];
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
   102
                response.add(Joining(room.name.clone()).send_self());
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
   103
                super::common::enter_room(server, client_id, *room_id, response);
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
   104
            }
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
   105
        }
14783
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   106
        SetServerVar(var) => {
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   107
            if !server.clients[client_id].is_admin() {
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   108
                response.add(Warning("Access denied.".to_string()).send_self());
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   109
            } else {
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   110
                match var {
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   111
                    ServerVar::MOTDNew(msg) => server.greetings.for_latest_protocol = msg,
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   112
                    ServerVar::MOTDOld(msg) => server.greetings.for_old_protocols = msg,
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   113
                    ServerVar::LatestProto(n) => server.latest_protocol = n,
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   114
                }
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   115
            }
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   116
        }
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   117
        GetServerVar => {
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   118
            if !server.clients[client_id].is_admin() {
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   119
                response.add(Warning("Access denied.".to_string()).send_self());
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   120
            } else {
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   121
                let vars: Vec<_> = [
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   122
                    ServerVar::MOTDNew(server.greetings.for_latest_protocol.clone()),
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   123
                    ServerVar::MOTDOld(server.greetings.for_old_protocols.clone()),
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   124
                    ServerVar::LatestProto(server.latest_protocol),
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   125
                ]
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   126
                .iter()
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   127
                .flat_map(|v| v.to_protocol())
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   128
                .collect();
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   129
                response.add(ServerVars(vars).send_self());
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   130
            }
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
   131
        }
13445
d3c86ade3d4d Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents: 13444
diff changeset
   132
        Rnd(v) => {
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
   133
            response.add(rnd_reply(&v).send_self());
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
   134
        }
14789
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   135
        Stats => {
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   136
            let mut protocols: HashSet<_> = server
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   137
                .clients
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   138
                .iter()
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   139
                .map(|(_, c)| c.protocol_number)
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   140
                .chain(server.rooms.iter().map(|(_, r)| r.protocol_number))
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   141
                .collect();
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   142
            let mut protocols: Vec<_> = protocols.drain().collect();
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   143
            protocols.sort();
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   144
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   145
            let mut html = Vec::with_capacity(protocols.len() + 2);
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   146
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   147
            html.push("<table>".to_string());
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   148
            for protocol in protocols {
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   149
                html.push(format!(
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   150
                    "<tr><td>{}</td><td>{}</td><td>{}</td></tr>",
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   151
                    super::utils::protocol_version_string(protocol),
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   152
                    server.protocol_clients(protocol).count(),
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   153
                    server.protocol_rooms(protocol).count()
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   154
                ));
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   155
            }
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   156
            html.push("</table>".to_string());
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   157
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   158
            response.add(Warning(html.join("")).send_self());
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   159
        }
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
   160
        List => warn!("Deprecated LIST message received"),
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
   161
        _ => warn!("Incorrect command in lobby state"),
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
   162
    }
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
   163
}