gameServer2/src/server/client.rs
author unc0rr
Wed, 18 Jan 2017 22:54:02 +0300
changeset 12146 78925eff02c2
parent 12144 f3121d7dedec
child 12147 4d7d41be1993
permissions -rw-r--r--
Basic support for NICK message
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
12133
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
     1
use slab;
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
     2
use mio::tcp::*;
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
     3
use mio::*;
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
     4
use std::io::Write;
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
     5
use std::io;
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
     6
use netbuf;
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
     7
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
     8
use utils;
12141
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
     9
use protocol::ProtocolDecoder;
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    10
use protocol::messages;
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    11
use protocol::messages::HWProtocolMessage::*;
12143
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12142
diff changeset
    12
use server::actions::Action::*;
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12142
diff changeset
    13
use server::actions::Action;
12142
193dfdcb0620 - Use logging facilities instead of plain println!
unc0rr
parents: 12141
diff changeset
    14
use log;
12133
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    15
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    16
pub struct HWClient {
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    17
    sock: TcpStream,
12141
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    18
    decoder: ProtocolDecoder,
12146
78925eff02c2 Basic support for NICK message
unc0rr
parents: 12144
diff changeset
    19
    buf_out: netbuf::Buf,
78925eff02c2 Basic support for NICK message
unc0rr
parents: 12144
diff changeset
    20
    pub nick: String,
78925eff02c2 Basic support for NICK message
unc0rr
parents: 12144
diff changeset
    21
    roomId: Token,
12133
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    22
}
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    23
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    24
impl HWClient {
12146
78925eff02c2 Basic support for NICK message
unc0rr
parents: 12144
diff changeset
    25
    pub fn new(sock: TcpStream, roomId: &Token) -> HWClient {
12133
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    26
        HWClient {
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    27
            sock: sock,
12141
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    28
            decoder: ProtocolDecoder::new(),
12133
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    29
            buf_out: netbuf::Buf::new(),
12146
78925eff02c2 Basic support for NICK message
unc0rr
parents: 12144
diff changeset
    30
            nick: String::new(),
78925eff02c2 Basic support for NICK message
unc0rr
parents: 12144
diff changeset
    31
            roomId: roomId.clone(),
12133
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    32
        }
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    33
    }
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    34
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    35
    pub fn register(&mut self, poll: &Poll, token: Token) {
12142
193dfdcb0620 - Use logging facilities instead of plain println!
unc0rr
parents: 12141
diff changeset
    36
        poll.register(&self.sock, token, Ready::all(),
12133
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    37
                      PollOpt::edge())
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    38
            .ok().expect("could not register socket with event loop");
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    39
12142
193dfdcb0620 - Use logging facilities instead of plain println!
unc0rr
parents: 12141
diff changeset
    40
        self.send_msg(Connected(utils::PROTOCOL_VERSION));
12133
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    41
    }
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    42
12144
f3121d7dedec - Handle errors
unc0rr
parents: 12143
diff changeset
    43
    pub fn deregister(&mut self, poll: &Poll) {
f3121d7dedec - Handle errors
unc0rr
parents: 12143
diff changeset
    44
        poll.deregister(&self.sock);
f3121d7dedec - Handle errors
unc0rr
parents: 12143
diff changeset
    45
    }
f3121d7dedec - Handle errors
unc0rr
parents: 12143
diff changeset
    46
12143
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12142
diff changeset
    47
    pub fn send_raw_msg(&mut self, msg: &[u8]) {
12133
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    48
        self.buf_out.write(msg).unwrap();
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    49
        self.flush();
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    50
    }
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    51
12143
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12142
diff changeset
    52
    pub fn send_string(&mut self, msg: &String) {
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12142
diff changeset
    53
        self.send_raw_msg(&msg.as_bytes());
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12142
diff changeset
    54
    }
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12142
diff changeset
    55
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12142
diff changeset
    56
    pub fn send_msg(&mut self, msg: messages::HWProtocolMessage) {
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12142
diff changeset
    57
        self.send_string(&msg.to_raw_protocol());
12141
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    58
    }
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    59
12133
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    60
    fn flush(&mut self) {
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    61
        self.buf_out.write_to(&mut self.sock).unwrap();
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    62
        self.sock.flush();
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    63
    }
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    64
12143
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12142
diff changeset
    65
    pub fn readable(&mut self, poll: &Poll) -> Vec<Action> {
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12142
diff changeset
    66
        let v = self.decoder.read_from(&mut self.sock).unwrap();
12142
193dfdcb0620 - Use logging facilities instead of plain println!
unc0rr
parents: 12141
diff changeset
    67
        debug!("Read {} bytes", v);
12141
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    68
        let mut response = Vec::new();
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    69
        {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    70
            let msgs = self.decoder.extract_messages();
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    71
            for msg in msgs {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    72
                match msg {
12143
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12142
diff changeset
    73
                    Ping => response.push(SendMe(Pong.to_raw_protocol())),
12144
f3121d7dedec - Handle errors
unc0rr
parents: 12143
diff changeset
    74
                    Quit(Some(msg)) => response.push(ByeClient("User quit: ".to_string() + msg)),
f3121d7dedec - Handle errors
unc0rr
parents: 12143
diff changeset
    75
                    Quit(None) => response.push(ByeClient("User quit".to_string())),
12146
78925eff02c2 Basic support for NICK message
unc0rr
parents: 12144
diff changeset
    76
                    Nick(nick) => if self.nick.len() == 0 {
78925eff02c2 Basic support for NICK message
unc0rr
parents: 12144
diff changeset
    77
                        response.push(SetNick(nick.to_string()));
78925eff02c2 Basic support for NICK message
unc0rr
parents: 12144
diff changeset
    78
                    },
12142
193dfdcb0620 - Use logging facilities instead of plain println!
unc0rr
parents: 12141
diff changeset
    79
                    Malformed => warn!("Malformed/unknown message"),
193dfdcb0620 - Use logging facilities instead of plain println!
unc0rr
parents: 12141
diff changeset
    80
                    Empty => warn!("Empty message"),
193dfdcb0620 - Use logging facilities instead of plain println!
unc0rr
parents: 12141
diff changeset
    81
                    _ => unimplemented!(),
12141
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    82
                }
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    83
            }
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    84
        }
e25a82ce2374 - Render messages to string
unc0rr
parents: 12134
diff changeset
    85
        self.decoder.sweep();
12143
e0bf51609062 Introduce actions, just like in the old server
unc0rr
parents: 12142
diff changeset
    86
        response
12133
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    87
    }
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    88
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    89
    pub fn writable(&mut self, poll: &Poll) -> io::Result<()> {
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    90
        self.buf_out.write_to(&mut self.sock)?;
12144
f3121d7dedec - Handle errors
unc0rr
parents: 12143
diff changeset
    91
12133
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    92
        Ok(())
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    93
    }
12142
193dfdcb0620 - Use logging facilities instead of plain println!
unc0rr
parents: 12141
diff changeset
    94
12144
f3121d7dedec - Handle errors
unc0rr
parents: 12143
diff changeset
    95
    pub fn error(&mut self, poll: &Poll) -> Vec<Action> {
f3121d7dedec - Handle errors
unc0rr
parents: 12143
diff changeset
    96
        return vec![ByeClient("Connection reset".to_string())]
12142
193dfdcb0620 - Use logging facilities instead of plain println!
unc0rr
parents: 12141
diff changeset
    97
    }
12133
f50876f3eff8 Refactor modules layout
unc0rr
parents:
diff changeset
    98
}