🙃

backend/src/main.rs

@@ -3,13 +3,15 @@ mod evaluator;
 mod parser;
 mod tokenizer;
 
-use std::io;
-
-use websocket::server::WsServer;
+use futures_util::{StreamExt, TryStreamExt, future};
+use log::info;
+use std::{env, io::Error};
+use tokio::net::{TcpListener, TcpStream};
 
 use crate::{cell::CellRef, evaluator::Evaluator};
 
-fn main() {
+#[tokio::main]
+async fn main() -> Result<(), Error> {
     // let mut input = String::new();
     // io::stdin().read_line(&mut input).expect("Expected input.");
 
@@ -21,49 +23,79 @@ fn main() {
     // evaluator.set_cell(a1, input).unwrap();
     // println!("{:?}", evaluator.get_cell(a1).unwrap());
 
-    println!("CMDS : set <cell_ref>, get <cell_ref>");
-    loop {
-        let mut input = String::new();
-        io::stdin().read_line(&mut input).expect("Expected input.");
+    let addr = env::args()
+        .nth(1)
+        .unwrap_or_else(|| "127.0.0.1:7050".to_string());
 
-        let cmds = ["set", "get"];
-        let cmd = &input[0..3];
-        if !cmds.iter().any(|c| c == &cmd) {
-            println!("{} is an invalid command!", cmd);
-            println!("CMDS : set <cell_ref>, get <cell_ref>");
-            continue;
-        }
+    // Create the event loop and TCP listener we'll accept connections on.
+    let try_socket = TcpListener::bind(&addr).await;
+    let listener = try_socket.expect("Failed to bind");
+    info!("Listening on: {}", addr);
 
-        let rest = &input[4..];
-        let mut parts = rest.splitn(2, char::is_whitespace);
-
-        let raw_ref = parts.next().unwrap_or("").trim(); // cell reference
-        let raw_str = parts.next().unwrap_or("").trim(); // rest of the string (value)
-        // println!("{} {}", raw_ref, raw_str);
-
-        if let Ok(cell_ref) = CellRef::new(raw_ref.to_owned()) {
-            match cmd {
-                "set" => match evaluator.set_cell(cell_ref, raw_str.to_owned()) {
-                    Ok(_) => println!("Successfully set cell {} to {}.", raw_ref, raw_str),
-                    Err(e) => println!("{}", e),
-                },
-                "get" => match evaluator.get_cell(cell_ref) {
-                    Ok(res) => println!("{:?}", res),
-                    Err(e) => println!("{}", e),
-                },
-                _ => {
-                    panic!("Impossible.");
-                }
-            }
-        } else {
-            println!("{} is an invalid cell reference!", raw_ref);
-            continue;
-        }
+    while let Ok((stream, _)) = listener.accept().await {
+        tokio::spawn(accept_connection(stream));
     }
 
-    // let rt = Runtime::new().unwrap();
-    let handle = rt.handle().clone();
-    // let addr = "127.0.0.1:7050";
+    Ok(())
+
+    // println!("CMDS : set <cell_ref>, get <cell_ref>");
+    // loop {
+    //     let mut input = String::new();
+    //     io::stdin().read_line(&mut input).expect("Expected input.");
     //
-    // let socket = WsServer::bind(addr, handle);
+    //     let cmds = ["set", "get"];
+    //     let cmd = &input[0..3];
+    //     if !cmds.iter().any(|c| c == &cmd) {
+    //         println!("{} is an invalid command!", cmd);
+    //         println!("CMDS : set <cell_ref>, get <cell_ref>");
+    //         continue;
+    //     }
+    //
+    //     let rest = &input[4..];
+    //     let mut parts = rest.splitn(2, char::is_whitespace);
+    //
+    //     let raw_ref = parts.next().unwrap_or("").trim(); // cell reference
+    //     let raw_str = parts.next().unwrap_or("").trim(); // rest of the string (value)
+    //     // println!("{} {}", raw_ref, raw_str);
+    //
+    //     if let Ok(cell_ref) = CellRef::new(raw_ref.to_owned()) {
+    //         match cmd {
+    //             "set" => match evaluator.set_cell(cell_ref, raw_str.to_owned()) {
+    //                 Ok(_) => println!("Successfully set cell {} to {}.", raw_ref, raw_str),
+    //                 Err(e) => println!("{}", e),
+    //             },
+    //             "get" => match evaluator.get_cell(cell_ref) {
+    //                 Ok(res) => println!("{:?}", res),
+    //                 Err(e) => println!("{}", e),
+    //             },
+    //             _ => {
+    //                 panic!("Impossible.");
+    //             }
+    //         }
+    //     } else {
+    //         println!("{} is an invalid cell reference!", raw_ref);
+    //         continue;
+    //     }
+    // }
+}
+
+async fn accept_connection(stream: TcpStream) {
+    let addr = stream
+        .peer_addr()
+        .expect("connected streams should have a peer address");
+    info!("Peer address: {}", addr);
+
+    let ws_stream = tokio_tungstenite::accept_async(stream)
+        .await
+        .expect("Error during the websocket handshake occurred");
+
+    info!("New WebSocket connection: {}", addr);
+
+    let (write, read) = ws_stream.split();
+
+    // We should not forward messages other than text or binary.
+    read.try_filter(|msg| future::ready(msg.is_text() || msg.is_binary()))
+        .forward(write)
+        .await
+        .expect("Failed to forward messages");
 }