Adding tag notifications

main.py

@@ -12,6 +12,7 @@ import uvicorn
 import threading
 import asyncio
 from slack_sdk import WebClient
+import re
 
 app = FastAPI()
 
@@ -94,6 +95,106 @@ def send_notification(title, body, token, id, post_id, notif_type):
     except requests.exceptions.RequestException as e:
         print(f'Failed to send notification: {e}')
 
+def handle_list_comment_tags_update(event_payload):
+    print('Received update in comment tags:')
+    print(event_payload)
+
+    record = event_payload['record']
+
+    try:
+        content = record.get('comment', '[]')
+        tagged_user_ids = re.findall(r'@<([^>]+)>', content)
+    except json.JSONDecodeError:
+        print('Failed to parse tags.')
+        return
+
+    author_id = record['user_id']
+    username = getUsername(author_id).lower()
+    title = 'Picturelock'
+    body = f'@{username} tagged you in a comment!'
+    post_id = record['post_id']
+
+    for tagged_user_id in tagged_user_ids:
+        if tagged_user_id != author_id:
+            token = getAuthorNotifToken(tagged_user_id)
+            if token:
+                send_notification(title, body, token, author_id, post_id, "list")
+
+def handle_list_reply_tags_update(event_payload):
+    print('Received update in reply tags:')
+    print(event_payload)
+
+    record = event_payload['record']
+
+    try:
+        content = record.get('comment', '[]')
+        tagged_user_ids = re.findall(r'@<([^>]+)>', content)
+    except json.JSONDecodeError:
+        print('Failed to parse tags.')
+        return
+
+    author_id = record['author_id']
+    username = getUsername(author_id).lower()
+    title = 'Picturelock'
+    body = f'@{username} tagged you in a reply!'
+    post_id = record['post_id']
+
+    for tagged_user_id in tagged_user_ids:
+        if tagged_user_id != author_id:
+            token = getAuthorNotifToken(tagged_user_id)
+            if token:
+                send_notification(title, body, token, author_id, post_id, "list")
+
+def handle_comment_tags_update(event_payload):
+    print('Received update in comment tags:')
+    print(event_payload)
+
+    record = event_payload['record']
+
+    try:
+        content = record.get('comment', '[]')
+        tagged_user_ids = re.findall(r'@<([^>]+)>', content)
+    except json.JSONDecodeError:
+        print('Failed to parse tags.')
+        return
+
+    author_id = record['user_id']
+    username = getUsername(author_id).lower()
+    title = 'Picturelock'
+    body = f'@{username} tagged you in a comment!'
+    post_id = record['post_id']
+
+    for tagged_user_id in tagged_user_ids:
+        if tagged_user_id != author_id:
+            token = getAuthorNotifToken(tagged_user_id)
+            if token:
+                send_notification(title, body, token, author_id, post_id, "post")
+
+def handle_reply_tags_update(event_payload):
+    print('Received update in reply tags:')
+    print(event_payload)
+
+    record = event_payload['record']
+
+    try:
+        content = record.get('comment', '[]')
+        tagged_user_ids = re.findall(r'@<([^>]+)>', content)
+    except json.JSONDecodeError:
+        print('Failed to parse tags.')
+        return
+
+    author_id = record['author_id']
+    username = getUsername(author_id).lower()
+    title = 'Picturelock'
+    body = f'@{username} tagged you in a reply!'
+    post_id = record['post_id']
+
+    for tagged_user_id in tagged_user_ids:
+        if tagged_user_id != author_id:
+            token = getAuthorNotifToken(tagged_user_id)
+            if token:
+                send_notification(title, body, token, author_id, post_id, "post")
+
 def send_slack_message(event_type: str, category: str, username: str, post_id: str = None, list_id: str = None):
     try:
         print("reach 1")
@@ -139,10 +240,12 @@ def handle_tags_update(event_payload):
     print('Received update in post tags:')
     print(event_payload)
 
+
     record = event_payload['record']
-    tags_str = record.get('tags', '[]')
+
     try:
-        tagged_user_ids = json.loads(tags_str)
+        content = record.get('content', '[]')
+        tagged_user_ids = re.findall(r'@<([^>]+)>', content)
     except json.JSONDecodeError:
         print('Failed to parse tags.')
         return
@@ -395,6 +498,22 @@ def setup_likes_subscription(SUPABASE_ID, SUPABASE_KEY):
                 channel_posts_for_tags = s.set_channel("realtime:public:posts")
                 channel_posts_for_tags.join().on(
                     "INSERT", lambda payload: executor.submit(handle_tags_update, payload))
+                
+                channel_posts_for_tags = s.set_channel("realtime:public:comments")
+                channel_posts_for_tags.join().on(
+                    "INSERT", lambda payload: executor.submit(handle_comment_tags_update, payload))
+                
+                channel_posts_for_tags = s.set_channel("realtime:public:replies")
+                channel_posts_for_tags.join().on(
+                    "INSERT", lambda payload: executor.submit(handle_reply_tags_update, payload))
+                
+                channel_posts_for_tags = s.set_channel("realtime:public:listcomments")
+                channel_posts_for_tags.join().on(
+                    "INSERT", lambda payload: executor.submit(handle_list_comment_tags_update, payload))
+                
+                channel_posts_for_tags = s.set_channel("realtime:public:listreplies")
+                channel_posts_for_tags.join().on(
+                    "INSERT", lambda payload: executor.submit(handle_list_reply_tags_update, payload))
 
                 channel_support = s.set_channel("realtime:public:Support")
                 channel_support.join().on(