aboutsummaryrefslogtreecommitdiff
path: root/pixel-client/src/network.rs
diff options
context:
space:
mode:
authormetamuffin <metamuffin@disroot.org>2024-08-11 00:22:49 +0200
committermetamuffin <metamuffin@disroot.org>2024-08-11 00:30:43 +0200
commitf4499b7a0e808d84157e6148afef4f844a7007cf (patch)
tree354d970af16a4b3d6217575d259cf1afc0faf291 /pixel-client/src/network.rs
parent0f6a824cb4c6d36dda485a939d9c640ec234d802 (diff)
downloadhurrycurry-f4499b7a0e808d84157e6148afef4f844a7007cf.tar
hurrycurry-f4499b7a0e808d84157e6148afef4f844a7007cf.tar.bz2
hurrycurry-f4499b7a0e808d84157e6148afef4f844a7007cf.tar.zst
factor out client network code to client-lib
Diffstat (limited to 'pixel-client/src/network.rs')
-rw-r--r--pixel-client/src/network.rs148
1 files changed, 0 insertions, 148 deletions
diff --git a/pixel-client/src/network.rs b/pixel-client/src/network.rs
deleted file mode 100644
index e8aa08de..00000000
--- a/pixel-client/src/network.rs
+++ /dev/null
@@ -1,148 +0,0 @@
-/*
- Hurry Curry! - a game about cooking
- Copyright 2024 metamuffin
-
- This program is free software: you can redistribute it and/or modify
- it under the terms of the GNU Affero General Public License as published by
- the Free Software Foundation, version 3 of the License only.
-
- This program is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- GNU Affero General Public License for more details.
-
- You should have received a copy of the GNU Affero General Public License
- along with this program. If not, see <https://www.gnu.org/licenses/>.
-
-*/
-use anyhow::Result;
-use hurrycurry_protocol::{PacketC, PacketS, BINCODE_CONFIG, VERSION};
-use log::{debug, info, warn};
-use std::{collections::VecDeque, net::TcpStream};
-use tungstenite::{
- client::{uri_mode, IntoClientRequest},
- client_tls_with_config,
- handshake::client::Request,
- stream::{MaybeTlsStream, Mode},
- util::NonBlockingError,
- Message, WebSocket,
-};
-
-pub struct Network {
- sock: WebSocket<MaybeTlsStream<TcpStream>>,
- pub queue_in: VecDeque<PacketC>,
- pub queue_out: VecDeque<PacketS>,
- use_bincode: bool,
-}
-
-impl Network {
- pub fn connect(addr: &str) -> Result<Self> {
- let (parts, _) = addr.into_client_request().unwrap().into_parts();
- let mut builder = Request::builder()
- .uri(parts.uri.clone().clone())
- .method(parts.method.clone())
- .version(parts.version);
- *builder.headers_mut().unwrap() = parts.headers.clone();
- let request = builder.body(()).unwrap();
-
- let host = request.uri().host().unwrap();
- let host = if host.starts_with('[') {
- &host[1..host.len() - 1]
- } else {
- host
- };
- let port = request
- .uri()
- .port_u16()
- .unwrap_or(match uri_mode(request.uri())? {
- Mode::Plain => 27032,
- Mode::Tls => 443,
- });
-
- info!("Connecting: host={host:?} port={port}");
- let stream = TcpStream::connect((host, port))?;
- stream.set_nodelay(true).unwrap();
-
- let (mut sock, _) = client_tls_with_config(request, stream, None, None).unwrap();
-
- match sock.get_mut() {
- MaybeTlsStream::Plain(s) => s.set_nonblocking(true).unwrap(),
- MaybeTlsStream::Rustls(s) => s.sock.set_nonblocking(true).unwrap(),
- _ => todo!(),
- };
-
- info!("Handshake complete.");
- Ok(Self {
- sock,
- use_bincode: false,
- queue_in: VecDeque::new(),
- queue_out: VecDeque::new(),
- })
- }
-
- pub fn poll(&mut self) {
- loop {
- self.queue_in.extend(match self.sock.read() {
- Ok(Message::Text(packet)) => match serde_json::from_str(&packet) {
- Ok(packet) => {
- debug!("<- {packet:?}");
- if let PacketC::Version {
- minor,
- major,
- supports_bincode,
- } = &packet
- {
- if *minor == VERSION.0 && *major == VERSION.1 && *supports_bincode {
- info!("Binary protocol format enabled.");
- self.use_bincode = true;
- }
- }
- Some(packet)
- }
- Err(e) => {
- warn!("invalid json packet: {e:?}");
- None
- }
- },
- Ok(Message::Binary(packet)) => {
- match bincode::decode_from_slice(&packet, BINCODE_CONFIG) {
- Ok((packet, _)) => {
- debug!("<- {packet:?}");
- Some(packet)
- }
- Err(e) => {
- warn!("invalid bincode packet: {e:?}");
- None
- }
- }
- }
- Ok(_) => None,
- Err(e) => {
- if let Some(e) = e.into_non_blocking() {
- warn!("{e:?}");
- None
- } else {
- break;
- }
- }
- });
- }
-
- for packet in self.queue_out.drain(..) {
- debug!("-> {packet:?}");
- if self.use_bincode {
- self.sock
- .write(Message::Binary(
- bincode::encode_to_vec(&packet, BINCODE_CONFIG).unwrap(),
- ))
- .unwrap();
- } else {
- self.sock
- .write(Message::Text(serde_json::to_string(&packet).unwrap()))
- .unwrap();
- }
- }
-
- self.sock.flush().unwrap();
- }
-}