Add WIP websocket to controller and refactor settings

emgauwa-controller/src/main.rs

@@ -1,5 +1,53 @@
+use std::str;
+
+use futures::{future, pin_mut, StreamExt};
+use futures::channel::mpsc;
+use tokio::io::{AsyncReadExt, AsyncWriteExt};
+use tokio_tungstenite::{connect_async, tungstenite::protocol::Message};
+
 mod settings;
 
-fn main() {
-	let _settings = settings::init();
+#[tokio::main]
+async fn main() {
+	let settings = settings::init();
+
+	let url = format!(
+		"ws://{}:{}/api/v1/ws/controllers",
+		settings.core.host,
+		settings.core.port
+	);
+
+	let (stdin_tx, stdin_rx) = mpsc::unbounded();
+	tokio::spawn(read_stdin(stdin_tx));
+
+	let (ws_stream, _) = connect_async(url).await.expect("Failed to connect");
+	println!("WebSocket handshake has been successfully completed");
+
+	let (write, read) = ws_stream.split();
+
+	let stdin_to_ws = stdin_rx.map(Ok).forward(write);
+	let ws_to_stdout = {
+		read.for_each(|message| async {
+			let data = message.unwrap().into_text().unwrap();
+			println!("{}", data);
+		})
+	};
+
+	pin_mut!(stdin_to_ws, ws_to_stdout);
+	future::select(stdin_to_ws, ws_to_stdout).await;
 }
+
+// Our helper method which will read data from stdin and send it along the
+// sender provided.
+async fn read_stdin(tx: mpsc::UnboundedSender<Message>) {
+	let mut stdin = tokio::io::stdin();
+	loop {
+		let mut buf = vec![0; 1024];
+		let n = match stdin.read(&mut buf).await {
+			Err(_) | Ok(0) => break,
+			Ok(n) => n,
+		};
+		buf.truncate(n);
+		tx.unbounded_send(Message::text(str::from_utf8(&buf).unwrap())).unwrap();
+	}
+}