Awesome IRC bot
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.

113 lines
4.2 KiB

"""
.. module::PubSubClient
:synopsis: connect to a message bus and act as a message relay
.. moduleauthor::Dave Pedu <dave@davepedu.com>
"""
from pyircbot.modulebase import ModuleBase, hook
from msgbus.client import MsgbusSubClient # see http://gitlab.davepedu.com/dave/pymsgbus
from threading import Thread
from json import dumps, loads
from time import sleep
from zmq.error import Again
from traceback import print_exc
import re
COMMAND_RE = re.compile(r'\.(([a-zA-Z0-9]{1,16})(\s|$))(\s.+)?')
class PubSubClient(ModuleBase):
def __init__(self, bot, moduleName):
ModuleBase.__init__(self, bot, moduleName)
self.host, self.port = self.config.get("servers")[0].split(":")
self.bus = None
self.services = (self.bot.getmodulesbyservice("services") or [None]).pop(0)
self.bus_listener_thread = Thread(target=self.bus_listener)
self.bus_listener_thread.daemon = True
self.bus_listener_thread.start()
def bus_listener(self):
"""
Listen to the bus for send messages and act on recv
"""
sleep(3)
while True:
if not self.bus:
sleep(0.01)
continue
try:
channel, message = self.bus.recv(block=False)
except Again:
sleep(0.01)
continue
if channel == self.config.get("publish", "pyircbot_{}").format("meta_req"):
if self.services:
self.bus.pub(self.config.get("publish", "pyircbot_{}").format("meta_update"),
"{} {}".format(self.config.get("name", "default"),
dumps({"nick": self.services.nick()})))
else:
try:
name, subcommand, message = message.split(" ", 2)
if name != self.config.get("name", "default") and name != "default":
continue
if subcommand == "privmsg":
dest, message = loads(message)
self.bot.act_PRIVMSG(dest, message)
except:
print_exc()
def publish(self, subchannel, message):
"""
Abstracted callback for proxying irc messages to the bs
:param subchannel: event type such as "privmsg"
:type subchannel: str
:param message: message body
:type message: str
"""
self.bus.pub(self.config.get("publish", "pyircbot_{}").format(subchannel),
"{} {}".format(self.config.get("name", "default"), message))
@hook("PRIVMSG", "JOIN", "PART", "KICK", "MODE", "QUIT", "NICK")
def busdriver(self, msg, cmd):
"""
Relay a privmsg to the event bus
"""
self.publish(msg.command.lower(),
dumps([msg.args,
msg.prefix[0],
msg.trailing, {"prefix": msg.prefix}]))
@hook("PRIVMSG")
def bus_command(self, msg, cmd):
"""
Parse commands and publish as separate channels on the bus. Commands like `.seen nick` will be published
to channel `command_seen`.
"""
match = COMMAND_RE.match(msg.trailing)
if match:
cmd_name = match.groups()[1]
cmd_args = msg.trailing[len(cmd_name) + 1:].strip()
self.publish("command_{}".format(cmd_name),
dumps([msg.args,
msg.prefix[0],
cmd_args,
{"prefix": msg.prefix}]))
def onenable(self):
"""
Connect to the message bus when the module is enabled
"""
self.bus = MsgbusSubClient(self.host, int(self.port))
self.bus.sub(self.config.get("publish", "pyircbot_{}").format("meta_req"))
for channel in self.config.get("subscriptions"):
self.bus.sub(channel)
self.publish("sys", "online")
def ondisable(self):
"""
Disconnect to the message bus on shutdown
"""
self.log.warning("clean it up")
self.publish("sys", "offline")
self.bus.close() # This will crash the listener thread