Prusa working

1 files changed, 51 insertions, 31 deletions

diff --git a/src/main.rs b/src/main.rs
index 11a2e43..6c353e6 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -16,29 +16,40 @@ async fn main() {
 
     for printer in config.printers {
         match printer {
-            Printer::Prusa { name, host, api_key, .. } => {
+            Printer::Prusa {
+                name,
+                host,
+                api_key,
+                ..
+            } => {
                 let state_clone = state.clone();
                 tokio::spawn(async move {
-                    let client = Client::new();
-                    let response = client.get(format!("http://{}", host))
-                        .header("X-Api-Key", api_key)
-                        .send()
-                        .await
-                        .unwrap()
-                        .json::<serde_json::Value>()
-                        .await
-                        .unwrap();
-                    let mut lock = state_clone.lock().await;
-                    println!("{:?}", &response);
-                    /*
-                    if let Ok(msg) = serde_json::from_slice::<BambuMessage>(&p.payload) {
-                        lock.entry(name.clone())
-                            .and_modify(|prs| prs.bed_temp = msg.print.bed_temper)
-                            .or_insert_with(|| PrinterState { name: name.clone(), bed_temp: msg.print.bed_temper, ..Default::default() });
-                        println!("Updated state for {}: {:?}", &name, p.payload);
+                    loop {
+                        let result: Result<(), Box<dyn std::error::Error + Send + Sync>> = async {
+                            let client = Client::new();
+                            let response = client
+                                .get(format!("http://{}/api/v1/status", host))
+                                .header("X-Api-Key", &api_key)
+                                .send()
+                                .await?
+                                .json::<PrusaStatus>()
+                                .await?;
+                            let mut lock = state_clone.lock().await;
+                            lock.entry(name.clone())
+                                .and_modify(|prs| prs.bed_temp = response.printer.temp_bed)
+                                .or_insert_with(|| PrinterState {
+                                    name: name.clone(),
+                                    bed_temp: response.printer.temp_bed,
+                                    ..Default::default()
+                                });
+                            Ok(())
+                        }
+                        .await;
+                        if let Err(e) = result {
+                            eprintln!("Error polling Prusa printer {}: {}", name, e);
+                        }
+                        tokio::time::sleep(Duration::from_secs(5)).await;
                     }
-                    */
-                    println!("Found Prusa: {} at {}", name, host);
                 });
             }
             Printer::Bambu {
@@ -74,10 +85,16 @@ async fn main() {
                                 if let Event::Incoming(Packet::Publish(p)) = notification {
                                     let mut lock = state_clone.lock().await;
                                     println!("{:?}", &p.payload);
-                                    if let Ok(msg) = serde_json::from_slice::<BambuMessage>(&p.payload) {
+                                    if let Ok(msg) =
+                                        serde_json::from_slice::<BambuMessage>(&p.payload)
+                                    {
                                         lock.entry(name.clone())
                                             .and_modify(|prs| prs.bed_temp = msg.print.bed_temper)
-                                            .or_insert_with(|| PrinterState { name: name.clone(), bed_temp: msg.print.bed_temper, ..Default::default() });
+                                            .or_insert_with(|| PrinterState {
+                                                name: name.clone(),
+                                                bed_temp: msg.print.bed_temper,
+                                                ..Default::default()
+                                            });
                                         println!("Updated state for {}: {:?}", &name, p.payload);
                                     }
                                 }
@@ -94,15 +111,18 @@ async fn main() {
     }
 
     loop {
-        let state_clone = Arc::clone(&state);
-        println!("-- PRINTER STATE --");
-
-        let lock = state_clone.lock().await;
-
-        for (key, value) in lock.iter() {
-            println!("{}: {:?}", key, value);
+        tokio::select! {
+            _ = tokio::signal::ctrl_c() => {
+                println!("Shutting down...");
+                break;
+            }
+            _ = tokio::time::sleep(Duration::from_secs(1)) => {
+                println!("-- PRINTER STATE --");
+                let lock = state.lock().await;
+                for (key, value) in lock.iter() {
+                    println!("{}: {:?}", key, value);
+                }
+            }
         }
-
-        tokio::time::sleep(Duration::from_secs(1)).await;
     }
 }