From f6e5d5ca4f9b45bbba5b2e54107e986134b16eb0 Mon Sep 17 00:00:00 2001
From: Marco Crapts <mcrapts@gmail.com>
Date: Thu, 25 May 2023 22:29:40 +0200
Subject: [PATCH] delete notes & sync to firebase

---
 src/App.vue                                | 43 ------------------
 src/components/Search/SearchBar.vue        |  5 +--
 src/components/SideBar.vue                 |  2 +-
 src/components/SideBar/SideBarMenuItem.vue |  2 +-
 src/composables/useEncryption.ts           | 17 ++++++-
 src/composables/useNotes.ts                | 52 +++++++++++++++-------
 src/types.d.ts                             |  2 +-
 src/utils/crypto.ts                        |  4 ++
 src/utils/defaultNotes.ts                  |  1 +
 tailwind.config.ts                         |  2 +-
 10 files changed, 64 insertions(+), 66 deletions(-)

diff --git a/src/App.vue b/src/App.vue
index 7de5931..3b9ca21 100644
--- a/src/App.vue
+++ b/src/App.vue
@@ -89,49 +89,6 @@ provide('loading', loading)
         <Mindmap v-else-if="activeViewMode.name === 'Mindmap'" />
       </template>
       <SkeletonNote v-else />
-      <!-- <div>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-        <p>bla</p>
-      </div> -->
     </main>
   </div>
   <Modal :open="passphraseRequired" :persistent="true">
diff --git a/src/components/Search/SearchBar.vue b/src/components/Search/SearchBar.vue
index 2a5aefb..699afe1 100644
--- a/src/components/Search/SearchBar.vue
+++ b/src/components/Search/SearchBar.vue
@@ -20,8 +20,7 @@ const results = computed<Note[]>(() => {
   return query.value ? findNotes(query.value) : notes.value
 })
 
