feat: crawling path script dirs

Author: d-led

common/nix_candidate_source.go

@@ -26,6 +26,7 @@ func NewNixCandidateSource(fs Filesystem, key string) CandidateSource {
 	}
 	res.crawlKnownPaths()
 	res.crawlPathLists()
+	res.crawlPathScriptDirs()
 	return res
 }
 
@@ -111,6 +112,15 @@ func (s *NixCandidateSource) crawlPathLists() {
 	})
 }
 
+func (s *NixCandidateSource) crawlPathScriptDirs() {
+	if runtime.GOOS == "windows" {
+		return
+	}
+	ForEachScriptsDPath(func(script, expandedScript string) {
+		s.crawlSource(script, expandedScript)
+	})
+}
+
 // input is some path definition
 func (s *NixCandidateSource) harvestPaths(input string) []string {
 

common/script_list_source.go

@@ -0,0 +1,51 @@
+package common
+
+import (
+	"os"
+	"path/filepath"
+
+	"github.com/mitchellh/go-homedir"
+)
+
+var knownScriptDirs = []string{
+	"/etc/profile.d/",
+}
+
+var allowedExtensions = []string{
+	".sh",
+}
+
+func ForEachScriptsDPath(fn func(originalSource, expandedSource string)) {
+	for _, knownScriptDir := range knownScriptDirs {
+		_ = filepath.Walk(knownScriptDir,
+			func(originalSource string, info os.FileInfo, err error) error {
+				if err != nil || info.IsDir() || fileNotAllowed(originalSource) {
+					return nil
+				}
+
+				expanded, err := homedir.Expand(originalSource)
+				if err != nil {
+					return nil
+				}
+
+				fn(originalSource, expanded)
+
+				return nil
+			},
+		)
+	}
+}
+
+func fileNotAllowed(p string) bool {
+	return !extensionAllowed(p)
+}
+
+func extensionAllowed(p string) bool {
+	e := filepath.Ext(p)
+	for _, allowedExt := range allowedExtensions {
+		if allowedExt == e {
+			return true
+		}
+	}
+	return false
+}