1 files changed, 64 insertions, 2 deletions
diff --git a/mode.go b/mode.go
index 2ab966d..7b3bcad 100644
--- a/mode.go
+++ b/mode.go
@@ -4,7 +4,7 @@ package forgepb
 
 // TODO: implement i18n with the protobuf's
 func (f *Forge) GetMode() string {
-	switch f.Mode {
+	switch f.mode {
 	case ForgeMode_MASTER:
 		return "Release Mode (master branch)"
 	case ForgeMode_DEVEL:
@@ -12,6 +12,68 @@ func (f *Forge) GetMode() string {
 	case ForgeMode_USER:
 		return "Hack Mode (user branch)"
 	default:
-		return f.Mode.String()
+		return f.mode.String()
 	}
 }
+
+func (f *Forge) IsModeUnknown() bool {
+	if f.mode == ForgeMode_UNKNOWN {
+		return true
+	}
+	return false
+}
+
+func (f *Forge) IsModeNewUser() bool {
+	if f.mode == ForgeMode_NEWUSER {
+		return true
+	}
+	return false
+}
+
+func (f *Forge) IsModeDevel() bool {
+	if f.mode == ForgeMode_DEVEL {
+		return true
+	}
+	return false
+}
+
+func (f *Forge) SetModeNewUser() {
+	if err := f.SetMode(ForgeMode_NEWUSER); err != nil {
+		panic("SetMode failed")
+	}
+}
+
+func (f *Forge) IsModeMaster() bool {
+	if f.mode == ForgeMode_MASTER {
+		return true
+	}
+	return false
+}
+
+func (f *Forge) IsModeNormal() bool {
+	if f.mode == ForgeMode_NORMAL {
+		return true
+	}
+	return false
+}
+
+func (f *Forge) IsModeClean() bool {
+	if f.mode == ForgeMode_CLEAN {
+		return true
+	}
+	return false
+}
+
+func (f *Forge) IsModeUser() bool {
+	if f.mode == ForgeMode_USER {
+		return true
+	}
+	return false
+}
+
+func (f *Forge) IsModeGoLang() bool {
+	if f.mode == ForgeMode_NORMAL {
+		return true
+	}
+	return false
+}