diff --git a/vars.py b/vars.py index b339c86..52d4b2b 100644 --- a/vars.py +++ b/vars.py @@ -1,8 +1,13 @@ PING_WAIT = 300 # Seconds -MINIMUM_WAIT = 60 +MINIMUM_WAIT = 0 # debug, change to 60 for normal EXTRA_WAIT = 20 MAXIMUM_WAITED = 2 # limit for amount of !wait's MAX_SHOTS = 2 +NIGHT_TIME_LIMIT = 90 + +NO_VICTIMS_MESSAGES = ("The body of a young penguin pet is found", + "A pool of blood and wolf paw prints are found.", + "Traces of wolf fur are found") # These change ingame ROLES = {"person" : []} @@ -10,10 +15,13 @@ ORIGINAL_ROLES = None PHASE = "none" # "join", "day", or "night" LAST_PING = 0 CURSED = "" # nickname of cursed villager -GAME_START_TIME = 0 -CAN_START_TIME = 0 WAITED = 0 GUNNERS = {} +VICTIM = "" # nickname of to-be-killed villager +SEEN = [] # list of seers that have had visions +DEAD = [] # list of people who are dead +TRAITOR = "" +TIMERS = [None, None] # nightlimit, daylimit is_role = lambda plyr, rol: rol in ROLES and plyr in ROLES[rol] @@ -33,4 +41,10 @@ def list_players_and_roles(): for x in ROLES.keys(): for p in ROLES[x]: plr[p] = x - return plr \ No newline at end of file + return plr + +get_role = lambda plyr: list_players_and_roles()[plyr] + +def del_player(pname): + prole = get_role(pname) + ROLES[prole].remove(pname) \ No newline at end of file diff --git a/wolfgame.py b/wolfgame.py index 6689117..75b872a 100644 --- a/wolfgame.py +++ b/wolfgame.py @@ -2,7 +2,8 @@ from oyoyo.parse import parse_nick import vars import botconfig import decorators -import time +from datetime import datetime, timedelta +import threading import random COMMANDS = {} @@ -33,28 +34,29 @@ def say(cli, nick, rest): # To be removed later @pmcmd("!bye", admin_only=True) @cmd("!bye", admin_only=True) def forced_exit(cli, nick, *rest): # Admin Only - reset_game(cli, nick, rest[0], None) + reset_game(cli, nick, botconfig.CHANNEL, None) cli.quit("Forced quit from admin") raise SystemExit -@cmd("!exec") +@cmd("!exec", admin_only=True) def py(cli, nick, chan, rest): - if nick in botconfig.ADMINS: - exec(rest) + exec(rest) @cmd("!ping") def pinger(cli, nick, chan, rest): - if vars.LAST_PING + 300 > time.time(): - cli.notice(nick, "This command is ratelimited. \ -Please wait a while before using it again.") + if (vars.LAST_PING and + vars.LAST_PING + timedelta(seconds=300) > datetime.now()): + cli.notice(nick, ("This command is ratelimited. " + +"Please wait a while before using it again.")) return - vars.LAST_PING = time.time() + vars.LAST_PING = datetime.now() vars.PINGING = True TO_PING = [] @hook("whoreply") - def on_whoreply(server, dunno, chan, dunno1, dunno2, dunno3, user, status, dunno4): + 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. @@ -81,7 +83,7 @@ def join(cli, nick, chan, rest): cli.mode(chan, "+v", nick, nick+"!*@*") vars.ROLES["person"].append(nick) vars.PHASE = "join" - vars.CAN_START_TIME = time.time() + vars.MINIMUM_WAIT + vars.CAN_START_TIME = datetime.now() + timedelta(seconds=vars.MINIMUM_WAIT) 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)) @@ -117,14 +119,206 @@ def stats(cli, nick, chan, rest): message.append("\u0002(0}\u0002 {1}".format(count, vars.plural(role))) else: message.append("\u0002{0}\u0002 {1}".format(count, role)) + if len(vars.ROLES["wolf"]) > 1: + vb = "are" + else: + vb = "is" cli.msg(chan, - "{0}: There are {1}, and {2}.".format(nick, + "{0}: There {verb} {1}, and {2}.".format(nick, ", ".join(message[0:-1]), - message[-1])) + message[-1]), verb=vb) -def transition_night(cli, chan): - vars.PHASE = "night" +def del_player(cli, nick, died_in_game = True): + cli.mode(botconfig.CHANNEL, "-v", "{0} {0}!*@*".format(nick)) + if vars.PHASE != "join" and died_in_game: + cli.mode(botconfig.CHANNEL, "+q", "{0} {0}!*@*".format(nick)) + vars.DEAD.append(nick) + vars.del_player(nick) + +def leave(cli, what, nick): + if nick not in vars.list_players(): # not playing + return + msg = "" + if what in ("!quit", "!leave"): + msg = ("\u0002{0}\u0002 died of an unknown disease. "+ + "S/He was a \u0002{1}\u0002.") + died_in_game = True + elif what == "part": + msg = ("\u0002{0}\u0002 died due to eating poisonous berries. "+ + "Appears (s)he was a \u0002{1}\u0002.") + elif what == "quit": + msg = ("\u0002{0}\u0002 died due to a fatal attack by wild animals. "+ + "Appears (s)he was a \u0002{1}\u0002.") + elif what == "kick": + msg = ("\u0002{0}\u0002 died due to falling off a cliff. "+ + "Appears (s)he was a \u0002{1}\u0002.") + msg = msg.format(nick, vars.get_role(nick)) + cli.msg(botconfig.CHANNEL, msg) + del_player(cli, nick, died_in_game) + +cmd("!leave")(lambda cli, nick, chan, *rest: leave(cli, "!leave", nick)) +cmd("!quit")(lambda cli, nick, chan, *rest: leave(cli, "!quit", nick)) +hook("part")(lambda cli, nick, chan, *rest: leave(cli, "part", nick)) +hook("quit")(lambda cli, nick, chan, *rest: leave(cli, "quit", nick)) + +def transition_day(cli): + chan = botconfig.CHANNEL + + vars.PHASE = "day" + vars.DAY_START_TIME = datetime.now() + td = vars.DAY_START_TIME - vars.NIGHT_START_TIME + vars.NIGHT_TIMEDELTA += td + min, sec = td.seconds // 60, td.seconds % 60 + + message = "Night lasted \u0002{0:0>2}:{1:0>2}\u0002. It is now daytime. \ +The villagers awake, thankful for surviving the night, \ +and search the village... ".format(min, sec) + if not vars.VICTIM: + message += random.choice(vars.NO_VICTIMS_MESSAGES) + cli.msg(chan, message); + return + # TODO: check if visited is harlot + + dead = [] + + message += "The dead body of \u0002{0}\u0002, a \ +\u0002{1}\u0002, is found. Those remaining mourn his/her \ +death.".format(vars.VICTIM, vars.get_role(vars.VICTIM)) + dead.append(vars.VICTIM) + # TODO: check if harlot also died + cli.msg(chan, message) + + for deadperson in dead: + del_player(cli, deadperson, True) + + + +def chk_nightdone(cli): + if (len(vars.SEEN) == len(vars.ROLES["seer"]) and + vars.VICTIM and vars.PHASE == "night"): + if vars.TIMERS[0]: + vars.TIMERS[0].cancel() # cancel timer + vars.TIMERS[0] = None + transition_day(cli) + + +@pmcmd("!kill") +@pmcmd("kill") +def kill(cli, nick, rest): + if vars.PHASE == "none": + cli.msg(nick, "No game is currently running.") + return + if not nick in vars.list_players(): + cli.msg(nick, "You're currently playing") + return + if not (vars.is_role(nick, "wolf") or vars.is_role(nick, "traitor")): + cli.msg(nick, "Only a wolf may use this command") + return + if vars.PHASE != "night": + cli.msg(nick, "You may only kill people at night.") + return + victim = rest.split(" ")[0].strip() + if not victim: + cli.msg(nick, "Not enough parameters") + return + if victim not in vars.list_players(): + cli.msg(nick,"\u0002{0}\u0002 is currently not playing.".format(victim)) + return + if victim == nick: + cli.msg(nick, "Suicide is bad. Don't do it.") + return + if victim in vars.ROLES["wolf"]: + cli.msg(nick, "You may only kill villagers, not other wolves") + return + vars.VICTIM = victim + cli.msg(nick, "You have selected \u0002{0}\u0002 to be killed".format(victim)) + chk_nightdone(cli) + +@pmcmd("see") +@pmcmd("!see") +def see(cli, nick, rest): + if vars.PHASE == "none": + cli.msg(nick, "No game is currently running.") + return + if not nick in vars.list_players(): + cli.msg(nick, "You're currently playing") + return + if not vars.is_role(nick, "seer"): + cli.msg(nick, "Only a seer may use this command") + return + if vars.PHASE != "night": + cli.msg(nick, "You may have visions at night.") + return + if nick in vars.SEEN: + cli.msg(nick, "You may only have one vision per round.") + victim = rest.split(" ")[0].strip() + if not victim: + cli.msg(nick, "Not enough parameters") + return + if victim not in vars.list_players(): + cli.msg(nick,"\u0002{0}\u0002 is \ +currently not playing.".format(victim)) + return + if vars.CURSED == nick: + role = "wolf" + elif vars.TRAITOR == nick: + role = "villager" + else: + role = vars.get_role(victim) + cli.msg(nick, "You have a vision; in this vision, \ +you see that \u0002{0}\u0002 is a \u0002{1}\u0002!".format(victim, + role)) + vars.SEEN.append(nick) + chk_nightdone(cli) + +@pmcmd("") +def relay(cli, nick, rest): + badguys = vars.ROLES.get("wolf", []) + vars.ROLES.get("traitor", []) + if len(badguys) > 1: + if vars.is_role(nick, "wolf") or vars.is_role(nick, "traitor"): + badguys.remove(nick) # remove self from list + for badguy in badguys: + cli.msg(badguy, "{0} says: {1}".format(nick, rest)) + +def transition_night(cli): + vars.PHASE = "night" + vars.VICTIM = "" # nickname of cursed villager + vars.SEEN = [] # list of seers that have had visions + vars.NIGHT_START_TIME = datetime.now() + + chan = botconfig.CHANNEL + cli.msg(chan, "It is now nighttime. All players \ +check for PMs from me for instructions. If you did not receive \ +one, simply sit back, relax, and wait patiently for morning.") + + t = threading.Timer(vars.NIGHT_TIME_LIMIT, transition_day, [cli]) + vars.TIMERS[0] = t + t.start() + + # send PMs + pl = vars.list_players() + for wolf in vars.ROLES["wolf"]: + cli.msg(wolf, 'You are a \u0002wolf\u0002. It is your job to kill all the \ +villagers. Use "kill " to kill a villager. Also, if \ +you send a PM to me, it will be relayed to all other wolves.') + _pl = pl[:] + _pl.remove(wolf) # remove self from list + for i, player in enumerate(_pl): + if vars.is_role(player, "wolf"): + _pl[i] = player + " (wolf)" + elif vars.is_role(player, "traitor"): + _pl[i] = player + " (traitor)" + cli.msg(wolf, "Players: "+", ".join(_pl)) + + for seer in vars.ROLES["seer"]: + _pl = pl[:] + _pl.remove(seer) # remove self from list + cli.msg(seer, 'You are a \u0002seer\u0002. \ +It is your job to detect the wolves, you may have a vision once per night. \ +Use "see " to see the role of a player.') + cli.msg(seer, "Players: "+", ".join(_pl)) + @cmd("!start") def start(cli, nick, chan, rest): pl = vars.list_players() @@ -137,10 +331,10 @@ def start(cli, nick, chan, rest): if nick not in pl: cli.notice(nick, "You're currently not playing.") return - now = time.time() - if vars.CAN_START_TIME > now: - cli.msg(chan, "Please wait at least {0} more seconds.".format( - int(vars.CAN_START_TIME - now))) + now = datetime.now() + dur = int((vars.CAN_START_TIME - now).total_seconds()) + if dur > 0: + cli.msg(chan, "Please wait at least {0} more seconds.".format(dur)) return if len(pl) < 4: @@ -153,6 +347,7 @@ def start(cli, nick, chan, rest): nwolves = 0 ndrunk = 0 ncursed = 0 + ntraitor = 0 if len(pl) >= 8: nharlots = 1 @@ -189,17 +384,24 @@ def start(cli, nick, chan, rest): vars.ROLES["villager"] = pl if ncursed: - CURSED = random.choice(var.ROLES["villager"] + \ - var.ROLES.get("harlot", []) +\ - var.ROLES.get("village drunk", [])) + vars.CURSED = random.choice(vars.ROLES["villager"] + \ + vars.ROLES.get("harlot", []) +\ + vars.ROLES.get("village drunk", [])) + if ntraitor: + possible = vars.ROLES["villager"] + if ncursed: + possible.remove(vars.CURSED) # Cursed traitors are not allowed + vars.TRAITOR = random.choice(possible) cli.msg(chan, "{0}: Welcome to Werewolf, the popular detective/social \ party game (a theme of Mafia).".format(", ".join(vars.list_players()))) cli.mode(chan, "+m") - vars.GAME_START_TIME = time.time() vars.ORIGINAL_ROLES = dict(vars.ROLES) # Make a copy - transition_night(cli, chan) + vars.DAY_TIMEDELTA = timedelta(0) + vars.NIGHT_TIMEDELTA = timedelta(0) + vars.DEAD = [] + transition_night(cli) @cmd("!wait") def wait(cli, nick, chan, rest): @@ -217,11 +419,11 @@ def wait(cli, nick, chan, rest): cli.msg(chan, "Limit has already been reached for extending the wait time.") return - now = time.time() + now = datetime.now() if now > vars.CAN_START_TIME: - vars.CAN_START_TIME = now + vars.EXTRA_WAIT + vars.CAN_START_TIME = now + timedelta(seconds=vars.EXTRA_WAIT) else: - vars.CAN_START_TIME += vars.EXTRA_WAIT + vars.CAN_START_TIME += timedelta(seconds=vars.EXTRA_WAIT) vars.WAITED += 1 cli.msg(chan, "{0} increased the wait \ time by {1} seconds.".format(nick, vars.EXTRA_WAIT)) @@ -229,15 +431,21 @@ time by {1} seconds.".format(nick, vars.EXTRA_WAIT)) @cmd("!reset", admin_only = True) def reset_game(cli, nick, chan, rest): vars.PHASE = "none" + + if vars.TIMERS[0]: + vars.TIMERS[0].cancel() + vars.TIMERS[0] = None cli.mode(chan, "-m") - for pl in vars.list_players(): - cli.mode(chan, "-v", pl) - + for plr in vars.list_players(): + cli.mode(chan, "-v", "{0} {0}!*@*".format(plr)) + for deadguy in vars.DEAD: + cli.mode(chan, "-q", "{0} {0}!*@*".format(deadguy)) + vars.ROLES = {"person" : []} vars.ORIGINAL_ROLES = None vars.CURSED = "" - vars.GAME_START_TIME = 0 - vars.CAN_START_TIME = 0 + vars.CAN_START_TIME = timedelta(0) vars.GUNNERS = {} vars.WAITED = 0 + vars.VICTIM = "" \ No newline at end of file