better "ForgeMode" handling

1 files changed, 4 insertions, 7 deletions

diff --git a/main.go b/main.go
index d5fdf52..6c02138 100644
--- a/main.go
+++ b/main.go
@@ -82,7 +82,7 @@ func main() {
 	}
 
 	if argv.Checkout != nil {
-		me.forge.Config.Mode = forgepb.ForgeMode_MASTER
+		setForgeMode(forgepb.ForgeMode_MASTER)
 		if err := doCheckout(); err != nil {
 			badExit(err)
 		}
@@ -132,8 +132,7 @@ func main() {
 				log.Info("you are already in the normal state")
 				okExit("")
 			}
-			me.forge.Config.Mode = forgepb.ForgeMode_NORMAL
-			me.forge.Config.ConfigSave()
+			setForgeMode(forgepb.ForgeMode_NORMAL)
 			log.Info("normal mode on")
 			okExit("")
 		}
@@ -143,8 +142,7 @@ func main() {
 				log.Info("you were aleady not in the normal state")
 				okExit("")
 			}
-			me.forge.Config.Mode = forgepb.ForgeMode_MASTER
-			me.forge.Config.ConfigSave()
+			setForgeMode(forgepb.ForgeMode_MASTER)
 			log.Info("normal mode off")
 			okExit("")
 		}
@@ -153,8 +151,7 @@ func main() {
 			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 {
 				log.Infof("Forge has set the mode to 'Normal'\n")
-				me.forge.Config.Mode = forgepb.ForgeMode_NORMAL
-				me.forge.ConfigSave()
+				setForgeMode(forgepb.ForgeMode_NORMAL)
 			}
 			okExit("")
 		}