57 lines
1.7 KiB
Rust
57 lines
1.7 KiB
Rust
use std::{net::{TcpStream, Shutdown}, thread, io::Write, sync::mpsc::{Receiver, Sender, TryRecvError}, time::Duration};
|
|
|
|
use log::debug;
|
|
|
|
use crate::{protocol::{data::{PacketDecoder}, serverbound::*, clientbound::*, NetworkState}};
|
|
|
|
use super::Player;
|
|
|
|
pub struct NetworkClient {
|
|
pub id: i32,
|
|
pub verified: bool,
|
|
pub closed: bool,
|
|
pub stream: TcpStream,
|
|
pub serverbound: Receiver<ServerBoundPacket>,
|
|
pub player: Option<Player>,
|
|
}
|
|
|
|
impl NetworkClient {
|
|
pub fn listen(mut stream: TcpStream, send: Sender<ServerBoundPacket>) {
|
|
let mut state = NetworkState::Handshake;
|
|
let dur = Duration::from_millis(5);
|
|
loop {
|
|
match PacketDecoder::decode(&mut stream) {
|
|
Ok(decoder) => {
|
|
let packet = ServerBoundPacket::decode(&mut state, decoder);
|
|
send.send(packet).unwrap();
|
|
}
|
|
Err(_) => break
|
|
}
|
|
thread::sleep(dur)
|
|
}
|
|
let _ = stream.shutdown(Shutdown::Both);
|
|
}
|
|
|
|
pub fn recv_packet(&mut self, alive: &mut bool) -> Option<ServerBoundPacket> {
|
|
match self.serverbound.try_recv() {
|
|
Ok(packet) => Some(packet),
|
|
Err(TryRecvError::Empty) => None,
|
|
Err(TryRecvError::Disconnected) => {
|
|
*alive = false;
|
|
None
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn send_packet(&mut self, packet: impl ClientBoundPacket) -> std::io::Result<()> {
|
|
self.stream.write_all(&encode_packet(packet))?;
|
|
Ok(())
|
|
}
|
|
|
|
pub fn close(&mut self) {
|
|
debug!("Closed connection id {}", self.id);
|
|
let _ = self.stream.shutdown(Shutdown::Both);
|
|
self.closed = true;
|
|
}
|
|
}
|