X-Git-Url: http://www.git.cypherpunks.ru/?p=goredo.git;a=blobdiff_plain;f=cleanup.go;h=bdd295dd3ee6689f8ed93af330f4f30b7de1303a;hp=f75788aae270780f69df8b567fad0c40ab5df6fb;hb=b4eefdd675c9aef9ff8bd1089d031ee05733195b;hpb=edb601644735972fac3efcdc27849b5b9b1f5fb7

diff --git a/cleanup.go b/cleanup.go
index f75788a..bdd295d 100644
--- a/cleanup.go
+++ b/cleanup.go
@@ -1,6 +1,6 @@
 /*
-goredo -- redo implementation on pure Go
-Copyright (C) 2020-2021 Sergey Matveev <stargrave@stargrave.org>
+goredo -- djb's redo implementation on pure Go
+Copyright (C) 2020-2023 Sergey Matveev <stargrave@stargrave.org>
 
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
@@ -18,6 +18,7 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 package main
 
 import (
+	"flag"
 	"fmt"
 	"io"
 	"log"
@@ -31,8 +32,18 @@ const (
 	CleanupFull = "full"
 	CleanupLog  = "log"
 	CleanupTmp  = "tmp"
+	CleanupLock = "lock"
 )
 
+var DryRun *bool
+
+func init() {
+	if CmdName() != CmdNameRedoCleanup {
+		return
+	}
+	DryRun = flag.Bool("n", false, "do no delete files during cleanup, just show them")
+}
+
 func redoDirClean(root, what string) error {
 	root, err := filepath.Abs(root)
 	if err != nil {
@@ -42,6 +53,7 @@ func redoDirClean(root, what string) error {
 	if err != nil {
 		return err
 	}
+	defer dir.Close()
 	for {
 		fis, err := dir.Readdir(1 << 10)
 		if err != nil {
@@ -55,17 +67,31 @@ func redoDirClean(root, what string) error {
 			pth = cwdMustRel(root, fi.Name())
 			switch what {
 			case CleanupLog:
-				if strings.HasSuffix(fi.Name(), LogSuffix) {
+				if strings.HasSuffix(fi.Name(), LogSuffix) ||
+					strings.HasSuffix(fi.Name(), LogRecSuffix) {
 					fmt.Println(pth)
-					if err = os.Remove(pth); err != nil {
-						return err
+					if !*DryRun {
+						if err = os.Remove(pth); err != nil {
+							return err
+						}
+					}
+				}
+			case CleanupLock:
+				if strings.HasSuffix(fi.Name(), LockSuffix) {
+					fmt.Println(pth)
+					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:
@@ -73,7 +99,7 @@ func redoDirClean(root, what string) error {
 			}
 		}
 	}
-	return dir.Close()
+	return nil
 }
 
 func cleanupWalker(root, what string) error {
@@ -101,10 +127,18 @@ 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)
 					}
+				} else if (what == CleanupTmp || what == CleanupFull) &&
+					strings.HasPrefix(fi.Name(), TmpPrefix) {
+					fmt.Println(pthRel)
+					if !*DryRun {
+						err = os.RemoveAll(pth)
+					}
 				} else {
 					err = cleanupWalker(pth, what)
 				}
@@ -116,11 +150,13 @@ 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
+					}
 				}
 			}
 		}
 	}
-	return dir.Close()
+	return nil
 }