You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
451 lines
15 KiB
451 lines
15 KiB
from oyoyo.parse import parse_nick |
|
import vars |
|
import botconfig |
|
import decorators |
|
from datetime import datetime, timedelta |
|
import threading |
|
import random |
|
|
|
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) |
|
|
|
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", admin_only=True) |
|
@cmd("!bye", admin_only=True) |
|
def forced_exit(cli, nick, *rest): # Admin Only |
|
reset_game(cli, nick, botconfig.CHANNEL, None) |
|
cli.quit("Forced quit from admin") |
|
raise SystemExit |
|
|
|
@cmd("!exec", admin_only=True) |
|
def py(cli, nick, chan, rest): |
|
exec(rest) |
|
|
|
@cmd("!ping") |
|
def pinger(cli, nick, chan, rest): |
|
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 = datetime.now() |
|
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": |
|
cli.mode(chan, "+v", nick, nick+"!*@*") |
|
vars.ROLES["person"].append(nick) |
|
vars.PHASE = "join" |
|
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)) |
|
elif nick in vars.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: |
|
cli.mode(chan, "+v", nick, nick+"!*@*") |
|
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": |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
|
|
pl = vars.list_players() |
|
if len(pl) > 1: |
|
cli.msg(chan, '{0}: \u0002{1}\u0002 players: {2}'.format(nick, |
|
len(pl), ", ".join(pl))) |
|
else: |
|
cli.msg(chan, '{0}: \u00021\u0002 player: {1}'.format(nick, pl[0])) |
|
|
|
if vars.PHASE == "join": |
|
return |
|
|
|
message = [] |
|
for role in ("wolf", "seer", "harlot"): |
|
count = len(vars.ROLES.get(role,[])) |
|
if count > 1: |
|
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 {verb} {1}, and {2}.".format(nick, |
|
", ".join(message[0:-1]), |
|
message[-1]), verb=vb) |
|
|
|
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 <nick>" 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 <nick>" 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() |
|
if vars.PHASE == "none": |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
if vars.PHASE != "join": |
|
cli.notice(nick, "Werewolf is already in play.") |
|
return |
|
if nick not in pl: |
|
cli.notice(nick, "You're currently not playing.") |
|
return |
|
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: |
|
cli.msg(chan, "{0}: Four or more players are required to play.".format(nick)) |
|
return |
|
|
|
vars.ROLES = {} |
|
nharlots = 0 |
|
nseers = 0 |
|
nwolves = 0 |
|
ndrunk = 0 |
|
ncursed = 0 |
|
ntraitor = 0 |
|
|
|
if len(pl) >= 8: |
|
nharlots = 1 |
|
nseers = 1 |
|
nwolves = 2 |
|
ndrunk = 1 |
|
ncursed = 1 |
|
elif(len(pl)) >= 6: |
|
nseers = 1 |
|
nwolves = 1 |
|
ndrunk = 1 |
|
ncursed = 1 |
|
else: |
|
nseers = 1 |
|
nwolves = 1 |
|
|
|
seer = random.choice(pl) |
|
vars.ROLES["seer"] = [seer] |
|
pl.remove(seer) |
|
if nharlots: |
|
harlots = random.sample(pl, nharlots) |
|
vars.ROLES["harlot"] = harlots |
|
for h in harlots: |
|
pl.remove(h) |
|
if nwolves: |
|
wolves = random.sample(pl, nwolves) |
|
vars.ROLES["wolf"] = wolves |
|
for w in wolves: |
|
pl.remove(w) |
|
if ndrunk: |
|
drunk = random.choice(pl) |
|
vars.ROLES["village drunk"] = [drunk] |
|
pl.remove(drunk) |
|
vars.ROLES["villager"] = pl |
|
|
|
if ncursed: |
|
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.ORIGINAL_ROLES = dict(vars.ROLES) # Make a copy |
|
vars.DAY_TIMEDELTA = timedelta(0) |
|
vars.NIGHT_TIMEDELTA = timedelta(0) |
|
vars.DEAD = [] |
|
transition_night(cli) |
|
|
|
@cmd("!wait") |
|
def wait(cli, nick, chan, rest): |
|
pl = vars.list_players() |
|
if vars.PHASE == "none": |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
if vars.PHASE != "join": |
|
cli.notice(nick, "Werewolf is already in play.") |
|
return |
|
if nick not in pl: |
|
cli.notice(nick, "You're currently not playing.") |
|
return |
|
if vars.WAITED >= vars.MAXIMUM_WAITED: |
|
cli.msg(chan, "Limit has already been reached for extending the wait time.") |
|
return |
|
|
|
now = datetime.now() |
|
if now > vars.CAN_START_TIME: |
|
vars.CAN_START_TIME = now + timedelta(seconds=vars.EXTRA_WAIT) |
|
else: |
|
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)) |
|
|
|
@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 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.CAN_START_TIME = timedelta(0) |
|
vars.GUNNERS = {} |
|
vars.WAITED = 0 |
|
vars.VICTIM = "" |
|
|