cleanup -dry-run

[goredo.git] / cleanup.go

diff --git a/cleanup.go b/cleanup.go
index 2c4268c9a785df2c54eee40932d403ceeb4910cd..61f028f2d56dc6d600a1146ff1ac7d640f5cef56 100644 (file)
--- a/cleanup.go
+++ b/cleanup.go
@@ -18,6 +18,7 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 package main
 
 import (
+       "flag"
        "fmt"
        "io"
        "log"
@@ -33,6 +34,10 @@ const (
        CleanupTmp  = "tmp"
 )
 
+var (
+       DryRun = flag.Bool("dry-run", false, "do no delete files during cleanup, just show them")
+)
+
 func redoDirClean(root, what string) error {
        root, err := filepath.Abs(root)
        if err != nil {
@@ -57,15 +62,19 @@ func redoDirClean(root, what string) error {
                        case CleanupLog:
                                if strings.HasSuffix(fi.Name(), LogSuffix) {
                                        fmt.Println(pth)
-                                       if err = os.Remove(pth); err != nil {
-                                               return err
+                                       if !*DryRun {
+                                               if err = os.Remove(pth); err != nil {
+                                                       return err
+                                               }
                                        }
                                }
                        case CleanupTmp:
                                if strings.HasPrefix(fi.Name(), TmpPrefix) {
                                        fmt.Println(pth)
-                                       if err = os.Remove(pth); err != nil {
-                                               return err
+                                       if !*DryRun {
+                                               if err = os.Remove(pth); err != nil {
+                                                       return err
+                                               }
                                        }
                                }
                        default:
@@ -101,7 +110,9 @@ func cleanupWalker(root, what string) error {
                                if fi.Name() == RedoDir {
                                        if what == CleanupFull {
                                                fmt.Println(pthRel)
-                                               err = os.RemoveAll(pth)
+                                               if !*DryRun {
+                                                       err = os.RemoveAll(pth)
+                                               }
                                        } else {
                                                err = redoDirClean(pth, what)
                                        }
@@ -116,8 +127,10 @@ func cleanupWalker(root, what string) error {
                        if (what == CleanupTmp || what == CleanupFull) &&
                                strings.HasPrefix(fi.Name(), TmpPrefix) {
                                fmt.Println(pthRel)
-                               if err = os.Remove(pth); err != nil {
-                                       return err
+                               if !*DryRun {
+                                       if err = os.Remove(pth); err != nil {
+                                               return err
+                                       }
                                }
                        }
                }