106 lines
2.8 KiB
Rust
106 lines
2.8 KiB
Rust
|
pub use iced_core as core;
|
||
|
|
|
||
|
|
pub mod client;
|
||
|
|
pub mod timing;
|
||
|
|
|
||
|
|
use crate::timing::Timing;
|
||
|
|
|
||
|
|
use futures::future;
|
||
|
|
use futures::stream::{self, Stream, StreamExt};
|
||
|
|
use semver::Version;
|
||
|
|
use serde::{Deserialize, Serialize};
|
||
|
|
use tokio::io::{self, AsyncBufReadExt, BufStream};
|
||
|
|
use tokio::net;
|
||
|
|
|
||
|
|
pub const SOCKET_ADDRESS: &str = "127.0.0.1:9167";
|
||
|
|
|
||
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||
|
|
pub enum Input {
|
||
|
|
Connected(Version),
|
||
|
|
Reported(Report),
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Clone)]
|
||
|
|
pub enum Event {
|
||
|
|
Connected(Version),
|
||
|
|
Disconnected,
|
||
|
|
Reported(Report),
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||
|
|
pub enum Report {
|
||
|
|
Timing(Timing),
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn run() -> impl Stream<Item = Event> {
|
||
|
|
enum State {
|
||
|
|
Disconnected,
|
||
|
|
Connected(BufStream<net::TcpStream>),
|
||
|
|
}
|
||
|
|
|
||
|
|
stream::unfold(State::Disconnected, |state| async {
|
||
|
|
match state {
|
||
|
|
State::Disconnected => match connect().await {
|
||
|
|
Ok(stream) => {
|
||
|
|
let stream = BufStream::new(stream);
|
||
|
|
|
||
|
|
Some((None, State::Connected(stream)))
|
||
|
|
}
|
||
|
|
Err(_error) => Some((None, State::Disconnected)),
|
||
|
|
},
|
||
|
|
State::Connected(stream) => match receive(stream).await {
|
||
|
|
Ok((_, Event::Disconnected)) | Err(_) => {
|
||
|
|
Some((Some(Event::Disconnected), State::Disconnected))
|
||
|
|
}
|
||
|
|
Ok((stream, message)) => {
|
||
|
|
Some((Some(message), State::Connected(stream)))
|
||
|
|
}
|
||
|
|
},
|
||
|
|
}
|
||
|
|
})
|
||
|
|
.filter_map(future::ready)
|
||
|
|
}
|
||
|
|
|
||
|
|
async fn connect() -> Result<net::TcpStream, io::Error> {
|
||
|
|
let listener = net::TcpListener::bind(SOCKET_ADDRESS).await?;
|
||
|
|
|
||
|
|
let (stream, _) = listener.accept().await?;
|
||
|
|
|
||
|
|
stream.set_nodelay(true)?;
|
||
|
|
stream.readable().await?;
|
||
|
|
|
||
|
|
Ok(stream)
|
||
|
|
}
|
||
|
|
|
||
|
|
async fn receive(
|
||
|
|
mut stream: BufStream<net::TcpStream>,
|
||
|
|
) -> Result<(BufStream<net::TcpStream>, Event), io::Error> {
|
||
|
|
let mut input = String::new();
|
||
|
|
|
||
|
|
loop {
|
||
|
|
match stream.read_line(&mut input).await? {
|
||
|
|
0 => return Ok((stream, Event::Disconnected)),
|
||
|
|
n => {
|
||
|
|
match serde_json::from_str(&input[..n]) {
|
||
|
|
Ok(input) => {
|
||
|
|
return Ok((
|
||
|
|
stream,
|
||
|
|
match dbg!(input) {
|
||
|
|
Input::Connected(version) => {
|
||
|
|
Event::Connected(version)
|
||
|
|
}
|
||
|
|
Input::Reported(report) => {
|
||
|
|
Event::Reported(report)
|
||
|
|
}
|
||
|
|
},
|
||
|
|
))
|
||
|
|
}
|
||
|
|
Err(_) => {
|
||
|
|
// TODO: Log decoding error
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|