summaryrefslogtreecommitdiff
path: root/crocoite
diff options
context:
space:
mode:
Diffstat (limited to 'crocoite')
-rw-r--r--crocoite/cli.py4
-rw-r--r--crocoite/irc.py16
2 files changed, 17 insertions, 3 deletions
diff --git a/crocoite/cli.py b/crocoite/cli.py
index b73051b..be3538a 100644
--- a/crocoite/cli.py
+++ b/crocoite/cli.py
@@ -132,7 +132,7 @@ def recursive ():
return 0
def irc ():
- import json
+ import json, re
from .irc import Chromebot
logger = Logger (consumer=[DatetimeConsumer (), JsonPrintConsumer ()])
@@ -145,6 +145,7 @@ def irc ():
with open (args.config) as fd:
config = json.load (fd)
s = config['irc']
+ blacklist = dict (map (lambda x: (re.compile (x[0], re.I), x[1]), config['blacklist'].items ()))
loop = asyncio.get_event_loop()
bot = Chromebot (
@@ -157,6 +158,7 @@ def irc ():
destdir=config['destdir'],
processLimit=config['process_limit'],
logger=logger,
+ blacklist=blacklist,
loop=loop)
stop = lambda signum: bot.cancel ()
loop.add_signal_handler (signal.SIGINT, stop, signal.SIGINT)
diff --git a/crocoite/irc.py b/crocoite/irc.py
index 1b0fa1b..5351a85 100644
--- a/crocoite/irc.py
+++ b/crocoite/irc.py
@@ -368,11 +368,11 @@ def jobExists (func):
return inner
class Chromebot (ArgparseBot):
- __slots__ = ('jobs', 'tempdir', 'destdir', 'processLimit')
+ __slots__ = ('jobs', 'tempdir', 'destdir', 'processLimit', 'blacklist')
def __init__ (self, host, port, ssl, nick, logger, channels=[],
tempdir=tempfile.gettempdir(), destdir='.', processLimit=1,
- loop=None):
+ blacklist={}, loop=None):
super().__init__ (host=host, port=port, ssl=ssl, nick=nick,
logger=logger, channels=channels, loop=loop)
@@ -380,6 +380,7 @@ class Chromebot (ArgparseBot):
self.tempdir = tempdir
self.destdir = destdir
self.processLimit = asyncio.Semaphore (processLimit)
+ self.blacklist = blacklist
def getParser (self):
parser = NonExitingArgumentParser (prog=self.nick + ': ', add_help=False)
@@ -404,10 +405,21 @@ class Chromebot (ArgparseBot):
return parser
+ def isBlacklisted (self, url):
+ for k, v in self.blacklist.items():
+ if k.match (url):
+ return v
+ return False
+
@voice
async def handleArchive (self, user, args, reply):
""" Handle the archive command """
+ msg = self.isBlacklisted (args.url)
+ if msg:
+ reply (f'{args.url} cannot be queued: {msg}')
+ return
+
j = Job (args.url, user.name)
assert j.id not in self.jobs, 'duplicate job id'
self.jobs[j.id] = j