summaryrefslogtreecommitdiffstats
path: root/browser/components/webextensions/ext-omnibox.js
blob: 9b2f60ca4541f365c348605fb9887ea9033c202d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim: set sts=2 sw=2 et tw=80: */
"use strict";

Cu.import("resource://gre/modules/ExtensionUtils.jsm");

XPCOMUtils.defineLazyModuleGetter(this, "ExtensionSearchHandler",
                                  "resource://gre/modules/ExtensionSearchHandler.jsm");
var {
  SingletonEventManager,
} = ExtensionUtils;

// WeakMap[extension -> keyword]
let gKeywordMap = new WeakMap();

/* eslint-disable mozilla/balanced-listeners */
extensions.on("manifest_omnibox", (type, directive, extension, manifest) => {
  let keyword = manifest.omnibox.keyword;
  try {
    // This will throw if the keyword is already registered.
    ExtensionSearchHandler.registerKeyword(keyword, extension);
    gKeywordMap.set(extension, keyword);
  } catch (e) {
    extension.manifestError(e.message);
  }
});

extensions.on("shutdown", (type, extension) => {
  let keyword = gKeywordMap.get(extension);
  if (keyword) {
    ExtensionSearchHandler.unregisterKeyword(keyword);
    gKeywordMap.delete(extension);
  }
});
/* eslint-enable mozilla/balanced-listeners */

extensions.registerSchemaAPI("omnibox", "addon_parent", context => {
  let {extension} = context;
  return {
    omnibox: {
      setDefaultSuggestion(suggestion) {
        let keyword = gKeywordMap.get(extension);
        try {
          // This will throw if the keyword failed to register.
          ExtensionSearchHandler.setDefaultSuggestion(keyword, suggestion);
        } catch (e) {
          return Promise.reject(e.message);
        }
      },

      onInputStarted: new SingletonEventManager(context, "omnibox.onInputStarted", fire => {
        let listener = (eventName) => {
          fire();
        };
        extension.on(ExtensionSearchHandler.MSG_INPUT_STARTED, listener);
        return () => {
          extension.off(ExtensionSearchHandler.MSG_INPUT_STARTED, listener);
        };
      }).api(),

      onInputCancelled: new SingletonEventManager(context, "omnibox.onInputCancelled", fire => {
        let listener = (eventName) => {
          fire();
        };
        extension.on(ExtensionSearchHandler.MSG_INPUT_CANCELLED, listener);
        return () => {
          extension.off(ExtensionSearchHandler.MSG_INPUT_CANCELLED, listener);
        };
      }).api(),

      onInputEntered: new SingletonEventManager(context, "omnibox.onInputEntered", fire => {
        let listener = (eventName, text, disposition) => {
          fire(text, disposition);
        };
        extension.on(ExtensionSearchHandler.MSG_INPUT_ENTERED, listener);
        return () => {
          extension.off(ExtensionSearchHandler.MSG_INPUT_ENTERED, listener);
        };
      }).api(),
    },

    omnibox_internal: {
      addSuggestions(id, suggestions) {
        let keyword = gKeywordMap.get(extension);
        try {
          ExtensionSearchHandler.addSuggestions(keyword, id, suggestions);
        } catch (e) {
          // Silently fail because the extension developer can not know for sure if the user
          // has already invalidated the callback when asynchronously providing suggestions.
        }
      },

      onInputChanged: new SingletonEventManager(context, "omnibox_internal.onInputChanged", fire => {
        let listener = (eventName, text, id) => {
          fire(text, id);
        };
        extension.on(ExtensionSearchHandler.MSG_INPUT_CHANGED, listener);
        return () => {
          extension.off(ExtensionSearchHandler.MSG_INPUT_CHANGED, listener);
        };
      }).api(),
    },
  };
});