pub fn extract_comment(content: &str, ext: &str) -> Option<String> {
match ext {
"rs" | "c" | "cpp" | "js" | "ts" | "java" => {
if let Some(start) = content.find("/*") {
if let Some(end) = content.find("*/") {
let comment = &content[start + 2..end];
return Some(comment.trim().to_string());
}
}
if let Some(start) = content.find("//") {
let comment = &content[start + 2..];
return Some(comment.trim().to_string());
}
None
}
"yml" | "py" | "sh" | "rb" => {
if let Some(start) = content.find('#') {
let comment = &content[start + 1..];
return Some(comment.trim().to_string());
}
None
}
"md" | "html" => {
if let Some(start) = content.find("<!--") {
if let Some(end) = content.find("-->") {
let comment = &content[start + 4..end];
return Some(comment.trim().to_string());
}
}
None
}
_ => None,
}
}