rust/hedgewars-server/src/handlers/inlobby.rs
author alfadur
Tue, 01 Feb 2022 02:23:35 +0300
changeset 15831 7d0f747afcb8
parent 15804 747278149393
child 15848 3d05bada4799
permissions -rw-r--r--
move server network to tokio
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
     1
use super::{common::rnd_reply, strings::*};
15804
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15533
diff changeset
     2
use crate::handlers::actions::ToPendingMessage;
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13523
diff changeset
     3
use crate::{
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
     4
    core::{
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
     5
        client::HwClient,
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
     6
        server::{AccessError, CreateRoomError, HwServer, JoinRoomError},
15804
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15533
diff changeset
     7
        types::ClientId,
14782
50fcef24003f add typed protocol flags
alfadur
parents: 14694
diff changeset
     8
    },
15804
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15533
diff changeset
     9
    utils::is_name_illegal,
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15533
diff changeset
    10
};
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15533
diff changeset
    11
use hedgewars_network_protocol::{
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15533
diff changeset
    12
    messages::{
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
    13
        add_flags, remove_flags, server_chat, HwProtocolMessage, HwServerMessage::*,
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
    14
        ProtocolFlags as Flags,
14783
b3adc030104b implement server vars
alfadur
parents: 14782
diff changeset
    15
    },
15804
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15533
diff changeset
    16
    types::ServerVar,
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
    17
};
13805
0463a4221327 cleanup crate imports
alfadur
parents: 13666
diff changeset
    18
use log::*;
14789
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
    19
use std::{collections::HashSet, convert::identity};
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
    20
14671
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    21
pub fn handle(
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
    22
    server: &mut HwServer,
14671
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    23
    client_id: ClientId,
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    24
    response: &mut super::Response,
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
    25
    message: HwProtocolMessage,
14671
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    26
) {
15804
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15533
diff changeset
    27
    use hedgewars_network_protocol::messages::HwProtocolMessage::*;
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    28
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
    29
    match message {
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    30
        CreateRoom(name, password) => match server.create_room(client_id, name, password) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    31
            Err(CreateRoomError::InvalidName) => response.warn(ILLEGAL_ROOM_NAME),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    32
            Err(CreateRoomError::AlreadyExists) => response.warn(ROOM_EXISTS),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    33
            Ok((client, room)) => {
14671
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    34
                response.add(
14504
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    35
                    RoomAdd(room.info(Some(&client)))
6cc0fce249f9 Server action refactoring part 1 of N
alfadur <mail@none>
parents: 14457
diff changeset
    36
                        .send_all()
14671
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    37
                        .with_protocol(room.protocol_number),
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14504
diff changeset
    38
                );
14797
ce2268ae261f add missing room joined message
alfadur
parents: 14789
diff changeset
    39
                response.add(RoomJoined(vec![client.nick.clone()]).send_self());
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    40
                response.add(
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    41
                    ClientFlags(
15533
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15526
diff changeset
    42
                        add_flags(&[Flags::RoomMaster, Flags::Ready, Flags::InRoom]),
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    43
                        vec![client.nick.clone()],
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    44
                    )
15533
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15526
diff changeset
    45
                    .send_all(),
14782
50fcef24003f add typed protocol flags
alfadur
parents: 14694
diff changeset
    46
                );
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    47
            }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    48
        },
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
    49
        Chat(msg) => {
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    50
            response.add(
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    51
                ChatMsg {
15533
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15526
diff changeset
    52
                    nick: server.client(client_id).nick.clone(),
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    53
                    msg,
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    54
                }
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    55
                .send_all()
14694
25c564f77b7d remove lobby room
alfadur <mail@none>
parents: 14690
diff changeset
    56
                .in_lobby()
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    57
                .but_self(),
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    58
            );
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    59
        }
15533
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15526
diff changeset
    60
        JoinRoom(name, password) => {
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15526
diff changeset
    61
            match server.join_room_by_name(client_id, &name, password.as_deref()) {
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15526
diff changeset
    62
                Err(error) => super::common::get_room_join_error(error, response),
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15526
diff changeset
    63
                Ok((client, room, room_clients)) => {
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15526
diff changeset
    64
                    super::common::get_room_join_data(client, room, room_clients, response)
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15526
diff changeset
    65
                }
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    66
            }
15533
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15526
diff changeset
    67
        }
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    68
        Follow(nick) => {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    69
            if let Some(client) = server.find_client(&nick) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    70
                if let Some(room_id) = client.room_id {
15533
0606f89698e7 check the room passwords
alfadur <mail@none>
parents: 15526
diff changeset
    71
                    match server.join_room(client_id, room_id, None) {
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    72
                        Err(error) => super::common::get_room_join_error(error, response),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    73
                        Ok((client, room, room_clients)) => {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    74
                            super::common::get_room_join_data(client, room, room_clients, response)
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    75
                        }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    76
                    }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    77
                } else {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    78
                    response.warn(NO_ROOM);
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13523
diff changeset
    79
                }
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13523
diff changeset
    80
            } else {
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    81
                response.warn(NO_USER);
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    82
            }
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    83
        }
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    84
        SetServerVar(var) => match server.set_var(client_id, var) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    85
            Err(AccessError()) => response.warn(ACCESS_DENIED),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    86
            Ok(()) => response.add(server_chat(VARIABLE_UPDATED.to_string()).send_self()),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    87
        },
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    88
        GetServerVar => match server.get_vars(client_id) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    89
            Err(AccessError()) => response.warn(ACCESS_DENIED),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    90
            Ok(vars) => {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    91
                response.add(
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    92
                    ServerVars(vars.iter().flat_map(|v| v.to_protocol()).collect()).send_self(),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    93
                );
14787
0e64acbc3f8b implement follow message
alfadur
parents: 14785
diff changeset
    94
            }
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    95
        },
13445
d3c86ade3d4d Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents: 13444
diff changeset
    96
        Rnd(v) => {
14672
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14671
diff changeset
    97
            response.add(rnd_reply(&v).send_self());
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    98
        }
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
    99
        Stats => match server.get_used_protocols(client_id) {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   100
            Err(AccessError()) => response.warn(ACCESS_DENIED),
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   101
            Ok(protocols) => {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   102
                let mut html = Vec::with_capacity(protocols.len() + 2);
14789
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   103
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   104
                html.push("<table>".to_string());
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   105
                for protocol in protocols {
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   106
                    html.push(format!(
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   107
                        "<tr><td>{}</td><td>{}</td><td>{}</td></tr>",
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   108
                        super::utils::protocol_version_string(protocol),
15526
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15482
diff changeset
   109
                        server.protocol_client_ids(protocol).count(),
24f692e791d3 disallow mutable rooms outside the server
alfadur <mail@none>
parents: 15482
diff changeset
   110
                        server.protocol_room_ids(protocol).count()
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   111
                    ));
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   112
                }
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   113
                html.push("</table>".to_string());
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   114
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   115
                response.add(Warning(html.join("")).send_self());
14789
18240b308505 implement stats message
alfadur
parents: 14787
diff changeset
   116
            }
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15075
diff changeset
   117
        },
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12852
diff changeset
   118
        List => warn!("Deprecated LIST message received"),
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
   119
        _ => warn!("Incorrect command in lobby state"),
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
   120
    }
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
   121
}