From 588d044e9ee86659abb956f3575cfddd2c4d9d7a Mon Sep 17 00:00:00 2001
From: "Vgr E.Barry" <vgr255@live.ca>
Date: Sun, 15 Mar 2015 13:31:40 -0400
Subject: [PATCH] Allow !join to register game mode votes

---
 modules/wolfgame.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/modules/wolfgame.py b/modules/wolfgame.py
index 9bb85fa..b15571a 100644
--- a/modules/wolfgame.py
+++ b/modules/wolfgame.py
@@ -1118,6 +1118,16 @@ def join(cli, nick, chan, rest):
             cli.notice(nick, "You are not logged in to NickServ.")
             return
     join_player(cli, nick, chan)
+    if rest and not var.FGAMED:
+        gamemode = rest.lower().split()[0]
+        if gamemode not in var.GAME_MODES.keys():
+            match, _ = complete_match(gamemode, var.GAME_MODES.keys() - ["roles"])
+            if not match:
+                return
+            gamemode = match
+        if gamemode != "roles":
+            var.GAMEMODE_VOTES[nick] = gamemode
+            cli.msg(chan, "\002{0}\002 votes for the \002{1}\002 game mode.".format(nick, gamemode))
 
 def join_player(cli, player, chan, who = None, forced = False):
     if who is None: