rust/hedgewars-server/src/server/actions.rs
author alfadur <mail@none>
Wed, 06 Feb 2019 22:29:02 +0300
changeset 14711 f61ce544d436
parent 14710 aae29ba56aec
child 14715 25c564f77b7d
permissions -rw-r--r--
Server action refactoring part N of N
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
     1
use super::{
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
     2
    client::HWClient,
14395
e5db279308d7 dispose of server mods
alfadur
parents: 14371
diff changeset
     3
    core::HWServer,
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
     4
    coretypes::{ClientId, GameCfg, RoomId, VoteType},
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
     5
    handlers,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
     6
    room::HWRoom,
13494
8c5dd562c9f7 Add room flags
alfadur
parents: 13486
diff changeset
     7
    room::{GameInfo, RoomFlags},
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
     8
};
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13521
diff changeset
     9
use crate::{
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    10
    protocol::messages::{server_chat, HWProtocolMessage, HWServerMessage, HWServerMessage::*},
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    11
    utils::to_engine_msg,
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
    12
};
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    13
use rand::{distributions::Uniform, thread_rng, Rng};
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    14
use std::{io, io::Write, iter::once, mem::replace};
12138
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents:
diff changeset
    15
14477
a077aac9df01 Start database interaction implementation
unc0rr
parents: 14436
diff changeset
    16
#[cfg(feature = "official-server")]
a077aac9df01 Start database interaction implementation
unc0rr
parents: 14436
diff changeset
    17
use super::database;
a077aac9df01 Start database interaction implementation
unc0rr
parents: 14436
diff changeset
    18
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    19
pub enum Destination {
13426
f091f69d59e4 Additional round cleanup
alfadur
parents: 13424
diff changeset
    20
    ToId(ClientId),
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    21
    ToSelf,
13423
87a6cad20c90 Implement game start & engine messages
alfadur
parents: 13422
diff changeset
    22
    ToAll {
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    23
        room_id: Option<RoomId>,
13486
1ee192f13456 Better packing for clients
alfadur
parents: 13460
diff changeset
    24
        protocol: Option<u16>,
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    25
        skip_self: bool,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    26
    },
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    27
}
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    28
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    29
pub struct PendingMessage {
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    30
    pub destination: Destination,
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    31
    pub message: HWServerMessage,
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    32
}
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    33
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    34
impl PendingMessage {
13426
f091f69d59e4 Additional round cleanup
alfadur
parents: 13424
diff changeset
    35
    pub fn send(message: HWServerMessage, client_id: ClientId) -> PendingMessage {
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    36
        PendingMessage {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    37
            destination: Destination::ToId(client_id),
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    38
            message,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    39
        }
13426
f091f69d59e4 Additional round cleanup
alfadur
parents: 13424
diff changeset
    40
    }
f091f69d59e4 Additional round cleanup
alfadur
parents: 13424
diff changeset
    41
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    42
    pub fn send_self(message: HWServerMessage) -> PendingMessage {
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    43
        PendingMessage {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    44
            destination: Destination::ToSelf,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    45
            message,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    46
        }
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    47
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    48
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    49
    pub fn send_all(message: HWServerMessage) -> PendingMessage {
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    50
        let destination = Destination::ToAll {
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    51
            room_id: None,
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    52
            protocol: None,
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    53
            skip_self: false,
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    54
        };
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    55
        PendingMessage {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    56
            destination,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    57
            message,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    58
        }
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    59
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    60
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    61
    pub fn in_room(mut self, clients_room_id: RoomId) -> PendingMessage {
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    62
        if let Destination::ToAll {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    63
            ref mut room_id, ..
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    64
        } = self.destination
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    65
        {
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    66
            *room_id = Some(clients_room_id)
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    67
        }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    68
        self
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    69
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    70
13486
1ee192f13456 Better packing for clients
alfadur
parents: 13460
diff changeset
    71
    pub fn with_protocol(mut self, protocol_number: u16) -> PendingMessage {
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    72
        if let Destination::ToAll {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    73
            ref mut protocol, ..
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    74
        } = self.destination
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    75
        {
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    76
            *protocol = Some(protocol_number)
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    77
        }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    78
        self
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    79
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    80
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    81
    pub fn but_self(mut self) -> PendingMessage {
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    82
        if let Destination::ToAll {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    83
            ref mut skip_self, ..
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    84
        } = self.destination
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    85
        {
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    86
            *skip_self = true
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    87
        }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    88
        self
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    89
    }
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    90
}
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    91
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    92
impl HWServerMessage {
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    93
    pub fn send(self, client_id: ClientId) -> PendingMessage {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    94
        PendingMessage::send(self, client_id)
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    95
    }
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    96
    pub fn send_self(self) -> PendingMessage {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    97
        PendingMessage::send_self(self)
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    98
    }
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
    99
    pub fn send_all(self) -> PendingMessage {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
   100
        PendingMessage::send_all(self)
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14477
diff changeset
   101
    }
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
   102
}