From b3f740770a079168893489b6e599942fc7686217 Mon Sep 17 00:00:00 2001
From: Marco Crapts <mcrapts@gmail.com>
Date: Sun, 28 May 2023 22:23:41 +0200
Subject: [PATCH] slight refactoring

---
 src/App.vue                      |  6 +++--
 src/composables/useEncryption.ts |  4 +--
 src/composables/useNotes.ts      | 44 +++++++++++++-------------------
 3 files changed, 23 insertions(+), 31 deletions(-)

diff --git a/src/App.vue b/src/App.vue
index 17b0717..01c22c8 100644
--- a/src/App.vue
+++ b/src/App.vue
@@ -1,9 +1,10 @@
 <script setup lang="ts">
-import { activeNote, updateNote, notes, activeNotesSource } from '@/composables/useNotes'
+import { activeNote, updateNote, notes, activeNotesSource, getNotes } from '@/composables/useNotes'
 import { viewModes, activeViewMode } from '@/composables/useViewMode'
 import {
   getClientKey,
   getEncryptionKey,
+  encryptionKey,
   setClientKey,
   passphraseRequired
 } from '@/composables/useEncryption'
@@ -30,12 +31,13 @@ const firebaseAuthUI =
 provide('firebaseAuthUI', firebaseAuthUI)
 
 watch(
-  activeNotesSource,
+  [activeNotesSource, encryptionKey],
   () => {
     if (activeNotesSource.value === 'firebase') {
       getClientKey()
       getEncryptionKey()
     }
+    getNotes()
   },
   { immediate: true }
 )
diff --git a/src/composables/useEncryption.ts b/src/composables/useEncryption.ts
index d3c66de..c7ef641 100644
--- a/src/composables/useEncryption.ts
+++ b/src/composables/useEncryption.ts
@@ -59,9 +59,7 @@ async function getEncryptedEncryptionKey(): Promise<EncryptedEncryptionKey | voi
   return data?.key
 }
 
-async function setEncryptedEncryptionKey(
-  encryptedEncryptionKey: EncryptedEncryptionKey | null
-): Promise<void> {
+async function setEncryptedEncryptionKey(encryptedEncryptionKey: EncryptedEncryptionKey | null) {
   if (!user.value || !db.value) return
   const docRef = doc(db.value, 'encryptionKeys', user.value.uid)
   await setDoc(docRef, { key: encryptedEncryptionKey })
diff --git a/src/composables/useNotes.ts b/src/composables/useNotes.ts
index b188ce0..278026e 100644
--- a/src/composables/useNotes.ts
+++ b/src/composables/useNotes.ts
@@ -250,30 +250,22 @@ const parseBaseNotes = (notes: BaseNotes): BaseNotes => {
   )
 }
 
-watch(
-  [activeNotesSource, encryptionKey],
-  async () => {
-    if (!activeNotesSource.value) return
-    baseNotes.value = {}
-    let notes: BaseNotes = {}
-    if (activeNotesSource.value === 'local') {
-      try {
-        notes = JSON.parse(localStorage.getItem('notes') || '{}')
-      } catch (error) {
-        console.log(error)
-      }
-    } else if (activeNotesSource.value === 'firebase' && db.value) {
-      if (encryptionKey.value === undefined || !user.value) return
-      const firebaseNotes = (await getDoc(doc(db.value, 'pages', user.value.uid))).data() as {
-        [noteId: string]: any
-      }
-      notes = encryptionKey.value ? decryptNotes(firebaseNotes, encryptionKey.value) : firebaseNotes
-      console.log('get notes from firebase', notes)
+export const getNotes = async () => {
+  baseNotes.value = {}
+  let notes: BaseNotes = {}
+  if (activeNotesSource.value === 'local') {
+    try {
+      notes = JSON.parse(localStorage.getItem('notes') || '{}')
+    } catch (error) {
+      console.log(error)
     }
-    baseNotes.value = parseBaseNotes(notes)
-    if (!rootNote.value) insertDefaultNotes(defaultNotes)
-
-    setActiveNote(rootNote.value?.id)
-  },
-  { immediate: true }
-)
+  } else if (activeNotesSource.value === 'firebase') {
+    if (encryptionKey.value === undefined || !user.value || !db.value) return
+    const firebaseNotes = (await getDoc(doc(db.value, 'pages', user.value.uid))).data() as BaseNotes
+    notes = encryptionKey.value ? decryptNotes(firebaseNotes, encryptionKey.value) : firebaseNotes
+    console.log('get notes from firebase', notes)
+  }
+  baseNotes.value = parseBaseNotes(notes)
+  if (!rootNote.value) insertDefaultNotes(defaultNotes)
+  setActiveNote(rootNote.value?.id)
+}