Improve versions checking by looking at workdir files (#8)

cmd/gokart.go

@@ -27,13 +27,13 @@ func main() {
 }
 
 func ps1() {
+	wdFiles := internal.ListWdFiles()
+
 	fmt.Print("\n")
 	fmt.Print(ansi.Bold())
-
 	fmt.Print(internal.Path())
 	fmt.Print(git.Git())
-	fmt.Print(versions.All())
-
+	fmt.Print(versions.All(wdFiles))
 	fmt.Print(ansi.Reset())
 }
 

cmd/gokart_test.go

@@ -0,0 +1,9 @@
+package main
+
+import "testing"
+
+func BenchmarkPs1(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		ps1()
+	}
+}

internal/files.go

@@ -0,0 +1,26 @@
+package internal
+
+import (
+	"os"
+)
+
+func ListWdFiles() []string {
+	wd, err := os.Getwd()
+	if err != nil {
+		return nil
+	}
+
+	entries, err := os.ReadDir(wd)
+	if err != nil {
+		return nil
+	}
+
+	var files []string
+	for _, entry := range entries {
+		if !entry.IsDir() {
+			files = append(files, entry.Name())
+		}
+	}
+
+	return files
+}

internal/versions/all.go

@@ -1,7 +1,6 @@
 package versions
 
 import (
-	"strings"
 	"sync"
 )
 
@@ -17,21 +16,28 @@ var sections = []section{
 	Docker,
 }
 
-func All() string {
+func All(wdFiles []string) string {
 	var wg sync.WaitGroup
-	result := make([]string, len(sections))
+	parts := make([]string, len(sections))
 
 	for i, s := range sections {
 		wg.Add(1)
 
 		i := i
 		s := s
 		go func() {
-			result[i] = s.version()
+			parts[i] = s.version(wdFiles)
 			wg.Done()
 		}()
 	}
-
 	wg.Wait()
-	return strings.Join(result, " ")
+
+	result := ""
+	for _, part := range parts {
+		if part != "" {
+			result += " " + part
+		}
+	}
+
+	return result
 }

internal/versions/golang.go

@@ -12,6 +12,9 @@ var Go = section{
 	upsearchFiles: []string{
 		"go.mod",
 	},
+	wdFiles: []string{
+		"*.go",
+	},
 
 	command: []string{"go", "version"},
 	versionFunc: func(output string) string {

internal/versions/java.go

@@ -10,10 +10,16 @@ var Java = section{
 	color:  ansi.Cyan,
 
 	upsearchFiles: []string{
-		"pom.xml",
 		"build.gradle",
-		"settings.gradle",
 		"build.xml",
+		"pom.xml",
+		"settings.gradle",
+	},
+	wdFiles: []string{
+		"*.class",
+		"*.jar",
+		"*.java",
+		"*.war",
 	},
 
 	command: []string{"java", "--version"},

internal/versions/node.go

@@ -9,8 +9,12 @@ var Node = section{
 	color:  ansi.Green,
 
 	upsearchFiles: []string{
-		"package.json",
 		"node_modules",
+		"package.json",
+	},
+	wdFiles: []string{
+		"*.js",
+		"*.ts",
 	},
 
 	command: []string{"node", "-v"},

internal/versions/php.go

@@ -12,6 +12,9 @@ var Php = section{
 	upsearchFiles: []string{
 		"composer.json",
 	},
+	wdFiles: []string{
+		"*.php",
+	},
 
 	command: []string{"php", "-v"},
 	versionFunc: func(output string) string {

internal/versions/ruby.go

@@ -13,6 +13,9 @@ var Ruby = section{
 		"Gemfile",
 		"Rakefile",
 	},
+	wdFiles: []string{
+		"*.rb",
+	},
 
 	command: []string{"ruby", "-v"},
 	versionFunc: func(output string) string {

internal/versions/rust.go

@@ -12,6 +12,9 @@ var Rust = section{
 	upsearchFiles: []string{
 		"Cargo.toml",
 	},
+	wdFiles: []string{
+		"*.rs",
+	},
 
 	command: []string{"rustc", "--version"},
 	versionFunc: func(output string) string {

internal/versions/section.go

@@ -3,24 +3,42 @@ package versions
 import (
 	"gokart-prompt/internal"
 	"gokart-prompt/internal/ansi"
+	"path"
 )
 
 type section struct {
 	symbol        string
 	color         ansi.ColorCode
 	upsearchFiles []string
+	wdFiles       []string
 	command       []string
 	versionFunc   func(output string) string
 }
 
-func (s section) version() string {
-	if _, ok := internal.UpsearchWd(s.upsearchFiles); !ok {
+func (s section) version(wdFiles []string) string {
+	if !s.display(wdFiles) {
 		return ""
 	}
 
 	if output, ok := internal.Command(s.command[0], s.command[1:]...); ok {
-		return ansi.Color(s.color, " "+s.symbol+" "+s.versionFunc(output))
+		return ansi.Color(s.color, s.symbol+" "+s.versionFunc(output))
 	}
 
 	return ""
 }
+
+func (s section) display(wdFiles []string) bool {
+	for _, file := range wdFiles {
+		for _, pattern := range s.wdFiles {
+			if ok, _ := path.Match(pattern, file); ok {
+				return true
+			}
+		}
+	}
+
+	if _, ok := internal.UpsearchWd(s.upsearchFiles); ok {
+		return true
+	}
+
+	return false
+}