Merge pull request #4 from simar7/export-funcs

chore(rego): Export rego funcs
aquasecurity · Sep 15, 2023 · 0f6e136 · 0f6e136
2 parents cf704c9 + 33edb29
commit 0f6e136
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/pkg/rego/embed.go b/pkg/rego/embed.go
@@ -90,7 +90,7 @@ func RecurseEmbeddedModules(fs embed.FS, dir string) (map[string]*ast.Module, er
 			}
 			continue
 		}
-		if !isRegoFile(entry.Name()) || isDotFile(entry.Name()) {
+		if !IsRegoFile(entry.Name()) || IsDotFile(entry.Name()) {
 			continue
 		}
 		fullPath := strings.Join([]string{dir, entry.Name()}, "/")

diff --git a/pkg/rego/load.go b/pkg/rego/load.go
@@ -13,15 +13,15 @@ import (
 	"github.com/open-policy-agent/opa/bundle"
 )
 
-func isRegoFile(name string) bool {
+func IsRegoFile(name string) bool {
 	return strings.HasSuffix(name, bundle.RegoExt) && !strings.HasSuffix(name, "_test"+bundle.RegoExt)
 }
 
-func isDotFile(name string) bool {
+func IsDotFile(name string) bool {
 	return strings.HasPrefix(name, ".")
 }
 
-func isJSONFile(name string) bool {
+func IsJSONFile(name string) bool {
 	return strings.HasSuffix(name, ".json")
 }
 
@@ -42,7 +42,7 @@ func (s *Scanner) loadPoliciesFromDirs(target fs.FS, paths []string) (map[string
 			if info.IsDir() {
 				return nil
 			}
-			if !isRegoFile(info.Name()) || isDotFile(info.Name()) {
+			if !IsRegoFile(info.Name()) || IsDotFile(info.Name()) {
 				return nil
 			}
 			data, err := fs.ReadFile(target, filepath.ToSlash(path))

diff --git a/pkg/rego/metadata.go b/pkg/rego/metadata.go
@@ -427,7 +427,7 @@ func findSchemaInFS(paths []string, srcFS fs.FS, schemaName string) ([]byte, err
 			if info.IsDir() {
 				return nil
 			}
-			if !isJSONFile(info.Name()) {
+			if !IsJSONFile(info.Name()) {
 				return nil
 			}
 			if info.Name() == schemaName+".json" {