feat(cli): add entropy cli commands (#53)

* feat: add resource commands

* fix: forbidigo linter to ignore cli

* feat: add action & logs cmds

* fix: lint issues

* fix: remove context, goroutine etc

* feat: add yaml output & args support

* fix: remove UseEnumNumbers from MarshalOptions

* feat: add groupig & usage details

* chore: change filePath flag to file

* fix: allow only json & yaml output

* fix: error handling in default case

Author: ishanarya0

.golangci.yml

@@ -130,5 +130,9 @@ issues:
       linters:
         - gocritic
         - dupl
+    - path: cli
+      linters:
+        - forbidigo
+        - contextcheck
 severity:
   default-severity: error

cli/action.go

@@ -0,0 +1,80 @@
+package cli
+
+import (
+	"fmt"
+
+	"github.com/odpf/salt/term" //nolint
+	entropyv1beta1 "go.buf.build/odpf/gwv/odpf/proton/odpf/entropy/v1beta1"
+	"google.golang.org/protobuf/types/known/structpb"
+
+	"github.com/MakeNowJust/heredoc"
+	"github.com/odpf/salt/printer"
+	"github.com/spf13/cobra"
+)
+
+func cmdAction() *cobra.Command {
+	var urn, file, output string
+	var params *structpb.Value
+	cmd := &cobra.Command{
+		Use:     "action <action-name>",
+		Aliases: []string{"action"},
+		Short:   "Manage actions",
+		Example: heredoc.Doc(`
+			$ entropy action start --urn=<resource-urn> --file=<file-path> --out=json
+		`),
+		Annotations: map[string]string{
+			"group:core": "true",
+		},
+		Args: cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			spinner := printer.Spin("")
+			defer spinner.Stop()
+			cs := term.NewColorScheme()
+
+			var reqBody entropyv1beta1.ApplyActionRequest
+			if file != "" {
+				if err := parseFile(file, params); err != nil {
+					return err
+				}
+				reqBody.Params = params
+			}
+
+			reqBody.Urn = urn
+			reqBody.Action = args[0]
+
+			err := reqBody.ValidateAll()
+			if err != nil {
+				return err
+			}
+
+			client, cancel, err := createClient(cmd)
+			if err != nil {
+				return err
+			}
+			defer cancel()
+
+			res, err := client.ApplyAction(cmd.Context(), &reqBody)
+			if err != nil {
+				return err
+			}
+			spinner.Stop()
+
+			fmt.Println(cs.Greenf("Action applied successfully"))
+			if output == outputJSON || output == outputYAML || output == outputYML {
+				formattedString, err := formatOutput(res.GetResource(), output)
+				if err != nil {
+					return err
+				}
+				fmt.Println(cs.Bluef(formattedString))
+			}
+
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVarP(&urn, "urn", "u", "", "urn of the resource")
+	cmd.Flags().StringVarP(&file, "file", "f", "", "path to the params file")
+	cmd.Flags().StringVarP(&output, "out", "o", "", "output format, `-o json | yaml`")
+
+	return cmd
+}

cli/cli.go

@@ -17,12 +17,15 @@ familiar languages, tools, and engineering practices.`,
 func Execute(ctx context.Context) {
 	rootCmd.PersistentFlags().StringP(configFlag, "c", "", "Override config file")
 	rootCmd.AddCommand(
-		cmdServe(ctx),
-		cmdMigrate(ctx),
+		cmdServe(),
+		cmdMigrate(),
 		cmdVersion(),
 		cmdShowConfigs(),
+		cmdResource(),
+		cmdAction(),
+		cmdLogs(),
 	)
 
 	cmdx.SetHelp(rootCmd)
-	_ = rootCmd.Execute()
+	_ = rootCmd.ExecuteContext(ctx)
 }

cli/client.go

@@ -0,0 +1,47 @@
+package cli
+
+import (
+	"context"
+	"strconv"
+	"time"
+
+	"github.com/spf13/cobra"
+	entropyv1beta1 "go.buf.build/odpf/gwv/odpf/proton/odpf/entropy/v1beta1"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
+)
+
+const timeout = 2
+
+func createConnection(ctx context.Context, host string) (*grpc.ClientConn, error) {
+	opts := []grpc.DialOption{
+		grpc.WithTransportCredentials(insecure.NewCredentials()),
+		grpc.WithBlock(),
+	}
+
+	return grpc.DialContext(ctx, host, opts...)
+}
+
+func createClient(cmd *cobra.Command) (entropyv1beta1.ResourceServiceClient, func(), error) {
+	c, err := loadConfig(cmd)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	host := c.Service.Host + ":" + strconv.Itoa(c.Service.Port)
+
+	dialTimeoutCtx, dialCancel := context.WithTimeout(cmd.Context(), time.Second*timeout)
+	conn, err := createConnection(dialTimeoutCtx, host)
+	if err != nil {
+		dialCancel()
+		return nil, nil, err
+	}
+
+	cancel := func() {
+		dialCancel()
+		conn.Close()
+	}
+
+	client := entropyv1beta1.NewResourceServiceClient(conn)
+	return client, cancel, nil
+}

cli/logs.go

@@ -0,0 +1,81 @@
+package cli
+
+import (
+	"errors"
+	"io"
+	"log"
+	"strings"
+
+	"github.com/odpf/salt/term" //nolint
+	entropyv1beta1 "go.buf.build/odpf/gwv/odpf/proton/odpf/entropy/v1beta1"
+
+	"github.com/MakeNowJust/heredoc"
+	"github.com/odpf/salt/printer"
+	"github.com/spf13/cobra"
+)
+
+func cmdLogs() *cobra.Command {
+	var filter []string
+	filters := make(map[string]string)
+	cmd := &cobra.Command{
+		Use:     "logs <resource-urn>",
+		Aliases: []string{"logs"},
+		Short:   "Gets logs",
+		Example: heredoc.Doc(`
+			$ entropy logs <resource-urn> --filter="key1=value1" --filter="key2=value2"
+		`),
+		Annotations: map[string]string{
+			"group:core": "true",
+		},
+		Args: cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			spinner := printer.Spin("")
+			defer spinner.Stop()
+			cs := term.NewColorScheme()
+
+			client, cancel, err := createClient(cmd)
+			if err != nil {
+				return err
+			}
+			defer cancel()
+
+			var reqBody entropyv1beta1.GetLogRequest
+			for _, f := range filter {
+				keyValue := strings.Split(f, "=")
+				filters[keyValue[0]] = keyValue[1]
+			}
+
+			reqBody.Filter = filters
+			reqBody.Urn = args[0]
+
+			err = reqBody.ValidateAll()
+			if err != nil {
+				return err
+			}
+
+			stream, err := client.GetLog(cmd.Context(), &reqBody)
+			if err != nil {
+				return err
+			}
+			spinner.Stop()
+
+			for {
+				resp, err := stream.Recv()
+				if errors.Is(err, io.EOF) {
+					break
+				}
+				if err != nil {
+					log.Fatalf("cannot receive %v", err)
+				}
+				log.SetFlags(0)
+				log.Printf(cs.Bluef("%s", resp.GetChunk().GetData())) //nolint
+			}
+
+			return nil
+		},
+	}
+
+	cmd.Flags().StringArrayVarP(&filter, "filter", "f", nil, "Use filters. Example: --filter=\"key=value\"")
+
+	return cmd
+}

cli/migrate.go

@@ -8,10 +8,13 @@ import (
 	"github.com/odpf/entropy/internal/store/mongodb"
 )
 
-func cmdMigrate(ctx context.Context) *cobra.Command {
+func cmdMigrate() *cobra.Command {
 	cmd := &cobra.Command{
 		Use:   "migrate",
 		Short: "Run DB migrations",
+		Annotations: map[string]string{
+			"group:other": "server",
+		},
 	}
 
 	cmd.RunE = func(cmd *cobra.Command, args []string) error {
@@ -20,7 +23,7 @@ func cmdMigrate(ctx context.Context) *cobra.Command {
 			return err
 		}
 
-		return runMigrations(ctx, cfg.DB)
+		return runMigrations(cmd.Context(), cfg.DB)
 	}
 
 	return cmd