feat: add thermal state to rocket

Author: eltorio

rust/src/daemon/launch_daemon.rs

@@ -328,6 +328,7 @@ impl Daemon {
             self.audio_stream.clone(),
             Some(Arc::clone(&self.visualization_state)),
             Some(Arc::clone(&self.streaming_registry)),
+            Some(self.thermal_regulation_state.clone()),
         )
         .await;
 

rust/src/visualization/server/builder.rs

@@ -13,6 +13,7 @@ use crate::acquisition::SharedAudioStream;
 use crate::config::{Config, GenerixConfig};
 use crate::include_png_as_base64;
 use crate::processing::nodes::streaming_registry::StreamingNodeRegistry;
+use crate::thermal_regulation::SharedThermalState;
 use crate::visualization::api::graph::graph::*;
 use crate::visualization::api::*;
 use crate::visualization::auth::{
@@ -51,6 +52,7 @@ use tokio::sync::RwLock;
 /// * `audio_stream` - Optional shared audio stream for real-time audio endpoints
 /// * `visualization_state` - Optional shared visualization state for statistics
 /// * `streaming_registry` - Optional streaming node registry for audio processing
+/// * `thermal_state` - Optional shared thermal regulation state for temperature control
 ///
 /// ### Returns
 ///
@@ -72,7 +74,7 @@ use tokio::sync::RwLock;
 /// async fn example() {
 ///     let figment = Figment::from(rocket::Config::default());
 ///     let config = Arc::new(RwLock::new(Config::default()));
-///     let rocket = server::build_rocket(figment, config, None, None, None).await;
+///     let rocket = server::build_rocket(figment, config, None, None, None, None).await;
 ///     // Launch the server
 ///     // rocket.launch().await.expect("Failed to launch");
 /// }
@@ -83,6 +85,7 @@ pub async fn build_rocket(
     audio_stream: Option<Arc<SharedAudioStream>>,
     visualization_state: Option<Arc<SharedVisualizationState>>,
     streaming_registry: Option<Arc<StreamingNodeRegistry>>,
+    thermal_state: Option<SharedThermalState>,
 ) -> Rocket<Build> {
     // Load hmac secret from config
     let config_read = config.read().await;
@@ -202,6 +205,15 @@ pub async fn build_rocket(
         .manage(jwt_validator)
         .manage(config.clone()); // Add config as managed state for future dynamic configuration
 
+    // Add thermal regulation state if available
+    let rocket_builder = if let Some(thermal_state) = thermal_state {
+        debug!("Adding SharedThermalState to Rocket state management");
+        rocket_builder.manage(thermal_state)
+    } else {
+        debug!("No thermal state provided, thermal regulation API will return 404");
+        rocket_builder
+    };
+
     // Attach compression fairing if enabled in config and not using VITE_DEVELOPMENT
     // VITE_DEVELOPMENT is an environment variable for proxying Vite dev server
     let rocket_builder = if compression_config && !std::env::var("VITE_DEVELOPMENT").is_ok() {

rust/src/visualization/server/mod.rs

@@ -49,7 +49,7 @@
 //!         .merge(("hmac_secret", "your-secret-key".to_string()));
 //!     
 //!     let config = Arc::new(RwLock::new(Config::default()));
-//!     let rocket = server::build_rocket(figment, config, None, None, None).await;
+//!     let rocket = server::build_rocket(figment, config, None, None, None, None).await;
 //!     rocket.launch().await.expect("Failed to launch server");
 //! }
 //! ```

rust/tests/api_jwt_test.rs

@@ -52,6 +52,7 @@ async fn test_protected_api_with_jwt() {
         None,
         Some(visualization_state),
         None,
+        None,
     )
     .await;
     let client = rocket::local::asynchronous::Client::tracked(rocket)

rust/tests/rs256_jwt_test.rs

@@ -194,6 +194,7 @@ async fn test_oidc_endpoints_with_rs256() {
         None,
         None,
         None,
+        None,
     )
     .await;
 
@@ -287,6 +288,7 @@ async fn test_token_endpoint_with_rs256() {
         None,
         None,
         None,
+        None,
     )
     .await;
 

rust/tests/rs256_pkce_flow_test.rs

@@ -480,6 +480,7 @@ async fn test_rs256_pkce_flow_s256() {
         None,
         None,
         None,
+        None,
     )
     .await;
 
@@ -555,6 +556,7 @@ async fn test_rs256_pkce_flow_plain() {
         None,
         None,
         None,
+        None,
     )
     .await;
 
@@ -636,6 +638,7 @@ async fn test_rs256_pkce_flow() {
             None,
             None,
             None,
+            None,
         )
         .await;
 
@@ -682,6 +685,7 @@ async fn test_rs256_jwks_endpoint() {
         None,
         None,
         None,
+        None,
     )
     .await;
 
@@ -768,6 +772,7 @@ async fn test_rs256_pkce_invalid_challenge_method() {
         None,
         None,
         None,
+        None,
     )
     .await;
 

rust/tests/server_test_only.rs

@@ -122,6 +122,7 @@ async fn test_oauth2_pkce_flow() {
         None,
         None,
         None,
+        None,
     )
     .await;
     let client = Client::tracked(rocket)
@@ -390,6 +391,7 @@ async fn test_oauth2_invalid_credentials() {
         None,
         None,
         None,
+        None,
     )
     .await;
     let client = Client::tracked(rocket)