summaryrefslogtreecommitdiffhomepage
path: root/apioforum/webhooks.py
blob: 2407039aba1d0c0a0224b739fbe43187d8eccccc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
import urllib
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]+"..."
    else:
        return text

def send_discord_webhook(url,payload):
    headers = {
        "User-Agent":"apioforum (https://f.gh0.pw/apioforum/, v0.0)",
        "Content-Type":"application/json",
    }
    req = urllib.request.Request(
        url,
        json.dumps(payload).encode("utf-8"),
        headers
    )
    # todo: read response and things
    urllib.request.urlopen(req)
    #try:
    #    res = urllib.request.urlopen(req)
    #except urllib.error.HTTPError as e:
    #    print(f"error {e.code} {e.read()}")
    #else:
    #    print(f"succ {res.read()}")
    
# todo: object oriented thing for different kinds of webhook (not just discord)

def f(name,value):
    # discord embed field
    return {"name":name,"value":value,"inline":True}

def discord_on_new_thread(wh_url, thread):
    db = get_db()
    forum = db.execute("select * from forums where id = ?",(thread['forum'],)).fetchone()
    username = thread['creator']
    userpage = url_for('user.view_user',username=username,_external=True)

    forumpage = url_for('forum.view_forum',forum_id=forum['id'],_external=True)

    post = db.execute("select * from posts where thread = ? order by id asc limit 1",(thread['id'],)).fetchone()
    
    
    payload = {
        "username":"apioforum",
        "avatar_url":"https://d.gh0.pw/lib/exe/fetch.php?media=wiki:logo.png",
        "embeds":[
            {
                "title":thread['title'],
                "description":abridge_post(post['content']),
                "url": url_for('thread.view_thread',thread_id=thread['id'],_external=True),
                "color": 0xff00ff,
                "fields":[
                    f('author',f"[{username}]({userpage})"),
                    f('forum',f"[{forum['name']}]({forumpage})"),
                ],
                "footer":{
                    "text":thread['created'].isoformat(' '),
                },
            },
        ],
    }
    send_discord_webhook(wh_url,payload)

def discord_on_new_post(wh_url, post):
    from .thread import post_jump
    db = get_db()

    thread = db.execute("select * from threads where id = ?",(post['thread'],)).fetchone()
    threadpage = url_for('thread.view_thread',thread_id=thread['id'],_external=True)

    forum = db.execute("select * from forums where id = ?",(thread['forum'],)).fetchone()
    forumpage = url_for('forum.view_forum',forum_id=forum['id'],_external=True)

    username = post['author']
    userpage = url_for('user.view_user',username=username,_external=True)

    payload = {
        "username":"apioforum",
        "avatar_url":"https://d.gh0.pw/lib/exe/fetch.php?media=wiki:logo.png",
        "embeds":[
            {
                "title":"re: "+thread['title'],
                "description":abridge_post(post['content']),
                "url": post_jump(post['id'],external=True),
                "color": 0x00ffff,
                "fields":[
                    f('author',f"[{username}]({userpage})"),
                    f('thread',f"[{thread['title']}]({threadpage})"),
                    f('forum',f"[{forum['name']}]({forumpage})"),
                ],
                "footer":{
                    "text":post['created'].isoformat(' '),
                },
            },
        ],
    }
    send_discord_webhook(wh_url,payload)

def _do_webhooks(forum_id,thing,fn):
    db = get_db()
    # todo inheritance
    webhooks = db.execute("""
        WITH RECURSIVE fs AS
            (SELECT * FROM forums WHERE id = ?
             UNION ALL
             SELECT forums.* FROM forums, fs WHERE fs.parent=forums.id)
        SELECT * from webhooks
        WHERE
            webhooks.forum = ?
        OR
            webhooks.inherits AND webhooks.forum IN (SELECT id FROM fS);

    """,(forum_id,forum_id)).fetchall()
    for wh in webhooks:
        wh_url = wh['url']
        try:
            fn(wh_url,thing)
        except:
            pass # handle probably
        flash(f"wh {wh['id']}")

def do_webhooks_thread(forum_id,thread):
    _do_webhooks(forum_id,thread,discord_on_new_thread)

def do_webhooks_post(forum_id,post):
    _do_webhooks(forum_id,post,discord_on_new_post)