feat: Add find command for searching by regex

This commit is contained in:
Felix Pojtinger
2021-11-28 23:10:33 +01:00
parent fe21a93576
commit dc84733080

74
cmd/stbak/cmd/find.go Normal file
View File

@@ -0,0 +1,74 @@
package cmd
import (
"context"
"regexp"
"github.com/pojntfx/stfs/pkg/converters"
"github.com/pojntfx/stfs/pkg/formatting"
"github.com/pojntfx/stfs/pkg/persisters"
"github.com/spf13/cobra"
"github.com/spf13/viper"
"github.com/volatiletech/sqlboiler/v4/boil"
)
const (
expressionFlag = "expression"
)
var findCmd = &cobra.Command{
Use: "find",
Aliases: []string{"f"},
Short: "Find a file or directory by matching against a regex",
RunE: func(cmd *cobra.Command, args []string) error {
if err := viper.BindPFlags(cmd.PersistentFlags()); err != nil {
return err
}
if viper.GetBool(verboseFlag) {
boil.DebugMode = true
}
metadataPersister := persisters.NewMetadataPersister(viper.GetString(metadataFlag))
if err := metadataPersister.Open(); err != nil {
return err
}
headers, err := metadataPersister.GetHeaders(context.Background())
if err != nil {
return err
}
first := true
for _, dbhdr := range headers {
if regexp.MustCompile(viper.GetString(expressionFlag)).Match([]byte(dbhdr.Name)) {
if first {
if err := formatting.PrintCSV(formatting.TARHeaderCSV); err != nil {
return err
}
first = false
}
hdr, err := converters.DBHeaderToTarHeader(dbhdr)
if err != nil {
return err
}
if err := formatting.PrintCSV(formatting.GetTARHeaderAsCSV(dbhdr.Record, dbhdr.Block, hdr)); err != nil {
return err
}
}
}
return nil
},
}
func init() {
findCmd.PersistentFlags().StringP(expressionFlag, "e", "", "Regex to match the file/directory name against")
viper.AutomaticEnv()
rootCmd.AddCommand(findCmd)
}