From 6c73578b7f720ed0c107ba0bf0d09d5b4a8f89c4 Mon Sep 17 00:00:00 2001
From: Alexis <35.alexis.pele@gmail.com>
Date: Sun, 25 Aug 2024 14:18:11 +0200
Subject: [PATCH] Added subscriptions to world list

---
 pages/i/index.vue                   | 73 ++++++++++++++++++++++++++---
 supabase/migrations/202401_init.sql |  2 +
 2 files changed, 69 insertions(+), 6 deletions(-)

diff --git a/pages/i/index.vue b/pages/i/index.vue
index 12a6ab1..7416694 100644
--- a/pages/i/index.vue
+++ b/pages/i/index.vue
@@ -1,15 +1,18 @@
 <script lang="ts" setup>
-import type { World } from '~/models/World';
+import type { RealtimeChannel } from "@supabase/supabase-js"
+import type { World } from "~/models/World";
 
-const { data: res } = await useFetch('/api/worlds/query')
+const supabase = useSupabaseClient()
 
-const worlds = res.value?.data as World[]
+const { data: res } = await useFetch("/api/worlds/query")
+
+const worlds = ref<World[]>(res.value?.data as World[])
 
 useHead({
-  title: 'Profil'
+  title: "Profil"
 })
 definePageMeta({
-  middleware: ['auth-guard']
+  middleware: ["auth-guard"]
 })
 
 const user = useSupabaseUser()
@@ -17,9 +20,67 @@ const user = useSupabaseUser()
 // Redirect user back home when they log out on the page
 watch(user, (n, _o) => {
   if (!n) {
-    navigateTo('/')
+    navigateTo("/")
   }
 })
+
+/**
+ * === World subscriptions ===
+ */
+/** Active world channel */
+let worldChannel: RealtimeChannel
+
+/** Handles world insertion realtime events */
+function handleInsertedWorld(newWorld: World) {
+  try {
+    worlds.value.push(newWorld)
+  } catch (err) {
+    console.log(err)
+  }
+}
+
+/** Handles world deletion realtime events */
+function handleDeletedWorld(id: number) {
+  try {
+    worlds.value.splice(worlds.value.findIndex(w => w.id === id))
+  } catch (err) {
+    console.log(err)
+  }
+}
+
+onMounted(() => {
+  worldChannel = supabase.channel("custom-insert-channel")
+    .on(
+      "postgres_changes",
+      { event: "*", schema: "public", table: "worlds" },
+      async (payload) => {
+        switch (payload.eventType) {
+          case "INSERT":
+            handleInsertedWorld(payload.new as World)
+            break
+
+          case "DELETE":
+            handleDeletedWorld(payload.old.id)
+            break
+
+          case "UPDATE":
+            worlds.value = (await $fetch("/api/worlds/query")).data as World[]
+            break
+
+          default:
+            console.log("Unknown event has been triggered. This should not happen unless Supabase added one somehow.")
+            console.log(payload)
+            break
+        }
+      }
+    )
+    .subscribe()
+})
+
+onUnmounted(() => {
+  // Unsubscribe from realtime
+  supabase.removeChannel(worldChannel)
+})
 </script>
 
 <template>
diff --git a/supabase/migrations/202401_init.sql b/supabase/migrations/202401_init.sql
index 058a91c..ca78c9d 100644
--- a/supabase/migrations/202401_init.sql
+++ b/supabase/migrations/202401_init.sql
@@ -161,6 +161,8 @@ alter table public.calendar_event_categories enable row level security;
 alter table public.calendar_event_categories_links enable row level security;
 
 -- Add realtime
+alter publication supabase_realtime add table users;
+alter publication supabase_realtime add table worlds;
 alter publication supabase_realtime add table calendars;
 
 -- User policies