added tag notifications

main.py

@@ -92,6 +92,30 @@ 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_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)
+    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 post!'
+    post_id = record['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_listlikes_update(event_payload):
     print('Received update in listlikes:')
@@ -325,6 +349,10 @@ def setup_likes_subscription(SUPABASE_ID, SUPABASE_KEY):
                 channel_posts.join().on(
                     "INSERT", lambda payload: executor.submit(handle_groupreviewrequests_update, payload))
 
+                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))
+
                 s.listen()
 
             except Exception as e: