From 8100e5e6df76dc8a7b57bc118a0a68c185fcd40f Mon Sep 17 00:00:00 2001
From: Jordan Orelli <jordanorelli@gmail.com>
Date: Thu, 13 Nov 2014 13:30:26 -0500
Subject: [PATCH] added --debug flag

---
 commands.go | 18 ++++++++++++++----
 main.go     |  4 ++++
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/commands.go b/commands.go
index 4304ea8..d4f6c1b 100644
--- a/commands.go
+++ b/commands.go
@@ -14,6 +14,7 @@ type Command struct {
 	help    string
 	handler func(*Connection, ...string)
 	mobile  bool
+	debug   bool // marks command as a debug mode command
 }
 
 var infoCommand = &Command{
@@ -188,8 +189,9 @@ var colonizeCommand = &Command{
 }
 
 var winCommand = &Command{
-	name: "win",
-	help: "win the game.",
+	name:  "win",
+	help:  "win the game.",
+	debug: true,
 	handler: func(conn *Connection, args ...string) {
 		conn.Win("win-command")
 	},
@@ -271,10 +273,18 @@ func runCommand(conn *Connection, name string, args ...string) {
 }
 
 func registerCommand(c *Command) {
-	commandRegistry[c.name] = c
+	if c.debug {
+		if options.debug {
+			commandRegistry[c.name] = c
+			log_info("registered debug command: %s", c.name)
+		}
+	} else {
+		commandRegistry[c.name] = c
+		log_info("registered command: %s", c.name)
+	}
 }
 
-func init() {
+func setupCommands() {
 	commandRegistry = make(map[string]*Command, 16)
 	registerCommand(bombCommand)
 	registerCommand(broadcastCommand)
diff --git a/main.go b/main.go
index 1b4e436..9ef9287 100644
--- a/main.go
+++ b/main.go
@@ -20,6 +20,7 @@ var options struct {
 	playerSpeed float64
 	bombSpeed   float64
 	economic    int
+	debug       bool
 }
 
 var (
@@ -90,11 +91,13 @@ func handleConnection(conn *Connection) {
 func main() {
 	flag.Parse()
 	dbconnect()
+
 	rand.Seed(time.Now().UnixNano())
 	info_log = log.New(os.Stdout, "[INFO] ", 0)
 	error_log = log.New(os.Stderr, "[ERROR] ", 0)
 
 	setupDb()
+	setupCommands()
 	listener, err := net.Listen("tcp", ":9220")
 	if err != nil {
 		bail(E_No_Port, "unable to start server: %v", err)
@@ -121,4 +124,5 @@ func init() {
 	flag.IntVar(&options.economic, "economic", 25000, "amount of money needed to win economic victory")
 	flag.Float64Var(&options.moneyMean, "money-mean", 10000, "mean amount of money on a system")
 	flag.Float64Var(&options.moneySigma, "money-sigma", 1500, "standard deviation in money per system")
+	flag.BoolVar(&options.debug, "debug", false, "puts the game in debug mode")
 }