fix file suggestions for app entrypoints

foomo · Sep 21, 2023 · e8650e8 · e8650e8
1 parent 027115d
commit e8650e8
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/config/attach.go b/config/attach.go
@@ -60,7 +60,7 @@ func (c AttachConfig) MarshalYAML() (interface{}, error) {
 
 func (c AttachConfig) SourcePathSuggest(d prompt.Document) []prompt.Suggest {
 	return suggest.Completer(d, suggest.MustList(func() ([]string, error) {
-		return find(".", "-type", "f", "-name", "main.go")
+		return findContaining("package main", ".", "-type", "f", "-name", "*.go")
 	}))
 }
 

diff --git a/config/config.go b/config/config.go
@@ -100,3 +100,7 @@ func loadYaml(path string, data interface{}) error {
 func find(args ...string) ([]string, error) {
 	return script.Exec(fmt.Sprintf("find %v", strings.Join(args, " "))).Slice()
 }
+
+func findContaining(v string, args ...string) ([]string, error) {
+	return script.Exec(fmt.Sprintf("find %v -exec grep -lr %q {} +", strings.Join(args, " "), v)).Slice()
+}
diff --git a/config/patch.go b/config/patch.go
@@ -61,7 +61,7 @@ func (c PatchConfig) MarshalYAML() (interface{}, error) {
 
 func (c PatchConfig) SourcePathSuggest(d prompt.Document) []prompt.Suggest {
 	return suggest.Completer(d, suggest.MustList(func() ([]string, error) {
-		return find(".", "-type", "f", "-name", "main.go")
+		return findContaining("package main", ".", "-type", "f", "-name", "*.go")
 	}))
 }