1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
use axum::{Json, Router, extract::State, routing::get};
use native_tls::TlsConnector;
use printstats::*;
use reqwest::Client;
use rumqttc::{AsyncClient, Event, MqttOptions, Packet, QoS, Transport};
use std::collections::HashMap;
use std::fs;
use std::sync::Arc;
use std::time::Duration;
use tokio::sync::Mutex;
type StateMap = Arc<Mutex<HashMap<String, PrinterState>>>;
#[tokio::main]
async fn main() {
tracing_subscriber::fmt::init();
let content = fs::read_to_string("config.toml").expect("Couldn't read config.toml");
let config = Config::load(&content);
let state: StateMap = Arc::new(Mutex::new(HashMap::new()));
for printer in config.printers {
match printer {
Printer::Prusa { name, host, api_key } => {
tracing::info!(name, host, "Found Prusa");
let state_clone = state.clone();
tokio::spawn(poll_prusa(name, host, api_key, state_clone));
}
Printer::Bambu { name, host, serial_number, access_code } => {
tracing::info!(name, host, "Found Bambu");
let state_clone = state.clone();
tokio::spawn(poll_bambu(name, host, serial_number, access_code, state_clone));
}
}
}
let app = Router::new()
.route("/", get(root))
.with_state(state.clone());
let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
let _ = axum::serve(listener, app).await;
}
async fn poll_prusa(
name: String,
host: String,
api_key: String,
state: StateMap,
) {
let client = Client::new();
loop {
let result: Result<(), Box<dyn std::error::Error + Send + Sync>> = async {
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.lock().await;
let entry = lock.entry(name.clone()).or_default();
extract_status_from_prusa(&response, entry);
Ok(())
}
.await;
if let Err(e) = result {
tracing::error!(name, error = %e, "Error polling Prusa printer");
}
tokio::time::sleep(Duration::from_secs(5)).await;
}
}
async fn poll_bambu(
name: String,
host: String,
serial_number: String,
access_code: String,
state: StateMap,
) {
let mut mqttoptions = MqttOptions::new(&name, &host, 8883);
mqttoptions.set_keep_alive(Duration::from_secs(5));
mqttoptions.set_credentials("bblp", &access_code);
// Bambu printers use TLS with a self-signed certificate
let connector = TlsConnector::builder()
.danger_accept_invalid_certs(true)
.build()
.unwrap();
mqttoptions.set_transport(Transport::tls_with_config(connector.into()));
let (client, mut eventloop) = AsyncClient::new(mqttoptions, 10);
client
.subscribe(format!("device/{}/report", serial_number), QoS::AtMostOnce)
.await
.unwrap();
loop {
// eventloop.poll() yields back to Tokio when there's no data
match eventloop.poll().await {
Ok(Event::Incoming(Packet::Publish(p))) => {
tracing::debug!(payload = ?p.payload, "Received Bambu payload");
match serde_json::from_slice::<BambuStatus>(&p.payload) {
Ok(msg) => {
let mut lock = state.lock().await;
let entry = lock.entry(name.clone()).or_default();
extract_status_from_bambu(&msg, entry);
tracing::debug!(name, payload = ?p.payload, "Updated state");
}
Err(e) => tracing::error!(error = %e, "Failed to deserialize BambuStatus"),
}
}
Ok(_) => {}
Err(e) => {
tracing::error!(error = ?e, "MQTT error");
tokio::time::sleep(Duration::from_secs(5)).await; // Simple retry
}
}
}
}
async fn root(
State(state): State<StateMap>,
) -> Json<HashMap<String, PrinterState>> {
let lock = state.lock().await;
Json(lock.clone())
}
|
