mirror of
https://github.com/jlu5/PyLink.git
synced 2024-11-01 09:19:23 +01:00
59 lines
2.3 KiB
Python
59 lines
2.3 KiB
Python
# fantasy.py: Adds FANTASY command support, to allow calling commands in channels
|
|
import sys
|
|
import os
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
import utils
|
|
import world
|
|
from log import log
|
|
|
|
def handle_fantasy(irc, source, command, args):
|
|
"""Fantasy command handler."""
|
|
|
|
if not irc.connected.is_set():
|
|
# Break if the IRC network isn't ready.
|
|
return
|
|
|
|
try: # First, try to fetch the config-defined prefix.
|
|
prefixes = [irc.botdata["prefix"]]
|
|
except KeyError: # Config option is missing.
|
|
prefixes = []
|
|
|
|
if irc.botdata.get("respondtonick"):
|
|
# If responding to nick is enabled, add variations of the current nick
|
|
# to the prefix list: "<nick>," and "<nick>:"
|
|
nick = irc.pseudoclient.nick
|
|
prefixes += [nick+',', nick+':']
|
|
|
|
if not prefixes:
|
|
# We finished with an empty prefixes list, meaning fantasy is misconfigured!
|
|
log.warning("(%s) Fantasy prefix was not set in configuration - "
|
|
"fantasy commands will not work!", irc.name)
|
|
return
|
|
|
|
channel = args['target']
|
|
orig_text = args['text']
|
|
|
|
if utils.isChannel(channel) and not irc.isInternalClient(source):
|
|
# The following conditions must be met for an incoming message for
|
|
# fantasy to trigger:
|
|
# 1) The message target is a channel.
|
|
# 2) A PyLink service client exists in the channel.
|
|
# 3) The message starts with one of our fantasy prefixes.
|
|
# 4) The sender is NOT a PyLink client (this prevents infinite
|
|
# message loops).
|
|
for botname, sbot in world.services.items():
|
|
log.debug('(%s) fantasy: checking bot %s', irc.name, botname)
|
|
if sbot.uids.get(irc.name) in irc.channels[channel].users:
|
|
for prefix in prefixes: # Cycle through the prefixes list we finished with.
|
|
if orig_text.startswith(prefix):
|
|
|
|
# Cut off the length of the prefix from the text.
|
|
text = orig_text[len(prefix):]
|
|
|
|
# Finally, call the bot command and loop to the next bot.
|
|
sbot.call_cmd(irc, source, text, called_by=channel, notice=False)
|
|
continue
|
|
|
|
utils.add_hook(handle_fantasy, 'PRIVMSG')
|