forked from nrdxp/sukr
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmermaid.rs
More file actions
63 lines (55 loc) · 1.79 KB
/
mermaid.rs
File metadata and controls
63 lines (55 loc) · 1.79 KB
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
//! Mermaid diagram rendering via mermaid-rs-renderer.
//!
//! Converts Mermaid diagram definitions to SVG at build-time.
use mermaid_rs_renderer::RenderOptions;
use std::panic;
/// Render a Mermaid diagram to SVG.
///
/// # Arguments
/// * `code` - The Mermaid diagram definition
///
/// # Returns
/// The rendered SVG string, or an error message on failure.
///
/// # Note
/// Uses catch_unwind to handle panics in upstream dependencies gracefully.
pub fn render_diagram(code: &str) -> Result<String, String> {
let code = code.to_owned();
let result = panic::catch_unwind(move || {
let opts = RenderOptions::modern();
mermaid_rs_renderer::render_with_options(&code, opts)
});
match result {
Ok(Ok(svg)) => Ok(svg),
Ok(Err(e)) => Err(e.to_string()),
Err(_) => Err("mermaid rendering panicked (upstream bug)".to_string()),
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_simple_flowchart() {
let result = render_diagram("flowchart LR; A-->B-->C").unwrap();
assert!(result.contains("<svg"));
assert!(result.contains("</svg>"));
}
#[test]
fn test_sequence_diagram() {
let result = render_diagram("sequenceDiagram\n Alice->>Bob: Hello").unwrap();
assert!(result.contains("<svg"));
}
#[test]
fn test_invalid_syntax_no_panic() {
// Should not panic, returns error
let result = render_diagram("invalid diagram syntax ???");
// May succeed with error node or fail gracefully
assert!(result.is_ok() || result.is_err());
}
#[test]
fn test_state_diagram() {
let result =
render_diagram("stateDiagram-v2\n [*] --> Idle\n Idle --> Processing: Start");
assert!(result.is_ok() || result.is_err());
}
}