rust/hedgewars-server/src/handlers/inlobby.rs
author Wuzzy <Wuzzy2@mail.ru>
Thu, 14 Nov 2019 17:12:31 +0100
changeset 15526 266ff128a65a
parent 15504 4cc9ec732392
child 15548 24f692e791d3
permissions -rw-r--r--
ASA moon02: Disable hog input after victory Player could briefly move Crazy Runner when skipping final cutscene.
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
     1
use super::{common::rnd_reply, strings::*};
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13494
diff changeset
     2
use crate::{
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
     3
    core::{
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
     4
        client::HwClient,
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
     5
        server::{AccessError, CreateRoomError, HwServer, JoinRoomError},
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
     6
        types::{ClientId, ServerVar},
14803
50fcef24003f add typed protocol flags
alfadur
parents: 14715
diff changeset
     7
    },
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
     8
    protocol::messages::{
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
     9
        add_flags, remove_flags, server_chat, HwProtocolMessage, HwServerMessage::*,
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    10
        ProtocolFlags as Flags,
14804
b3adc030104b implement server vars
alfadur
parents: 14803
diff changeset
    11
    },
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
    12
    utils::is_name_illegal,
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
    13
};
13810
0463a4221327 cleanup crate imports
alfadur
parents: 13666
diff changeset
    14
use log::*;
14810
18240b308505 implement stats message
alfadur
parents: 14808
diff changeset
    15
use std::{collections::HashSet, convert::identity};
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
    16
14692
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14525
diff changeset
    17
pub fn handle(
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    18
    server: &mut HwServer,
14692
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14525
diff changeset
    19
    client_id: ClientId,
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14525
diff changeset
    20
    response: &mut super::Response,
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    21
    message: HwProtocolMessage,
14692
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14525
diff changeset
    22
) {
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    23
    use crate::protocol::messages::HwProtocolMessage::*;
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    24
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
    25
    match message {
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    26
        CreateRoom(name, password) => match server.create_room(client_id, name, password) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    27
            Err(CreateRoomError::InvalidName) => response.warn(ILLEGAL_ROOM_NAME),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    28
            Err(CreateRoomError::AlreadyExists) => response.warn(ROOM_EXISTS),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    29
            Ok((client, room)) => {
14692
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14525
diff changeset
    30
                response.add(
14525
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
    31
                    RoomAdd(room.info(Some(&client)))
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14478
diff changeset
    32
                        .send_all()
14692
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14525
diff changeset
    33
                        .with_protocol(room.protocol_number),
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14525
diff changeset
    34
                );
14818
ce2268ae261f add missing room joined message
alfadur
parents: 14810
diff changeset
    35
                response.add(RoomJoined(vec![client.nick.clone()]).send_self());
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    36
                response.add(
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    37
                    ClientFlags(
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    38
                        add_flags(&[Flags::RoomMaster, Flags::Ready]),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    39
                        vec![client.nick.clone()],
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    40
                    )
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    41
                    .send_self(),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    42
                );
14803
50fcef24003f add typed protocol flags
alfadur
parents: 14715
diff changeset
    43
                response.add(
50fcef24003f add typed protocol flags
alfadur
parents: 14715
diff changeset
    44
                    ClientFlags(add_flags(&[Flags::InRoom]), vec![client.nick.clone()]).send_self(),
50fcef24003f add typed protocol flags
alfadur
parents: 14715
diff changeset
    45
                );
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    46
            }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    47
        },
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    48
        Chat(msg) => {
14693
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14692
diff changeset
    49
            response.add(
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14692
diff changeset
    50
                ChatMsg {
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    51
                    nick: server.get_client_nick(client_id).to_string(),
14693
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14692
diff changeset
    52
                    msg,
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14692
diff changeset
    53
                }
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14692
diff changeset
    54
                .send_all()
14715
25c564f77b7d remove lobby room
alfadur <mail@none>
parents: 14711
diff changeset
    55
                .in_lobby()
14693
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14692
diff changeset
    56
                .but_self(),
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14692
diff changeset
    57
            );
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
    58
        }
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    59
        JoinRoom(name, _password) => match server.join_room_by_name(client_id, &name) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    60
            Err(error) => super::common::get_room_join_error(error, response),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    61
            Ok((client, room, room_clients)) => {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    62
                super::common::get_room_join_data(client, room, room_clients, response)
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    63
            }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    64
        },
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    65
        Follow(nick) => {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    66
            if let Some(client) = server.find_client(&nick) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    67
                if let Some(room_id) = client.room_id {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    68
                    match server.join_room(client_id, room_id) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    69
                        Err(error) => super::common::get_room_join_error(error, response),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    70
                        Ok((client, room, room_clients)) => {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    71
                            super::common::get_room_join_data(client, room, room_clients, response)
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    72
                        }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    73
                    }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    74
                } else {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    75
                    response.warn(NO_ROOM);
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13494
diff changeset
    76
                }
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13494
diff changeset
    77
            } else {
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    78
                response.warn(NO_USER);
14693
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14692
diff changeset
    79
            }
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
    80
        }
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    81
        SetServerVar(var) => match server.set_var(client_id, var) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    82
            Err(AccessError()) => response.warn(ACCESS_DENIED),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    83
            Ok(()) => response.add(server_chat(VARIABLE_UPDATED.to_string()).send_self()),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    84
        },
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    85
        GetServerVar => match server.get_vars(client_id) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    86
            Err(AccessError()) => response.warn(ACCESS_DENIED),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    87
            Ok(vars) => {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    88
                response.add(
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    89
                    ServerVars(vars.iter().flat_map(|v| v.to_protocol()).collect()).send_self(),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    90
                );
14808
0e64acbc3f8b implement follow message
alfadur
parents: 14806
diff changeset
    91
            }
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    92
        },
13445
d3c86ade3d4d Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents: 13444
diff changeset
    93
        Rnd(v) => {
14693
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14692
diff changeset
    94
            response.add(rnd_reply(&v).send_self());
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
    95
        }
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    96
        Stats => match server.get_used_protocols(client_id) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    97
            Err(AccessError()) => response.warn(ACCESS_DENIED),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    98
            Ok(protocols) => {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
    99
                let mut html = Vec::with_capacity(protocols.len() + 2);
14810
18240b308505 implement stats message
alfadur
parents: 14808
diff changeset
   100
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
   101
                html.push("<table>".to_string());
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
   102
                for protocol in protocols {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
   103
                    html.push(format!(
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
   104
                        "<tr><td>{}</td><td>{}</td><td>{}</td></tr>",
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
   105
                        super::utils::protocol_version_string(protocol),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
   106
                        server.protocol_clients(protocol).count(),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
   107
                        server.protocol_rooms(protocol).count()
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
   108
                    ));
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
   109
                }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
   110
                html.push("</table>".to_string());
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
   111
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
   112
                response.add(Warning(html.join("")).send_self());
14810
18240b308505 implement stats message
alfadur
parents: 14808
diff changeset
   113
            }
15463
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15096
diff changeset
   114
        },
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
   115
        List => warn!("Deprecated LIST message received"),
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
   116
        _ => warn!("Incorrect command in lobby state"),
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
   117
    }
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
   118
}