weerwolven/wolfgame.py

from oyoyo.parse import parse_nick
import vars
import botconfig
import decorators

COMMANDS = {}
PM_COMMANDS = {}
HOOKS = {}    

cmd = decorators.generate(COMMANDS)
pmcmd = decorators.generate(PM_COMMANDS)
hook = decorators.generate(HOOKS)

# Game Logic Begins:

def connect_callback(cli):
    cli.identify(botconfig.PASS)
    cli.join(botconfig.CHANNEL)
    cli.msg("ChanServ", "op "+botconfig.CHANNEL)
    cli.msg(botconfig.CHANNEL, "\u0002Wolfbot2 is here.\u0002")

def reset_game():
    vars.ROLES = {"person" : []}
    vars.PHASE = "none"

# Command Handlers:
@cmd("!say")
def say(cli, nick, rest):  # To be removed later
    cli.msg(botconfig.CHANNEL, "{0} says: {1}".format(nick, rest))
    
    
@pmcmd("!bye")
@cmd("!bye")
def forced_exit(cli, nick, *rest):  # Admin Only
    if nick in botconfig.ADMINS:
        cli.quit("Forced quit from admin")
        raise SystemExit
        
@cmd("!exec")
def py(cli, nick, chan, rest):
    if nick in botconfig.ADMINS:
        exec(rest)

@cmd("!ping")
def pinger(cli, nick, chan, rest):
    vars.PINGING = True
    TO_PING = []

    @hook("whoreply")
    def on_whoreply(server, dunno, chan, dunno1, dunno2, dunno3, user, status, dunno4):
        if not vars.PINGING: return
        if user in (botconfig.NICK, nick): return  # Don't ping self.
        
        if vars.PINGING and 'G' not in status and '+' not in status:
            # TODO: check if the user has AWAY'D himself
            TO_PING.append(user)

    @hook("endofwho")
    def do_ping(*args):
        if not vars.PINGING: return
        
        chan = args[2]
        cli.msg(chan, "PING! "+" ".join(TO_PING))
        vars.PINGING = False
        
        HOOKS.pop("whoreply")
        HOOKS.pop("endofwho")
    
    cli.send("WHO "+chan)
        
@cmd("!join")
def join(cli, nick, chan, rest):
    if vars.PHASE == "none":
        vars.ROLES["person"].append(nick)
        vars.PHASE = "join"
        cli.msg(chan, '\u0002{0}\u0002 has started a game of Werewolf. \
Type "!join" to join. Type "!start" to start the game. \
Type "!wait" to increase join wait time.'.format(nick))
    elif nick in list_players():
        cli.notice(nick, "You're already playing!")
    elif vars.PHASE != "join":
        cli.notice(nick, "Sorry but the game is already running.  Try again next time.")
    else:
        vars.ROLES["person"].append(nick)
        cli.msg(chan, '\u0002{0}\u0002 has joined the game.'.format(nick))

@cmd("!stats")
def stats(cli, nick, chan, rest):
    if vars.PHASE == "none":
        return
    pl = list_players()
    if len(pl) > 1:
        cli.msg(chan, '{0}: {1} players: {2}'.format(nick,
            len(pl), ", ".join(pl)))
    else:
        cli.msg(chan, '{0}: 1 player: {1}'.format(nick, pl[0]))
    
    if vars.PHASE == "join":
        return
    
    msg = []
    for role in vars.ROLES.keys():
        num = len(vars.ROLES[role])
        if num > 1:
            msg.append("{0} {1}".format(num, plural(role)))
        else:
            msg.append("{0} {1}".format(num, role))
    if len(msg) > 2:  # More than 2 roles to say
        msg[-1] = "and "+msg[-1]+"."
        msg[0] = "{0}: There are ".format(nick) + msg[0]
        cli.msg(chan, ", ".join(msg))
    elif len(msg) == 2:  # 2 roles to say
        cli.msg(chan, "{0}: There are ".format(nick) + msg[0],
                "and", msg[1] + ".")
    elif len(msg) == 1:
        cli.msg(chan, "{0}: There is ".format(nick) + msg[0] + ".")        

@cmd("!start")
def start(cli, nick, chan, rest):
    pl = list_players()

    if(len(pl)) < 4:
        cli.msg(chan, "{0}: Four or more players are required to play.".format(nick))
        return