File size: 2,047 Bytes
17f328f | 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 64 | pub fn parse_argument_comment(text: &str) -> Option<&str> {
let trimmed = text.trim_end();
let comment_start = trimmed.rfind("/*")?;
let comment = &trimmed[comment_start..];
let name = comment.strip_prefix("/*")?.strip_suffix("*/")?;
is_identifier(name).then_some(name)
}
pub fn parse_argument_comment_prefix(text: &str) -> Option<&str> {
let trimmed = text.trim_start();
let comment = trimmed.strip_prefix("/*")?;
let (name, _) = comment.split_once("*/")?;
is_identifier(name).then_some(name)
}
fn is_identifier(text: &str) -> bool {
let mut chars = text.chars();
let Some(first) = chars.next() else {
return false;
};
if !(first == '_' || first.is_ascii_alphabetic()) {
return false;
}
chars.all(|ch| ch == '_' || ch.is_ascii_alphanumeric())
}
#[cfg(test)]
mod tests {
use super::parse_argument_comment;
use super::parse_argument_comment_prefix;
#[test]
fn parses_trailing_comment() {
assert_eq!(parse_argument_comment("(/*base_url*/ "), Some("base_url"));
assert_eq!(
parse_argument_comment(", /*timeout_ms*/ "),
Some("timeout_ms")
);
assert_eq!(
parse_argument_comment(".method::<u8>(/*base_url*/ "),
Some("base_url")
);
}
#[test]
fn rejects_non_matching_shapes() {
assert_eq!(parse_argument_comment("(\n"), None);
assert_eq!(parse_argument_comment("(/* base_url*/ "), None);
assert_eq!(parse_argument_comment("(/*base_url */ "), None);
assert_eq!(parse_argument_comment("(/*base_url=*/ "), None);
assert_eq!(parse_argument_comment("(/*1base_url*/ "), None);
assert_eq!(parse_argument_comment_prefix("/*env=*/ None"), None);
}
#[test]
fn parses_prefix_comment() {
assert_eq!(parse_argument_comment_prefix("/*env*/ None"), Some("env"));
assert_eq!(
parse_argument_comment_prefix("\n /*retry_count*/ 3"),
Some("retry_count")
);
}
}
|