wip: graph topology

Author: aeltorio

rust/examples/test_serializable_graph.rs

@@ -1,29 +1,29 @@
 // Test file to verify serializable ProcessingGraph functionality
-use rust_photoacoustic::processing::{ProcessingGraph, SerializableProcessingGraph};
 use rust_photoacoustic::processing::nodes::{InputNode, ProcessingNode};
+use rust_photoacoustic::processing::{ProcessingGraph, SerializableProcessingGraph};
 
 fn main() -> anyhow::Result<()> {
     // Create a simple processing graph
     let mut graph = ProcessingGraph::new();
-    
+
     // Add an input node
     let input_node = Box::new(InputNode::new("main_input".to_string()));
     graph.add_node(input_node)?;
-    
+
     // Convert to serializable format
     let serializable_graph: SerializableProcessingGraph = graph.into();
-    
+
     // Verify the conversion
     println!("Serializable Graph Created Successfully!");
     println!("Number of nodes: {}", serializable_graph.nodes.len());
     println!("Input node: {:?}", serializable_graph.input_node);
     println!("Graph is valid: {}", serializable_graph.is_valid);
-    
+
     if let Some(node) = serializable_graph.nodes.first() {
         println!("First node: {} (type: {})", node.id, node.node_type);
         println!("Accepts input types: {:?}", node.accepts_input_types);
     }
-    
+
     // Try serializing to JSON
     match serde_json::to_string_pretty(&serializable_graph) {
         Ok(json) => {
@@ -34,8 +34,8 @@ fn main() -> anyhow::Result<()> {
             println!("JSON serialization failed: {}", e);
         }
     }
-    
+
     println!("\n✅ All serializable ProcessingGraph functionality working correctly!");
-    
+
     Ok(())
 }

rust/src/visualization/api/get/graph_statistics.rs

@@ -10,8 +10,11 @@
 
 use rocket::serde::json::Json;
 use rocket::{get, response::status, State};
+use rocket_okapi::okapi::openapi3::OpenApi;
+use rocket_okapi::openapi_get_routes_spec;
 
 use crate::processing::graph::ProcessingGraphStatistics;
+use crate::processing::SerializableProcessingGraph;
 use crate::visualization::shared_state::SharedVisualizationState;
 use auth_macros::openapi_protect_get;
 
@@ -73,7 +76,7 @@ use auth_macros::openapi_protect_get;
 /// - `403 Forbidden`: Token lacks required scope
 /// - `404 Not Found`: No processing is currently active
 /// - `500 Internal Server Error`: Server error accessing statistics
-#[openapi_protect_get("/api/graph-statistics", "read:api")]
+#[openapi_protect_get("/api/graph-statistics", "read:api", tag = "Processing")]
 pub async fn get_graph_statistics(
     state: &State<SharedVisualizationState>,
 ) -> Result<Json<ProcessingGraphStatistics>, status::NotFound<String>> {
@@ -85,3 +88,21 @@ pub async fn get_graph_statistics(
         )),
     }
 }
+
+/// Get processing graph information
+///
+/// Returns a JSON object representing SerializableProcessingGraph
+#[openapi_protect_get("/api/graph", "read:api", tag = "Processing")]
+pub async fn get_graph(
+    state: &State<SharedVisualizationState>,
+) -> Result<Json<SerializableProcessingGraph>, status::NotFound<String>> {
+    // Get the current processing graph from shared state
+    Err(status::NotFound(
+        "No processing graph is currently available".to_string(),
+    ))
+}
+
+/// Centralized function to get all graph routes with OpenAPI documentation
+pub fn get_graph_routes() -> (Vec<rocket::Route>, OpenApi) {
+    openapi_get_routes_spec![get_graph_statistics, get_graph]
+}

rust/src/visualization/server/builder.rs

@@ -170,16 +170,20 @@ pub async fn build_rocket(
     let rocket_builder = rocket::custom(figment).attach(CORS);
 
     // Add visualization state if available (before mounting routes that need it)
-    let rocket_builder = if let Some(vis_state) = visualization_state {
+    let (rocket_builder, openapi_spec_graph) = if let Some(vis_state) = visualization_state {
         debug!("Adding SharedVisualizationState to Rocket state management");
         // Extract the value from Arc to match the expected type for State<SharedVisualizationState>
         let shared_state = (*vis_state).clone();
-        rocket_builder
-            .manage(shared_state)
-            .mount("/", routes![get_graph_statistics,])
+        let (openapi_routes_graph, openapi_spec_graph) = get_graph_routes(); // Get graph routes and OpenAPI spec
+        (
+            rocket_builder
+                .manage(shared_state)
+                .mount("/", openapi_routes_graph),
+            openapi_spec_graph,
+        )
     } else {
         debug!("No visualization state provided, API will return 404 for statistics");
-        rocket_builder
+        (rocket_builder, OpenApi::default())
     };
 
     let openapi_spec_audio: OpenApi;
@@ -245,7 +249,7 @@ pub async fn build_rocket(
         (openapi_routes_audio, openapi_spec_audio) = get_audio_streaming_routes();
 
         // Merge the audio OpenAPI spec with the base spec
-        let merged_spec = marge_spec_list(&[("/".to_string(), openapi_spec_base), ("/".to_string(),openapi_spec_audio)]).unwrap();
+        let merged_spec = marge_spec_list(&[("/".to_string(), openapi_spec_base), ("/".to_string(),openapi_spec_audio), ("/".to_string(),openapi_spec_graph)]).unwrap();
         let openapi_settings = OpenApiSettings::default();
         rocket_builder
             .mount(