1 files changed, 125 insertions, 0 deletions
diff --git a/main.go b/main.go
new file mode 100644
index 0000000..785494b
--- /dev/null
+++ b/main.go
@@ -0,0 +1,125 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"go/parser"
+	"go/printer"
+	"go/token"
+	"io/ioutil"
+	"os"
+	"path"
+	"path/filepath"
+	"strings"
+
+	"github.com/codegangsta/cli"
+
+	"golang.org/x/tools/go/ast/astutil"
+)
+
+func main() {
+	app := cli.NewApp()
+	app.Name = "gomove"
+	app.Usage = "Move Golang packages to a new path."
+	app.Version = "0.0.1"
+	app.ArgsUsage = "[old path] [new path]"
+	app.Author = "Kaushal Subedi <kaushal@subedi.co>"
+
+	app.Flags = []cli.Flag{
+		cli.StringFlag{
+			Name:  "dir, d",
+			Value: "./",
+			Usage: "directory to scan",
+		},
+		cli.StringFlag{
+			Name:  "file, f",
+			Usage: "only move imports in a file",
+		},
+	}
+
+	app.Action = func(c *cli.Context) {
+		file := c.String("file")
+		dir := c.String("dir")
+		from := c.Args().Get(0)
+		to := c.Args().Get(1)
+
+		if file != "" {
+			ProcessFile(file, from, to)
+		} else {
+			RunApp(dir, from, to, c)
+		}
+
+	}
+
+	app.Run(os.Args)
+}
+
+func RunApp(dir string, from string, to string, c *cli.Context) {
+
+	if from != "" && to != "" {
+		filepath.Walk(dir, func(filePath string, info os.FileInfo, err error) error {
+			ProcessFile(filePath, from, to)
+			return nil
+		})
+
+	} else {
+		cli.ShowAppHelp(c)
+	}
+
+}
+
+func ProcessFile(filePath string, from string, to string) {
+	// If the file is a go file scan it
+	if path.Ext(filePath) == ".go" {
+		// New FileSet to parse the go file to
+		fSet := token.NewFileSet()
+
+		// Parse the file
+		file, err := parser.ParseFile(fSet, filePath, nil, 0)
+		if err != nil {
+			fmt.Println(err)
+		}
+
+		// Get the list of imports from the ast
+		imports := astutil.Imports(fSet, file)
+
+		// Keep track of number of changes
+		numChanges := 0
+
+		// Iterate through the imports array
+		for _, mPackage := range imports {
+			for _, mImport := range mPackage {
+				// Since astutil returns the path string with quotes, remove those
+				importString := strings.Replace(mImport.Path.Value, "\"", "", -1)
+
+				// If the path matches the oldpath, replace it with the new one
+				if strings.Contains(importString, from) {
+					//If it needs to be replaced, increase numChanges so we can write the file later
+					numChanges++
+
+					// Join the path of the import package with the remainder from the old one after removing the old import package
+					replacePackage := path.Join(to, strings.Replace(importString, from, "", -1))
+
+					fmt.Println("Updating import " + importString + " from file " + filePath + " to " + replacePackage)
+
+					// Remove the old import and replace it with the replacement
+					astutil.DeleteImport(fSet, file, importString)
+					astutil.AddImport(fSet, file, replacePackage)
+				}
+			}
+		}
+
+		// If the number of changes are more than 0, write file
+		if numChanges > 0 {
+			// Print the new AST tree to a new output buffer
+			var outputBuffer bytes.Buffer
+			printer.Fprint(&outputBuffer, fSet, file)
+
+			ioutil.WriteFile(filePath, outputBuffer.Bytes(), os.ModePerm)
+
+			fmt.Printf("File "+filePath+" saved after %d changes.\n", numChanges)
+		} else {
+			fmt.Println("No changes needed on file " + filePath)
+		}
+	}
+}