cli: try fix issues reported in 'oracle' command

Try and fix errors issues reported here: open-policy-agent/vscode-opa#81 1. Only include .rego files in definition location checks 2. Don't hang if input not provided on stdin Signed-off-by: Anders Eknert <[email protected]>
anderseknert · Feb 22, 2023 · 514725f · 514725f
1 parent df1f4a2
commit 514725f
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 3 deletions.
diff --git a/cmd/oracle.go b/cmd/oracle.go
@@ -106,7 +106,14 @@ func dofindDefinition(params findDefinitionParams, stdin io.Reader, stdout io.Wr
 		if len(params.bundlePaths.v) > 1 {
 			return errors.New("not implemented: multiple bundle paths")
 		}
-		b, err = loader.NewFileLoader().WithSkipBundleVerification(true).AsBundle(params.bundlePaths.v[0])
+		b, err = loader.NewFileLoader().
+			WithSkipBundleVerification(true).
+			WithFilter(func(abspath string, info os.FileInfo, depth int) bool {
+				// While directories may contain other things of interest for OPA (json, yaml..),
+				// only .rego will work reliably for the purpose of finding definitions
+				return strings.HasPrefix(info.Name(), ".rego")
+			}).
+			AsBundle(params.bundlePaths.v[0])
 		if err != nil {
 			return err
 		}
@@ -123,10 +130,17 @@ func dofindDefinition(params findDefinitionParams, stdin io.Reader, stdout io.Wr
 	var bs []byte
 
 	if params.stdinBuffer {
-		bs, err = io.ReadAll(stdin)
+		stat, err := os.Stdin.Stat()
 		if err != nil {
 			return err
 		}
+		// Only read from stdin when there is something actually there
+		if (stat.Mode() & os.ModeCharDevice) == 0 {
+			bs, err = io.ReadAll(stdin)
+			if err != nil {
+				return err
+			}
+		}
 	}
 
 	result, err := oracle.New().FindDefinition(oracle.DefinitionQuery{

diff --git a/cmd/oracle_test.go b/cmd/oracle_test.go
@@ -27,7 +27,11 @@ p { q }
 
 q = true`)
 
-	files := map[string]string{"test.rego": onDiskModule}
+	files := map[string]string{
+		"test.rego":    onDiskModule,
+		"document.txt": "this should not be included",
+		"ignore.json":  `{"neither": "should this"}`,
+	}
 
 	test.WithTempFS(files, func(rootDir string) {