diff options
| -rw-r--r-- | apioforum/webhooks.py | 81 |
1 files changed, 71 insertions, 10 deletions
diff --git a/apioforum/webhooks.py b/apioforum/webhooks.py index a9b3a72..5543687 100644 --- a/apioforum/webhooks.py +++ b/apioforum/webhooks.py @@ -4,10 +4,10 @@ import json from .db import get_db from flask import url_for, flash -def abridge_post(text): - MAXLEN = 20 - if len(text) > MAXLEN+3: - return text[:MAXLEN]+"..." + +def abridge(text,maxlen=20): + if len(text) > maxlen+3: + return text[:maxlen]+"..." else: return text @@ -48,17 +48,19 @@ def do_webhooks_thread(forum_id,thread): for wh in get_webhooks(forum_id): try: wh.on_new_thread(thread) - except: - # should probably log the error somewhere + except Exception as e: + #raise e flash(f"error executing webhook with id {wh.wh_id}") def do_webhooks_post(forum_id,post): for wh in get_webhooks(forum_id): try: wh.on_new_post(post) - except: + except Exception as e: + #raise e flash(f"error executing webhook with id {wh.wh_id}") + @webhook_type("fake") class FakeWebhook(WebhookType): def on_new_post(self, post): @@ -108,7 +110,7 @@ class DiscordWebhook(WebhookType): "embeds":[ { "title":"new thread: "+thread['title'], - "description":abridge_post(post['content']), + "description":abridge(post['content']), "url": url_for('thread.view_thread',thread_id=thread['id'],_external=True), "color": 0xff00ff, "fields":[ @@ -143,7 +145,7 @@ class DiscordWebhook(WebhookType): "embeds":[ { "title":"re: "+thread['title'], - "description":abridge_post(post['content']), + "description":abridge(post['content']), "url": post_jump(post['id'],external=True), "color": 0x00ffff, "fields":[ @@ -159,4 +161,63 @@ class DiscordWebhook(WebhookType): } self.send(payload) - + +@webhook_type("apionet") +class ApionetWebhook(WebhookType): + # this is generally quite awful + + sockpath = "/srv/apiobot/bees.sock" + #sockpath = "/home/rebecca/programming/apiobot/bees.sock" + MAXMSGLEN = 420 + + # doesn't use url or anything + def send(self,payload): + # this is possibly terrible + import socket + s = socket.socket(socket.AF_UNIX,socket.SOCK_DGRAM) + s.sendto(payload.encode("utf-8"),self.sockpath) + s.close() + + def append_url(self,rest,url): + ub = url.encode("utf-8") + max_rlen = self.MAXMSGLEN - len(ub) - 1 + + while len(rest.encode("utf-8")) > max_rlen: + # chop off characters until short enough + rest = rest[:-1] + + return rest + " " + url + + def on_new_thread(self,thread): + # copy paste from above. the great refactor will fix this all + db = get_db() + forum = db.execute("select * from forums where id = ?",(thread['forum'],)).fetchone() + username = thread['creator'][:30] + post = db.execute("select * from posts where thread = ? order by id asc limit 1",(thread['id'],)).fetchone() + url = url_for('thread.view_thread',thread_id=thread['id'],_external=True) + + + p = "" + if forum['id'] != 1: + p = f" in {forum['name']}" + payload = f"new thread{p}: \"{abridge(thread['title'],100)}\" - \"{abridge(post['content'],100)}\" (author: {username}) ->" + self.send(self.append_url(payload,url)) + + def on_new_post(self,post): + from .thread import post_jump + db = get_db() + + thread = db.execute("select * from threads where id = ?",(post['thread'],)).fetchone() + forum = db.execute("select * from forums where id = ?",(thread['forum'],)).fetchone() + username = post['author'][:30] + + url = post_jump(post['id'],external=True) + + + payload = f"re: \"{abridge(thread['title'],100)}\" - \"{abridge(post['content'],100)}\" (author: {username}) ->" + self.send(self.append_url(payload,url)) + + + + + |