adding a --fork flag to allow contributors to use their own forks in ponzu-dev workflow & described this in readme

2 files changed, 22 insertions, 2 deletions

diff --git a/cmd/ponzu/main.go b/cmd/ponzu/main.go
index d919242..ba85f36 100644
--- a/cmd/ponzu/main.go
+++ b/cmd/ponzu/main.go
@@ -75,7 +75,8 @@ var (
 	tls  bool
 
 	// for ponzu internal / core development
-	dev bool
+	dev  bool
+	fork string
 )
 
 func init() {
@@ -88,6 +89,7 @@ func main() {
 	flag.IntVar(&port, "port", 8080, "port for ponzu to bind its listener")
 	flag.BoolVar(&tls, "tls", false, "enable automatic TLS/SSL certificate management")
 	flag.BoolVar(&dev, "dev", false, "modify environment for Ponzu core development")
+	flag.StringVar(&fork, "fork", "", "modify repo source for Ponzu core development")
 	flag.Parse()
 
 	args := flag.Args()
diff --git a/cmd/ponzu/options.go b/cmd/ponzu/options.go
index c5873c2..0090da7 100644
--- a/cmd/ponzu/options.go
+++ b/cmd/ponzu/options.go
@@ -215,11 +215,29 @@ func createProjInDir(path string) error {
 	}
 
 	if dev {
+		branch := exec.Command("git", "checkout", "ponzu-dev")
+		branch.Stdout = os.Stdout
+		branch.Stderr = os.Stderr
+
+		err := branch.Start()
+		if err != nil {
+			return err
+		}
+
+		err = branch.Wait()
+		if err != nil {
+			return err
+		}
+
+		if fork != "" {
+			local = filepath.Join(gopath, "src", fork)
+		}
+
 		devClone := exec.Command("git", "clone", local, "--branch", "ponzu-dev", "--single-branch", path)
 		devClone.Stdout = os.Stdout
 		devClone.Stderr = os.Stderr
 
-		err := devClone.Start()
+		err = devClone.Start()
 		if err != nil {
 			return err
 		}