From cb7cd6e03095ac7acbfeb2c74a587c4f053ef6bc Mon Sep 17 00:00:00 2001 From: Colin P. Mccabe Date: Thu, 22 Jul 2021 16:28:21 -0700 Subject: [PATCH] Add recursive_decompress.go --- .gitignore | 1 + recursive_decompress.go | 91 +++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 92 insertions(+), 0 deletions(-) create mode 100644 recursive_decompress.go diff --git a/.gitignore b/.gitignore index b17d6aa..b6a0926 100644 --- a/.gitignore +++ b/.gitignore @@ -17,6 +17,7 @@ msgpack-translate directory_merge pickrand random_word +recursive_decompress audiobooker # diff --git a/recursive_decompress.go b/recursive_decompress.go new file mode 100644 index 0000000..a642fa7 --- /dev/null +++ b/recursive_decompress.go @@ -0,0 +1,91 @@ +package main + +import ( + "flag" + "os/exec" + "path/filepath" + "fmt" + "os" + "strings" +) + +var removeArchives *bool + +func processPath(curPath string) error { + info, err := os.Stat(curPath) + if err != nil { + return fmt.Errorf("Unable to stat %s: %s", curPath, err.Error()) + } + if info.IsDir() { + entries, err := os.ReadDir(curPath) + if err != nil { + return fmt.Errorf("Unable to call readdir on %s: %s", curPath, err.Error()) + } + for i := range(entries) { + err := processPath(filepath.Join(curPath, entries[i].Name())) + if err != nil { + return err + } + } + } else { + tarBaseName := tarBaseName(curPath) + if tarBaseName != "" { + err = applyTarOn(curPath) + if err != nil { + return err + } + err = processPath(filepath.Join(filepath.Dir(curPath), tarBaseName)) + if err != nil { + return err + } + if *removeArchives { + err = os.Remove(curPath) + if err != nil { + return fmt.Errorf("Unable to remove archive file %s: %s", curPath, err.Error()) + } + fmt.Printf("== rm %s\n", curPath) + } + } + } + return nil +} + +func tarBaseName(curPath string) string { + if strings.HasSuffix(curPath, ".tgz") { + return filepath.Base(curPath[:len(curPath)-4]) + } + return "" +} + +func applyTarOn(curPath string) error { + fmt.Printf("== decompressing tar file %s\n", curPath) + cmd := exec.Command("tar", "xvf", filepath.Base(curPath)) + cmd.Dir = filepath.Dir(curPath) + //cmd.Stdout = os.Stdout + err := cmd.Run() + if err != nil { + return fmt.Errorf("error running tar xvf on %s: %s", curPath, err.Error()) + } + return nil +} + +func main() { + removeArchives = flag.Bool("r", false, "Remove archive files after decompressing them") + flag.Usage = func() { + fmt.Printf("recursive_decompress: recursively decompresses a path.\n") + fmt.Printf("usage: recursive_decompress [flags] [paths...]\n") + fmt.Printf("\n") + flag.PrintDefaults() + } + flag.Parse() + curPath := "." + if flag.Arg(0) != "" { + curPath = flag.Arg(0) + } + err := processPath(curPath) + if err != nil { + fmt.Printf("error: %s\n", err.Error()) + os.Exit(1) + } + os.Exit(0) +} -- 1.6.6.rc1.39.g9a42