improved tree colouring

1 files changed, 32 insertions, 11 deletions

diff --git a/light-client/src/network.rs b/light-client/src/network.rs
index e3cd1eb5..997a3344 100644
--- a/light-client/src/network.rs
+++ b/light-client/src/network.rs
@@ -1,3 +1,20 @@
+/*
+    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};
 use log::warn;
@@ -19,32 +36,36 @@ pub struct Network {
 impl Network {
     pub fn connect(addr: &str) -> Result<Self> {
         let (parts, _) = addr.into_client_request().unwrap().into_parts();
-        let uri = parts.uri.clone();
-
         let mut builder = Request::builder()
-            .uri(uri.clone())
+            .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 uri = request.uri();
-        let mode = uri_mode(uri)?;
         let host = request.uri().host().unwrap();
         let host = if host.starts_with('[') {
             &host[1..host.len() - 1]
         } else {
             host
         };
-        let port = uri.port_u16().unwrap_or(match mode {
-            Mode::Plain => 27032,
-            Mode::Tls => 443,
-        });
+        let port = request
+            .uri()
+            .port_u16()
+            .unwrap_or(match uri_mode(request.uri())? {
+                Mode::Plain => 27032,
+                Mode::Tls => 443,
+            });
         let stream = TcpStream::connect((host, port))?;
         stream.set_nodelay(true).unwrap();
-        // stream.set_nonblocking(true).unwrap();
 
-        let (sock, _) = client_tls_with_config(request, stream, None, None).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!(),
+        };
 
         Ok(Self {
             sock,