Merge pull request #28 from ebkalderon/add-x-shellscript-support

Add support for text/x-shellscript
bojand · Dec 1, 2020 · c4e825f · c4e825f
2 parents 5e85472 + 7791f61
commit c4e825f
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 2 deletions.
diff --git a/src/map.rs b/src/map.rs
@@ -465,5 +465,11 @@ matcher_map!(
         "html",
         matchers::text::is_html
     ),
-    (MatcherType::TEXT, "text/xml", "xml", matchers::text::is_xml)
+    (MatcherType::TEXT, "text/xml", "xml", matchers::text::is_xml),
+    (
+        MatcherType::TEXT,
+        "text/x-shellscript",
+        "sh",
+        matchers::text::is_shellscript
+    )
 );
diff --git a/src/matchers/text.rs b/src/matchers/text.rs
@@ -65,9 +65,14 @@ fn starts_with_ignore_ascii_case(buf: &[u8], needle: &[u8]) -> bool {
     buf.len() >= needle.len() && buf[..needle.len()].eq_ignore_ascii_case(needle)
 }
 
+/// Returns whether a buffer is a shell script.
+pub fn is_shellscript(buf: &[u8]) -> bool {
+    buf.len() > 2 && &buf[..2] == b"#!"
+}
+
 #[cfg(test)]
 mod tests {
-    use super::{is_html, trim_start_whitespaces};
+    use super::{is_html, is_shellscript, trim_start_whitespaces};
 
     #[test]
     fn trim_whitespaces() {
@@ -88,4 +93,9 @@ mod tests {
         assert_eq!(is_html(b"<HTML "), true);
         assert_eq!(is_html(b"   <BODY>"), true);
     }
+
+    #[test]
+    fn shellscript() {
+        assert_eq!(is_shellscript(b"#!"), false);
+    }
 }
diff --git a/testdata/sample.sh b/testdata/sample.sh
@@ -0,0 +1 @@
+#!/bin/bash
diff --git a/tests/text.rs b/tests/text.rs
@@ -3,3 +3,5 @@ mod common;
 test_format!(TEXT, "text/html", "html", html, "sample.html");
 
 test_format!(TEXT, "text/xml", "xml", xml, "sample.xml");
+
+test_format!(TEXT, "text/x-shellscript", "sh", sh, "sample.sh");