mirror of
https://codeberg.org/nobody/LocalCDN.git
synced 2024-12-25 09:40:38 +01:00
351 lines
8.9 KiB
JavaScript
351 lines
8.9 KiB
JavaScript
/**
|
||
* State Manager
|
||
* Belongs to LocalCDN (since 2020-02-26)
|
||
* (Origin: Decentraleyes)
|
||
*
|
||
* @author Thomas Rientjes
|
||
* @since 2017-03-10
|
||
*
|
||
* @author nobody
|
||
* @since 2020-02-26
|
||
*
|
||
* @license MPL 2.0
|
||
*
|
||
* 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';
|
||
|
||
/**
|
||
* State Manager
|
||
*/
|
||
|
||
var stateManager = {};
|
||
|
||
/**
|
||
* Public Methods
|
||
*/
|
||
|
||
stateManager.registerInjection = function (tabIdentifier, injection) {
|
||
|
||
let injectionIdentifier, registeredTab, injectionCount;
|
||
|
||
injectionIdentifier = injection.source + injection.path + injection.version;
|
||
registeredTab = stateManager.tabs[tabIdentifier];
|
||
|
||
registeredTab.injections[injectionIdentifier] = injection;
|
||
injectionCount = Object.keys(registeredTab.injections).length || 0;
|
||
|
||
if (injectionCount > 0) {
|
||
|
||
chrome.browserAction.setTitle({
|
||
'tabId': tabIdentifier,
|
||
'title': `LocalCDN (${injectionCount})`
|
||
});
|
||
|
||
if (stateManager.showIconBadge === true) {
|
||
|
||
wrappers.setBadgeText({
|
||
'tabId': tabIdentifier,
|
||
'text': injectionCount.toString()
|
||
});
|
||
}
|
||
}
|
||
|
||
if (isNaN(storageManager.amountInjected)) {
|
||
|
||
storageManager.type.get(Setting.AMOUNT_INJECTED, function (items) {
|
||
|
||
storageManager.amountInjected = items.amountInjected;
|
||
|
||
storageManager.type.set({
|
||
[Setting.AMOUNT_INJECTED]: ++storageManager.amountInjected
|
||
});
|
||
});
|
||
|
||
} else {
|
||
|
||
chrome.storage.local.set({
|
||
[Setting.AMOUNT_INJECTED]: ++storageManager.amountInjected
|
||
});
|
||
}
|
||
if (stateManager.internalStatistics) {
|
||
stats.setStats(injection);
|
||
}
|
||
};
|
||
|
||
stateManager.addDomainToWhitelist = function (domain) {
|
||
|
||
return new Promise((resolve) => {
|
||
|
||
let whitelistedDomains = requestAnalyzer.whitelistedDomains;
|
||
whitelistedDomains[domain] = true;
|
||
|
||
storageManager.type.set({whitelistedDomains}, resolve);
|
||
});
|
||
};
|
||
|
||
stateManager.removeDomainFromWhitelist = function (domain) {
|
||
|
||
return new Promise((resolve) => {
|
||
|
||
let whitelistedDomains = requestAnalyzer.whitelistedDomains;
|
||
delete whitelistedDomains[domain];
|
||
|
||
storageManager.type.set({whitelistedDomains}, resolve);
|
||
});
|
||
};
|
||
|
||
stateManager.addDomainToManipulateDOMlist = function (domain) {
|
||
|
||
return new Promise((resolve) => {
|
||
|
||
let domainsManipulateDOM = requestAnalyzer.domainsManipulateDOM;
|
||
domainsManipulateDOM[domain] = true;
|
||
|
||
storageManager.type.set({domainsManipulateDOM}, resolve);
|
||
});
|
||
};
|
||
|
||
stateManager.removeDomainFromManipulateDOMlist = function (domain) {
|
||
|
||
return new Promise((resolve) => {
|
||
|
||
let domainsManipulateDOM = requestAnalyzer.domainsManipulateDOM;
|
||
delete domainsManipulateDOM[domain];
|
||
|
||
storageManager.type.set({domainsManipulateDOM}, resolve);
|
||
});
|
||
};
|
||
|
||
|
||
/**
|
||
* Private Methods
|
||
*/
|
||
|
||
stateManager._createTab = function (tab) {
|
||
|
||
let tabIdentifier, requestFilters;
|
||
|
||
tabIdentifier = tab.id;
|
||
|
||
stateManager.tabs[tabIdentifier] = {
|
||
'injections': {}
|
||
};
|
||
|
||
requestFilters = {
|
||
'tabId': tabIdentifier,
|
||
'urls': stateManager.validHosts
|
||
};
|
||
|
||
chrome.webRequest.onBeforeRequest.addListener(function (requestDetails) {
|
||
|
||
tab = stateManager.tabs[tabIdentifier].details || {};
|
||
return interceptor.handleRequest(requestDetails, tabIdentifier, tab);
|
||
|
||
}, requestFilters, [WebRequest.BLOCKING]);
|
||
};
|
||
|
||
stateManager._removeTab = function (tabIdentifier) {
|
||
delete stateManager.tabs[tabIdentifier];
|
||
};
|
||
|
||
stateManager._updateTab = function (details) {
|
||
|
||
let tabDomain, domainIsWhitelisted, frameIdentifier, tabIdentifier;
|
||
|
||
tabDomain = helpers.extractDomainFromUrl(details.url, true);
|
||
domainIsWhitelisted = stateManager._domainIsListed(tabDomain);
|
||
|
||
frameIdentifier = details.frameId;
|
||
tabIdentifier = details.tabId;
|
||
|
||
if (frameIdentifier !== 0 || tabIdentifier === -1) {
|
||
return;
|
||
}
|
||
|
||
chrome.browserAction.setTitle({
|
||
'tabId': tabIdentifier,
|
||
'title': 'LocalCDN (0)'
|
||
});
|
||
|
||
if (domainIsWhitelisted) {
|
||
|
||
stateManager._setIconDisabled(tabIdentifier);
|
||
|
||
chrome.browserAction.setTitle({
|
||
'tabId': tabIdentifier,
|
||
'title': 'LocalCDN (–)'
|
||
});
|
||
}
|
||
|
||
if (stateManager.showIconBadge === true) {
|
||
stateManager._clearBadgeText(tabIdentifier);
|
||
}
|
||
|
||
if (stateManager.tabs[tabIdentifier]) {
|
||
stateManager.tabs[tabIdentifier].injections = {};
|
||
}
|
||
};
|
||
|
||
stateManager._handleStorageChanged = function (changes) {
|
||
|
||
if (Setting.SHOW_ICON_BADGE in changes) {
|
||
|
||
stateManager.showIconBadge = changes.showIconBadge.newValue;
|
||
|
||
if (changes.showIconBadge.newValue !== true) {
|
||
|
||
chrome.tabs.query({}, function (tabs) {
|
||
tabs.forEach(stateManager._removeIconBadgeFromTab);
|
||
});
|
||
}
|
||
}
|
||
|
||
if (Setting.STRIP_METADATA in changes) {
|
||
|
||
requestSanitizer.disable();
|
||
|
||
if (changes.stripMetadata.newValue !== false) {
|
||
requestSanitizer.enable();
|
||
}
|
||
}
|
||
if (Setting.NEGATE_HTML_FILTER_LIST in changes) {
|
||
stateManager.getInvertOption = changes.negateHtmlFilterList.newValue;
|
||
}
|
||
if (Setting.SELECTED_ICON in changes) {
|
||
stateManager.selectedIcon = changes.selectedIcon.newValue;
|
||
}
|
||
if (Setting.INTERNAL_STATISTICS in changes) {
|
||
stateManager.internalStatistics = changes.internalStatistics.newValue;
|
||
}
|
||
if (Setting.INTERNAL_STATISTICS_DATA in changes) {
|
||
stats.data = changes.internalStatisticsData.newValue;
|
||
}
|
||
};
|
||
|
||
stateManager._clearBadgeText = function (tabIdentifier) {
|
||
|
||
wrappers.setBadgeText({
|
||
'tabId': tabIdentifier,
|
||
'text': ''
|
||
});
|
||
};
|
||
|
||
stateManager._removeIconBadgeFromTab = function (tab) {
|
||
stateManager._clearBadgeText(tab.id);
|
||
};
|
||
|
||
stateManager._domainIsListed = function (domain, listname) {
|
||
|
||
if (domain !== null) {
|
||
|
||
let whitelistRecord, isWhitelisted;
|
||
|
||
if (listname === "manipulate-dom") {
|
||
whitelistRecord = requestAnalyzer.domainsManipulateDOM[domain];
|
||
isWhitelisted = Boolean(whitelistRecord);
|
||
} else {
|
||
whitelistRecord = requestAnalyzer.whitelistedDomains[domain];
|
||
isWhitelisted = Boolean(whitelistRecord);
|
||
}
|
||
|
||
return isWhitelisted;
|
||
}
|
||
|
||
return false;
|
||
};
|
||
|
||
stateManager._setIconDisabled = function (tabIdentifier) {
|
||
|
||
wrappers.setIcon({
|
||
'path': stateManager.selectedIcon,
|
||
'tabId': tabIdentifier
|
||
}, 'Disabled');
|
||
};
|
||
|
||
|
||
/**
|
||
* Initializations
|
||
*/
|
||
|
||
stateManager.requests = {};
|
||
stateManager.tabs = {};
|
||
stateManager.getInvertOption = false;
|
||
stateManager.validHosts = [];
|
||
stateManager.selectedIcon = 'Default';
|
||
stateManager.internalStatistics = false;
|
||
|
||
for (let mapping in mappings) {
|
||
|
||
let supportedHost = Address.ANY_PROTOCOL + mapping + Address.ANY_PATH;
|
||
stateManager.validHosts.push(supportedHost);
|
||
}
|
||
|
||
chrome.tabs.query({}, function (tabs) {
|
||
tabs.forEach(stateManager._createTab);
|
||
});
|
||
|
||
storageManager.type.get([Setting.SHOW_ICON_BADGE, Setting.SELECTED_ICON], function (items) {
|
||
|
||
if (items.showIconBadge === undefined) {
|
||
items.showIconBadge = true;
|
||
}
|
||
if (items.selectedIcon === undefined) {
|
||
stateManager.selectedIcon = 'Default';
|
||
}
|
||
stateManager.showIconBadge = items.showIconBadge;
|
||
stateManager.selectedIcon = items.selectedIcon;
|
||
});
|
||
|
||
chrome.storage.local.get([Setting.INTERNAL_STATISTICS], function (items) {
|
||
stateManager.internalStatistics = items.internalStatistics;
|
||
});
|
||
|
||
|
||
/**
|
||
* Event Handlers
|
||
*/
|
||
|
||
chrome.tabs.onCreated.addListener(stateManager._createTab);
|
||
chrome.tabs.onRemoved.addListener(stateManager._removeTab);
|
||
|
||
chrome.webRequest.onBeforeRequest.addListener(function (requestDetails) {
|
||
|
||
if (requestDetails.tabId !== -1 && stateManager.tabs[requestDetails.tabId]) {
|
||
|
||
stateManager.tabs[requestDetails.tabId].details = {
|
||
'url': requestDetails.url
|
||
};
|
||
}
|
||
|
||
}, {'types': [WebRequestType.MAIN_FRAME], 'urls': [Address.ANY]}, [WebRequest.BLOCKING]);
|
||
|
||
chrome.webNavigation.onCommitted.addListener(stateManager._updateTab, {
|
||
'url': [{'urlContains': ':'}]
|
||
});
|
||
|
||
chrome.webRequest.onErrorOccurred.addListener(function (requestDetails) {
|
||
|
||
if (stateManager.requests[requestDetails.requestId]) {
|
||
delete stateManager.requests[requestDetails.requestId];
|
||
}
|
||
|
||
}, {'urls': [Address.ANY]});
|
||
|
||
chrome.webRequest.onBeforeRedirect.addListener(function (requestDetails) {
|
||
|
||
let knownRequest = stateManager.requests[requestDetails.requestId];
|
||
|
||
if (knownRequest) {
|
||
|
||
stateManager.registerInjection(knownRequest.tabIdentifier, knownRequest.targetDetails);
|
||
delete stateManager.requests[requestDetails.requestId];
|
||
}
|
||
|
||
}, {'urls': [Address.ANY]});
|
||
|
||
chrome.storage.onChanged.addListener(stateManager._handleStorageChanged);
|