Support looking up syntax via extension on a dotfile

Author: Nemo157

src/web/highlight.rs

@@ -38,12 +38,8 @@ fn try_with_syntax(syntax: &SyntaxReference, code: &str) -> Result<String> {
 fn select_syntax(name: Option<&str>, code: &str) -> &'static SyntaxReference {
     name.and_then(|name| {
         SYNTAXES.find_syntax_by_token(name).or_else(|| {
-            if name.starts_with('.') {
-                None
-            } else {
-                name.rsplit_once('.')
-                    .and_then(|(_, ext)| SYNTAXES.find_syntax_by_token(ext))
-            }
+            name.rsplit_once('.')
+                .and_then(|(_, ext)| SYNTAXES.find_syntax_by_token(ext))
         })
     })
     .or_else(|| SYNTAXES.find_syntax_by_first_line(code))
@@ -75,4 +71,11 @@ mod tests {
         assert_eq!(select_syntax(Some("Cargo.toml.orig"), "").name, toml.name);
         assert_eq!(select_syntax(Some("Cargo.lock"), "").name, toml.name);
     }
+
+    #[test]
+    fn dotfile_with_extension() {
+        let toml = select_syntax(Some("toml"), "");
+
+        assert_eq!(select_syntax(Some(".rustfmt.toml"), "").name, toml.name);
+    }
 }

src/web/source.rs

@@ -457,4 +457,25 @@ mod tests {
             Ok(())
         });
     }
+
+    #[test]
+    fn dotfiles_with_extension_are_highlighted() {
+        wrapper(|env| {
+            env.fake_release()
+                .name("fake")
+                .version("0.1.0")
+                .source_file(".rustfmt.toml", b"[rustfmt]")
+                .create()?;
+
+            let web = env.frontend();
+
+            let response = web
+                .get("/crate/fake/0.1.0/source/.rustfmt.toml")
+                .send()?
+                .text()?;
+            assert!(response.contains(r#"<span class="syntax-source syntax-toml">"#));
+
+            Ok(())
+        });
+    }
 }