feat: Implement stcache list command

2021-11-21 01:44:08 +01:00
parent 43053d83b9
commit 4ad796bb17
3 changed files with 98 additions and 8 deletions
@@ -0,0 +1,86 @@
+package cmd
+
+import (
+	"archive/tar"
+	"context"
+	"encoding/json"
+	"os"
+	"path/filepath"
+
+	"github.com/pojntfx/stfs/pkg/formatting"
+	"github.com/pojntfx/stfs/pkg/persisters"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+var list = &cobra.Command{
+	Use:     "list",
+	Aliases: []string{"l"},
+	Short:   "List contents of index",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		if err := viper.BindPFlags(cmd.PersistentFlags()); err != nil {
+			return err
+		}
+
+		metadataPersister := persisters.NewMetadataPersister(viper.GetString(dbFlag))
+		if err := metadataPersister.Open(); err != nil {
+			return err
+		}
+
+		headers, err := metadataPersister.GetHeaders(context.Background())
+		if err != nil {
+			return err
+		}
+
+		for i, hdr := range headers {
+			if i == 0 {
+				if err := formatting.PrintCSV(formatting.TARHeaderCSV); err != nil {
+					return err
+				}
+			}
+
+			paxRecords := map[string]string{}
+			if err := json.Unmarshal([]byte(hdr.Paxrecords), &paxRecords); err != nil {
+				return err
+			}
+
+			if err := formatting.PrintCSV(formatting.GetTARHeaderAsCSV(hdr.Record, hdr.Block, &tar.Header{
+				Typeflag:   byte(hdr.Typeflag),
+				Name:       hdr.Name,
+				Linkname:   hdr.Linkname,
+				Size:       hdr.Size,
+				Mode:       hdr.Mode,
+				Uid:        int(hdr.UID),
+				Gid:        int(hdr.Gid),
+				Uname:      hdr.Uname,
+				Gname:      hdr.Gname,
+				ModTime:    hdr.Modtime,
+				AccessTime: hdr.Accesstime,
+				ChangeTime: hdr.Changetime,
+				Devmajor:   hdr.Devmajor,
+				Devminor:   hdr.Devminor,
+				PAXRecords: paxRecords,
+				Format:     tar.Format(hdr.Format),
+			})); err != nil {
+				return err
+			}
+		}
+
+		return nil
+	},
+}
+
+func init() {
+	// Get default working dir
+	home, err := os.UserHomeDir()
+	if err != nil {
+		panic(err)
+	}
+	workingDirDefault := filepath.Join(home, ".local", "share", "stcache", "var", "lib", "stcache")
+
+	list.PersistentFlags().StringP(dbFlag, "d", filepath.Join(workingDirDefault, "index.sqlite"), "Database to use")
+
+	viper.AutomaticEnv()
+
+	rootCmd.AddCommand(list)
+}