|
| 1 | +//! CLI: round-trips a `.graphite` file through `NodeNetwork → Registry → NodeNetwork`. |
| 2 | +
|
| 3 | +use std::fs; |
| 4 | +use std::path::PathBuf; |
| 5 | +use std::process::ExitCode; |
| 6 | + |
| 7 | +use graph_craft::document::NodeNetwork; |
| 8 | +use graph_storage::Registry; |
| 9 | + |
| 10 | +fn main() -> ExitCode { |
| 11 | + if let Err(error) = run() { |
| 12 | + eprintln!("{error}"); |
| 13 | + return ExitCode::FAILURE; |
| 14 | + } |
| 15 | + ExitCode::SUCCESS |
| 16 | +} |
| 17 | + |
| 18 | +fn run() -> Result<(), String> { |
| 19 | + let args: Vec<String> = std::env::args().collect(); |
| 20 | + let [_, input, output] = args.as_slice() else { |
| 21 | + return Err(format!("Usage: {} <input.graphite> <output.graphite>", args.first().map(String::as_str).unwrap_or("round_trip"))); |
| 22 | + }; |
| 23 | + |
| 24 | + let input_path = PathBuf::from(input); |
| 25 | + let output_path = PathBuf::from(output); |
| 26 | + |
| 27 | + println!("Loading artwork from: {}", input_path.display()); |
| 28 | + |
| 29 | + let json_content = fs::read_to_string(&input_path).map_err(|e| format!("Error reading input: {e}"))?; |
| 30 | + let mut doc: serde_json::Value = serde_json::from_str(&json_content).map_err(|e| format!("Error parsing JSON: {e}"))?; |
| 31 | + |
| 32 | + let original_network: NodeNetwork = serde_json::from_value(doc["network_interface"]["network"].clone()).map_err(|e| format!("Error deserializing NodeNetwork: {e}"))?; |
| 33 | + println!("Original network: {} nodes", original_network.nodes.len()); |
| 34 | + |
| 35 | + // No byte store here: keep the extracted declaration bytes in hand and rebuild the `Declarations` |
| 36 | + // map from them so the back-conversion can resolve proto-node identifiers. |
| 37 | + let conversion = |
| 38 | + Registry::convert_from_runtime(&original_network, &graph_storage::NoMetadata, &Default::default(), graph_storage::PeerId(0)).map_err(|e| format!("Error converting to Registry: {e}"))?; |
| 39 | + let declarations = conversion.declarations().map_err(|e| format!("Error rebuilding declarations: {e}"))?; |
| 40 | + let registry = conversion.registry; |
| 41 | + println!("Registry: {} node instances, {} networks", registry.node_instances.len(), registry.networks.len()); |
| 42 | + |
| 43 | + let mut node_ids: Vec<_> = registry.node_instances.keys().copied().collect(); |
| 44 | + node_ids.sort(); |
| 45 | + println!("Registry node IDs: {node_ids:?}"); |
| 46 | + |
| 47 | + let (converted_network, _entries) = registry.to_runtime_with_metadata(&declarations).map_err(|e| format!("Error converting back to NodeNetwork: {e}"))?; |
| 48 | + println!("Converted network: {} nodes", converted_network.nodes.len()); |
| 49 | + |
| 50 | + doc["network_interface"]["network"] = serde_json::to_value(&converted_network).map_err(|e| format!("Error serializing converted network: {e}"))?; |
| 51 | + let output_json = serde_json::to_string_pretty(&doc).map_err(|e| format!("Error serializing output JSON: {e}"))?; |
| 52 | + fs::write(&output_path, output_json).map_err(|e| format!("Error writing output: {e}"))?; |
| 53 | + |
| 54 | + println!("Wrote round-tripped artwork to: {}", output_path.display()); |
| 55 | + Ok(()) |
| 56 | +} |
0 commit comments