add reactions support to thread

app/routes/api.py

@@ -1,7 +1,8 @@
 from flask import Blueprint, request, url_for
 from ..lib.babycode import babycode_to_html
+from ..constants import REACTION_EMOJI
 from .users import is_logged_in, get_active_user
-from ..models import APIRateLimits, Threads
+from ..models import APIRateLimits, Threads, Reactions
 from ..db import db
 
 bp = Blueprint("api", __name__, url_prefix="/api/")
@@ -41,3 +42,57 @@ def babycode_preview():
         return {'error': 'markup field missing or invalid type'}, 400
     rendered = babycode_to_html(markup)
     return {'html': rendered}
+
+
+@bp.post('/add-reaction/<post_id>')
+def add_reaction(post_id):
+    if not is_logged_in():
+        return {'error': 'not authorized', 'error_code': 401}, 401
+    user = get_active_user()
+    reaction_text = request.json.get('emoji')
+    if not reaction_text or not isinstance(reaction_text, str):
+        return {'error': 'emoji field missing or invalid type', 'error_code': 400}, 400
+    if reaction_text not in REACTION_EMOJI:
+        return {'error': 'unsupported reaction', 'error_code': 400}, 400
+
+    reaction = Reactions.find({
+        'user_id': user.id,
+        'post_id': int(post_id),
+        'reaction_text': reaction_text,
+    })
+
+    if reaction:
+        return {'error': 'reaction already exists', 'error_code': 409}, 409
+
+    reaction = Reactions.create({
+        'user_id': user.id,
+        'post_id': int(post_id),
+        'reaction_text': reaction_text,
+    })
+
+    return {'status': 'added'}
+
+
+@bp.post('/remove-reaction/<post_id>')
+def remove_reaction(post_id):
+    if not is_logged_in():
+        return {'error': 'not authorized'}, 401
+    user = get_active_user()
+    reaction_text = request.json.get('emoji')
+    if not reaction_text or not isinstance(reaction_text, str):
+        return {'error': 'emoji field missing or invalid type'}, 400
+    if reaction_text not in REACTION_EMOJI:
+        return {'error': 'unsupported reaction'}, 400
+
+    reaction = Reactions.find({
+        'user_id': user.id,
+        'post_id': int(post_id),
+        'reaction_text': reaction_text,
+    })
+
+    if not reaction:
+        return {'error': 'reaction does not exist'}, 404
+
+    reaction.delete()
+
+    return {'status': 'removed'}

app/routes/threads.py

@@ -3,7 +3,7 @@ from flask import (
     )
 from .users import login_required, mod_only, get_active_user, is_logged_in
 from ..db import db
-from ..models import Threads, Topics, Posts, Subscriptions
+from ..models import Threads, Topics, Posts, Subscriptions, Reactions
 from ..constants import InfoboxKind
 from .posts import create_post
 from slugify import slugify
@@ -61,6 +61,7 @@ def thread(slug):
         topic = topic,
         topics = other_topics,
         is_subscribed = is_subscribed,
+        Reactions = Reactions,
     )