Basic support for views

Author: anagrius

api/views.go

@@ -0,0 +1,38 @@
+package api
+
+import "github.com/shurcooL/graphql"
+
+type Views struct {
+	client *Client
+}
+
+type RolePermission struct {
+	Role struct {
+		Name string
+	}
+	View struct {
+		Name string
+	}
+	QueryPrefix string
+}
+
+type View struct {
+	Name  string
+	Roles []RolePermission
+}
+
+func (c *Client) Views() *Views { return &Views{client: c} }
+
+func (c *Views) Get(name string) (View, error) {
+	var q struct {
+		View View `graphql:"searchDomain(name: $name)"`
+	}
+
+	variables := map[string]interface{}{
+		"name": graphql.String(name),
+	}
+
+	graphqlErr := c.client.Query(&q, variables)
+
+	return q.View, graphqlErr
+}

cmd/root.go

@@ -65,6 +65,7 @@ or
 Common commands:
   users <subcommand>
   parsers <subcommand>
+	views <subcommand>
 		`,
 	}
 
@@ -85,6 +86,7 @@ Common commands:
 	rootCmd.AddCommand(newIngestCmd())
 	rootCmd.AddCommand(newLoginCmd())
 	rootCmd.AddCommand(newIngestTokensCmd())
+	rootCmd.AddCommand(newViewsCmd())
 }
 
 // initConfig reads in config file and ENV variables if set.

cmd/users_show.go

@@ -34,7 +34,7 @@ func newUsersShowCmd() *cobra.Command {
 			user, err := client.Users().Get(username)
 
 			if err != nil {
-				return fmt.Errorf("Error fetching token list: %s", err)
+				return fmt.Errorf("Error fetching user: %s", err)
 			}
 
 			printUserTable(user)

cmd/views.go

@@ -0,0 +1,80 @@
+// Copyright © 2018 Humio Ltd.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package cmd
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/humio/cli/api"
+	"github.com/olekukonko/tablewriter"
+	"github.com/spf13/cobra"
+)
+
+func newViewsCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "views",
+		Short: "Manage views",
+	}
+
+	cmd.AddCommand(newViewsShowCmd())
+
+	return cmd
+}
+
+func printViewTable(view api.View) {
+
+	data := [][]string{
+		[]string{"Name", view.Name},
+	}
+
+	w := tablewriter.NewWriter(os.Stdout)
+	w.AppendBulk(data)
+	w.SetBorder(false)
+	w.SetColumnSeparator(":")
+	w.SetColumnAlignment([]int{tablewriter.ALIGN_RIGHT, tablewriter.ALIGN_LEFT})
+
+	fmt.Println()
+	w.Render()
+	fmt.Println()
+}
+
+func printViewRoleTable(view api.View) {
+
+	data := [][]string{}
+
+	for _, role := range view.Roles {
+		data = append(data, []string{role.Role.Name, role.QueryPrefix})
+	}
+
+	w := tablewriter.NewWriter(os.Stdout)
+	w.AppendBulk(data)
+	w.SetBorder(true)
+	w.SetHeader([]string{"Role", "Query Prefix"})
+	w.SetColumnSeparator(":")
+	w.SetColumnAlignment([]int{tablewriter.ALIGN_RIGHT, tablewriter.ALIGN_LEFT})
+
+	fmt.Println()
+	w.Render()
+	fmt.Println()
+}
+
+func viewRoleNames(user api.User) []string {
+	names := make([]string, len(user.Roles))
+	for i, r := range user.Roles {
+		names[i] = r.Name
+	}
+	return names
+}

cmd/views_show.go

@@ -0,0 +1,49 @@
+// Copyright © 2018 Humio Ltd.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/spf13/cobra"
+)
+
+func newViewsShowCmd() *cobra.Command {
+	cmd := cobra.Command{
+		Use:   "show [flags] <view>",
+		Short: "Show details about a view.",
+		Args:  cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			viewName := args[0]
+
+			// Get the HTTP client
+			client := NewApiClient(cmd)
+
+			view, err := client.Views().Get(viewName)
+
+			if err != nil {
+				return fmt.Errorf("Error fetching view: %s", err)
+			}
+
+			printViewTable(view)
+
+			printViewRoleTable(view)
+
+			return nil
+		},
+	}
+
+	return &cmd
+}