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.
120 lines
3.4 KiB
120 lines
3.4 KiB
14 years ago
|
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.GAME_STARTED = False
|
||
|
|
vars.ROLES = {"person" : []}
|
||
|
|
vars.PHASE = "none"
|
||
|
|
|
||
|
|
# Command Handlers:
|
||
|
|
@cmd("!say")
|
||
|
|
def say(cli, rawnick, rest): # To be removed later
|
||
|
|
cli.msg(botconfig.CHANNEL, "{0} says: {1}".format(parse_nick(rawnick)[0], rest))
|
||
|
|
|
||
|
|
|
||
|
|
@pmcmd("!bye")
|
||
|
|
@cmd("!bye")
|
||
|
|
def forced_exit(cli, rawnick, *rest): # Admin Only
|
||
|
|
if parse_nick(rawnick)[0] in botconfig.ADMINS:
|
||
|
|
cli.quit("Forced quit from admin")
|
||
|
|
raise SystemExit
|
||
|
|
|
||
|
|
@cmd("!exec")
|
||
|
|
def py(cli, rawnick, chan, rest):
|
||
|
|
if parse_nick(rawnick)[0] in botconfig.ADMINS:
|
||
|
|
exec(rest)
|
||
|
|
|
||
|
|
@cmd("!ping")
|
||
|
|
def pinger(cli, rawnick, 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, parse_nick(rawnick)[0]): 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, rawnick, chan, rest):
|
||
|
|
if vars.PHASE != "none":
|
||
|
|
return
|
||
|
|
|
||
|
|
vars.GAME_STARTED = True
|
||
|
|
|
||
|
|
nick = parse_nick(rawnick)[0]
|
||
|
|
cli.msg(chan, '{0} 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))
|
||
|
|
|
||
|
|
vars.ROLES["person"].append(nick)
|
||
|
|
vars.PHASE = "join"
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
@cmd("!stats")
|
||
|
|
def stats(cli, rawnick, chan, rest):
|
||
|
|
if vars.PHASE == "none":
|
||
|
|
return
|
||
|
|
nick = parse_nick(rawnick)[0]
|
||
|
|
pl = []
|
||
|
|
for x in vars.ROLES.values(): pl.extend(x)
|
||
|
|
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]))
|
||
|
|
|
||
|
|
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] + ".")
|
||
|
|
|
||
|
|
# Game Logic Ends
|