Add support for printing rego messages on CLI

secmon-lab · Jan 15, 2024 · c029c28 · c029c28
1 parent 681c188
commit c029c28
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/pkg/chain/core/policy.go b/pkg/chain/core/policy.go
@@ -2,6 +2,7 @@ package core
 
 import (
 	"errors"
+	"fmt"
 
 	"log/slog"
 
@@ -51,11 +52,15 @@ func (x *Core) QueryActionPolicy(ctx *model.Context, in, out any) error {
 
 func makeRegoPrint(ctx *model.Context) policy.RegoPrint {
 	return func(file string, row int, msg string) error {
-		ctx.Logger().Info("rego print",
-			slog.String("file", file),
-			slog.Int("row", row),
-			slog.String("msg", msg),
-		)
+		if ctx.OnCLI() {
+			fmt.Printf("	%s:%d: %s\n", file, row, msg)
+		} else {
+			ctx.Logger().Info("rego print",
+				slog.String("file", file),
+				slog.Int("row", row),
+				slog.String("msg", msg),
+			)
+		}
 		return nil
 	}
 }
diff --git a/pkg/controller/cli/play.go b/pkg/controller/cli/play.go
@@ -67,6 +67,7 @@ func cmdPlay() *cli.Command {
 
 			ctx := model.NewContext(
 				model.WithBase(c.Context),
+				model.WithCLI(),
 			)
 			ctx.Logger().Info("starting alertchain with play mode", slog.Any("playbook", playbookPath))