Hotfix for uploading avatar images for new users

fingerthief · Jun 24, 2024 · fc42525 · fc42525
1 parent 379d0ba
commit fc42525
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 18 deletions.
diff --git a/src/components/configuration-sections/GeneralConfigSection.vue b/src/components/configuration-sections/GeneralConfigSection.vue
@@ -104,15 +104,28 @@ const isSavedPromptsOpen = ref(true);
 
 const fetchStoredFiles = async () => {
     try {
-        const request = indexedDB.open('UserFilesDB', 5);
+        const dbName = 'UserFilesDB';
+        const dbVersion = 5;
+        const storeName = 'userFiles';
 
         const db = await new Promise((resolve, reject) => {
+            const request = indexedDB.open(dbName, dbVersion);
+
+            request.onerror = (event) => reject(`IndexedDB error: ${event.target.error}`);
+
             request.onsuccess = (event) => resolve(event.target.result);
-            request.onerror = reject;
+
+            request.onupgradeneeded = (event) => {
+                const db = event.target.result;
+                if (!db.objectStoreNames.contains(storeName)) {
+                    db.createObjectStore(storeName, { keyPath: 'fileName' });
+                    console.log(`Created new object store: ${storeName}`);
+                }
+            };
         });
 
-        const transaction = db.transaction(['userFiles'], 'readonly');
-        const store = transaction.objectStore('userFiles');
+        const transaction = db.transaction([storeName], 'readonly');
+        const store = transaction.objectStore(storeName);
         const getAllRequest = store.getAll();
 
         const result = await new Promise((resolve, reject) => {

diff --git a/src/components/dialogs/StoredFilesDialog.vue b/src/components/dialogs/StoredFilesDialog.vue
@@ -93,36 +93,43 @@ const closeStoredFiles = () => {
 
 const fetchStoredFiles = async () => {
     try {
-        const request = indexedDB.open('UserFilesDB', 5);
+        const dbName = 'UserFilesDB';
+        const dbVersion = 5;
+        const storeName = 'userFiles';
 
-        request.onupgradeneeded = (event) => {
-            const db = event.target.result;
-            if (!db.objectStoreNames.contains('userFiles')) {
-                db.createObjectStore('userFiles', { keyPath: 'id', autoIncrement: true });
-            }
-        };
+        const db = await new Promise((resolve, reject) => {
+            const request = indexedDB.open(dbName, dbVersion);
+
+            request.onerror = (event) => reject(`IndexedDB error: ${event.target.error}`);
+
+            request.onsuccess = (event) => resolve(event.target.result);
 
-        const event = await new Promise((resolve, reject) => {
-            request.onsuccess = resolve;
-            request.onerror = reject;
+            request.onupgradeneeded = (event) => {
+                const db = event.target.result;
+                if (!db.objectStoreNames.contains(storeName)) {
+                    db.createObjectStore(storeName, { keyPath: 'fileName' });
+                    console.log(`Created new object store: ${storeName}`);
+                }
+            };
         });
 
-        const db = event.target.result;
-        const transaction = db.transaction(['userFiles'], 'readonly');
-        const store = transaction.objectStore('userFiles');
+        const transaction = db.transaction([storeName], 'readonly');
+        const store = transaction.objectStore(storeName);
         const getAllRequest = store.getAll();
 
         const result = await new Promise((resolve, reject) => {
             getAllRequest.onsuccess = () => resolve(getAllRequest.result);
             getAllRequest.onerror = reject;
         });
 
-        return result;
+        return result.filter(file => file.fileType && file.fileType.startsWith('image/'));
     } catch (error) {
         console.error(`Error Fetching Stored Files: ${error}`);
+        return [];
     }
 };
 
+
 const addStoredFileToContext = (file) => {
     let messageContent;
     if (file.fileType.startsWith('image/')) {