feat(lss): adds --no-header and --short flags, and lowercases all flag descriptions

cmd/list-dbs.go

@@ -24,6 +24,8 @@ package cmd
 
 import (
 	"fmt"
+	"os"
+
 	"github.com/spf13/cobra"
 )
 
@@ -43,12 +45,77 @@ func listStores(cmd *cobra.Command, args []string) error {
 	if err != nil {
 		return fmt.Errorf("cannot list stores: %v", err)
 	}
+
+	short, err := cmd.Flags().GetBool("short")
+	if err != nil {
+		return fmt.Errorf("cannot list stores: %v", err)
+	}
+
+	if short {
+		for _, db := range dbs {
+			fmt.Println("@" + db)
+		}
+		return nil
+	}
+
+	type storeInfo struct {
+		name  string
+		keys  int
+		size  string
+	}
+
+	rows := make([]storeInfo, 0, len(dbs))
+	nameW, keysW, sizeW := len("Store"), len("Keys"), len("Size")
+
 	for _, db := range dbs {
-		fmt.Println("@" + db)
+		p, err := store.storePath(db)
+		if err != nil {
+			return fmt.Errorf("cannot list stores: %v", err)
+		}
+		fi, err := os.Stat(p)
+		if err != nil {
+			return fmt.Errorf("cannot list stores: %v", err)
+		}
+		entries, err := readStoreFile(p, nil)
+		if err != nil {
+			return fmt.Errorf("cannot list stores: %v", err)
+		}
+		name := "@" + db
+		keysStr := fmt.Sprintf("%d", len(entries))
+		sizeStr := formatSize(int(fi.Size()))
+		if len(name) > nameW {
+			nameW = len(name)
+		}
+		if len(keysStr) > keysW {
+			keysW = len(keysStr)
+		}
+		if len(sizeStr) > sizeW {
+			sizeW = len(sizeStr)
+		}
+		rows = append(rows, storeInfo{name: name, keys: len(entries), size: sizeStr})
+	}
+
+	underline := func(s string) string {
+		if stdoutIsTerminal() {
+			return "\033[4m" + s + "\033[0m"
+		}
+		return s
+	}
+	noHeader, _ := cmd.Flags().GetBool("no-header")
+	if !noHeader {
+		fmt.Printf("%*s%s %*s%s %s\n",
+			keysW-len("Keys"), "", underline("Keys"),
+			sizeW-len("Size"), "", underline("Size"),
+			underline("Store"))
+	}
+	for _, r := range rows {
+		fmt.Printf("%*d %*s %s\n", keysW, r.keys, sizeW, r.size, r.name)
 	}
 	return nil
 }
 
 func init() {
+	listStoresCmd.Flags().Bool("short", false, "only print store names")
+	listStoresCmd.Flags().Bool("no-header", false, "suppress the header row")
 	rootCmd.AddCommand(listStoresCmd)
 }