Works with shebangs

diff --git a/extractor.go b/extractor.go
index 19bc27472cea37b8466450dc342317bfa9e85650..a0d43dd822b4561fed3c59ccbb0da5d5c5463d89 100644 (file)
--- a/extractor.go
+++ b/extractor.go
@@ -9,7 +9,7 @@ type MatchOptions struct {
 }
 
 func pathExtractor(input string) [][][]byte {
-       surroundRegex := "[^]()[ \\t:'\"]*"
+       surroundRegex := "[^]()!#[ \\t:'\"]*"
        r := regexp.MustCompile("(" + surroundRegex + "[\\./]" + surroundRegex + ")")
        temp := [][][]byte{}
        temp = r.FindAllSubmatch([]byte(input), -1)

diff --git a/pe_test.go b/pe_test.go
index acb93c13b04ede10c558f6e517ecb49d5ebcfc73..c88aa0d5dee7ba52f7ff90bb214e1f5f5d60c1ea 100644 (file)
--- a/pe_test.go
+++ b/pe_test.go
@@ -13,6 +13,14 @@ func TestEverything(t *testing.T) {
                t.Errorf("Matches sentence", output)
        }
 
+       output = GetAllMatches("!#/usr/bin/env", MatchOptions{})
+       if len(output) != 1 {
+               t.Errorf("Doesn't match shebang", output)
+       }
+       if output[0] != "/usr/bin/env" {
+               t.Errorf("Doesn't match shebang", output)
+       }
+
        output = GetAllMatches("hello .gitignore", MatchOptions{})
        if output[0] != ".gitignore" {
                t.Errorf("Doesnt match hidden files", output)