pub fn strip_markers(text: &str) -> String {
let mut result = text.to_string();
let leading_markers = ["<!--", "///", "/*", "//", "#"];
if let Some(non_ws_idx) = result.find(|c: char| !c.is_whitespace()) {
for marker in &leading_markers {
if result[non_ws_idx..].starts_with(marker) {
let marker_end = non_ws_idx + marker.len();
let remove_space = if result[marker_end..].starts_with(' ') {
1
} else {
0
};
result.replace_range(non_ws_idx..(marker_end + remove_space), "");
break;
}
}
}
let trailing_markers = ["*/", "-->"];
for marker in &trailing_markers {
let pattern = format!(" {marker}");
if result.ends_with(&pattern) {
let new_len = result.len() - pattern.len();
result.truncate(new_len);
break;
} else if result.ends_with(marker) {
let new_len = result.len() - marker.len();
result.truncate(new_len);
break;
}
}
result
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_strip_markers() {
let input = "/// TODO: Fix this issue";
let output = strip_markers(input);
assert_eq!(output, "TODO: Fix this issue");
let input2 = "/* TODO: Refactor code */";
let output2 = strip_markers(input2);
assert_eq!(output2, "TODO: Refactor code");
}
#[test]
fn test_strip_markers_different_markers() {
let input_hash = "# Note: This is a test";
assert_eq!(strip_markers(input_hash), "Note: This is a test");
let input_html = "<!-- Important comment -->";
assert_eq!(strip_markers(input_html), "Important comment");
}
#[test]
fn test_strip_markers_with_indent() {
let input = " // Indented comment";
let output = strip_markers(input);
assert_eq!(output, " Indented comment");
}
}