mirror of
https://github.com/open-policy-agent/opa.git
synced 2026-08-22 08:14:48 -06:00
Patrick East
f1b9c7586b
Previously if the errors passed into the presentation Output were not structured w/ JSON tags for marshaling the error would be an empty string. This changes to wrap the errors with a struct in cases where they would otherwise not be formatted. We do this by forcing every error into a structure and translating known error types into it. Fixes: #1726 Fixes: #1724 Signed-off-by: Patrick East <east.patrick@gmail.com>
78 lines
1.6 KiB
Go
78 lines
1.6 KiB
Go
// Copyright 2018 The OPA Authors. All rights reserved.
|
|
// Use of this source code is governed by an Apache2
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/open-policy-agent/opa/ast"
|
|
pr "github.com/open-policy-agent/opa/internal/presentation"
|
|
"github.com/open-policy-agent/opa/loader"
|
|
"github.com/open-policy-agent/opa/util"
|
|
)
|
|
|
|
const (
|
|
parseFormatPretty = "pretty"
|
|
parseFormatJSON = "json"
|
|
)
|
|
|
|
var parseParams = struct {
|
|
format *util.EnumFlag
|
|
}{
|
|
format: util.NewEnumFlag(parseFormatPretty, []string{parseFormatPretty, parseFormatJSON}),
|
|
}
|
|
|
|
var parseCommand = &cobra.Command{
|
|
Use: "parse <path>",
|
|
Short: "Parse Rego source file",
|
|
Long: `Parse Rego source file and print AST.`,
|
|
PreRunE: func(Cmd *cobra.Command, args []string) error {
|
|
if len(args) == 0 {
|
|
return fmt.Errorf("no source file specified")
|
|
}
|
|
return nil
|
|
},
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
os.Exit(parse(args))
|
|
},
|
|
}
|
|
|
|
func parse(args []string) int {
|
|
|
|
if len(args) == 0 {
|
|
return 0
|
|
}
|
|
|
|
result, err := loader.Rego(args[0])
|
|
|
|
switch parseParams.format.String() {
|
|
case parseFormatJSON:
|
|
if err != nil {
|
|
pr.JSON(os.Stderr, pr.Output{Errors: pr.NewOutputErrors(err)})
|
|
return 1
|
|
}
|
|
|
|
bs, err := json.MarshalIndent(result.Parsed, "", " ")
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
return 1
|
|
}
|
|
fmt.Println(string(bs))
|
|
default:
|
|
ast.Pretty(os.Stdout, result.Parsed)
|
|
}
|
|
|
|
return 0
|
|
}
|
|
|
|
func init() {
|
|
parseCommand.Flags().VarP(parseParams.format, "format", "f", "set output format")
|
|
RootCommand.AddCommand(parseCommand)
|
|
}
|