feat: make cli client take stdin async

This commit is contained in:
2026-08-17 18:38:30 +02:00
parent ed6fc3583a
commit 1be6ece665

View File

@@ -4,7 +4,7 @@ use std::{
}; };
use checkers::network::client::Client; use checkers::network::client::Client;
use log::{debug, error, info}; use log::{debug, error, info, warn};
use tokio::sync::mpsc; use tokio::sync::mpsc;
const QUIT_COMMAND: &str = "/quit"; const QUIT_COMMAND: &str = "/quit";
@@ -54,30 +54,40 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
let (client_sender, mut client_receiver) = mpsc::channel(1024); let (client_sender, mut client_receiver) = mpsc::channel(1024);
client.set_update_sender(client_sender); client.set_update_sender(client_sender);
client.update(); client.update();
let (stdin_sender, mut stdin_receiver) = mpsc::channel(1024);
tokio::task::spawn_blocking(move || {
let stdin = std::io::stdin();
loop {
let mut buffer = String::new();
if stdin.read_line(&mut buffer).is_ok() {
if stdin_sender.blocking_send(buffer).is_err() {
break;
}
}
}
});
let help_msg: String = format!( let help_msg: String = format!(
"\n┌────Help──── "\n┌────Help────────────────────────────
│\t {} - quit │\t {} - quit
│\t {} - show this message │\t {} - show this message
│\t {} - message the server you're ready │\t {} - message the server you're ready
│\t {} - send text message to the server │\t {} - send text message to the server
└───────────────────────────────────────
", ",
QUIT_COMMAND, HELP_COMMAND, READY_COMMAND, SEND_COMMAND, QUIT_COMMAND, HELP_COMMAND, READY_COMMAND, SEND_COMMAND,
); );
info!("====== Type /help for help ========"); info!("───────Type /help for help───────");
let mut buffer = String::new();
loop { loop {
print!("> "); tokio::select! {
io::stdout().flush()?; msg = client_receiver.recv() => {
match msg {
io::stdin().read_line(&mut buffer)?; Some(checkers::network::client::ClientData::GameStart {
buffer.pop(); // remove \n
match client_receiver.try_recv() {
Ok(checkers::network::client::ClientData::GameStart {
identity, identity,
board_view, board_view,
}) => { }) => {
@@ -88,28 +98,32 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
println!("{}", board_view.to_string(identity.id)); println!("{}", board_view.to_string(identity.id));
} }
Ok(checkers::network::client::ClientData::AvailableCaptures(capture_paths)) => { Some(checkers::network::client::ClientData::AvailableCaptures(capture_paths)) => {
todo!() todo!()
} }
Ok(checkers::network::client::ClientData::AvailableMoves(move_paths)) => todo!(), Some(checkers::network::client::ClientData::AvailableMoves(move_paths)) => todo!(),
Ok(checkers::network::client::ClientData::TextMessage(content)) => { Some(checkers::network::client::ClientData::TextMessage(content)) => {
info!("Got message from the server: {}", content); info!("Got message from the server: {}", content);
} }
Ok(checkers::network::client::ClientData::GameEnd(game_result)) => { Some(checkers::network::client::ClientData::GameEnd(game_result)) => {
info!("Game has ended with the result: {:?}", game_result); info!("Game has ended with the result: {:?}", game_result);
} }
_ => {} None => {
warn!("Connection broken! Stopping");
break Ok(());
}
}
} }
if buffer.is_empty() { buffer_opt = stdin_receiver.recv() => {
buffer.clear(); if let Some(mut buffer) = buffer_opt {
continue; buffer.pop(); // remove \n
}
if !buffer.is_empty() {
let words: Vec<&str> = buffer.split_whitespace().collect(); let words: Vec<&str> = buffer.split_whitespace().collect();
let cmd = words[0]; let cmd = words[0];
let mut args: String = words[1..].join(" "); let mut args: String = words[1..].join(" ");
@@ -119,25 +133,25 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
Ok(cmd) => match cmd { Ok(cmd) => match cmd {
CliCommands::Quit => { CliCommands::Quit => {
info!("Bye bye"); info!("Bye bye");
break; std::process::exit(0);
} }
CliCommands::Help => { CliCommands::Help => {
info!("{}", help_msg); info!("{}", help_msg);
buffer.clear();
} }
CliCommands::Ready => { CliCommands::Ready => {
client.signal_readiness().await; client.signal_readiness().await;
buffer.clear();
} }
CliCommands::Send => { CliCommands::Send => {
client.send_text_message(args).await; client.send_text_message(args).await;
buffer.clear();
} }
}, },
Err(e) => error!("{}", e), Err(e) => error!("{}", e),
} }
buffer.clear();
} }
print!("> ");
return Ok(()); io::stdout().flush()?;
}
}
}
}
} }