Merge pull request #10 from Connehito/working-hours

Add checking safety hour

cli.go

@@ -1,11 +1,12 @@
 package main
 
 import (
+	"bufio"
 	"flag"
 	"fmt"
-	"io"
-
 	"github.com/mitchellh/colorstring"
+	"io"
+	"os"
 )
 
 // CLI has stdout/stderr's writer and Gdp's interface.
@@ -27,6 +28,12 @@ const (
 	CommandPublish = "publish"
 )
 
+// Safety Hour.
+const (
+	SafetyHourStart  = 9
+	SafetyHourEnd = 19
+)
+
 // Run invokes depoloy and publish's process.
 func (cli *CLI) Run(args []string) int {
 	var version bool
@@ -126,6 +133,14 @@ func (cli *CLI) Run(args []string) int {
 
 	// execution
 	if subCommand == CommandDeploy {
+		if !isSafetyHour() {
+			fmt.Fprintln(cli.outStream, "It's past the regular time. Is this a hot-fix release?")
+			fmt.Fprint(cli.outStream, "> ")
+			if !yesOrNo(cli) {
+				return ExitError
+			}
+		}
+
 		if err := cli.gdp.Deploy(tag); err != nil {
 			printError(cli.errStream, fmt.Sprintf("Deploy execution error: %s.", err.Error()))
 			return ExitError
@@ -173,3 +188,31 @@ func validate(cli *CLI, subCommand string, tag string) bool {
 
 	return true
 }
+
+func isSafetyHour() bool {
+	t := Now()
+	if t.Hour() > SafetyHourStart && t.Hour() < SafetyHourEnd {
+		return  true
+	}
+
+	return false
+}
+
+func yesOrNo(cli *CLI) bool {
+	reader := bufio.NewReader(os.Stdin)
+	s, err := reader.ReadByte()
+	if err != nil {
+		return false
+	}
+
+	if s == []byte("Y")[0] || s == []byte("y")[0] {
+		fmt.Fprintln(cli.outStream, "OK. Take time.")
+		return true
+	} else if s == []byte("N")[0] || s == []byte("n")[0] {
+		printError(cli.errStream, fmt.Sprintf("Good choice."))
+		return false
+	}
+
+	printError(cli.errStream, fmt.Sprintf("Please enter y or n."))
+	return false
+}

cli_test.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"strings"
 	"testing"
+	"time"
 )
 
 // Tests for flag
@@ -183,6 +184,7 @@ func TestRun_Deploy(t *testing.T) {
 		errStream: err,
 		gdp:       &FakeGdpDeploy{},
 	}
+	Set(time.Date(2020, 4, 1, 17, 00, 00, 0, time.Local))
 
 	args := strings.Split("gdp deploy -t v1.2.4", " ")
 	code := cli.Run(args)
@@ -223,6 +225,7 @@ func TestRun_DeployNoSpecifiedTag(t *testing.T) {
 		errStream: err,
 		gdp:       &FakeGdpDeploy{},
 	}
+	Set(time.Date(2020, 4, 1, 17, 00, 00, 0, time.Local))
 
 	args := strings.Split("gdp deploy", " ")
 
@@ -258,6 +261,7 @@ func TestRun_DeployForce(t *testing.T) {
 		errStream: err,
 		gdp:       &FakeGdpDeployForce{},
 	}
+	Set(time.Date(2020, 4, 1, 17, 00, 00, 0, time.Local))
 
 	args := strings.Split("gdp deploy -t v1.2.4 -f", " ")
 	code := cli.Run(args)
@@ -432,6 +436,7 @@ func TestRun_DeployErrorInDeploy(t *testing.T) {
 		errStream: err,
 		gdp:       &FakeGdpDeployErrorInDeploy{},
 	}
+	Set(time.Date(2020, 4, 1, 17, 00, 00, 0, time.Local))
 
 	args := strings.Split("gdp deploy -t v1.2.4", " ")
 	code := cli.Run(args)

main.go

@@ -6,7 +6,7 @@ import (
 )
 
 // Version is GDP's version.
-const Version string = "v0.2.2"
+const Version string = "v0.2.3"
 
 // Usage is GDP's usage.
 const Usage string = "usage: gdp deploy|publish [options]"

time.go

@@ -0,0 +1,16 @@
+package main
+
+import "time"
+
+var fakeTime time.Time
+
+func Now() time.Time {
+	if !fakeTime.IsZero() {
+		return fakeTime
+	}
+	return time.Now()
+}
+
+func Set(t time.Time) {
+	fakeTime = t
+}