95 lines
No EOL
2.5 KiB
Rust
95 lines
No EOL
2.5 KiB
Rust
use std::str;
|
|
|
|
use futures::{future, pin_mut, SinkExt, StreamExt};
|
|
use futures::channel::mpsc;
|
|
use sqlx::pool::PoolConnection;
|
|
use sqlx::Sqlite;
|
|
use tokio::io::AsyncReadExt;
|
|
use tokio_tungstenite::{connect_async, tungstenite::protocol::Message};
|
|
use tokio_tungstenite::tungstenite::Error;
|
|
use emgauwa_lib::db;
|
|
use emgauwa_lib::db::Controller;
|
|
use emgauwa_lib::db::types::ControllerUid;
|
|
use crate::relay_loop::run_relay_loop;
|
|
use crate::settings::Settings;
|
|
|
|
mod settings;
|
|
mod driver;
|
|
mod relay_loop;
|
|
|
|
fn create_this_controller(conn: &mut PoolConnection<Sqlite>, settings: &Settings) -> Controller {
|
|
futures::executor::block_on(async {
|
|
Controller::create(
|
|
conn,
|
|
&ControllerUid::new(),
|
|
&settings.name,
|
|
i64::try_from(settings.relays.len()).expect("Too many relays"),
|
|
true
|
|
).await.expect("Failed to create controller")
|
|
})
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let settings = settings::init();
|
|
|
|
let pool = db::init(&settings.database).await;
|
|
|
|
let mut conn = pool.acquire().await.unwrap();
|
|
|
|
let this = Controller::get_all(&mut conn)
|
|
.await
|
|
.expect("Failed to get controller from database")
|
|
.pop()
|
|
.unwrap_or_else(|| create_this_controller(&mut conn, &settings));
|
|
|
|
let this_json = serde_json::to_string(&this).unwrap();
|
|
|
|
println!("{:?}", settings.relays);
|
|
println!("{:?}", this);
|
|
println!("{}", this_json);
|
|
|
|
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");
|
|
|
|
let (mut write, read) = ws_stream.split();
|
|
|
|
write.send(Message::text(this_json)).await.unwrap();
|
|
let ws_to_stdout = read.for_each(handle_message);
|
|
let stdin_to_ws = stdin_rx.map(Ok).forward(write);
|
|
|
|
tokio::spawn(run_relay_loop(settings));
|
|
|
|
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();
|
|
}
|
|
}
|
|
|
|
pub async fn handle_message(message_result: Result<Message, Error>) {
|
|
match message_result {
|
|
Ok(message) => println!("{}", message.into_text().unwrap()),
|
|
Err(err) => println!("Error: {}", err)
|
|
}
|
|
} |