Merge pull request #16 from remind101/abort-prod-push

Ask for confirmation before deploying to prod. Fix #13.

deploy.go

@@ -1,6 +1,7 @@
 package deploy
 
 import (
+	"bufio"
 	"errors"
 	"fmt"
 	"io"
@@ -70,6 +71,10 @@ var flags = []cli.Flag{
 	},
 }
 
+var ProtectedEnvironments = map[string]bool{
+	"production": true,
+}
+
 // NewApp returns a new cli.App for the deploy command.
 func NewApp() *cli.App {
 	app := cli.NewApp()
@@ -182,6 +187,13 @@ func newDeploymentRequest(c *cli.Context) (*github.DeploymentRequest, error) {
 		return nil, fmt.Errorf("--env flag is required")
 	}
 
+	if ProtectedEnvironments[env] {
+		yes := askYN("Are you sure you want to push to production?")
+		if !yes {
+			return nil, fmt.Errorf("Deployment aborted.")
+		}
+	}
+
 	var contexts *[]string
 	if c.Bool("force") {
 		s := []string{}
@@ -341,3 +353,10 @@ func SplitRepo(nwo, defaultOrg string) (owner string, repo string, err error) {
 
 	return
 }
+
+func askYN(prompt string) bool {
+	r := bufio.NewReader(os.Stdin)
+	fmt.Printf("%s (y/N)\n", prompt)
+	a, _ := r.ReadString('\n')
+	return strings.ToUpper(a) == "Y\n"
+}