Merge pull request #361 from magodo/not_remove_mapping_file

Allow resource mapping file in the output directory in `map` mode

internal/utils/remove_everything.go

@@ -6,14 +6,26 @@ import (
 	"path/filepath"
 )
 
-func RemoveEverythingUnder(path string) error {
+//	RemoveEverythingUnder removes everything under a path.
+//
+// The top level directory entries whose name matches any "skipps" will be skipped.
+func RemoveEverythingUnder(path string, skipps ...string) error {
 	// #nosec G304
 	dir, err := os.Open(path)
 	if err != nil {
 		return fmt.Errorf("failed to read directory %s: %v", path, err)
 	}
+
+	skipMap := map[string]bool{}
+	for _, v := range skipps {
+		skipMap[v] = true
+	}
+
 	entries, _ := dir.Readdirnames(0)
 	for _, entry := range entries {
+		if skipMap[entry] {
+			continue
+		}
 		if err := os.RemoveAll(filepath.Join(path, entry)); err != nil {
 			return fmt.Errorf("failed to remove %s: %v", entry, err)
 		}

main.go

@@ -13,6 +13,7 @@ import (
 	"strings"
 
 	internalconfig "github.com/Azure/aztfy/internal/config"
+	"github.com/Azure/aztfy/internal/meta"
 	"github.com/pkg/profile"
 
 	"github.com/Azure/aztfy/pkg/config"
@@ -130,7 +131,7 @@ func main() {
 		if !empty {
 			switch {
 			case flagOverwrite:
-				if err := utils.RemoveEverythingUnder(flagOutputDir); err != nil {
+				if err := utils.RemoveEverythingUnder(flagOutputDir, meta.ResourceMappingFileName); err != nil {
 					return fmt.Errorf("failed to clean up output directory %q: %v", flagOutputDir, err)
 				}
 			case flagAppend:
@@ -154,7 +155,7 @@ The output directory is not empty. Please choose one of actions below:
 				fmt.Scanf("%s", &ans)
 				switch strings.ToLower(ans) {
 				case "y":
-					if err := utils.RemoveEverythingUnder(flagOutputDir); err != nil {
+					if err := utils.RemoveEverythingUnder(flagOutputDir, meta.ResourceMappingFileName); err != nil {
 						return err
 					}
 				case "n":