ircbot: add shorturl plugin

ircbot/plugin/help.py

@@ -10,7 +10,7 @@
 def register(bot):
     threading.Thread(target=help_server, args=(bot,), daemon=True).start()
     bot.listen(r'^help$', help, require_mention=True)
-    bot.listen(r'^macros$', help_macro, require_mention=True)
+    bot.listen(r'^macros?$', help_macro, require_mention=True)
 
 
 def help(bot, msg):
@@ -39,6 +39,11 @@ def render_response(self, template, **context):
             self.end_headers()
             self.wfile.write(rendered)
 
+        def render_404(self):
+            self.send_response(404, 'File not found')
+            self.end_headers()
+            self.wfile.write(b'404 File not found')
+
         def do_GET(self):
             if self.path == '/':
                 plugins = collections.defaultdict(set)
@@ -55,9 +60,7 @@ def do_GET(self):
                     macros=bot.plugins['macros'].list(bot),
                 )
             else:
-                self.send_response(404, 'File not found')
-                self.end_headers()
-                self.wfile.write(b'404 File not found')
+                self.render_404()
 
     return RequestHandler
 

ircbot/plugin/shorturls.py

@@ -0,0 +1,120 @@
+"""Control ocfweb shorturls through ircbot."""
+from ocflib.misc.shorturls import get_connection as shorturl_db
+
+KEYWORDS = {'add', 'delete', 'rename', 'replace'}
+
+
+def register(bot):
+    # [!-~] is all printable ascii except spaces
+    bot.listen(r'^!shorturl ([\w/]+)', show)
+    bot.listen(r'^!shorturl add ([\w/]+) (.+)$', add, require_privileged_oper=True)
+    bot.listen(r'^!shorturl delete ([\w/]+)$', delete, require_privileged_oper=True)
+    bot.listen(r'^!shorturl rename ([\w/]+) ([\w/]+)$', rename, require_privileged_oper=True)
+    bot.listen(r'^!shorturl replace ([\w/]+) (.+)$', replace, require_privileged_oper=True)
+
+
+def retrieve(bot, slug):
+    """Reusable function to retrieve a shorturl by slug from the DB."""
+
+    with shorturl_db() as c:
+        c.execute(
+            'SELECT target FROM shorturls_public WHERE slug = %s',
+            (slug,),
+        )
+        target = c.fetchone()
+
+        return target['target'] if target else None
+
+
+def show(bot, msg):
+    """Return a shorturl by slug."""
+
+    slug = msg.match.group(1)
+
+    # special case these so show doesn't trigger on add/delete
+    # while still letting the trigger appear anywhere in the msg
+    if slug in KEYWORDS:
+        return
+
+    target = retrieve(bot, slug)
+    if not target:
+        msg.respond('shorturl `{}` does not exist.'.format(slug))
+    else:
+        msg.respond(target, ping=False)
+
+
+def add(bot, msg):
+    """Add a new shorturl."""
+
+    slug = msg.match.group(1)
+    target = msg.match.group(2)
+
+    if slug in KEYWORDS:
+        msg.respond('`{}` is a reserved keyword.'.format(slug))
+        return
+
+    if len(slug) > 255:
+        msg.respond('shorturl slugs must be <= 255 characters')
+        return
+
+    with shorturl_db(user='ocfircbot', password=bot.mysql_password) as c:
+
+        c.execute('SELECT * FROM shorturls WHERE slug = %s', (slug,))
+        result = c.fetchone()
+        if result is not None:
+            msg.respond(
+                'shorturl `{}` already exists as {}'.format(
+                    result['slug'],
+                    result['target'],
+                ),
+            )
+        else:
+            c.execute(
+                'INSERT INTO shorturls (slug, target) VALUES (%s, %s)',
+                (slug, target),
+            )
+            msg.respond('shorturl added as `{}`'.format(slug))
+
+
+def delete(bot, msg):
+    """Delete a shorturl."""
+
+    slug = msg.match.group(1)
+    with shorturl_db(user='ocfircbot', password=bot.mysql_password) as c:
+        c.execute(
+            'DELETE FROM shorturls WHERE slug = %s',
+            (slug,),
+        )
+        msg.respond('shorturl `{}` has been deleted.'.format(slug))
+
+
+def rename(bot, msg):
+    """Rename a shorturl."""
+
+    old_slug = msg.match.group(1)
+    new_slug = msg.match.group(2)
+
+    if new_slug in KEYWORDS:
+        msg.respond('`{}` is a reserved keyword.'.format(new_slug))
+        return
+
+    with shorturl_db(user='ocfircbot', password=bot.mysql_password) as c:
+        c.execute(
+            'UPDATE shorturls SET slug = %s WHERE slug = %s',
+            (new_slug, old_slug),
+        )
+        msg.respond('shorturl `{}` has been renamed to `{}`'.format(old_slug, new_slug))
+
+
+def replace(bot, msg):
+    """Replace the target of a shorturl slug."""
+
+    slug = msg.match.group(1)
+    new_target = msg.match.group(2)
+
+    with shorturl_db(user='ocfircbot', password=bot.mysql_password) as c:
+        c.execute(
+            'UPDATE shorturls SET target = %s WHERE slug = %s',
+            (new_target, slug),
+        )
+        msg.respond('shorturl `{}` updated'.format(slug))