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.
1666 lines
61 KiB
1666 lines
61 KiB
from oyoyo.parse import parse_nick |
|
import var |
|
import botconfig |
|
import decorators |
|
from datetime import datetime, timedelta |
|
import threading |
|
import random |
|
import copy |
|
from time import sleep |
|
from time import time as timetime |
|
import re |
|
import logging |
|
import sys |
|
import os |
|
|
|
COMMANDS = {} |
|
PM_COMMANDS = {} |
|
HOOKS = {} |
|
|
|
cmd = decorators.generate(COMMANDS) |
|
pmcmd = decorators.generate(PM_COMMANDS) |
|
hook = decorators.generate(HOOKS, raw_nick=True) |
|
|
|
# Game Logic Begins: |
|
|
|
def connect_callback(cli): |
|
cli.identify(botconfig.PASS) |
|
cli.join(botconfig.CHANNEL) |
|
cli.msg("ChanServ", "op "+botconfig.CHANNEL) |
|
|
|
var.USERS = [] |
|
@hook("whoreply") |
|
def on_whoreply(cli, server, dunno, chan, dunno1, |
|
dunno2, dunno3, user, status, dunno4): |
|
if user in var.USERS: return # Don't add someone who is already there |
|
|
|
var.USERS.append(user) |
|
cli.who(botconfig.CHANNEL) |
|
|
|
var.LAST_PING = 0 # time of last !ping |
|
var.ROLES = {"person" : []} |
|
var.PHASE = "none" # "join", "day", or "night" |
|
var.TIMERS = [None, None] |
|
var.DEAD = [] |
|
|
|
var.ORIGINAL_SETTINGS = {} |
|
var.DENIED_SETTINGS_CHANGE = [] |
|
var.SETTINGS_CHANGE_OPPOSITION = [] |
|
var.SETTINGS_CHANGE_REQUESTER = None |
|
|
|
var.LAST_SAID_TIME = {} |
|
|
|
var.GAME_START_TIME = datetime.now() # for idle checker only |
|
var.GRAVEYARD = [] |
|
var.GRAVEYARD_LOCK = threading.Lock() |
|
|
|
|
|
@pmcmd("!say") |
|
def say(cli, nick, rest): # To be removed later |
|
cli.msg(botconfig.CHANNEL, "{0} says: {1}".format(nick, rest)) |
|
|
|
|
|
|
|
def mass_mode(cli, md): |
|
""" Example: mass_mode((('+v', 'asdf'), ('-v','wobosd'))) """ |
|
lmd = len(md) # store how many mode changes to do |
|
for start_i in range(0, lmd, 4): # 4 mode-changes at a time |
|
if start_i + 4 > lmd: # If this is a remainder (mode-changes < 4) |
|
z = list(zip(*md[start_i:])) # zip this remainder |
|
ei = lmd % 4 # len(z) |
|
else: |
|
z = list(zip(*md[start_i:start_i+4])) # zip four |
|
ei = 4 # len(z) |
|
# Now z equal something like [('+v', '-v'), ('asdf', 'wobosd')] |
|
arg1 = "".join(z[0]) |
|
arg2 = " ".join(z[1]) + " " + " ".join([x+"!*@*" for x in z[1]]) |
|
cli.mode(botconfig.CHANNEL, arg1, arg2) |
|
|
|
|
|
|
|
def reset_settings(): |
|
for attr in list(var.ORIGINAL_SETTINGS.keys()): |
|
setattr(var, attr, var.ORIGINAL_SETTINGS[attr]) |
|
dict.clear(var.ORIGINAL_SETTINGS) |
|
|
|
var.SETTINGS_CHANGE_OPPOSITION = [] |
|
var.SETTINGS_CHANGE_REQUESTER = None |
|
|
|
|
|
def reset(cli): |
|
chan = botconfig.CHANNEL |
|
var.PHASE = "none" |
|
|
|
if var.TIMERS[0]: |
|
var.TIMERS[0].cancel() |
|
var.TIMERS[0] = None |
|
if var.TIMERS[1]: |
|
var.TIMERS[1].cancel() |
|
var.TIMERS[1] = None |
|
|
|
cli.mode(chan, "-m") |
|
cmodes = [] |
|
for plr in var.list_players(): |
|
cmodes.append(("-v", plr)) |
|
for deadguy in var.DEAD: |
|
cmodes.append(("-q", deadguy)) |
|
mass_mode(cli, cmodes) |
|
var.DEAD = [] |
|
|
|
var.ROLES = {"person" : []} |
|
|
|
reset_settings() |
|
var.DENIED_SETTINGS_CHANGE = [] |
|
|
|
dict.clear(var.LAST_SAID_TIME) |
|
|
|
|
|
@pmcmd("!bye", admin_only=True) |
|
@cmd("!bye", admin_only=True) |
|
def forced_exit(cli, nick, *rest): # Admin Only |
|
reset(cli) |
|
print("Quitting in 5 seconds.") |
|
dict.clear(COMMANDS) |
|
dict.clear(PM_COMMANDS) |
|
dict.clear(PM_COMMANDS) |
|
cli.quit("Forced quit from admin") |
|
raise SystemExit |
|
|
|
|
|
|
|
@cmd("!exec", owner_only = True) |
|
def py(cli, nick, chan, rest): |
|
try: |
|
exec(rest) |
|
except Exception as e: |
|
cli.msg(chan, str(type(e))+":"+str(e)) |
|
|
|
|
|
|
|
@cmd("!eval", owner_only = True) |
|
def pyeval(cli, nick, chan, rest): |
|
try: |
|
a = str(eval(rest)) |
|
if len(a) < 500: |
|
cli.msg(chan, a) |
|
else: |
|
cli.msg(chan, a[0:500]) |
|
except Exception as e: |
|
cli.msg(chan, str(type(e))+":"+str(e)) |
|
|
|
|
|
@cmd("!restart", admin_only=True) |
|
def restart_program(cli, nick, chan, rest): |
|
try: |
|
forced_exit(cli, nick, chan, rest) |
|
finally: |
|
print("RESTARTING!") |
|
python = sys.executable |
|
os.execl(python, python, *sys.argv) |
|
|
|
|
|
|
|
@cmd("!op", admin_only=True) |
|
def give_op(cli, nick, chan, rest): |
|
if not rest.strip(): |
|
rest = nick |
|
cli.msg("ChanServ", " ".join(("op",chan,rest.strip()))) |
|
@pmcmd("!op", admin_only=True) |
|
def give_op_pm(cli, nick, rest): |
|
give_op(cli, nick, botconfig.CHANNEL, rest) |
|
|
|
|
|
|
|
@cmd("!deop", admin_only=True) |
|
def take_op(cli, nick, chan, rest): |
|
if not rest.strip(): |
|
rest = nick |
|
cli.msg("ChanServ", " ".join(("deop",chan,rest.strip()))) |
|
|
|
|
|
@cmd("!sudo revoke", owner_only=True) |
|
def revoke(cli, nick, chan, rest): |
|
r = rest.strip() |
|
if r in botconfig.ADMINS: |
|
ladmins = list(botconfig.ADMINS) |
|
ladmins.remove(r) |
|
botconfig.ADMINS = tuple(ladmins) |
|
cli.msg(chan, "Revoked admin rights for \u0002{0}\u0002.".format(r)) |
|
|
|
|
|
|
|
@cmd("!ping") |
|
def pinger(cli, nick, chan, rest): |
|
if (var.LAST_PING and |
|
var.LAST_PING + timedelta(seconds=300) > datetime.now()): |
|
cli.notice(nick, ("This command is ratelimited. " + |
|
"Please wait a while before using it again.")) |
|
return |
|
|
|
var.LAST_PING = datetime.now() |
|
var.PINGING = True |
|
TO_PING = [] |
|
|
|
|
|
|
|
@hook("whoreply") |
|
def on_whoreply(cli, server, dunno, chan, dunno1, |
|
dunno2, dunno3, user, status, dunno4): |
|
if not var.PINGING: return |
|
if user in (botconfig.NICK, nick): return # Don't ping self. |
|
|
|
if var.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 var.PINGING: return |
|
|
|
cli.msg(chan, "PING! "+" ".join(TO_PING)) |
|
var.PINGING = False |
|
|
|
HOOKS.pop("whoreply") |
|
HOOKS.pop("endofwho") |
|
|
|
cli.who(chan) |
|
|
|
|
|
|
|
#def chk_bed |
|
|
|
|
|
|
|
@cmd("!fping", admin_only=True) |
|
def fpinger(cli, nick, chan, rest): |
|
var.LAST_PING = None |
|
pinger(cli, nick, chan, rest) |
|
|
|
|
|
|
|
@cmd("!join") |
|
def join(cli, nick, chan, rest): |
|
if var.PHASE == "none": |
|
cli.mode(chan, "+v", nick, nick+"!*@*") |
|
var.ROLES["person"].append(nick) |
|
var.PHASE = "join" |
|
var.WAITED = 0 |
|
var.CAN_START_TIME = datetime.now() + timedelta(seconds=var.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 var.list_players(): |
|
cli.notice(nick, "You're already playing!") |
|
elif var.PHASE != "join": |
|
cli.notice(nick, "Sorry but the game is already running. Try again next time.") |
|
else: |
|
cli.mode(chan, "+v", nick, nick+"!*@*") |
|
var.ROLES["person"].append(nick) |
|
cli.msg(chan, '\u0002{0}\u0002 has joined the game.'.format(nick)) |
|
|
|
|
|
@cmd("!fjoin", admin_only=True) |
|
def fjoin(cli, nick, chan, rest): |
|
noticed = False |
|
if not rest.strip(): |
|
return |
|
for a in re.split("\s+",rest): |
|
a = a.strip() |
|
if not a: |
|
continue |
|
if a[0].isalpha() or a[0] == "!" and a[0] != "\\": |
|
if not a.lower().endswith("serv"): |
|
if not noticed: |
|
cli.msg(chan, nick+": You may only fjoin fake people for now.") |
|
noticed = True |
|
continue |
|
if a != botconfig.NICK and a: |
|
join(cli, a.strip(), chan, "") |
|
else: |
|
cli.notice(nick, "No, that won't be allowed.") |
|
|
|
@cmd("!fleave", admin_only=True) |
|
def fleave(cli, nick, chan, rest): |
|
for a in re.split("\s+",rest): |
|
a = a.strip() |
|
if not a: |
|
continue |
|
pll = [x.lower() for x in var.list_players()] |
|
if a.lower() != botconfig.NICK.lower() and a.lower() in pll: |
|
del_player(cli, a.strip()) |
|
cli.msg(chan, ("\u0002{0}\u0002 has used fleave"+ |
|
" on \u0002{1}\u0002.").format(nick, a.strip())) |
|
elif a.lower() not in pll: |
|
cli.msg(chan, nick+": That could not be done.") |
|
|
|
@cmd("!fstart", admin_only=True) |
|
def fstart(cli, nick, chan, rest): |
|
var.CAN_START_TIME = datetime.now() |
|
start(cli, nick, chan, rest) |
|
|
|
@cmd("!chankick", admin_only=True) |
|
def chankick(cli, nick, chan, rest): |
|
rest = rest.split(" ", 1) |
|
if rest[0] != botconfig.NICK: |
|
cli.kick(chan, *rest) |
|
else: |
|
cli.kick(chan, nick, "No.") |
|
|
|
@hook("kick") |
|
def on_kicked(cli, nick, chan, victim, reason): |
|
if victim == botconfig.NICK: |
|
cli.join(botconfig.CHANNEL) |
|
cli.kick(chan, nick, "No.") |
|
else: |
|
print("WHAT", nick, chan, victim, reason) |
|
|
|
|
|
@cmd("!stats") |
|
def stats(cli, nick, chan, rest): |
|
if var.PHASE == "none": |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
|
|
pl = var.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 var.PHASE == "join": |
|
return |
|
|
|
message = [] |
|
f = False |
|
l1 = [k for k in var.ROLES.keys() |
|
if var.ROLES[k]] |
|
l2 = [k for k in var.ORIGINAL_ROLES.keys() |
|
if var.ORIGINAL_ROLES[k]] |
|
for role in set(l1+l2): |
|
count = len(var.ROLES[role]) |
|
if not f and count>1: |
|
vb = "are" |
|
f = True |
|
else: |
|
vb = "is" |
|
if count > 1: |
|
message.append("\u0002{0}\u0002 {1}".format(count, var.plural(role))) |
|
else: |
|
message.append("\u0002{0}\u0002 {1}".format(count if count else "no", role)) |
|
cli.msg(chan, "{0}: There {3} {1}, and {2}.".format(nick, |
|
", ".join(message[0:-1]), |
|
message[-1], |
|
vb)) |
|
|
|
|
|
|
|
def hurry_up(cli): |
|
if var.PHASE != "day": return |
|
|
|
chan = botconfig.CHANNEL |
|
pl = var.list_players() |
|
avail = len(pl) - len(var.WOUNDED) |
|
votesneeded = avail // 2 + 1 |
|
|
|
found_dup = False |
|
maxfound = (0, "") |
|
for votee, voters in iter(var.VOTES.items()): |
|
if len(voters) > maxfound[0]: |
|
maxfound = (len(voters), votee) |
|
found_dup = False |
|
elif len(voters) == maxfound[0]: |
|
found_dup = True |
|
if maxfound[0] > 0 and not found_dup: |
|
cli.msg(chan, "The sun sets.") |
|
var.VOTES[maxfound[1]] = [None] * votesneeded |
|
chk_decision(cli) # Induce a lynch |
|
else: |
|
cli.msg(chan, "The sun is almost setting.") |
|
for plr in pl: |
|
var.VOTES[plr] = [None] * (votesneeded - 1) |
|
|
|
|
|
|
|
@cmd("!fnight", admin_only=True) |
|
def fnight(cli, nick, chan, rest): |
|
if var.PHASE != "day": |
|
cli.notice(nick, "It is not daytime.") |
|
else: |
|
hurry_up(cli) |
|
|
|
|
|
@cmd("!fday", admin_only=True) |
|
def fday(cli, nick, chan, rest): |
|
if var.PHASE != "night": |
|
cli.notice(nick, "It is not nighttime.") |
|
else: |
|
transition_day(cli) |
|
|
|
|
|
|
|
def chk_decision(cli): |
|
chan = botconfig.CHANNEL |
|
pl = var.list_players() |
|
avail = len(pl) - len(var.WOUNDED) |
|
votesneeded = avail // 2 + 1 |
|
for votee, voters in iter(var.VOTES.items()): |
|
if len(voters) >= votesneeded: |
|
cli.msg(botconfig.CHANNEL, |
|
random.choice(var.LYNCH_MESSAGES).format( |
|
votee, var.get_role(votee))) |
|
if del_player(cli, votee, True): |
|
transition_night(cli) |
|
|
|
|
|
|
|
@cmd("!votes") |
|
def show_votes(cli, nick, chan, rest): |
|
if var.PHASE in ("none", "join"): |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
elif nick not in var.list_players(): |
|
cli.notice(nick, "You're not currently playing.") |
|
return |
|
if var.PHASE != "day": |
|
cli.notice(nick, "Voting is only during the day.") |
|
return |
|
elif not var.VOTES.values(): |
|
cli.msg(chan, nick+": No votes yet.") |
|
return |
|
if None in [x for voter in var.VOTES.values() for x in voter]: |
|
cli.msg(chan, (nick+": Tiebreaker conditions. Whoever "+ |
|
"receives the next vote will be lynched.")) |
|
return |
|
|
|
votelist = ["{0}: {1} ({2})".format(votee, |
|
len(var.VOTES[votee]), |
|
" ".join(var.VOTES[votee])) |
|
for votee in var.VOTES.keys()] |
|
cli.msg(chan, "{0}: {1}".format(nick, ", ".join(votelist))) |
|
|
|
pl = var.list_players() |
|
avail = len(pl) - len(var.WOUNDED) |
|
votesneeded = avail // 2 + 1 |
|
cli.msg(chan, ("{0}: \u0002{1}\u0002 players, \u0002{2}\u0002 votes "+ |
|
"required to lynch, \u0002{3}\u0002 players available " + |
|
"to vote.").format(nick, len(pl), votesneeded, avail)) |
|
|
|
|
|
|
|
def chk_traitor(cli): |
|
for tt in var.ROLES["traitor"]: |
|
var.ROLES["wolf"].append(tt) |
|
var.ROLES["traitor"].remove(tt) |
|
cli.msg(tt, ('HOOOOOOOOOWL. You have become... a wolf!\n'+ |
|
'It is up to you to avenge your fallen leaders!')) |
|
|
|
|
|
|
|
def chk_win(cli): |
|
""" Returns True if someone won """ |
|
|
|
chan = botconfig.CHANNEL |
|
lpl = len(var.list_players()) |
|
if lpl == 0: |
|
cli.msg(chan, "No more players remaining. Game ended.") |
|
reset(cli) |
|
return True |
|
if var.PHASE == "join": |
|
return False |
|
elif (len(var.ROLES["wolf"])+ |
|
len(var.ROLES["traitor"])+ |
|
len(var.ROLES["werecrow"])) == lpl / 2: |
|
cli.msg(chan, ("Game over! There are the same number of wolves as "+ |
|
"villagers. The wolves eat everyone, and win.")) |
|
elif (len(var.ROLES["wolf"])+ |
|
len(var.ROLES["traitor"])+ |
|
len(var.ROLES["werecrow"])) > lpl / 2: |
|
cli.msg(chan, ("Game over! There are more wolves than "+ |
|
"villagers. The wolves eat everyone, and win.")) |
|
elif (not var.ROLES["wolf"] and |
|
not var.ROLES["traitor"] and |
|
not var.ROLES["werecrow"]): |
|
cli.msg(chan, ("Game over! All the wolves are dead! The villagers "+ |
|
"chop them up, BBQ them, and have a hearty meal.")) |
|
elif not len(var.ROLES["wolf"]) and var.ROLES["traitor"]: |
|
chk_traitor(cli) |
|
cli.msg(chan, ('\u0002The villagers, during their celebrations, are '+ |
|
'frightened as they hear a loud howl. The wolves are '+ |
|
'not gone!\u0002')) |
|
return False |
|
else: |
|
return False |
|
|
|
if var.DAY_START_TIME: |
|
now = datetime.now() |
|
td = now - var.DAY_START_TIME |
|
var.DAY_TIMEDELTA += td |
|
if var.NIGHT_START_TIME: |
|
now = datetime.now() |
|
td = now - var.NIGHT_START_TIME |
|
var.NIGHT_TIMEDELTA += td |
|
|
|
daymin, daysec = var.DAY_TIMEDELTA.seconds // 60, var.DAY_TIMEDELTA.seconds % 60 |
|
nitemin, nitesec = var.NIGHT_TIMEDELTA.seconds // 60, var.NIGHT_TIMEDELTA.seconds % 60 |
|
total = var.DAY_TIMEDELTA + var.NIGHT_TIMEDELTA |
|
tmin, tsec = total.seconds // 60, total.seconds % 60 |
|
cli.msg(chan, ("Game lasted \u0002{0:0>2}:{1:0>2}\u0002. " + |
|
"\u0002{2:0>2}:{3:0>2}\u0002 was day. " + |
|
"\u0002{4:0>2}:{5:0>2}\u0002 was night. ").format(tmin, tsec, |
|
daymin, daysec, |
|
nitemin, nitesec)) |
|
|
|
roles_msg = [] |
|
var.ORIGINAL_ROLES["cursed villager"] = var.CURSED |
|
for role in var.ORIGINAL_ROLES.keys(): |
|
if len(var.ORIGINAL_ROLES[role]) == 0 or role == "villager": |
|
continue |
|
elif len(var.ORIGINAL_ROLES[role]) == 2: |
|
msg = "The {1} were \u0002{0[0]}\u0002 and \u0002{0[1]}\u0002." |
|
roles_msg.append(msg.format(var.ORIGINAL_ROLES[role], var.plural(role))) |
|
elif len(var.ORIGINAL_ROLES[role]) == 1: |
|
roles_msg.append("The {1} was \u0002{0[0]}\u0002.".format(var.ORIGINAL_ROLES[role], |
|
role)) |
|
else: |
|
msg = "The {2} were {0}, and \u0002{1}\u0002." |
|
nickslist = ["\u0002"+x+"\u0002" for x in var.ORIGINAL_ROLES[role][0:-1]] |
|
roles_msg.append(msg.format(", ".join(nickslist), |
|
var.ORIGINAL_ROLES[role][-1], |
|
var.plural(role))) |
|
cli.msg(chan, " ".join(roles_msg)) |
|
|
|
reset(cli) |
|
return True |
|
|
|
|
|
|
|
def del_player(cli, nick, forced_death = False): |
|
""" |
|
Returns: False if one side won. |
|
arg: forced_death = True when lynched. |
|
""" |
|
|
|
cmode = [] |
|
cmode.append(("-v", nick)) |
|
var.del_player(nick) |
|
ret = True |
|
if var.PHASE == "join": |
|
# Died during the joining process as a person |
|
mass_mode(cli, cmode) |
|
return not chk_win(cli) |
|
if var.PHASE != "join" and ret: |
|
# Died during the game, so quiet! |
|
if nick[0].isalpha() or nick[0] == "\\": |
|
cmode.append(("+q", nick)) |
|
mass_mode(cli, cmode) |
|
var.DEAD.append(nick) |
|
ret = not chk_win(cli) |
|
if var.PHASE in ("night", "day") and ret: |
|
# remove him from variables if he is in there |
|
if var.VICTIM == nick: |
|
var.VICTIM = "" |
|
for x in (var.OBSERVED, var.HVISITED): |
|
keys = list(x.keys()) |
|
for k in keys: |
|
if k == nick: |
|
del x[k] |
|
elif x[k] == nick: |
|
del x[k] |
|
if nick in var.GUNNERS.keys(): |
|
del var.GUNNERS[nick] |
|
if nick in var.CURSED: |
|
var.CURSED.remove(nick) |
|
if var.PHASE == "day" and not forced_death and ret: # didn't die from lynching |
|
if nick in var.VOTES.keys(): |
|
del var.VOTES[nick] # Delete his votes |
|
for k in var.VOTES.keys(): |
|
if nick in var.VOTES[k]: |
|
var.VOTES[k].remove(nick) |
|
chk_decision(cli) |
|
return ret |
|
|
|
|
|
@hook("ping") |
|
def on_ping(cli, prefix, server): |
|
cli.send('PONG', server) |
|
check_graveyard(cli) |
|
|
|
|
|
def reaper(cli): |
|
# check to see if idlers need to be killed. |
|
var.IDLE_KILLED = [] |
|
var.IDLE_WARNED = [] |
|
|
|
if not var.WARN_IDLE_TIME or not var.KILL_IDLE_TIME: |
|
return |
|
|
|
while var.PHASE != "none": |
|
to_warn = [] |
|
for nick in var.list_players(): |
|
lst = var.LAST_SAID_TIME.get(nick, var.GAME_START_TIME) |
|
tdiff = datetime.now() - lst |
|
if (tdiff > timedelta(seconds=var.WARN_IDLE_TIME) and |
|
nick not in var.IDLE_WARNED): |
|
to_warn.append(nick) |
|
var.IDLE_WARNED.append(nick) |
|
var.LAST_SAID_TIME[nick] = (datetime.now() - |
|
timedelta(seconds=var.KILL_IDLE_TIME)) # Give him a chance |
|
elif (tdiff > timedelta(seconds=var.KILL_IDLE_TIME) and |
|
nick not in var.IDLE_KILLED): |
|
with var.GRAVEYARD_LOCK: |
|
var.GRAVEYARD.append(("kill",nick)) |
|
var.IDLE_KILLED.append(nick) |
|
if to_warn: |
|
with var.GRAVEYARD_LOCK: |
|
var.GRAVEYARD.append(("warn", to_warn)) |
|
sleep(10) |
|
|
|
|
|
|
|
def check_graveyard(cli): |
|
if not var.GRAVEYARD: return |
|
chan = botconfig.CHANNEL |
|
with var.GRAVEYARD_LOCK: |
|
for action, x in var.GRAVEYARD: |
|
if action=="kill": |
|
if x not in var.list_players(): |
|
continue |
|
cli.msg(chan, ("\u0002{0}\u0002 didn't get out of bed "+ |
|
"for a very long time. S/He is declared dead. Appears "+ |
|
"(s)he was a \u0002{1}\u0002").format(x, var.get_role(x))) |
|
del_player(cli, x) |
|
elif action=="warn": |
|
pl = var.list_players() |
|
x = [a for a in x if a in pl] |
|
cli.msg(chan, ("{0}: \u0002You have been idling for a while. "+ |
|
"Please remember to say something soon or you "+ |
|
"might be declared dead.\u0002").format(", ".join(x))) |
|
var.GRAVEYARD = [] |
|
|
|
|
|
@cmd("") # update last said |
|
def update_last_said(cli, nick, *rest): |
|
if var.PHASE not in ("join", "none"): |
|
var.LAST_SAID_TIME[nick] = datetime.now() |
|
check_graveyard(cli) |
|
|
|
|
|
|
|
@hook("join") |
|
def on_join(cli, raw_nick, chan): |
|
nick = parse_nick(raw_nick)[0] |
|
if nick not in var.USERS and nick != botconfig.NICK: |
|
var.USERS.append(nick) |
|
|
|
@cmd("!goat") |
|
def goat(cli, nick, chan, rest): |
|
if var.PHASE in ("none", "join"): |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
elif nick not in var.list_players(): |
|
cli.notice(nick, "You're not currently playing.") |
|
return |
|
if var.PHASE != "day": |
|
cli.notice(nick, "You can only do that in the day.") |
|
return |
|
if var.GOATED: |
|
cli.notice(nick, "You can only do that once per day.") |
|
return |
|
if rest.strip() in var.USERS: |
|
cli.msg(chan, ("\u0002{0}\u0002's goat walks by "+ |
|
"and kicks \u0002{1}\u0002.").format(nick, |
|
rest.strip())) |
|
|
|
|
|
|
|
@hook("nick") |
|
def on_nick(cli, prefix, nick): |
|
prefix = parse_nick(prefix)[0] |
|
|
|
if prefix in var.DENIED_SETTINGS_CHANGE: |
|
var.DENIED_SETTINGS_CHANGE.append(nick) |
|
var.DENIED_SETTINGS_CHANGE.remove(prefix) |
|
|
|
if prefix in var.USERS: |
|
var.USERS.remove(prefix) |
|
var.USERS.append(nick) |
|
|
|
if prefix in var.list_players(): |
|
r = var.ROLES[var.get_role(prefix)] |
|
r.append(nick) |
|
r.remove(prefix) |
|
|
|
if var.PHASE in ("night", "day"): |
|
if var.VICTIM == prefix: |
|
var.VICTIM = nick |
|
kvp = [] |
|
for dictvar in (var.HVISITED, var.OBSERVED): |
|
for a,b in dictvar.items(): |
|
if a == prefix: |
|
a = nick |
|
if b == prefix: |
|
b = nick |
|
kvp.append((a,b)) |
|
dictvar.update(kvp) |
|
if prefix in dictvar.keys(): |
|
del dictvar[prefix] |
|
if prefix in var.SEEN: |
|
var.SEEN.remove(prefix) |
|
var.SEEN.append(nick) |
|
if nick in var.GUNNERS.keys(): |
|
del var.GUNNERS[nick] |
|
if nick in var.CURSED: |
|
var.CURSED.remove(nick) |
|
|
|
if var.PHASE == "day": |
|
if prefix in var.WOUNDED: |
|
var.WOUNDED.remove(prefix) |
|
var.WOUNDED.append(nick) |
|
if prefix in var.VOTES: |
|
var.VOTES[nick] = var.VOTES.pop(prefix) |
|
for v in var.VOTES.values(): |
|
if prefix in v: |
|
v.remove(prefix) |
|
v.append(nick) |
|
else: |
|
return |
|
|
|
|
|
def leave(cli, what, nick): |
|
if var.PHASE == "none" and what.startswith("!"): |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
elif var.PHASE == "none": |
|
return |
|
if nick not in var.list_players() and what.startswith("!"): # not playing |
|
cli.notice(nick, "You're not currently playing.") |
|
return |
|
elif nick not in var.list_players(): |
|
return |
|
msg = "" |
|
if what in ("!quit", "!leave"): |
|
msg = ("\u0002{0}\u0002 died of an unknown disease. "+ |
|
"S/He was a \u0002{1}\u0002.") |
|
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, var.get_role(nick)) |
|
cli.msg(botconfig.CHANNEL, msg) |
|
del_player(cli, nick) |
|
|
|
cmd("!leave")(lambda cli, nick, *rest: leave(cli, "!leave", nick)) |
|
cmd("!quit")(lambda cli, nick, *rest: leave(cli, "!quit", nick)) |
|
#Functions decorated with hook do not parse the nick by default |
|
hook("part")(lambda cli, nick, *rest: leave(cli, "part", parse_nick(nick)[0])) |
|
hook("quit")(lambda cli, nick, *rest: leave(cli, "quit", parse_nick(nick)[0])) |
|
hook("kick")(lambda cli, nick, *rest: leave(cli, "kick", parse_nick(rest[1])[0])) |
|
|
|
|
|
|
|
def begin_day(cli): |
|
chan = botconfig.CHANNEL |
|
|
|
# Reset nighttime variables |
|
var.VICTIM = "" # nickname of kill victim |
|
var.GUARDED = "" |
|
var.KILLER = "" # nickname of who chose the victim |
|
var.SEEN = [] # list of seers that have had visions |
|
var.OBSERVED = {} # those whom werecrows have observed |
|
var.HVISITED = {} |
|
|
|
cli.msg(chan, ("The villagers must now vote for whom to lynch. "+ |
|
'Use "!lynch <nick>" to cast your vote. 3 votes '+ |
|
'are required to lynch.')) |
|
|
|
if var.DAY_TIME_LIMIT > 0: # Time limit enabled |
|
t = threading.Timer(var.DAY_TIME_LIMIT, hurry_up, [cli]) |
|
var.TIMERS[1] = t |
|
t.start() |
|
|
|
|
|
|
|
def transition_day(cli, gameid=0): |
|
if gameid: |
|
if gameid != var.NIGHT_ID: |
|
return |
|
print("Day is starting...") |
|
var.PHASE = "day" |
|
var.GOATED = False |
|
chan = botconfig.CHANNEL |
|
|
|
if var.DAY_TIME_LIMIT > 0: # Time limit enabled |
|
t = threading.Timer(var.DAY_TIME_LIMIT, hurry_up, [cli]) |
|
var.TIMERS[1] = t |
|
t.start() |
|
|
|
# Reset daytime variables |
|
var.VOTES = {} |
|
var.WOUNDED = [] |
|
var.DAY_START_TIME = datetime.now() |
|
|
|
td = var.DAY_START_TIME - var.NIGHT_START_TIME |
|
var.NIGHT_START_TIME = None |
|
var.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)] |
|
dead = [] |
|
crowonly = var.ROLES["werecrow"] and not var.ROLES["wolf"] |
|
for crow, target in iter(var.OBSERVED.items()): |
|
if target in var.ROLES["harlot"]+var.ROLES["seer"]: |
|
cli.msg(crow, ("As the sun rises, you conclude that \u0002{0}\u0002 was not in "+ |
|
"bed at night, and you fly back to your house.").format(target)) |
|
elif target not in var.ROLES["village drunk"]: |
|
cli.msg(crow, ("As the sun rises, you conclude that \u0002{0}\u0002 was sleeping "+ |
|
"all night long, and you fly back to your house.").format(target)) |
|
if not var.VICTIM: |
|
message.append(random.choice(var.NO_VICTIMS_MESSAGES) + |
|
" All villagers, however, have survived.") |
|
elif var.VICTIM in var.ROLES["harlot"]: # Attacked harlot, yay no deaths |
|
if var.HVISITED.get(var.VICTIM): |
|
message.append("The wolves' selected victim was a harlot, "+ |
|
"but she wasn't home.") |
|
if var.VICTIM and (var.VICTIM not in var.ROLES["harlot"] or # not a harlot |
|
not var.HVISITED.get(var.VICTIM)): # harlot stayed home |
|
message.append(("The dead body of \u0002{0}\u0002, a "+ |
|
"\u0002{1}\u0002, is found. Those remaining mourn his/her "+ |
|
"death.").format(var.VICTIM, var.get_role(var.VICTIM))) |
|
dead.append(var.VICTIM) |
|
if var.VICTIM in var.HVISITED.values(): # victim was visited by some harlot |
|
for hlt in var.HVISITED.keys(): |
|
if var.HVISITED[hlt] == var.VICTIM: |
|
message.append(("\u0002{0}\u0002, a harlot, made the unfortunate mistake of "+ |
|
"visiting the victim's house last night and is "+ |
|
"now dead.").format(hlt)) |
|
dead.append(hlt) |
|
# TODO: check if harlot also died |
|
for harlot in var.ROLES["harlot"]: |
|
if var.HVISITED.get(harlot) in var.ROLES["wolf"]: |
|
message.append(("\u0002{0}\u0002, a harlot, made the unfortunate mistake of "+ |
|
"visiting a wolf's house last night and is "+ |
|
"now dead.").format(harlot)) |
|
dead.append(harlot) |
|
for crow, target in iter(var.OBSERVED.items()): |
|
if (target in var.ROLES["harlot"] and |
|
target in var.HVISITED.keys() and |
|
target not in dead): |
|
# Was visited by a crow |
|
cli.msg(target, ("You suddenly remember that you were startled by the loud "+ |
|
"sound of the flapping of wings during the walk back home.")) |
|
elif target in var.ROLES["village drunk"]: |
|
# Crow dies because of tiger (HANGOVER) |
|
cli.msg(chan, ("The bones of \u0002{0}\u0002, a werecrow, "+ |
|
"were found near the village drunk's house. "+ |
|
"The drunk's pet tiger probably ate him.").format(crow)) |
|
dead.append(crow) |
|
for deadperson in dead: |
|
if not del_player(cli, deadperson): |
|
return |
|
cli.msg(chan, "\n".join(message)) |
|
begin_day(cli) |
|
|
|
|
|
def chk_nightdone(cli): |
|
if (len(var.SEEN) == len(var.ROLES["seer"]) and # Seers have seen. |
|
len(var.HVISITED.keys()) == len(var.ROLES["harlot"]) and # harlots have visited. |
|
(var.VICTIM or |
|
(var.ROLES["werecrow"] == 1 and # Wolves have done their stuff |
|
not var.ROLES["wolf"] and var.OBSERVED) or |
|
(not var.ROLES["wolf"] + var.ROLES["werecrow"])) and |
|
var.PHASE == "night"): # no wolves |
|
if var.TIMERS[0]: |
|
if var.TIMERS[0].is_alive(): |
|
return |
|
var.TIMERS[0].cancel() # cancel timer |
|
var.TIMERS[0] = None |
|
if var.PHASE == "night": # Double check |
|
transition_day(cli) |
|
|
|
|
|
|
|
@cmd("!lynch", "!vote") |
|
def vote(cli, nick, chan, rest): |
|
if var.PHASE in ("none", "join"): |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
elif nick not in var.list_players(): |
|
cli.notice(nick, "You're not currently playing.") |
|
return |
|
if var.PHASE != "day": |
|
cli.notice(nick, ("Lynching is only allowed during the day. "+ |
|
"Please wait patiently for morning.")) |
|
return |
|
pl = var.list_players() |
|
pl_l = [x.strip().lower() for x in pl] |
|
rest = re.split("\s+",rest)[0].strip().lower() |
|
if rest in pl_l: |
|
if nick in var.WOUNDED: |
|
cli.msg(chan, ("{0}: You are wounded and resting, "+ |
|
"thus you are unable to vote for the day.")) |
|
voted = pl[pl_l.index(rest)] |
|
lcandidates = list(var.VOTES.keys()) |
|
for voters in lcandidates: # remove previous vote |
|
if nick in var.VOTES[voters]: |
|
var.VOTES[voters].remove(nick) |
|
if not var.VOTES.get(voters) and voters != voted: |
|
del var.VOTES[voters] |
|
break |
|
if voted not in var.VOTES.keys(): |
|
var.VOTES[voted] = [nick] |
|
else: |
|
var.VOTES[voted].append(nick) |
|
cli.msg(chan, ("\u0002{0}\u0002 votes for "+ |
|
"\u0002{1}\u0002.").format(nick, voted)) |
|
chk_decision(cli) |
|
elif not rest: |
|
cli.notice(nick, "Not enough parameters.") |
|
else: |
|
cli.notice(nick, "\u0002{0}\u0002 is currently not playing.".format(rest)) |
|
|
|
|
|
|
|
@cmd("!retract") |
|
def retract(cli, nick, chan, rest): |
|
if var.PHASE in ("none", "join"): |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
elif nick not in var.list_players(): |
|
cli.notice(nick, "You're not currently playing.") |
|
return |
|
if var.PHASE != "day": |
|
cli.notice(nick, ("Lynching is only allowed during the day. "+ |
|
"Please wait patiently for morning.")) |
|
return |
|
|
|
candidates = var.VOTES.keys() |
|
for voter in list(candidates): |
|
if nick in var.VOTES[voter]: |
|
var.VOTES[voter].remove(nick) |
|
if not var.VOTES[voter]: |
|
del var.VOTES[voter] |
|
cli.msg(chan, "\u0002{0}\u0002 retracted his/her vote.".format(nick)) |
|
break |
|
else: |
|
cli.notice(nick, "You haven't voted yet.") |
|
|
|
|
|
|
|
@cmd("!shoot", "shoot") |
|
def shoot(cli, nick, chan, rest): |
|
if var.PHASE in ("none", "join"): |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
elif nick not in var.list_players(): |
|
cli.notice(nick, "You're not currently playing.") |
|
return |
|
if var.PHASE != "day": |
|
cli.notice(nick, ("Shooting is only allowed during the day. "+ |
|
"Please wait patiently for morning.")) |
|
return |
|
if nick not in var.GUNNERS.keys(): |
|
cli.msg(nick, "You don't have a gun.") |
|
return |
|
elif not var.GUNNERS[nick]: |
|
cli.msg(nick, "You don't have any more bullets.") |
|
return |
|
victim = re.split("\s+",rest)[0].strip().lower() |
|
if not victim: |
|
cli.notice(nick, "Not enough parameters") |
|
return |
|
pl = var.list_players() |
|
pll = [x.lower() for x in pl] |
|
if victim not in pll: |
|
cli.notice(nick,"\u0002{0}\u0002 is currently not playing.".format(victim)) |
|
return |
|
victim = pl[pll.index(victim)] |
|
rand = random.random() |
|
if nick in var.ROLES["village drunk"]: |
|
chances = var.DRUNK_GUN_CHANCES |
|
else: |
|
chances = var.GUN_CHANCES |
|
if rand <= chances[0]: |
|
cli.msg(chan, ("\u0002{0}\u0002 shoots \u0002{1}\u0002 with "+ |
|
"a silver bullet!").format(nick, victim)) |
|
victimrole = var.get_role(victim) |
|
if victimrole in ("wolf", "werecrow"): |
|
cli.msg(chan, ("\u0002{0}\u0002 is a wolf, and is dying from "+ |
|
"the silver bullet.").format(victim)) |
|
if not del_player(cli, victim): |
|
return |
|
elif random.random() <= var.MANSLAUGHTER_CHANCE: |
|
cli.msg(chan, ("\u0002{0}\u0002 is a not a wolf "+ |
|
"but was accidentally fatally injured.").format(victim)) |
|
cli.msg(chan, "Appears (s)he was a \u0002{0}\u0002.".format(victimrole)) |
|
if not del_player(cli, victim): |
|
return |
|
else: |
|
cli.msg(chan, ("\u0002{0}\u0002 is a villager and is injured but "+ |
|
"will have a full recovery. S/He will be resting "+ |
|
"for the day.").format(victim)) |
|
var.WOUNDED.append(victim) |
|
chk_decision(cli) |
|
elif rand <= chances[0] + chances[1]: |
|
cli.msg(chan, "\u0002{0}\u0002 is a lousy shooter. S/He missed!".format(nick)) |
|
var.GUNNERS[nick] -= 1 |
|
else: |
|
cli.msg(chan, ("\u0002{0}\u0002 should clean his/her weapons more often. "+ |
|
"The gun exploded and killed him/her!").format(nick)) |
|
cli.msg(chan, "Appears that (s)he was a \u0002{0}\u0002.".format(var.get_role(nick))) |
|
if not del_player(cli, nick): |
|
return # Someone won. |
|
|
|
|
|
|
|
@pmcmd("!kill", "kill") |
|
def kill(cli, nick, rest): |
|
if var.PHASE in ("none", "join"): |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
elif nick not in var.list_players(): |
|
cli.notice(nick, "You're not currently playing.") |
|
return |
|
role = var.get_role(nick) |
|
if role not in ('wolf', 'werecrow'): |
|
cli.msg(nick, "Only a wolf may use this command.") |
|
return |
|
if var.PHASE != "night": |
|
cli.msg(nick, "You may only kill people at night.") |
|
return |
|
victim = re.split("\s+",rest)[0].strip().lower() |
|
if not victim: |
|
cli.msg(nick, "Not enough parameters") |
|
return |
|
if role == "werecrow": # Check if flying to observe |
|
if var.OBSERVED.get(nick): |
|
cli.msg(nick, ("You are flying to \u0002{0}'s\u0002 house, and "+ |
|
"therefore you don't have the time "+ |
|
"and energy to kill a villager.").format(var.OBSERVED[nick])) |
|
return |
|
pl = var.list_players() |
|
pll = [x.lower() for x in pl] |
|
if victim not in pll: |
|
cli.msg(nick,"\u0002{0}\u0002 is currently not playing.".format(victim)) |
|
return |
|
if victim == nick.lower(): |
|
cli.msg(nick, "Suicide is bad. Don't do it.") |
|
return |
|
if victim in var.ROLES["wolf"]+var.ROLES["traitor"]+var.ROLES["werecrow"]: |
|
cli.msg(nick, "You may only kill villagers, not other wolves") |
|
return |
|
var.VICTIM = pl[pll.index(victim)] |
|
cli.msg(nick, "You have selected \u0002{0}\u0002 to be killed".format(var.VICTIM)) |
|
chk_nightdone(cli) |
|
|
|
|
|
@pmcmd("!guard", "guard") |
|
def kill(cli, nick, rest): |
|
if var.PHASE in ("none", "join"): |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
elif nick not in var.list_players(): |
|
cli.notice(nick, "You're not currently playing.") |
|
return |
|
role = var.get_role(nick) |
|
if role != 'guardian angel': |
|
cli.msg(nick, "Only a wolf may use this command.") |
|
return |
|
if var.PHASE != "night": |
|
cli.msg(nick, "You may only guard people at night.") |
|
return |
|
victim = re.split("\s+",rest)[0].strip().lower() |
|
if not victim: |
|
cli.msg(nick, "Not enough parameters") |
|
return |
|
return #TODO |
|
if role == "werecrow": # Check if flying to observe |
|
if var.OBSERVED.get(nick): |
|
cli.msg(nick, ("You are flying to \u0002{0}'s\u0002 house, and "+ |
|
"therefore you don't have the time "+ |
|
"and energy to kill a villager.").format(var.OBSERVED[nick])) |
|
return |
|
pl = var.list_players() |
|
pll = [x.lower() for x in pl] |
|
if victim not in pll: |
|
cli.msg(nick,"\u0002{0}\u0002 is currently not playing.".format(victim)) |
|
return |
|
if victim == nick.lower(): |
|
cli.msg(nick, "Suicide is bad. Don't do it.") |
|
return |
|
if victim in var.ROLES["wolf"]+var.ROLES["traitor"]+var.ROLES["werecrow"]: |
|
cli.msg(nick, "You may only kill villagers, not other wolves") |
|
return |
|
var.VICTIM = pl[pll.index(victim)] |
|
cli.msg(nick, "You have selected \u0002{0}\u0002 to be killed".format(var.VICTIM)) |
|
chk_nightdone(cli) |
|
|
|
|
|
|
|
@pmcmd("observe", "!observe") |
|
def observe(cli, nick, rest): |
|
if var.PHASE in ("none", "join"): |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
elif nick not in var.list_players(): |
|
cli.notice(nick, "You're not currently playing.") |
|
return |
|
if not var.is_role(nick, "werecrow"): |
|
cli.msg(nick, "Only a werecrow may use this command.") |
|
return |
|
if var.PHASE != "night": |
|
cli.msg(nick, "You may only transform into a crow at night.") |
|
return |
|
victim = re.split("\s+", rest)[0].strip().lower() |
|
if not victim: |
|
cli.msg(nick, "Not enough parameters") |
|
return |
|
pl = var.list_players() |
|
pll = [x.lower() for x in pl] |
|
if victim not in pll: |
|
cli.msg(nick, "\u0002{0}\u0002 is currently not playing.".format(victim)) |
|
return |
|
victim = pl[pll.index(victim)] |
|
if victim == nick.lower(): |
|
cli.msg(nick, "Instead of doing that, you should probably go kill someone.") |
|
return |
|
if var.get_role(victim) in ("werecrow", "traitor", "wolf"): |
|
cli.msg(nick, "Flying to another wolf's house is a waste of time.") |
|
return |
|
var.OBSERVED[nick] = victim |
|
cli.msg(nick, ("You transform into a large crow and start your flight "+ |
|
"to \u0002{0}'s\u0002 house. You will return after "+ |
|
"collecting your observations when day begins.").format(victim)) |
|
|
|
|
|
|
|
@pmcmd("visit", "!visit") |
|
def hvisit(cli, nick, rest): |
|
if var.PHASE in ("none", "join"): |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
elif nick not in var.list_players(): |
|
cli.notice(nick, "You're not currently playing.") |
|
return |
|
if not var.is_role(nick, "harlot"): |
|
cli.msg(nick, "Only a harlot may use this command.") |
|
return |
|
if var.PHASE != "night": |
|
cli.msg(nick, "You may only visit someone at night.") |
|
return |
|
if var.HVISITED.get(nick): |
|
cli.msg(nick, ("You are already spending the night "+ |
|
"with \u0002{0}\u0002.").format(var.HVISITED[nick])) |
|
return |
|
victim = re.split("\s+",rest)[0].strip().lower() |
|
if not victim: |
|
cli.msg(nick, "Not enough parameters") |
|
return |
|
pl = [x.lower() for x in var.list_players()] |
|
if victim not in pl: |
|
cli.msg(nick,"\u0002{0}\u0002 is currently not playing.".format(victim)) |
|
return |
|
if nick.lower() == victim: # Staying home |
|
var.HVISITED[nick] = None |
|
cli.msg(nick, "You have chosen to stay home for the night.") |
|
else: |
|
var.HVISITED[nick] = var.list_players()[pl.index(victim)] |
|
cli.msg(nick, ("You are spending the night with \u0002{0}\u0002. "+ |
|
"Have a good time!").format(victim)) |
|
if var.HVISITED[nick] not in var.ROLES["wolf"]: |
|
cli.msg(var.HVISITED[nick], ("You are spending the night with \u0002{0}"+ |
|
"\u0002. Have a good time!").format(nick)) |
|
chk_nightdone(cli) |
|
|
|
|
|
|
|
@cmd("!frole", admin_only=True) |
|
def frole(cli, nick, chan, rest): |
|
rst = re.split("\s+",rest) |
|
if len(rst) < 2: |
|
cli.msg(chan, "The syntax is incorrect.") |
|
who = rst.pop(0).strip() |
|
rol = " ".join(rst).strip() |
|
if who not in var.USERS: |
|
if ((who[0].isalpha() or (who[0] == "!" or who[0] == "\\")) and |
|
not who.lower().endswith("serv")): |
|
cli.msg(chan, "Could not be done.") |
|
cli.msg(chan, "The target needs to be in this channel or a fake name.") |
|
return |
|
elif who == botconfig.NICK or not who: |
|
cli.msg(chan, "No.") |
|
return |
|
if rol not in var.ROLES.keys(): |
|
if var.PHASE not in ("night", "day"): |
|
cli.msg(chan, "This is only allowed in game.") |
|
if rol == "gunner": |
|
var.GUNNERS[who] = var.MAX_SHOTS |
|
elif rol == "cursed": |
|
var.CURSED.append(who) |
|
else: |
|
cli.msg(chan, "Not a valid role.") |
|
cli.msg(chan, "Operation successful.") |
|
return |
|
if who in var.list_players(): |
|
var.del_player(who) |
|
var.ROLES[rol].append(who) |
|
cli.msg(chan, "Operation successful.") |
|
if var.PHASE not in ('none','join'): |
|
chk_win(cli) |
|
|
|
|
|
@cmd("!force", admin_only=True) |
|
def forcepm(cli, nick, chan, rest): |
|
rst = re.split("\s+",rest) |
|
if len(rst) < 2: |
|
cli.msg(chan, "The syntax is incorrect.") |
|
return |
|
who = rst.pop(0).strip() |
|
if not who or who == botconfig.NICK: |
|
cli.msg(chan, "That won't work.") |
|
return |
|
if who[0].isalpha() or who[0] == "!" and who[0] != "\\": |
|
if not who.lower().endswith("serv"): |
|
if who not in var.USERS: |
|
cli.msg(chan, "This can only be done on fake nicks.") |
|
return |
|
cmd = rst.pop(0) |
|
if cmd.lower() in PM_COMMANDS.keys(): |
|
PM_COMMANDS[cmd.lower()](cli, who, " ".join(rst)) |
|
cli.msg(chan, "Operation successful.") |
|
if var.PHASE == "night": |
|
chk_nightdone(cli) |
|
elif cmd.lower() in COMMANDS.keys(): |
|
COMMANDS[cmd.lower()](cli, who, chan, " ".join(rst)) |
|
cli.msg(chan, "Operation successful.") |
|
else: |
|
cli.msg(chan, "That command was not found.") |
|
|
|
@pmcmd("see", "!see") |
|
def see(cli, nick, rest): |
|
if var.PHASE in ("none", "join"): |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
elif nick not in var.list_players(): |
|
cli.notice(nick, "You're not currently playing.") |
|
return |
|
if not var.is_role(nick, "seer"): |
|
cli.msg(nick, "Only a seer may use this command") |
|
return |
|
if var.PHASE != "night": |
|
cli.msg(nick, "You may only have visions at night.") |
|
return |
|
if nick in var.SEEN: |
|
cli.msg(nick, "You may only have one vision per round.") |
|
victim = re.split("\s+",rest)[0].strip().lower() |
|
pl = var.list_players() |
|
pll = [x.lower() for x in pl] |
|
if not victim: |
|
cli.msg(nick, "Not enough parameters") |
|
return |
|
if victim not in pll: |
|
cli.msg(nick,"\u0002{0}\u0002 is currently not playing.".format(victim)) |
|
return |
|
victim = pl[pll.index(victim)] |
|
if nick in var.CURSED: |
|
role = "wolf" |
|
elif var.get_role(victim) == "traitor": |
|
role = "villager" |
|
else: |
|
role = var.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)) |
|
var.SEEN.append(nick) |
|
chk_nightdone(cli) |
|
|
|
|
|
|
|
@pmcmd("") |
|
def relay(cli, nick, rest): |
|
if var.PHASE != "night": |
|
return |
|
badguys = var.ROLES["wolf"] + var.ROLES["traitor"] + var.ROLES["werecrow"] |
|
if len(badguys) > 1: |
|
if var.get_role(nick) in ("wolf","traitor","werecrow"): |
|
badguys.remove(nick) # remove self from list |
|
for badguy in badguys: |
|
cli.msg(badguy, "{0} says: {1}".format(nick, rest)) |
|
|
|
|
|
|
|
def transition_night(cli): |
|
var.PHASE = "night" |
|
|
|
if var.TIMERS[1]: # cancel daytime-limit timer |
|
var.TIMERS[1].cancel() |
|
var.TIMERS[1] = None |
|
|
|
# Reset nighttime variables |
|
var.VICTIM = "" # nickname of kill victim |
|
var.GUARDED = "" |
|
var.KILLER = "" # nickname of who chose the victim |
|
var.SEEN = [] # list of seers that have had visions |
|
var.OBSERVED = {} # those whom werecrows have observed |
|
var.HVISITED = {} |
|
var.NIGHT_START_TIME = datetime.now() |
|
|
|
daydur_msg = "" |
|
|
|
if var.NIGHT_TIMEDELTA or var.START_WITH_DAY: # transition from day |
|
td = var.NIGHT_START_TIME - var.DAY_START_TIME |
|
var.DAY_START_TIME = None |
|
var.DAY_TIMEDELTA += td |
|
min, sec = td.seconds // 60, td.seconds % 60 |
|
daydur_msg = "Day lasted \u0002{0:0>2}:{1:0>2}\u0002. ".format(min,sec) |
|
|
|
chan = botconfig.CHANNEL |
|
|
|
if var.NIGHT_TIME_LIMIT > 0: |
|
t = threading.Timer(var.NIGHT_TIME_LIMIT, transition_day, [cli, var.NIGHT_ID]) |
|
var.TIMERS[0] = t |
|
t.start() |
|
|
|
# send PMs |
|
ps = var.list_players() |
|
wolves = var.ROLES["wolf"]+var.ROLES["traitor"]+var.ROLES["werecrow"] |
|
for wolf in wolves: |
|
if wolf in var.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.')) |
|
elif wolf in var.ROLES["traitor"]: |
|
cli.msg(wolf, ('You are a \u0002traitor\u0002. You are exactly like a '+ |
|
'villager and not even a seer can see your true identity. '+ |
|
'Only detectives can. ')) |
|
else: |
|
cli.msg(wolf, ('You are a \u0002werecrow\u0002. You are able to fly at night. '+ |
|
'Use "kill <nick>" to kill a a villager. Alternatively, you can '+ |
|
'use "observe <nick>" to check if someone is in bed or not. '+ |
|
'Observing will prevent you participating in a killing.')) |
|
if len(wolves) > 1: |
|
cli.msg(wolf, 'Also, if you PM me, your message will be relayed to other wolves.') |
|
pl = ps[:] |
|
pl.remove(wolf) # remove self from list |
|
for i, player in enumerate(pl): |
|
if player in var.ROLES["wolf"]: |
|
pl[i] = player + " (wolf)" |
|
elif player in var.ROLES["traitor"]: |
|
pl[i] = player + " (traitor)" |
|
elif player in var.ROLES["werecrow"]: |
|
pl[i] = player + " (werecrow)" |
|
cli.msg(wolf, "\u0002Players:\u0002 "+", ".join(pl)) |
|
|
|
for seer in var.ROLES["seer"]: |
|
pl = ps[:] |
|
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)) |
|
|
|
for harlot in var.ROLES["harlot"]: |
|
pl = ps[:] |
|
pl.remove(harlot) |
|
cli.msg(harlot, ('You are a \u0002harlot\u0002. '+ |
|
'You may spend the night with one person per round. '+ |
|
'If you visit a victim of a wolf, or visit a wolf, '+ |
|
'you will die. Use !visit to visit a player.')) |
|
cli.msg(harlot, "Players: "+", ".join(pl)) |
|
|
|
for g_angel in var.ROLES["guardian angel"]: |
|
pl = ps[:] |
|
pl.remove(g_angel) |
|
cli.msg(g_angel, ('You are a \u0002guardian angel\u0002. '+ |
|
'It is your job to protect the villagers. If you guard a'+ |
|
' wolf, there is a 50/50 chance of you dying, if you guard '+ |
|
'a victim, they will live. Use !guard to guard a player.')); |
|
cli.msg(g_angel, "Players: " + ", ".join(pl)) |
|
|
|
for d in var.ROLES["village drunk"]: |
|
cli.msg(d, 'You have been drinking too much! You are the \u0002village drunk\u0002.') |
|
|
|
for g in tuple(var.GUNNERS.keys()): |
|
gun_msg = ("You hold a gun that shoots special silver bullets. You may only use it "+ |
|
"during the day. If you shoot a wolf, (s)he will die instantly, but if you "+ |
|
"shoot a villager, that villager will likely survive. You get {0}.") |
|
if var.GUNNERS[g] == 1: |
|
gun_msg = gun_msg.format("1 bullet") |
|
elif var.GUNNERS[g] > 1: |
|
gun_msg = gun_msg.format(str(var.GUNNERS[g]) + " bullets") |
|
else: |
|
continue |
|
cli.msg(g, gun_msg) |
|
|
|
cli.msg(chan, (daydur_msg + "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.")) |
|
if not var.ROLES["wolf"]: # Probably something interesting going on. |
|
chk_nightdone(cli) |
|
chk_traitor(cli) # TODO: Remove this nonsense and add |
|
# a startWithDay custom setting |
|
|
|
|
|
|
|
def cgamemode(cli, *args): |
|
chan = botconfig.CHANNEL |
|
for arg in args: |
|
modeargs = arg.split("=", 1) |
|
modeargs[0] = modeargs[0].strip() |
|
if modeargs[0] in var.GAME_MODES.keys(): |
|
md = modeargs.pop(0) |
|
modeargs[0] = modeargs[0].strip() |
|
try: |
|
gm = var.GAME_MODES[md](modeargs[0]) |
|
for attr in dir(gm): |
|
val = getattr(gm, attr) |
|
if (hasattr(var, attr) and not callable(val) |
|
and not attr.startswith("_")): |
|
var.ORIGINAL_SETTINGS[attr] = getattr(var, attr) |
|
setattr(var, attr, val) |
|
return True |
|
except var.InvalidModeException as e: |
|
cli.msg(botconfig.CHANNEL, "Invalid mode: "+str(e)) |
|
return False |
|
else: |
|
cli.msg(chan, "Mode \u0002{0}\u0002 not found.".format(modeargs[0])) |
|
|
|
|
|
@cmd("!start") |
|
def start(cli, nick, chan, rest): |
|
villagers = var.list_players() |
|
|
|
if var.PHASE == "none": |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
if var.PHASE != "join": |
|
cli.notice(nick, "Werewolf is already in play.") |
|
return |
|
if nick not in villagers: |
|
cli.notice(nick, "You're currently not playing.") |
|
return |
|
now = datetime.now() |
|
var.GAME_START_TIME = now # Only used for the idler checker |
|
dur = int((var.CAN_START_TIME - now).total_seconds()) |
|
if dur > 0: |
|
cli.msg(chan, "Please wait at least {0} more seconds.".format(dur)) |
|
return |
|
|
|
if len(villagers) < 4: |
|
cli.msg(chan, "{0}: Four or more players are required to play.".format(nick)) |
|
return |
|
|
|
for pcount in range(len(villagers), 3, -1): |
|
addroles = var.ROLES_GUIDE.get(pcount) |
|
if addroles: |
|
break |
|
|
|
if var.ORIGINAL_SETTINGS: # Custom settings |
|
while True: |
|
wvs = (addroles[var.INDEX_OF_ROLE["wolf"]] + |
|
addroles[var.INDEX_OF_ROLE["traitor"]]) |
|
if len(villagers) < (sum(addroles) - addroles[var.INDEX_OF_ROLE["gunner"]] - |
|
addroles[var.INDEX_OF_ROLE["cursed"]]): |
|
cli.msg(chan, "There are too few players in the "+ |
|
"game to use the custom roles.") |
|
elif not wvs: |
|
cli.msg(chan, "There has to be at least one wolf!") |
|
elif wvs > (len(villagers) / 2): |
|
cli.msg(chan, "Too many wolves.") |
|
else: |
|
break |
|
reset_settings() |
|
cli.msg(chan, "The default settings have been restored. Please !start again.") |
|
var.PHASE = "join" |
|
return |
|
|
|
|
|
var.ROLES = {} |
|
var.CURSED = [] |
|
var.GUNNERS = {} |
|
|
|
villager_roles = ("gunner", "cursed") |
|
for i, count in enumerate(addroles): |
|
role = var.ROLE_INDICES[i] |
|
if role in villager_roles: |
|
var.ROLES[role] = [None] * count |
|
continue # We deal with those later, see below |
|
selected = random.sample(villagers, count) |
|
var.ROLES[role] = selected |
|
for x in selected: |
|
villagers.remove(x) |
|
|
|
# Now for the villager roles |
|
# Select cursed (just a villager) |
|
if var.ROLES["cursed"]: |
|
var.CURSED = random.sample((villagers + # harlot and drunk can be cursed |
|
var.ROLES["harlot"] + |
|
var.ROLES["village drunk"]), |
|
len(var.ROLES["cursed"])) |
|
del var.ROLES["cursed"] |
|
# Select gunner (also a villager) |
|
if var.ROLES["gunner"]: |
|
possible = (villagers + |
|
var.ROLES["harlot"] + |
|
var.ROLES["village drunk"] + |
|
var.ROLES["seer"]) |
|
for csd in var.CURSED: |
|
if csd in possible: |
|
possible.remove(csd) |
|
for gnr in random.sample(possible, len(var.ROLES["gunner"])): |
|
if var.ROLES["village drunk"] == gnr: |
|
var.GUNNERS[gnr] = var.DRUNK_SHOTS_MULTIPLIER * var.MAX_SHOTS |
|
else: |
|
var.GUNNERS[gnr] = var.MAX_SHOTS |
|
del var.ROLES["gunner"] |
|
|
|
var.ROLES["villager"] = villagers |
|
|
|
cli.msg(chan, ("{0}: Welcome to Werewolf, the popular detective/social party "+ |
|
"game (a theme of Mafia).").format(", ".join(var.list_players()))) |
|
cli.mode(chan, "+m") |
|
|
|
var.ORIGINAL_ROLES = copy.deepcopy(var.ROLES) # Make a copy |
|
var.DAY_TIMEDELTA = timedelta(0) |
|
var.NIGHT_TIMEDELTA = timedelta(0) |
|
var.DAY_START_TIME = None |
|
var.NIGHT_START_TIME = None |
|
|
|
|
|
if not var.START_WITH_DAY: |
|
var.NIGHT_ID = timetime() |
|
transition_night(cli) |
|
else: |
|
#todo: notify roles |
|
var.DAY_ID = timetime() |
|
transition_day(cli) |
|
|
|
# DEATH TO IDLERS! |
|
reapertimer = threading.Thread(None, reaper, args=(cli,)) |
|
reapertimer.daemon = True |
|
reapertimer.start() |
|
|
|
|
|
@cmd("!game", admin_only=var.GAME_COMMAND_ADMIN_ONLY) |
|
def game(cli, nick, chan, rest): |
|
pl = var.list_players() |
|
if var.PHASE == "none": |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
if var.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 nick in var.DENIED_SETTINGS_CHANGE: |
|
cli.notice(nick, "You cannot vote because your previous "+ |
|
"settings change was denied by vote.") |
|
return |
|
if var.SETTINGS_CHANGE_REQUESTER: |
|
cli.notice(nick, "There is already an existing "+ |
|
"settings change request.") |
|
return |
|
rest = rest.strip().lower() |
|
if rest: |
|
if cgamemode(cli, *re.split("\s+",rest)): |
|
var.SETTINGS_CHANGE_REQUESTER = nick |
|
cli.msg(chan, ("\u0002{0}\u0002 has changed the "+ |
|
"game settings successfully. To "+ |
|
'oppose this change, use "!no".').format(nick)) |
|
if var.CAN_START_TIME <= datetime.now(): |
|
var.CAN_START_TIME = datetime.now() + timedelta(seconds=var.EXTRA_WAIT) * 2 |
|
cli.msg(chan, "The wait time has also been extended.") |
|
|
|
|
|
@cmd("!no") |
|
def nay(cli, nick, chan, rest): |
|
pl = var.list_players() |
|
if var.PHASE != "join" or not var.SETTINGS_CHANGE_REQUESTER: |
|
cli.notice(nick, "This command is only allowed if there is "+ |
|
"a game settings change request in effect.") |
|
return |
|
if nick not in pl: |
|
cli.notice(nick, "You're not currently playing.") |
|
return |
|
if var.SETTINGS_CHANGE_REQUESTER in pl: |
|
pl.remove(var.SETTINGS_CHANGE_REQUESTER) |
|
if nick in var.SETTINGS_CHANGE_OPPOSITION: |
|
cli.notice(nick, "You are already in the opposition.") |
|
return |
|
var.SETTINGS_CHANGE_OPPOSITION.append(nick) |
|
needed = max(len(pl)//2 + 1, 2) |
|
if len(var.SETTINGS_CHANGE_OPPOSITION) >= needed: |
|
cli.msg(chan, "The settings change request has been downvoted "+ |
|
"to oblivion. The default settings are restored.") |
|
var.DENIED_SETTINGS_CHANGE.append(var.SETTINGS_CHANGE_REQUESTER) |
|
reset_settings() |
|
else: |
|
cli.msg(chan, ("\u0002{0}\u0002 has voted \u0002no\u0002. {1} more "+ |
|
"vote{2} are needed to deny the change.").format(nick, |
|
needed - len(var.SETTINGS_CHANGE_OPPOSITION), |
|
"s" if needed > len(var.SETTINGS_CHANGE_OPPOSITION) + 1 else "")) |
|
|
|
|
|
|
|
@cmd("!wait") |
|
def wait(cli, nick, chan, rest): |
|
pl = var.list_players() |
|
if var.PHASE == "none": |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
if var.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 var.WAITED >= var.MAXIMUM_WAITED: |
|
cli.msg(chan, "Limit has already been reached for extending the wait time.") |
|
return |
|
|
|
now = datetime.now() |
|
if now > var.CAN_START_TIME: |
|
var.CAN_START_TIME = now + timedelta(seconds=var.EXTRA_WAIT) |
|
else: |
|
var.CAN_START_TIME += timedelta(seconds=var.EXTRA_WAIT) |
|
var.WAITED += 1 |
|
cli.msg(chan, ("\u0002{0}\u0002 increased the wait time by "+ |
|
"{1} seconds.").format(nick, var.EXTRA_WAIT)) |
|
|
|
|
|
|
|
@cmd("!fwait", admin_only=True) |
|
def fwait(cli, nick, chan, rest): |
|
pl = var.list_players() |
|
if var.PHASE == "none": |
|
cli.notice(nick, "No game is currently running.") |
|
return |
|
if var.PHASE != "join": |
|
cli.notice(nick, "Werewolf is already in play.") |
|
return |
|
|
|
now = datetime.now() |
|
if now > var.CAN_START_TIME: |
|
var.CAN_START_TIME = now + timedelta(seconds=var.EXTRA_WAIT) |
|
else: |
|
var.CAN_START_TIME += timedelta(seconds=var.EXTRA_WAIT) |
|
var.WAITED += 1 |
|
cli.msg(chan, ("\u0002{0}\u0002 increased the wait time by "+ |
|
"{1} seconds.").format(nick, var.EXTRA_WAIT)) |
|
|
|
|
|
@cmd("!reset") |
|
def reset_game(cli, nick, chan, rest): |
|
if nick in ("nyuszika7h", "jcao219"): |
|
reset(cli) |