Better listing of repos

api/repositories.go

@@ -0,0 +1,47 @@
+package api
+
+import "github.com/shurcooL/graphql"
+
+type Repositories struct {
+	client *Client
+}
+
+type Repository struct {
+	Name            string
+	RetentionDays   int64 `graphql:"timeBasedRetention"`
+	RetentionSizeGB int64 `graphql:"storageSizeBasedRetention"`
+	SpaceUsed       int64 `graphql:"compressedByteSize"`
+}
+
+func (c *Client) Repositories() *Repositories { return &Repositories{client: c} }
+
+func (c *Repositories) Get(name string) (Repository, error) {
+	var q struct {
+		Repository Repository `graphql:"repository(name: $name)"`
+	}
+
+	variables := map[string]interface{}{
+		"name": graphql.String(name),
+	}
+
+	graphqlErr := c.client.Query(&q, variables)
+
+	return q.Repository, graphqlErr
+}
+
+type RepoListItem struct {
+	Name      string
+	SpaceUsed int64 `graphql:"compressedByteSize"`
+}
+
+func (c *Repositories) List() ([]RepoListItem, error) {
+	var q struct {
+		Repositories []RepoListItem `graphql:"repositories"`
+	}
+
+	variables := map[string]interface{}{}
+
+	graphqlErr := c.client.Query(&q, variables)
+
+	return q.Repositories, graphqlErr
+}

cmd/output.go

@@ -1,6 +1,8 @@
 package cmd
 
 import (
+	"fmt"
+
 	"github.com/ryanuber/columnize"
 	"github.com/spf13/cobra"
 )
@@ -34,3 +36,16 @@ func valueOrEmpty(v string) string {
 	}
 	return v
 }
+
+func ByteCountDecimal(b int64) string {
+	const unit = 1000
+	if b < unit {
+		return fmt.Sprintf("%d B", b)
+	}
+	div, exp := int64(unit), 0
+	for n := b / unit; n >= unit; n /= unit {
+		div *= unit
+		exp++
+	}
+	return fmt.Sprintf("%.1f %cB", float64(b)/float64(div), "kMGTPE"[exp])
+}

cmd/repos.go

@@ -0,0 +1,52 @@
+// 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/humio/cli/api"
+	"github.com/olekukonko/tablewriter"
+	"github.com/spf13/cobra"
+)
+
+func newReposCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "repos",
+		Short: "Manage repositories",
+	}
+
+	cmd.AddCommand(newReposShowCmd())
+	cmd.AddCommand(newReposListCmd())
+
+	return cmd
+}
+
+func printRepoTable(cmd *cobra.Command, repo api.Repository) {
+
+	data := [][]string{
+		[]string{"Name", repo.Name},
+		[]string{"Space Used", ByteCountDecimal(repo.SpaceUsed)},
+		[]string{"Retention (Size)", ByteCountDecimal(repo.RetentionSizeGB / 1000000000)},
+		[]string{"Retention (Days)", fmt.Sprintf("%d", repo.RetentionDays)},
+	}
+
+	w := tablewriter.NewWriter(cmd.OutOrStdout())
+	w.AppendBulk(data)
+	w.SetBorder(false)
+	w.SetColumnSeparator(":")
+	w.SetColumnAlignment([]int{tablewriter.ALIGN_RIGHT, tablewriter.ALIGN_LEFT})
+	w.Render()
+}

cmd/repos_list.go

@@ -0,0 +1,74 @@
+// 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 (
+	"sort"
+
+	"github.com/humio/cli/api"
+	"github.com/olekukonko/tablewriter"
+	"github.com/spf13/cobra"
+)
+
+func newReposListCmd() *cobra.Command {
+	var orderBySize, reverse bool
+
+	cmd := cobra.Command{
+		Use:   "list [flags]",
+		Short: "List repositories.",
+		Args:  cobra.ExactArgs(0),
+		Run: func(cmd *cobra.Command, args []string) {
+			client := NewApiClient(cmd)
+
+			repos, apiErr := client.Repositories().List()
+			exitOnError(cmd, apiErr, "error fetching repository")
+
+			sort.Slice(repos, func(i, j int) bool {
+				var a, b api.RepoListItem
+				if reverse {
+					a = repos[i]
+					b = repos[j]
+				} else {
+					a = repos[j]
+					b = repos[j]
+				}
+
+				if !orderBySize {
+					return a.Name < b.Name
+				} else {
+					return a.SpaceUsed > b.SpaceUsed
+				}
+			})
+
+			rows := make([][]string, len(repos))
+			for i, view := range repos {
+				rows[i] = []string{view.Name, ByteCountDecimal(view.SpaceUsed)}
+			}
+
+			w := tablewriter.NewWriter(cmd.OutOrStdout())
+			w.SetHeader([]string{"Name", "Space Used"})
+			w.AppendBulk(rows)
+			w.SetBorder(false)
+
+			w.Render()
+			cmd.Println()
+		},
+	}
+
+	cmd.Flags().BoolVarP(&orderBySize, "size", "s", false, "Order by size instead of name")
+	cmd.Flags().BoolVarP(&reverse, "reverse", "r", true, "Reverse sorting order")
+
+	return &cmd
+}

cmd/repos_show.go

@@ -0,0 +1,43 @@
+// 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 newReposShowCmd() *cobra.Command {
+	cmd := cobra.Command{
+		Use:   "show [flags] <repo>",
+		Short: "Show details about a repository.",
+		Args:  cobra.ExactArgs(1),
+		Run: func(cmd *cobra.Command, args []string) {
+			repoName := args[0]
+
+			client := NewApiClient(cmd)
+
+			repo, apiErr := client.Repositories().Get(repoName)
+			exitOnError(cmd, apiErr, "error fetching repository")
+
+			printRepoTable(cmd, repo)
+
+			fmt.Println()
+		},
+	}
+
+	return &cmd
+}

cmd/root.go

@@ -111,6 +111,7 @@ Common Management Commands:
 	rootCmd.AddCommand(newViewsCmd())
 	rootCmd.AddCommand(newCompletionCmd())
 	rootCmd.AddCommand(newLicenseCmd())
+	rootCmd.AddCommand(newReposCmd())
 }
 
 // initConfig reads in config file and ENV variables if set.