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.
91 lines
2.5 KiB
91 lines
2.5 KiB
14 years ago
|
from oyoyo.client import IRCClient
|
||
|
|
from oyoyo.cmdhandler import DefaultCommandHandler
|
||
|
|
from oyoyo import helpers
|
||
|
|
from oyoyo.parse import parse_nick
|
||
|
|
import logging
|
||
|
|
import botconfig
|
||
|
|
|
||
|
|
def connect_callback(cli):
|
||
|
|
helpers.identify(cli, botconfig.PASS)
|
||
|
|
helpers.join(cli, botconfig.CHANNEL)
|
||
|
|
helpers.msg(cli, "ChanServ", "op "+botconfig.CHANNEL)
|
||
|
|
helpers.msg(cli, botconfig.CHANNEL, "\u0002Wolfbot2 is here.\u0002")
|
||
|
|
|
||
|
|
G_PM_COMMANDS = []
|
||
|
|
G_COMMANDS = []
|
||
|
|
COMMANDS = {}
|
||
|
|
PM_COMMANDS = {}
|
||
|
|
|
||
|
|
HOOKS = {}
|
||
|
|
|
||
|
|
def cmd(s, pmOnly = False):
|
||
|
|
def dec(f):
|
||
|
|
if s is None and pmOnly:
|
||
|
|
G_PM_COMMANDS.append(f)
|
||
|
|
elif s is None and not pmOnly:
|
||
|
|
G_COMMANDS.append(f)
|
||
|
|
elif pmOnly:
|
||
|
|
if s in PM_COMMANDS:
|
||
|
|
PM_COMMANDS[s].append(f)
|
||
|
|
else: PM_COMMANDS[s] = [f]
|
||
|
|
else:
|
||
|
|
if s in COMMANDS:
|
||
|
|
COMMANDS[s].append(f)
|
||
|
|
else: COMMANDS[s] = [f]
|
||
|
|
return f
|
||
|
|
return dec
|
||
|
|
|
||
|
|
def hook(s):
|
||
|
|
def dec(f):
|
||
|
|
HOOKS[s] = f
|
||
|
|
return f
|
||
|
|
return dec
|
||
|
|
|
||
|
|
class WolfBotHandler(DefaultCommandHandler):
|
||
|
|
def __init__(self, client):
|
||
|
|
super().__init__(client)
|
||
|
|
|
||
|
|
def privmsg(self, rawnick, chan, msg):
|
||
|
|
print("{0} in {1} said: {2}".format(rawnick, chan, msg))
|
||
|
|
|
||
|
|
if chan != botconfig.NICK: #not a PM
|
||
|
|
for x in COMMANDS:
|
||
|
|
if msg.startswith(x):
|
||
|
|
msg = msg.replace(x, "", 1)
|
||
|
|
for f in COMMANDS[x]:
|
||
|
|
f(self.client, rawnick, chan, msg.lstrip())
|
||
|
|
else:
|
||
|
|
for x in PM_COMMANDS:
|
||
|
|
if msg.startswith(x):
|
||
|
|
msg = msg.replace(x, "", 1)
|
||
|
|
for f in PM_COMMANDS[x]:
|
||
|
|
f(self.client, rawnick, msg.lstrip())
|
||
|
|
|
||
|
|
def nick(self, fro, to):
|
||
|
|
print(fro, to)
|
||
|
|
|
||
|
|
def main():
|
||
|
|
cli = IRCClient(WolfBotHandler, host="irc.freenode.net", port=6667, nick="wolfbot2-alpha",
|
||
|
|
connect_cb=connect_callback)
|
||
|
|
|
||
|
|
conn = cli.connect()
|
||
|
|
while True:
|
||
|
|
next(conn)
|
||
|
|
|
||
|
|
#Game Logic Begins:
|
||
|
|
|
||
|
|
@cmd("!say", True)
|
||
|
|
def join(cli, rawnick, rest):
|
||
|
|
helpers.msg(cli, botconfig.CHANNEL, "{0} says: {1}".format(parse_nick(rawnick)[0], rest))
|
||
|
|
|
||
|
|
@cmd("!bye", True)
|
||
|
|
@cmd("!bye", False)
|
||
|
|
def forced_exit(cli, rawnick, *rest):
|
||
|
|
if parse_nick(rawnick)[0] in botconfig.ADMINS:
|
||
|
|
helpers.quit(cli, "Forced quit from admin")
|
||
|
|
raise SystemExit
|
||
|
|
|
||
|
|
#Game Logic Ends
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|