[erebus.git] / modules / subtext.py

# Erebus IRC bot - Author: Erebus Team
# vim: fileencoding=utf-8
# module for 's/regex/replacement/' style correction
# warning: arbitrary regex's are generally capable of various DoS attacks on CPU/memory usage. use with caution.
# see for usage examples: https://github.com/zonidjan/erebus/commit/d7e9802778477f1faa26a03078cb1b3c018a5e5c
# This file is released into the public domain; see http://unlicense.org/

# module info
modinfo = {
	'author': 'Erebus Team',
	'license': 'public domain',
	'compatible': [0], # compatible module API versions
	'depends': [], # other modules required to work properly?
	'softdeps': [], # modules which are preferred but not required
}

# preamble
import modlib
lib = modlib.modlib(__name__)
modstart = lib.modstart
modstop = lib.modstop

# module code
import re
from collections import namedtuple
re_findsub = re.compile(r"""s([/'";:!@#$%^&-_=`~])(?P<search>.+?)\1(?P<replace>.*?)(?:\1(?P<global>g)?|$)""")
Line = namedtuple('Line', ['sender', 'msg'])
lastline = {}
@lib.hooknum("PRIVMSG")
def privmsg_hook(bot, line):
	pieces = line.split(None, 3)
	fromnick = pieces[0][1:].split('!')[0]
	chan = pieces[2]
	if chan[0] != "#": return
	msg = pieces[3][1:]

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