forked from mirrors/gecko-dev
104 lines
2.7 KiB
JavaScript
104 lines
2.7 KiB
JavaScript
/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
|
|
/* vim: set sts=2 sw=2 et tw=80: */
|
|
/* 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";
|
|
|
|
var { ExtensionError } = ExtensionUtils;
|
|
|
|
this.webRequest = class extends ExtensionAPI {
|
|
STREAM_FILTER_INACTIVE_STATUSES = ["closed", "disconnected", "failed"];
|
|
|
|
hasActiveStreamFilter(filtersWeakSet) {
|
|
const iter = ChromeUtils.nondeterministicGetWeakSetKeys(filtersWeakSet);
|
|
for (let filter of iter) {
|
|
if (!this.STREAM_FILTER_INACTIVE_STATUSES.includes(filter.status)) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
watchStreamFilterSuspendCancel({
|
|
context,
|
|
filters,
|
|
onSuspend,
|
|
onSuspendCanceled,
|
|
}) {
|
|
if (
|
|
!context.isBackgroundContext ||
|
|
context.extension.persistentBackground !== false
|
|
) {
|
|
return;
|
|
}
|
|
|
|
const { extension } = context;
|
|
const cancelSuspendOnActiveStreamFilter = () =>
|
|
this.hasActiveStreamFilter(filters);
|
|
context.callOnClose({
|
|
close() {
|
|
extension.off(
|
|
"internal:stream-filter-suspend-cancel",
|
|
cancelSuspendOnActiveStreamFilter
|
|
);
|
|
extension.off("background-script-suspend", onSuspend);
|
|
extension.off("background-script-suspend-canceled", onSuspend);
|
|
},
|
|
});
|
|
extension.on(
|
|
"internal:stream-filter-suspend-cancel",
|
|
cancelSuspendOnActiveStreamFilter
|
|
);
|
|
extension.on("background-script-suspend", onSuspend);
|
|
extension.on("background-script-suspend-canceled", onSuspendCanceled);
|
|
}
|
|
|
|
getAPI(context) {
|
|
let filters = new WeakSet();
|
|
|
|
context.callOnClose({
|
|
close() {
|
|
for (let filter of ChromeUtils.nondeterministicGetWeakSetKeys(
|
|
filters
|
|
)) {
|
|
try {
|
|
filter.disconnect();
|
|
} catch (e) {
|
|
// Ignore.
|
|
}
|
|
}
|
|
},
|
|
});
|
|
|
|
let isSuspending = false;
|
|
this.watchStreamFilterSuspendCancel({
|
|
context,
|
|
filters,
|
|
onSuspend: () => (isSuspending = true),
|
|
onSuspendCanceled: () => (isSuspending = false),
|
|
});
|
|
|
|
return {
|
|
webRequest: {
|
|
filterResponseData(requestId) {
|
|
if (isSuspending) {
|
|
throw new ExtensionError(
|
|
"filterResponseData method calls forbidden while background extension global is suspending"
|
|
);
|
|
}
|
|
requestId = parseInt(requestId, 10);
|
|
|
|
let streamFilter = context.cloneScope.StreamFilter.create(
|
|
requestId,
|
|
context.extension.id
|
|
);
|
|
|
|
filters.add(streamFilter);
|
|
return streamFilter;
|
|
},
|
|
},
|
|
};
|
|
}
|
|
};
|