From 8b3f1aa14b1f7240da5dde965655347f86c73f6d Mon Sep 17 00:00:00 2001
From: padreug <padreug@aiolabs.dev>
Date: Wed, 12 Feb 2025 01:42:04 +0100
Subject: [PATCH] improve subscribe to messages

---
 dev-dist/sw.js                 |  2 +-
 src/components/SupportChat.vue | 21 ++++++++++++++++++++-
 src/stores/nostr.ts            | 26 ++++++++++++++++++++++++--
 3 files changed, 45 insertions(+), 4 deletions(-)

diff --git a/dev-dist/sw.js b/dev-dist/sw.js
index f02da09..a0afe47 100644
--- a/dev-dist/sw.js
+++ b/dev-dist/sw.js
@@ -79,7 +79,7 @@ define(['./workbox-54d0af47'], (function (workbox) { 'use strict';
    */
   workbox.precacheAndRoute([{
     "url": "index.html",
-    "revision": "0.d4m96v8cgs"
+    "revision": "0.i6hkd1cnpb"
   }], {});
   workbox.cleanupOutdatedCaches();
   workbox.registerRoute(new workbox.NavigationRoute(workbox.createHandlerBoundToURL("index.html"), {
diff --git a/src/components/SupportChat.vue b/src/components/SupportChat.vue
index 05ddd99..e09473c 100644
--- a/src/components/SupportChat.vue
+++ b/src/components/SupportChat.vue
@@ -1,5 +1,5 @@
 <script setup lang="ts">
-import { ref, computed, nextTick, onMounted, watch } from 'vue'
+import { ref, computed, nextTick, onMounted, watch, onUnmounted } from 'vue'
 import { useNostrStore } from '@/stores/nostr'
 import { npubToHex } from '@/lib/nostr'
 import type { DirectMessage } from '@/types/nostr'
@@ -79,6 +79,25 @@ onMounted(async () => {
   }
 })
 
+// Add cleanup on unmount
+onUnmounted(() => {
+  if (nostrStore.activeChat) {
+    nostrStore.unsubscribeFromMessages()
+  }
+})
+
+// Watch for changes in activeChat
+watch(() => nostrStore.activeChat, async (newChat) => {
+  if (newChat) {
+    try {
+      await nostrStore.subscribeToMessages()
+    } catch (err) {
+      console.error('Failed to subscribe to messages:', err)
+      error.value = 'Failed to connect to chat. Please try again later.'
+    }
+  }
+})
+
 function scrollToBottom() {
   if (messagesEndRef.value) {
     messagesEndRef.value.scrollIntoView({ behavior: 'smooth' })
diff --git a/src/stores/nostr.ts b/src/stores/nostr.ts
index 29d93be..002c696 100644
--- a/src/stores/nostr.ts
+++ b/src/stores/nostr.ts
@@ -89,6 +89,7 @@ export const useNostrStore = defineStore('nostr', () => {
   const activeChat = ref<string | null>(null)
   const relayPool = ref<any[]>([])
   const processedMessageIds = ref(new Set<string>())
+  const currentSubscription = ref<any | null>(null)
 
   // Watch account changes and persist to localStorage
   watch(account, (newAccount) => {
@@ -245,9 +246,14 @@ export const useNostrStore = defineStore('nostr', () => {
     await publishEvent(event, account.value.relays)
   }
 
-  const subscribeToMessages = async () => {
+  async function subscribeToMessages() {
     if (!account.value) return
 
+    // Cleanup existing subscription if any
+    if (currentSubscription.value) {
+      unsubscribeFromMessages()
+    }
+
     // Filter for received messages with history
     const receivedFilter = {
       kinds: [4],
@@ -349,6 +355,14 @@ export const useNostrStore = defineStore('nostr', () => {
             resolve(true)
           }
         })
+
+        // Store subscriptions for cleanup
+        currentSubscription.value = {
+          unsub: () => {
+            receivedSub.unsub()
+            sentSub.unsub()
+          }
+        }
       })
     }
 
@@ -356,6 +370,13 @@ export const useNostrStore = defineStore('nostr', () => {
     await Promise.all(relayPool.value.map(relay => subscribeToRelay(relay)))
   }
 
+  function unsubscribeFromMessages() {
+    if (currentSubscription.value) {
+      currentSubscription.value.unsub()
+      currentSubscription.value = null
+    }
+  }
+
   return {
     account,
     profiles,
@@ -367,6 +388,7 @@ export const useNostrStore = defineStore('nostr', () => {
     login,
     logout,
     sendMessage,
-    subscribeToMessages
+    subscribeToMessages,
+    unsubscribeFromMessages
   }
 }) 
\ No newline at end of file