ralsina/tartrazine
1
0
Fork 0
mirror of https://github.com/ralsina/tartrazine.git synced 2025-06-18 22:23:07 -03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Improve shebang parsing to detect correct interpreter

Browse Source
This commit is contained in:
Lauris BH
2021-09-25 19:17:49 +03:00
parent 7c24e3d5d2
commit 4686615d9e
2 changed files with 55 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
common.go
View File

@ -111,13 +111,6 @@ func getFirstLanguageAndSafe(languages []string) (language string, safe bool) {
return
}
// getLanguageBySpecificClassifier returns the most probably language for the given content using
// classifier to detect language.
func getLanguageBySpecificClassifier(content []byte, candidates []string, classifier classifier) (language string, safe bool) {
languages := getLanguagesBySpecificClassifier(content, candidates, classifier)
return getFirstLanguageAndSafe(languages)
}
// GetLanguages applies a sequence of strategies based on the given filename and content
// to find out the most probable languages to return.
//
@ -300,9 +293,11 @@ func GetLanguagesByShebang(_ string, content []byte, _ []string) (languages []st
var (
shebangExecHack = regex.MustCompile(`exec (\w+).+\$0.+\$@`)
pythonVersion = regex.MustCompile(`python\d\.\d+`)
envOptArgs = regex.MustCompile(`-[i0uCSv]*|--\S+`)
envVarArgs = regex.MustCompile(`\S+=\S+`)
)
func getInterpreter(data []byte) (interpreter string) {
func getInterpreter(data []byte) string {
line := getFirstLine(data)
if !hasShebang(line) {
return ""
@ -317,7 +312,7 @@ func getInterpreter(data []byte) (interpreter string) {
// Extract interpreter name from path. Use path.Base because
// shebang on Cygwin/Windows still use a forward slash
interpreter = path.Base(string(splitted[0]))
interpreter := path.Base(string(splitted[0]))
// #!/usr/bin/env [...]
if interpreter == "env" {
@ -325,6 +320,13 @@ func getInterpreter(data []byte) (interpreter string) {
// /usr/bin/env with no arguments
return ""
}
for len(splitted) > 2 {
if envOptArgs.Match(splitted[1]) || envVarArgs.Match(splitted[1]) {
splitted = append(splitted[:1], splitted[2:]...)
continue
}
break
}
interpreter = path.Base(string(splitted[1]))
}
@ -342,7 +344,7 @@ func getInterpreter(data []byte) (interpreter string) {
interpreter = ""
}
return
return interpreter
}
func getFirstLines(content []byte, count int) []byte {