Skip to content

Commit

Permalink
Main flag parsing fixed
Browse files Browse the repository at this point in the history
  • Loading branch information
CDimonaco committed Jan 8, 2025
1 parent f649aa3 commit abb7f56
Showing 1 changed file with 10 additions and 9 deletions.
19 changes: 10 additions & 9 deletions cmd/workbench.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,27 +11,28 @@ import (
"github.com/trento-project/workbench/pkg/operator"
)

type opts struct {
type cliOptions struct {
Arguments string `long:"arguments" short:"a" description:"Json arguments of an operator" required:"true"`
Verbose bool `long:"verbose" short:"v" description:"Log verbosity"`
}

var options cliOptions
var flagParser = flags.NewParser(&options, flags.Default) //nolint

func main() {
ctx := context.Background()

var opts opts
_, err := flags.Parse(&opts)
args, err := flagParser.Parse()
if err != nil {
os.Exit(1)
}

logger := logrus.StandardLogger()
if opts.Verbose {
if options.Verbose {
logger.SetLevel(logrus.DebugLevel)
}

operatorName := os.Args[1]

operatorName := args[0]
registry := operator.StandardRegistry(operator.WithCustomLogger(logger))

builder, err := registry.GetOperatorBuilder(operatorName)
Expand All @@ -40,15 +41,15 @@ func main() {
}

opArgs := make(operator.OperatorArguments)
err = json.Unmarshal([]byte(opts.Arguments), &opArgs)
err = json.Unmarshal([]byte(options.Arguments), &opArgs)
if err != nil {
logger.Fatalf("could not unmarhsal %s into arguments", opts.Arguments)
logger.Fatalf("could not unmarhsal %s into arguments", options.Arguments)
}

logger.Infof(
"starting execution with operator: %s - arguments: %s",
operatorName,
opts.Arguments,
options.Arguments,
)

op := builder("test-cli", opArgs)
Expand Down

0 comments on commit abb7f56

Please sign in to comment.