From 96b3af48b5107395f5b8745c1b99ebbbee59f8ed Mon Sep 17 00:00:00 2001
From: Kaushal Subedi
Date: Mon, 26 Oct 2015 21:10:37 -0600
Subject: added new flag for safe mode

---
 main.go | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index 3d5d856..d235495 100644
--- a/main.go
+++ b/main.go
@@ -26,6 +26,10 @@ func main() {
 			Name:  "file, f",
 			Usage: "only move imports in a file",
 		},
+		cli.StringFlag{
+			Name:  "safe-mode, s",
+			Usage: "run program in safe mode (comments will be wiped)",
+		},
 	}
 
 	app.Action = func(c *cli.Context) {
@@ -35,7 +39,7 @@ func main() {
 		to := c.Args().Get(1)
 
 		if file != "" {
-			ProcessFileNative(file, from, to)
+			ProcessFile(file, from, to, c)
 		} else {
 			ScanDir(dir, from, to, c)
 		}
@@ -54,7 +58,7 @@ func ScanDir(dir string, from string, to string, c *cli.Context) {
 		filepath.Walk(dir, func(filePath string, info os.FileInfo, err error) error {
 			// Only process go files
 			if path.Ext(filePath) == ".go" {
-				ProcessFileNative(filePath, from, to)
+				ProcessFile(filePath, from, to, c)
 			}
 			return nil
 		})
@@ -64,3 +68,12 @@ func ScanDir(dir string, from string, to string, c *cli.Context) {
 	}
 
 }
+
+// ProcessFile processes file according to what mode is chosen
+func ProcessFile(filePath string, from string, to string, c *cli.Context) {
+	if c.String("safe-mode") == "true" {
+		ProcessFileAST(filePath, from, to)
+	} else {
+		ProcessFileNative(filePath, from, to)
+	}
+}
-- 
cgit v1.2.3