Set state on profile

cmd/actor/mode.go

@@ -0,0 +1,20 @@
+package main
+
+import (
+	"github.com/bahner/go-ma-actor/config"
+	"github.com/spf13/pflag"
+)
+
+const (
+	defaultPongMode  = false
+	defaultRelayMode = false
+)
+
+func init() {
+
+	// NB! Other mode pflags are in the proper mode packages.
+	pflag.Bool("pong", defaultPongMode, "Pong mode with automatic replies and no UI.")
+	pflag.Bool("pong-fortune", config.DEFAULT_PONG_FORTUNE_MODE, "Reply with a fortune cookie, instead of a static message, if availble.")
+	pflag.Bool("relay", defaultRelayMode, "Relay mode with no actor, to just listen and relay messages.")
+
+}

config/mode.go

@@ -54,10 +54,16 @@ func Mode() string {
 	}
 
 	if PongMode() {
+
+		// This only sets fallback when not specified on command line
+		SetProfile(pong)
 		return pong
 	}
 
 	if RelayMode() {
+
+		// This only sets fallback when not specified on command line
+		SetProfile(relay)
 		return relay
 	}
 

config/profile.go

@@ -4,6 +4,8 @@ import "github.com/spf13/pflag"
 
 // NB! This file is used early in the initialization process, so it can't depend on other packages.
 
+var profile string = "actor"
+
 // Profile is the mode unless overridden by the profile flag.
 func Profile() string {
 
@@ -14,5 +16,9 @@ func Profile() string {
 		}
 	}
 
-	return Mode()
+	return profile
+}
+
+func SetProfile(p string) {
+	profile = p
 }