fix setnitter

[erebus.git] / modules / nitterize.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
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]
	if chan[0] != "#": return
	msg = pieces[3][1:]

	replaced = ""
	for mo in re_findtwitter.finditer(msg):
		replaced += re_findtwitter.sub(bot.parent.cfg.get('nitterize', 'instance', 'https://nitter.net') + r"\1", mo.group(0)) + " "
	if replaced:
		bot.msg(chan, "%s" % (replaced))

@lib.hook(glevel=lib.STAFF, needchan=False)
@lib.argsEQ(1)
def setnitter(bot, user, chan, realtarget, *args):
	newnitter = args[0]
	if newnitter[-1] != "/":
		newnitter += "/"
	bot.parent.cfg.set('nitterize', 'instance', newnitter)
	return 'Done'