From a90fdfc70178038599bfc7247aedc83bfe2e88f0 Mon Sep 17 00:00:00 2001
From: Eric Huss <eric@huss.org>
Date: Sun, 29 Nov 2020 07:57:55 -0800
Subject: [PATCH] lint-docs: Use strip-prefix to simplify.

---
 src/tools/lint-docs/src/lib.rs | 7 +++----
 1 file 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 @@ fn lints_from_file(&self, path: &Path) -> Result<Vec<Lint>, Box<dyn Error>> {
                 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 @@ fn generate_lint_output(
         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("#!") {