modules/subtext.py

   1 # Erebus IRC bot - Author: Erebus Team
   2 # vim: fileencoding=utf-8
   3 # module for 's/regex/replacement/' style correction
   4 # warning: arbitrary regex's are generally capable of various DoS attacks on CPU/memory usage. use with caution.
   5 # This file is released into the public domain; see http://unlicense.org/
   6
   7 # module info
   8 modinfo = {
   9         'author': 'Erebus Team',
  10         'license': 'public domain',
  11         'compatible': [0], # compatible module API versions
  12         'depends': [], # other modules required to work properly?
  13         'softdeps': [], # modules which are preferred but not required
  14 }
  15
  16 # preamble
  17 import modlib
  18 lib = modlib.modlib(__name__)
  19 modstart = lib.modstart
  20 modstop = lib.modstop
  21
  22 # module code
  23 import re
  24 from collections import namedtuple
  25 re_findsub = re.compile(r"""s([/'";:!@#$%^&-_=`~])(?P<search>.+?)\1(?P<replace>.*?)(?:\1(?P<global>g)?|$)""")
  26 Line = namedtuple('Line', ['sender', 'msg'])
  27 lastline = {}
  28 @lib.hooknum("PRIVMSG")
  29 def privmsg_hook(bot, line):
  30         pieces = line.split(None, 3)
  31         fromnick = pieces[0][1:].split('!')[0]
  32         chan = pieces[2]
  33         if chan[0] != "#": return
  34         msg = pieces[3][1:]
  35
  36         mo = re_findsub.match(msg)
  37         if mo:
  38                 if mo.group('global') is not None:
  39                         count = 0 # unlimited
  40                 else:
  41                         count = 1 # only first
  42                 try:
  43                         newline = re.sub(mo.group('search'), mo.group('replace'), lastline[chan].msg, count)
  44                 except Exception as e: return # ignore it if it doesn't work
  45                 if newline != lastline[chan].msg:
  46                         if lastline[chan].sender == fromnick:
  47                                 bot.msg(chan, "<%s> %s" % (lastline[chan].sender, newline))
  48                         else:
  49                                 bot.msg(chan, "%s: <%s> %s" % (fromnick, lastline[chan].sender, newline))
  50         else:
  51                 lastline[chan] = Line(sender=fromnick, msg=msg)