Initial Commit

src/main.rs

@@ -0,0 +1,62 @@
+use ::std::sync::Arc;
+use clap::Parser;
+
+use cli::Cli;
+use structs::communication::RadarData;
+use tokio::sync::RwLock;
+
+mod dma;
+mod sdk;
+mod structs;
+mod cli;
+mod webserver;
+
+#[tokio::main]
+async fn main() -> anyhow::Result<()> {
+    let cli = Cli::parse();
+
+    if std::env::var("RADARFLOW_LOG").is_err() {
+        std::env::set_var("RADARFLOW_LOG", "warn")
+    }
+
+    simple_logger::SimpleLogger::new()
+        .with_level(cli.loglevel.into())
+        .init()
+        .expect("Initializing logger");
+
+    let rwlock = Arc::new(
+        RwLock::new(
+            RadarData::empty()
+        )
+    );
+
+    let rwlock_clone = rwlock.clone();
+    let dma_handle = tokio::spawn(async move {
+        if let Err(err) = dma::run(cli.connector, cli.pcileech_device, cli.poll_rate, rwlock_clone).await {
+            log::error!("Error in dma thread: {}", err.to_string());
+        }
+
+    });
+
+    tokio::spawn(async move {
+        let future = webserver::run(cli.web_path, cli.port, rwlock);
+
+        if let Ok(my_local_ip) = local_ip_address::local_ip() {
+            let address = format!("http://{}:{}", my_local_ip, cli.port);
+            println!("Launched webserver at {}", address);
+        } else {
+            let address = format!("http://0.0.0.0:{}", cli.port);
+            println!("launched webserver at! {}", address);
+        }
+
+        if let Err(err) = future.await {
+            log::error!("Error in websocket server: {}", err.to_string());
+        }
+    });
+
+    if let Err(err) = dma_handle.await {
+        log::error!("Error when waiting for dma thread: {}", err.to_string());
+    }
+
+    Ok(())
+}