cli user interface

1 files changed, 21 insertions, 0 deletions

diff --git a/main.go b/main.go
index 7aa68c1..8a8f2c4 100644
--- a/main.go
+++ b/main.go
@@ -140,6 +140,27 @@ func main() {
 	}
 
 	if argv.Normal != nil {
+		if argv.Normal.On != nil {
+			if me.forge.Config.Mode == forgepb.ForgeMode_NORMAL {
+				log.Info("you are already in the normal state")
+				okExit("")
+			}
+			me.forge.Config.Mode = forgepb.ForgeMode_NORMAL
+			me.forge.Config.ConfigSave()
+			log.Info("normal mode on")
+			okExit("")
+		}
+		if argv.Normal.Off != nil {
+			if me.forge.Config.Mode != forgepb.ForgeMode_NORMAL {
+				log.Info("you were aleady not in the normal state")
+				okExit("")
+			}
+			me.forge.Config.Mode = forgepb.ForgeMode_MASTER
+			me.forge.Config.ConfigSave()
+			log.Info("normal mode off")
+			okExit("")
+		}
+
 		if doNormal() {
 			log.Infof("all %d repos are on your user branch. It is safe to write code now.\n", me.forge.Repos.Len())
 			if me.forge.Config.Mode != forgepb.ForgeMode_NORMAL {