X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/b736b336bbec2f05e576e13d06619dc9552a7c91..cf400e0917f9b214cd7642e410ac0a13a9a8c726:/modules/nitterize.py

diff --git a/modules/nitterize.py b/modules/nitterize.py
new file mode 100644
index 0000000..2b4251f
--- /dev/null
+++ b/modules/nitterize.py
@@ -0,0 +1,37 @@
+# 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_findtwitter = re.compile(r"""https?://(?:www\.)?(?:twitter|x)\.com/([-.a-z0-9/]+)""", re.I)
+
+@lib.hooknum("PRIVMSG")
+def privmsg_hook(bot, line):
+	pieces = line.split(None, 3)
+	chan = pieces[2]
+	msg = pieces[3][1:]
+	replaced = ""
+	for mo in re_findtwitter.finditer(msg):
+		replaced += re_findtwitter.sub(r"https://nitter.projectsegfau.lt/\1", mo.group(0)) + " "
+	if replaced:
+		bot.msg(chan, "%s" % (replaced))