-const goToNote = (note: Note, element?: HTMLElement) => {
-  console.log(element)
+const goToNote = (note: Note) => {
   setActiveNote(note.id)
   active.value = false
   if (queryElem.value) queryElem.value.blur()
@@ -75,7 +74,7 @@ const resultsRefs = ref<InstanceType<typeof SearchResult>[]>([])
               :key="result.id"
               :result="result"
               :active-result="activeResult"
-              @go-to-note="(element) => goToNote(result, element)"
+              @go-to-note="goToNote(result)"
               ref="resultsRefs"
             />
           </template>
diff --git a/src/components/SideBar.vue b/src/components/SideBar.vue
index f32f334..e30e56c 100644
--- a/src/components/SideBar.vue
+++ b/src/components/SideBar.vue
@@ -28,7 +28,7 @@ const setViewMode = (viewMode: ViewMode) => {
 <template>
   <div
     id="sidebar"
-    class="fixed bottom-0 top-0 flex max-sm:w-sidebar-mobile sm:w-sidebar flex-col gap-4 overflow-y-auto px-2 py-3 text-[90%] max-sm:text-[120%]"
+    class="fixed bottom-0 top-0 flex flex-col gap-4 overflow-y-auto px-2 py-3 text-[90%] max-sm:w-sidebar-mobile max-sm:gap-6 max-sm:text-[110%] sm:w-sidebar"
   >
     <SideBarMenu>
       <template #header>Root note</template>
diff --git a/src/components/SideBar/SideBarMenuItem.vue b/src/components/SideBar/SideBarMenuItem.vue
index f83fd71..0048c51 100644
--- a/src/components/SideBar/SideBarMenuItem.vue
+++ b/src/components/SideBar/SideBarMenuItem.vue
@@ -6,7 +6,7 @@ const props = defineProps<{
 </script>
 <template>
   <a
-    class="mt-1 block w-full cursor-pointer overflow-x-hidden text-ellipsis whitespace-nowrap rounded hover:bg-gray-200 active:bg-primary active:text-primary-content"
+    class="max-sm:mt-2 mt-1 block w-full cursor-pointer overflow-x-hidden text-ellipsis whitespace-nowrap rounded hover:bg-gray-200 active:bg-primary active:text-primary-content"
     :class="props.active ? 'font-bold text-primary' : 'text-secondary'"
   >
     <i :class="props.icon" class="mr-2" v-if="props.icon"></i>
diff --git a/src/composables/useEncryption.ts b/src/composables/useEncryption.ts
index 81b4bc3..88d9389 100644
--- a/src/composables/useEncryption.ts
+++ b/src/composables/useEncryption.ts
@@ -1,6 +1,6 @@
 import { doc, getDoc } from 'firebase/firestore'
 import { user, db } from '@/composables/useFirebase'
-import { decrypt, calculateClientKey } from '@/utils/crypto'
+import { decrypt, encrypt, calculateClientKey } from '@/utils/crypto'
 
 function getClientKeysFromLocalStorage(): { [uid: string]: string } {
   try {
@@ -76,3 +76,18 @@ export const decryptNotes = (notes: BaseNotes, encryptionKey: EncryptionKey) =>
   )
   return decryptedNotes
 }
+
+const encryptNote = (note: BaseNote, key: EncryptionKey) => {
+  return {
+    ...note,
+    title: encrypt(note.title, key),
+    content: encrypt(note.content, key)
+  }
+}
+
+export const encryptNotes = (notes: BaseNotes, encryptionKey: EncryptionKey) => {
+  const encryptedNotes = Object.fromEntries(
+    Object.entries(notes).map(([noteId, note]) => [noteId, { ...encryptNote(note, encryptionKey) }])
+  )
+  return encryptedNotes
+}
diff --git a/src/composables/useNotes.ts b/src/composables/useNotes.ts
index f6de2e7..a3847b7 100644
--- a/src/composables/useNotes.ts
+++ b/src/composables/useNotes.ts
@@ -1,9 +1,9 @@
 import shortid from 'shortid'
 import { useTitle } from '@vueuse/core'
-import { doc, getDoc } from 'firebase/firestore'
+import { doc, getDoc, updateDoc, deleteField } from 'firebase/firestore'
 import { viewModes, activeViewMode } from '@/composables/useViewMode'
 import { initialized, user, db } from '@/composables/useFirebase'
-import { decryptNotes, encryptionKey } from '@/composables/useEncryption'
+import { decryptNotes, encryptNotes, encryptionKey } from '@/composables/useEncryption'
 import { defaultNotes } from '@/utils/defaultNotes'
 import { mdToHtml } from '@/utils/markdown'
 import { getAllMatches } from '@/utils/helpers'
@@ -39,18 +39,37 @@ watchEffect(() => {
   activeNotesSource.value = getSource()
 })
 
-const baseNotes = ref<{ [noteId: string]: BaseNote }>({})
+const baseNotes = ref<BaseNotes>({})
 watch(
   baseNotes,
-  () => {
+  async (newBaseNotes, oldBaseNotes) => {
     if (!activeNotesSource.value || Object.keys(baseNotes.value).length === 0) return
+    console.log()
+
     if (activeNotesSource.value === 'local') {
-      console.log('sync with local')
       localStorage.setItem('notes', JSON.stringify(baseNotes.value))
     } else if (activeNotesSource.value === 'firebase') {
-      console.log('sync with firebase')
+      if (!db.value) throw Error("Database undefined, can't sync to Firebase")
+      if (!user.value) throw Error("User undefined, can't sync to Firebase")
+      const notes = encryptionKey.value
+        ? encryptNotes(baseNotes.value, encryptionKey.value)
+        : baseNotes.value
+
+      const notesToDelete = Object.keys(oldBaseNotes).filter(
+        (x) => !Object.keys(newBaseNotes).includes(x)
+      )
+      try {
+        const docRef = doc(db.value, 'pages', user.value.uid)
+        await Promise.all(
+          notesToDelete.map((noteId: string) => {
+            return updateDoc(docRef, { [noteId]: deleteField() })
+          })
+        )
+        await updateDoc(docRef, notes)
+      } catch (error: any) {
+        console.error(error)
+      }
     }
-    console.log(`Sync base notes with ${activeNotesSource.value}`, baseNotes.value)
   },
   { deep: true }
 )
@@ -144,6 +163,7 @@ export const addNote = (title: string, content: string, goToNote: boolean = fals
     id,
     title,
     content,
+    isRoot: false,
     created: new Date().getTime(),
     modified: new Date().getTime()
   }
@@ -153,7 +173,9 @@ export const addNote = (title: string, content: string, goToNote: boolean = fals
 }
 
 export const deleteNote = (noteId: string) => {
-  delete baseNotes.value[noteId]
+  const baseNotesClone: BaseNotes = structuredClone(toRaw(baseNotes.value))
+  delete baseNotesClone[noteId]
+  baseNotes.value = baseNotesClone
 }
 
 const getNoteLinksByNoteId = (noteId: string): string[] => {
@@ -213,9 +235,9 @@ const parseBaseNotes = (notes: BaseNotes): BaseNotes => {
           id: noteId,
           title: note.title,
           content: note.content,
-          created: note.created,
-          modified: note.modified,
-          isRoot: note.isRoot
+          isRoot: Boolean(note.isRoot),
+          created: note.created || note.modified || new Date().getTime(),
+          modified: note.modified || note.created || new Date().getTime()
         }
       ]
     })
@@ -235,10 +257,10 @@ watch(
         console.log(error)
       }
     } else if (activeNotesSource.value === 'firebase' && db.value) {
-      if (encryptionKey.value === undefined) return
-      const firebaseNotes = (
-        await getDoc(doc(db.value, 'pages', user.value?.uid || ''))
-      ).data() as { [noteId: string]: any }
+      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)
     }
diff --git a/src/types.d.ts b/src/types.d.ts
index d865b2f..8bb3103 100644
--- a/src/types.d.ts
+++ b/src/types.d.ts
@@ -5,7 +5,7 @@ declare global {
     content: string
     created: number
     modified: number
-    isRoot?: boolean
+    isRoot: boolean
   }
 
   interface Note extends BaseNote {
diff --git a/src/utils/crypto.ts b/src/utils/crypto.ts
index 8a9d122..6cd3d05 100644
--- a/src/utils/crypto.ts
+++ b/src/utils/crypto.ts
@@ -13,3 +13,7 @@ export const decrypt = (encryptedMessage: string, key: string): string => {
     throw new Error("Message doesn't have valid encryption")
   return decryptedMessage.substring(encryptionPrefix.length)
 }
+
+export const encrypt = (unencryptedMessage: string, key: string): string => {
+  return CryptoJS.AES.encrypt(encryptionPrefix + unencryptedMessage, key).toString()
+}
diff --git a/src/utils/defaultNotes.ts b/src/utils/defaultNotes.ts
index cf88dc8..c47a164 100644
--- a/src/utils/defaultNotes.ts
+++ b/src/utils/defaultNotes.ts
@@ -14,6 +14,7 @@ Let's get started!`,
   {
     title: 'brackets',
     content: `If you type square brackets around text a link is created automatically. Like magic!`,
+    isRoot: false
   },
 ].map((note) => ({
   id: shortid.generate(),
diff --git a/tailwind.config.ts b/tailwind.config.ts
index 3705e94..564e585 100644
--- a/tailwind.config.ts
+++ b/tailwind.config.ts
@@ -1,7 +1,7 @@
 const colors = require('tailwindcss/colors')
 const appWidth = '1280px'
 const sideBarWidth = '220px'
-const sideBarWidthMobile ='320px'
+const sideBarWidthMobile ='260px'
 const primary = '#1E4BC4'
 const secondary = colors.gray[500]