def __init__(self, autojoin, autojoin_list=config["irc"]["channels"], nick=config["irc"]["nick"]):
self.autojoin_list = autojoin_list
self.autojoin = autojoin
self.nickname = nick
self.redmine_instance = RedmineTicketFetcher(config["redmine"]["url"], config["redmine"]["api_key"])
self.logger = Logger("main.log")
#self.logger.addLogfile("raw", "raw.log")
self.handlers = {}
self.config = config
self.callbacks = []
self.users = LibUnderscore.loadUserList()
#self.addHandler(EasterEggHandler())
for plugin in config['core']['plugins']['autoload']:
self.addHandler(plugin[0], plugin[1])
def __init__(self,
autojoin,
autojoin_list=config["irc"]["channels"],
nick=config["irc"]["nick"]):
self.autojoin_list = autojoin_list
self.autojoin = autojoin
self.nickname = nick
self.redmine_instance = RedmineTicketFetcher(
config["redmine"]["url"], config["redmine"]["api_key"])
self.logger = Logger("main.log")
#self.logger.addLogfile("raw", "raw.log")
self.handlers = {}
self.callbacks = []
self.users = LibUnderscore.loadUserList()
#self.addHandler(EasterEggHandler())
for plugin in config['core']['plugins']['autoload']:
self.addHandler(plugin[0], plugin[1])
def handleCommand(client, user, channel, msg):
command = parseCommand(client.nickname, msg)
if command:
client.logger.write("Command in %s from %s: %s: '%s'" % (channel, user, command["command"], command["args"]))
if command["command"] == "help":
client.msg(channel, Help.getHelp(command["args"].strip()))
# client.msg(channel,
#"""Type `snot <ticketNumber>` to get the contents of a ticket.
#snot <ticketNumber> <formatString> to customize the output.
#Example: $number | $summary_email | $assigned_to | $subject | $flags""")
elif command["command"] == "snot":
snotCommand = re.match("\s*#?(?P<ticketNumber>\d+)\s*(?P<fString>.*)", command["args"])
number = snotCommand.group("ticketNumber")
#fString = snotCommand.group("fString")
fString = None
if fString:
formattedString = sp.formatTicket(int(number), fString)
else:
formattedString = sp.formatTicket(int(number), "$number | $from_line | $assigned_to | $subject | $flags")
#client.msg(channel,"SNOT COMMAND TIME: %s" % snotCommand.groups("ticketNumber"))
client.msg(channel, formattedString)
elif command["command"] == "join":
channeljoin = re.match("(#?\S*)\s*(.*)", command["args"])
# client.msg(channel, str(channeljoin.groups()))
# for item in channeljoin.groups():
# client.msg(channel, item)
chan = channeljoin.group(1)
key = channeljoin.group(2)
if (key):
client.msg(channel, "Joining %s with key \"%s\"" % (chan, key))
client.join(chan, key)
else:
client.msg(channel, "Joining %s (no key)" % (chan,))
client.join(chan)
elif command["command"] in ("part", "leave"):
channelPart = re.match("(#?\S*)\s*", command["args"])
client.leave(channelPart.group(1), "Parting is such sweet sorrow")
elif command["command"] in ("listHandlers", "lh"):
client.msg(channel, "Current handlers:")
for handler in client.handlers:
client.msg(channel, handler)
elif command["command"] in ("reload", "rel"):
client.msg(channel, client.reloadModule(command["args"].strip()))
elif command["command"] in ("reloadHandler", "rh"):
client.msg(channel, client.reloadHandler(command["args"].strip()))
# elif command["command"] in ("reloadConfig", "rc"):
# CONFIG_FILE = "config.yaml"
# with open(CONFIG_FILE) as cfgFile:
# UnderscoreBot.config = yaml.load(cfgFile)
elif command["command"] in ("startSNOTMonitoring", "ssm"):
client.logger.write("Calling snot monitoring in subthread")
client.msg(channel, "Restarting snot monitoring")
reactor.callInThread(SNOTMagic.monitorLogs, client)
elif command["command"] in ("ticketHistory", "th"):
try:
lines = sp.getTicketHistory(command["args"])
for line in lines:
client.msg(channel, line.strip())
except ValueError as e:
client.msg(channel, "Invalid argument")
# elif command["command"] in ("authstat"):
# targetNick = command["args"]
# def callback(prefix, command, params):
# if prefix == "330" and params[1].lower() == targetNick.lower():
# client.msg(channel, "%s is logged in as %s" % (params[1], params[2]))
# return True
# elif prefix == "ERR_NOSUCHNICK" and params[1].lower() == targetNick.lower():
# client.msg(channel, "%s does not appear to be a current user" % (params[1]))
# return True
# elif prefix == "RPL_ENDOFWHOIS" and params[1].lower() == targetNick.lower():
# client.msg(channel, "%s does not appear to be logged in." % (params[1]))
# else:
# return False
# client.addCallback(callback)
# client.sendLine("WHOIS %s" % command["args"])
elif command["command"] in ("authstat"):
targetNick = command["args"]
LibUnderscore.checkAuthStatus(client, targetNick,
lambda nick,account: client.msg(channel, "%s is logged in as %s" % (nick, account)),
lambda nick : client.msg(channel, "%s does not appear to be a current user" % (nick)),
lambda nick : client.msg(channel, "%s does not appear to be logged in" % (nick)))
elif command["command"] in ("comp"):
validUsers = client.users
print validUsers
tickets = command["args"].split()
def authCallback(nick,account):
if account in validUsers:
for ticket in tickets:
if ticket.isdigit():
SNOTMagic.completeTicket(int(ticket), "*****@*****.**" % validUsers[account], "Ticket comped by %s in %s" % (user,channel))
client.msg(channel, "Emails sent.")
else:
client.msg(channel, "Sorry, you are not authorized to perform that action")
LibUnderscore.checkAuthStatus(client, user, authCallback,
lambda nick: client.msg(channel, "You don't exist. How is this even possible? Wren ^^"),
lambda nick: client.msg(channel, "You must be authenticated to NickServ to perform this operation"))
elif command["command"] in ("chronicle", "chron"):
ticketCommand = re.match("\s*#?(?P<ticketNumber>\d+)\s*(?P<fString>.*)", command["args"])
number = int(ticketCommand.group("ticketNumber"))
try:
d = client.redmine_instance.getTicket(number)
client.msg(channel, str(Template("#$id ($project) | $author | $assigned_to | $subject | $tracker").safe_substitute(d)))
except urllib2.HTTPError as e:
client.logger.write(str(type(e)))
client.msg(channel, str(e))
else:
print "Unrecognized command"
def handleCommand(client, user, channel, msg):
command = parseCommand("_", msg)
if command:
client.logger.write("Command in %s from %s: %s: '%s'" % (channel, user, command["command"], command["args"]))
if command["command"] == "help":
client.msg(channel, Help.getHelp(command["args"].strip()))
# client.msg(channel,
#"""Type `snot <ticketNumber>` to get the contents of a ticket.
#snot <ticketNumber> <formatString> to customize the output.
#Example: $number | $summary_email | $assigned_to | $subject | $flags""")
elif command["command"] == "snot":
snotCommand = re.match("\s*#?(?P<ticketNumber>\d+)\s*(?P<fString>.*)", command["args"])
number = snotCommand.group("ticketNumber")
#fString = snotCommand.group("fString")
fString = None
if fString:
formattedString = sp.formatTicket(int(number), fString)
else:
formattedString = sp.formatTicket(int(number), "$number | $from_line | $assigned_to | $subject | $flags")
#client.msg(channel,"SNOT COMMAND TIME: %s" % snotCommand.groups("ticketNumber"))
client.msg(channel, formattedString)
elif command["command"] == "join":
channeljoin = re.match("(#?\S*)\s*(.*)", command["args"])
# client.msg(channel, str(channeljoin.groups()))
# for item in channeljoin.groups():
# client.msg(channel, item)
chan = channeljoin.group(1)
key = channeljoin.group(2)
if (key):
client.msg(channel, "Joining %s with key \"%s\"" % (chan, key))
client.join(chan, key)
else:
client.msg(channel, "Joining %s (no key)" % (chan,))
client.join(chan)
elif command["command"] in ("part", "leave"):
channelPart = re.match("(#?\S*)\s*", command["args"])
client.leave(channelPart.group(1), "Parting is such sweet sorrow")
elif command["command"] in ("listHandlers", "lh"):
client.msg(channel, "Current handlers:")
for handler in client.handlers:
client.msg(channel, handler)
elif command["command"] in ("reload", "rel"):
client.msg(channel, client.reloadModule(command["args"].strip()))
elif command["command"] in ("reloadHandler", "rh"):
client.msg(channel, client.reloadHandler(command["args"].strip()))
elif command["command"] in ("reloadConfig", "rc"):
CONFIG_FILE = "config.yaml"
with open(CONFIG_FILE) as cfgFile:
client.config.update(yaml.load(cfgFile))
elif command["command"] in ("startSNOTMonitoring", "ssm"):
client.logger.write("Calling snot monitoring in subthread")
client.msg(channel, "Restarting snot monitoring")
reactor.callInThread(SNOTMagic.monitorLogs, client)
elif command["command"] in ("ticketHistory", "th"):
try:
lines = sp.getTicketHistory(command["args"])
for line in lines:
client.msg(channel, line.strip())
except ValueError as e:
client.msg(channel, "Invalid argument")
# elif command["command"] in ("authstat"):
# targetNick = command["args"]
# def callback(prefix, command, params):
# if prefix == "330" and params[1].lower() == targetNick.lower():
# client.msg(channel, "%s is logged in as %s" % (params[1], params[2]))
# return True
# elif prefix == "ERR_NOSUCHNICK" and params[1].lower() == targetNick.lower():
# client.msg(channel, "%s does not appear to be a current user" % (params[1]))
# return True
# elif prefix == "RPL_ENDOFWHOIS" and params[1].lower() == targetNick.lower():
# client.msg(channel, "%s does not appear to be logged in." % (params[1]))
# else:
# return False
# client.addCallback(callback)
# client.sendLine("WHOIS %s" % command["args"])
elif command["command"] in ("authstat",):
targetNick = command["args"]
LibUnderscore.checkAuthStatus(client, targetNick,
lambda nick,account: client.msg(channel, "%s is logged in as %s" % (nick, account)),
lambda nick : client.msg(channel, "%s does not appear to be a current user" % (nick)),
lambda nick : client.msg(channel, "%s does not appear to be logged in" % (nick)))
elif command["command"] in ("comp",):
validUsers = client.users
argSplit = command["args"].split(' ', 1)
tickets = argSplit[0].split(',')
if len(argSplit) >= 2:
userMessage = argSplit[1]
else:
userMessage = None
if tickets == None or tickets == ['']:
client.msg(channel, "USAGE: comp tkt1[,tkt2,tkt3...] [message]")
return
def authCallback(nick,account):
if account in validUsers:
for ticket in tickets:
if ticket.isdigit():
ticketDict = sp.parseTicket(int(ticket), client.config['snot']['defaultCommand'])
if (ticketDict):
client.msg(channel, "Completing %s (%s)" % (ticket, ticketDict['subject']))
message = "(Ticket comped by %s in %s)"
if userMessage:
message = userMessage + "\n\n" + message
SNOTMagic.completeTicket(int(ticket), "*****@*****.**" % validUsers[account], client.config, message % (user,channel))
else:
client.msg(channel, "%s does not appear to be a valid ticket" % (ticket,))
else:
client.msg(channel, "Sorry, you are not authorized to perform that action")
LibUnderscore.checkAuthStatus(client, user, authCallback,
lambda nick: client.msg(channel, "You don't exist. How is this even possible? Wren ^^"),
lambda nick: client.msg(channel, "You must be authenticated to NickServ to perform this operation"))
elif command["command"] in ("flag", "flags"):
validUsers = client.users
argSplit = command["args"].split(' ', 1)
tickets = argSplit[0].split(',')
if len(argSplit) >= 2:
flags = argSplit[1]
flagSplit = flags.split(',')
print tickets, flags, flagSplit
p = subprocess.Popen([client.config['snot']['defaultCommand'], '-hF'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
print type(p.stdout)
validFlags = [s.strip() for s in iter(p.stdout.readline,'\n')]
print validFlags
for flag in flagSplit:
if flag not in validFlags:
client.msg(channel, "%s is not a valid flag" % flag)
return
flagsArg = flags
else:
client.msg(channel, "USAGE: flags tkt1,[tkt2,tkt3...] flag1[,flag2,flag3...]")
return
def authCallback(nick,account):
if account in validUsers:
for ticket in tickets:
if ticket.isdigit():
ticketDict = sp.parseTicket(int(ticket), client.config['snot']['defaultCommand'])
if (ticketDict):
client.msg(channel, "Flagging %s (%s) as %s" % (ticket, ticketDict['subject'], flags))
message = "(Ticket flagged as %s by %s in %s)" % (flags,user,channel)
SNOTMagic.flagTicket(int(ticket), "*****@*****.**" % validUsers[account], client.config, flagsArg, message)
else:
client.msg(channel, "%s does not appear to be a valid ticket" % (ticket,))
else:
client.msg(channel, "Sorry, you are not authorized to perform that action")
LibUnderscore.checkAuthStatus(client, user, authCallback,
lambda nick: client.msg(channel, "You don't exist. How is this even possible? Wren ^^"),
lambda nick: client.msg(channel, "You must be authenticated to NickServ to perform this operation"))
# if (ticket):
elif command["command"] in ("nick",):
def authCallback(nick,account):
if account.lower() == "wren":
client.setNick(command["args"])
else:
client.msg(channel, "Sorry, you are not authorized to perform that action")
LibUnderscore.checkAuthStatus(client, user, authCallback,
lambda nick: client.msg(channel, "You don't exist. How is this even possible? Wren ^^"),
lambda nick: client.msg(channel, "You must be authenticated to NickServ to perform this operation"))
elif command["command"] in ("chronicle", "chron"):
ticketCommand = re.match("\s*#?(?P<ticketNumber>\d+)\s*(?P<fString>.*)", command["args"])
number = int(ticketCommand.group("ticketNumber"))
try:
d = client.redmine_instance.getTicket(number)
client.msg(channel, str(Template("#$id ($project) | $author | $assigned_to | $subject | $tracker").safe_substitute(d)))
except urllib2.HTTPError as e:
client.logger.write(str(type(e)))
client.msg(channel, str(e))
elif command["command"] in ("horse",):
with open("horse_combined", 'r') as f:
random_line = str(random.choice(f.readlines()))
client.msg(channel, random_line)
elif command["command"] in ("riker",):
with open("riker.txt", 'r') as f:
random_line = str(random.choice(f.readlines()))
client.msg(channel, random_line)
else:
print "Unrecognized command"
