39 lines
942 B
Rust
39 lines
942 B
Rust
|
use ra_ide_api::Documentation;
|
||
|
|
|
||
|
|
pub(crate) fn sanitize_markdown(docs: Documentation) -> Documentation {
|
||
|
|
let docs: String = docs.into();
|
||
|
|
|
||
|
|
// Massage markdown
|
||
|
|
let mut processed_lines = Vec::new();
|
||
|
|
let mut in_code_block = false;
|
||
|
|
for line in docs.lines() {
|
||
|
|
if line.starts_with("```") {
|
||
|
|
in_code_block = !in_code_block;
|
||
|
|
}
|
||
|
|
|
||
|
|
let line = if in_code_block && line.starts_with("```") && !line.contains("rust") {
|
||
|
|
"```rust".into()
|
||
|
|
} else {
|
||
|
|
line.to_string()
|
||
|
|
};
|
||
|
|
|
||
|
|
processed_lines.push(line);
|
||
|
|
}
|
||
|
|
|
||
|
|
Documentation::new(&processed_lines.join("\n"))
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(test)]
|
||
|
|
mod tests {
|
||
|
|
use super::*;
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn test_codeblock_adds_rust() {
|
||
|
|
let comment = "```\nfn some_rust() {}\n```";
|
||
|
|
assert_eq!(
|
||
|
|
sanitize_markdown(Documentation::new(comment)).contents(),
|
||
|
|
"```rust\nfn some_rust() {}\n```"
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|