summaryrefslogtreecommitdiffstats
path: root/toolkit/jetpack/sdk/content/mod.js
blob: 81fe9ee42ed05f013aaef410a0bbb4e011e68e73 (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
/* 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/. */
"use strict";

module.metadata = {
  "stability": "experimental"
};

const { Ci } = require("chrome");
const { dispatcher } = require("../util/dispatcher");
const { add, remove, iterator } = require("../lang/weak-set");

var getTargetWindow = dispatcher("getTargetWindow");

getTargetWindow.define(function (target) {
  if (target instanceof Ci.nsIDOMWindow)
    return target;
  if (target instanceof Ci.nsIDOMDocument)
    return target.defaultView || null;

  return null;
});

exports.getTargetWindow = getTargetWindow;

var attachTo = dispatcher("attachTo");
exports.attachTo = attachTo;

var detachFrom = dispatcher("detatchFrom");
exports.detachFrom = detachFrom;

function attach(modification, target) {
  if (!modification)
    return;

  let window = getTargetWindow(target);

  attachTo(modification, window);

  // modification are stored per content; `window` reference can still be the
  // same even if the content is changed, therefore `document` is used instead.
  add(modification, window.document);
}
exports.attach = attach;

function detach(modification, target) {
  if (!modification)
    return;

  if (target) {
    let window = getTargetWindow(target);
    detachFrom(modification, window);
    remove(modification, window.document);
  }
  else {
    let documents = iterator(modification);
    for (let document of documents) {
      let window = document.defaultView;
      // The window might have already gone away
      if (!window)
        continue;
      detachFrom(modification, document.defaultView);
      remove(modification, document);
    }
  }
}
exports.detach = detach;