1 files changed, 118 insertions, 0 deletions
diff --git a/browser/components/extensions/parent/ext-search.js b/browser/components/extensions/parent/ext-search.js
new file mode 100644
index 0000000000..be51f5a17a
--- /dev/null
+++ b/browser/components/extensions/parent/ext-search.js
@@ -0,0 +1,118 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
+/* vim: set sts=2 sw=2 et tw=80: */
+
+"use strict";
+
+var { ExtensionError } = ExtensionUtils;
+
+const dispositionMap = {
+  CURRENT_TAB: "current",
+  NEW_TAB: "tab",
+  NEW_WINDOW: "window",
+};
+
+this.search = class extends ExtensionAPI {
+  getAPI(context) {
+    function getTarget({ tabId, disposition, defaultDisposition }) {
+      let tab, where;
+      if (disposition) {
+        if (tabId) {
+          throw new ExtensionError(`Cannot set both 'disposition' and 'tabId'`);
+        }
+        where = dispositionMap[disposition];
+      } else if (tabId) {
+        tab = tabTracker.getTab(tabId);
+      } else {
+        where = dispositionMap[defaultDisposition];
+      }
+      return { tab, where };
+    }
+
+    return {
+      search: {
+        async get() {
+          await searchInitialized;
+          let visibleEngines = await Services.search.getVisibleEngines();
+          let defaultEngine = await Services.search.getDefault();
+          return Promise.all(
+            visibleEngines.map(async engine => {
+              let favIconUrl;
+              if (engine.iconURI) {
+                // Convert moz-extension:-URLs to data:-URLs to make sure that
+                // extensions can see icons from other extensions, even if they
+                // are not web-accessible.
+                // Also prevents leakage of extension UUIDs to other extensions..
+                if (
+                  engine.iconURI.schemeIs("moz-extension") &&
+                  engine.iconURI.host !== context.extension.uuid
+                ) {
+                  favIconUrl = await ExtensionUtils.makeDataURI(
+                    engine.iconURI.spec
+                  );
+                } else {
+                  favIconUrl = engine.iconURI.spec;
+                }
+              }
+
+              return {
+                name: engine.name,
+                isDefault: engine.name === defaultEngine.name,
+                alias: engine.alias || undefined,
+                favIconUrl,
+              };
+            })
+          );
+        },
+
+        async search(searchProperties) {
+          await searchInitialized;
+          let engine;
+
+          if (searchProperties.engine) {
+            engine = Services.search.getEngineByName(searchProperties.engine);
+            if (!engine) {
+              throw new ExtensionError(
+                `${searchProperties.engine} was not found`
+              );
+            }
+          }
+
+          let { tab, where } = getTarget({
+            tabId: searchProperties.tabId,
+            disposition: searchProperties.disposition,
+            defaultDisposition: "NEW_TAB",
+          });
+
+          await windowTracker.topWindow.BrowserSearch.loadSearchFromExtension({
+            query: searchProperties.query,
+            where,
+            engine,
+            tab,
+            triggeringPrincipal: context.principal,
+          });
+        },
+
+        async query(queryProperties) {
+          await searchInitialized;
+
+          let { tab, where } = getTarget({
+            tabId: queryProperties.tabId,
+            disposition: queryProperties.disposition,
+            defaultDisposition: "CURRENT_TAB",
+          });
+
+          await windowTracker.topWindow.BrowserSearch.loadSearchFromExtension({
+            query: queryProperties.text,
+            where,
+            tab,
+            triggeringPrincipal: context.principal,
+          });
+        },
+      },
+    };
+  }
+};