From ea8bde34889888021d05482ce45e802d99ce9a00 Mon Sep 17 00:00:00 2001
From: trickypr <23250792+trickypr@users.noreply.github.com>
Date: Mon, 13 Nov 2023 14:11:16 +1100
Subject: [PATCH] :construction: Begin work on settings page for search engines

Sync with laptop
---
 src/content/settings/Settings.svelte          |  7 ++++
 src/content/shared/search/constants.ts        |  8 +++++
 .../{browser/lib => shared}/search/search.ts  | 33 ++++++++++++++-----
 3 files changed, 39 insertions(+), 9 deletions(-)
 create mode 100644 src/content/shared/search/constants.ts
 rename src/content/{browser/lib => shared}/search/search.ts (73%)

diff --git a/src/content/settings/Settings.svelte b/src/content/settings/Settings.svelte
index 1cd33a1..76204ce 100644
--- a/src/content/settings/Settings.svelte
+++ b/src/content/settings/Settings.svelte
@@ -3,6 +3,7 @@
    - file, You can obtain one at http://mozilla.org/MPL/2.0/. -->
 
 <script lang="ts">
+  import { SEARCH_ENGINE_PREF } from '../shared/search/constants'
   import Category from './components/Category.svelte'
   import SubCategory from './components/SubCategory.svelte'
   import StringPref from './components/pref/StringPref.svelte'
@@ -21,6 +22,12 @@
     </SubCategory>
   </Category>
 
+  <Category bind:sidebarItems title="Search">
+    <SubCategory title="Default Search Engine">
+      <StringPref pref={SEARCH_ENGINE_PREF}>Default search engine</StringPref>
+    </SubCategory>
+  </Category>
+
   <Category bind:sidebarItems title="Keybinds">
     <SubCategory title="Browser">
       <StringPref pref="browser.keybinds.toolbox">
diff --git a/src/content/shared/search/constants.ts b/src/content/shared/search/constants.ts
new file mode 100644
index 0000000..4d1de00
--- /dev/null
+++ b/src/content/shared/search/constants.ts
@@ -0,0 +1,8 @@
+export const DEFAULT_SEARCH_ENGINE: (typeof SEARCH_ENGINE_IDS)[number] =
+  'google@search.fushra.com'
+export const SEARCH_ENGINE_PREF = 'browser.search.engine.default'
+
+export const SEARCH_ENGINE_IDS = [
+  'ddg@search.fushra.com',
+  'google@search.fushra.com',
+] as const
diff --git a/src/content/browser/lib/search/search.ts b/src/content/shared/search/search.ts
similarity index 73%
rename from src/content/browser/lib/search/search.ts
rename to src/content/shared/search/search.ts
index ed39f6d..f9e1db1 100644
--- a/src/content/browser/lib/search/search.ts
+++ b/src/content/shared/search/search.ts
@@ -3,22 +3,24 @@
  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
 
 import type { AddonSearchEngine } from 'resource://gre/modules/AddonSearchEngine.sys.mjs'
-import { Deferred } from '../Deferred'
-import { lazyESModuleGetters } from '../../../shared/TypedImportUtilities'
+import { Deferred } from '../../browser/lib/Deferred'
+import { lazyESModuleGetters } from '../TypedImportUtilities'
+import {
+  SEARCH_ENGINE_IDS,
+  SEARCH_ENGINE_PREF,
+  DEFAULT_SEARCH_ENGINE,
+} from './constants'
 
 const lazy = lazyESModuleGetters({
   AddonSearchEngine: 'resource://gre/modules/AddonSearchEngine.sys.mjs',
   SearchUtils: 'resource://gre/modules/SearchUtils.sys.mjs',
 })
 
-const SEARCH_ENGINE_IDS = [
-  'ddg@search.fushra.com',
-  'google@search.fushra.com',
-] as const
-
 class SearchService {
   private initPromise: Deferred<void> | undefined
+
   private searchEngines: Deferred<AddonSearchEngine[]> = new Deferred()
+  private defaultEngine: Deferred<AddonSearchEngine> = new Deferred()
 
   constructor() {}
 
@@ -45,7 +47,6 @@ class SearchService {
 
     const locale = lazy.SearchUtils.DEFAULT_TAG
     const engines: AddonSearchEngine[] = []
-
     for (const extensionID of SEARCH_ENGINE_IDS) {
       const engine = new (lazy.AddonSearchEngine as any)({
         isAppProvided: true,
@@ -55,8 +56,17 @@ class SearchService {
 
       engines.push(engine)
     }
-
     this.searchEngines.resolve(engines)
+
+    const defaultEngineId = Services.prefs.getStringPref(
+      SEARCH_ENGINE_PREF,
+      DEFAULT_SEARCH_ENGINE,
+    )
+    const defaultEngine = engines.find(
+      (engine: any) => engine._extensionID === defaultEngineId,
+    )!
+    this.defaultEngine.resolve(defaultEngine)
+
     this.initPromise.resolve()
   }
 
@@ -64,6 +74,11 @@ class SearchService {
     await this.init()
     return await this.searchEngines.promise
   }
+
+  async getDefaultEngine() {
+    await this.init()
+    return await this.defaultEngine.promise
+  }
 }
 
 export const searchService = new SearchService()