diff options
| -rw-r--r-- | src/tools/lint-docs/src/lib.rs | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/src/tools/lint-docs/src/lib.rs b/src/tools/lint-docs/src/lib.rs index dc878b718ad..326b7948098 100644 --- a/src/tools/lint-docs/src/lib.rs +++ b/src/tools/lint-docs/src/lib.rs @@ -142,8 +142,8 @@ impl<'a> LintExtractor<'a> { match lines.next() { Some((lineno, line)) => { let line = line.trim(); - if line.starts_with("/// ") { - doc_lines.push(line.trim()[4..].to_string()); + if let Some(text) = line.strip_prefix("/// ") { + doc_lines.push(text.trim().to_string()); } else if line.starts_with("///") { doc_lines.push("".to_string()); } else if line.starts_with("// ") { @@ -347,8 +347,7 @@ impl<'a> LintExtractor<'a> { let mut source = String::new(); let needs_main = !example.iter().any(|line| line.contains("fn main")); // Remove `# ` prefix for hidden lines. - let unhidden = - example.iter().map(|line| if line.starts_with("# ") { &line[2..] } else { line }); + let unhidden = example.iter().map(|line| line.strip_prefix("# ").unwrap_or(line)); let mut lines = unhidden.peekable(); while let Some(line) = lines.peek() { if line.starts_with("#!") { |