From 3611884bc530b25670e09d2b6887a829210fc1f5 Mon Sep 17 00:00:00 2001
From: Simon Cambier <simon.cambier@protonmail.com>
Date: Thu, 27 Jun 2024 21:25:06 +0200
Subject: [PATCH] #349 - Added a "danger" setting to disable the cache
 killswitch

---
 src/main.ts     | 17 ++++++++++++-----
 src/settings.ts | 28 +++++++++++++++++++++++-----
 2 files changed, 35 insertions(+), 10 deletions(-)

diff --git a/src/main.ts b/src/main.ts
index 4817945..b68c195 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -137,7 +137,9 @@ export default class OmnisearchPlugin extends Plugin {
         })
       )
 
-      this.refreshIndexCallback = this.notesIndexer.refreshIndex.bind(this.notesIndexer)
+      this.refreshIndexCallback = this.notesIndexer.refreshIndex.bind(
+        this.notesIndexer
+      )
       addEventListener('blur', this.refreshIndexCallback)
       removeEventListener
 
@@ -263,15 +265,20 @@ export default class OmnisearchPlugin extends Plugin {
       indexingStep.set(IndexingStepType.WritingCache)
 
       // Disable settings.useCache while writing the cache, in case it freezes
-      this.settings.useCache = false
-      await saveSettings(this)
+      const cacheEnabled = this.settings.useCache
+      if (cacheEnabled && !this.settings.DANGER_forceSaveCache) {
+        this.settings.useCache = false
+        await saveSettings(this)
+      }
 
       // Write the cache
       await searchEngine.writeToCache()
 
       // Re-enable settings.caching
-      this.settings.useCache = true
-      await saveSettings(this)
+      if (cacheEnabled) {
+        this.settings.useCache = true
+        await saveSettings(this)
+      }
     }
 
     console.timeEnd('Omnisearch - Indexing total time')
diff --git a/src/settings.ts b/src/settings.ts
index 8f5fe31..303c5e9 100644
--- a/src/settings.ts
+++ b/src/settings.ts
@@ -11,7 +11,7 @@ import {
 import { writable } from 'svelte/store'
 import { K_DISABLE_OMNISEARCH } from './globals'
 import type OmnisearchPlugin from './main'
-import { enablePrintDebug } from "./tools/utils";
+import { enablePrintDebug } from './tools/utils'
 
 interface WeightingSettings {
   weightBasename: number
@@ -71,6 +71,7 @@ export interface OmnisearchSettings extends WeightingSettings {
   httpApiNotice: boolean
 
   DANGER_httpHost: string | null
+  DANGER_forceSaveCache: boolean
 }
 
 /**
@@ -139,9 +140,7 @@ export class SettingsTab extends PluginSettingTab {
       span.innerHTML = `Omnisearch will use Text Extractor to index the content of your PDFs.`
     })
     new Setting(containerEl)
-      .setName(
-        `PDFs content indexing ${textExtractor ? '' : '⚠️ Disabled'}`
-      )
+      .setName(`PDFs content indexing ${textExtractor ? '' : '⚠️ Disabled'}`)
       .setDesc(indexPDFsDesc)
       .addToggle(toggle =>
         toggle.setValue(settings.PDFIndexing).onChange(async v => {
@@ -663,7 +662,7 @@ export class SettingsTab extends PluginSettingTab {
         })
       )
 
-      new Setting(containerEl)
+    new Setting(containerEl)
       .setName('Ignore Arabic diacritics (beta)')
       .setDesc(diacriticsDesc)
       .addToggle(toggle =>
@@ -695,6 +694,23 @@ export class SettingsTab extends PluginSettingTab {
         })
       )
 
+    // Force save cache
+    const forceSaveCacheDesc = new DocumentFragment()
+    forceSaveCacheDesc.createSpan({}, span => {
+      span.innerHTML = `Omnisearch has a security feature that automatically disables cache writing if it cannot fully perform the operation.<br>
+        Use this option to force the cache to be saved, even if it causes a crash.<br>
+        ⚠️ <span style="color: var(--text-accent)">Enabling this setting could lead to crash loops</span>`
+    })
+    new Setting(containerEl)
+      .setName('Force save the cache')
+      .setDesc(forceSaveCacheDesc)
+      .addToggle(toggle =>
+        toggle.setValue(settings.DANGER_forceSaveCache).onChange(async v => {
+          settings.DANGER_forceSaveCache = v
+          await saveSettings(this.plugin)
+        })
+      )
+
     // Clear cache data
     if (isCacheEnabled()) {
       const resetCacheDesc = new DocumentFragment()
@@ -713,6 +729,7 @@ export class SettingsTab extends PluginSettingTab {
           })
         })
     }
+
     //#endregion Danger Zone
   }
 
@@ -769,6 +786,7 @@ export function getDefaultSettings(app: App): OmnisearchSettings {
     verboseLogging: false,
 
     DANGER_httpHost: null,
+    DANGER_forceSaveCache: false,
   }
 }