use fizyr_rpc::TcpPeer;

#[derive(clap::Parser)]
struct Options {
	#[clap(default_value = "localhost:12345")]
	address: String,
}

#[tokio::main(flavor = "current_thread")]
async fn main() {
	if let Err(e) = do_main(&clap::Parser::parse()).await {
		eprintln!("Error: {}", e);
		std::process::exit(1);
	}
}

async fn do_main(options: &Options) -> Result<(), String> {
	// Connect to a remote server.
	let (peer, info) = TcpPeer::connect(&options.address, Default::default()).await
		.map_err(|e| format!("failed to connect to {}: {}", options.address, e))?;
	eprintln!("Connected to: {}", info.remote_address());

	// Send a request to the remote peer.
	let mut request = peer
		.send_request(1, &b"Hello World!"[..])
		.await
		.map_err(|e| format!("failed to send request: {}", e))?;

	while let Some(update) = request.recv_update().await {
		let message = std::str::from_utf8(&update.body.data).map_err(|_| "invalid UTF-8 in update")?;
		eprintln!("Received update: {}", message);
	}

	let response = request
		.recv_response()
		.await
		.map_err(|e| format!("failed to read message: {}", e))?;
	// Parse the message body as UTF-8, print it and exit the loop.
	let message = std::str::from_utf8(&response.body.data).map_err(|_| "invalid UTF-8 in response")?;
	eprintln!("Received response: {}", message);

	Ok(())
}