LocalCDN-Firefox-Chrome-Brave/core/state-manager.js

324 lines
9.8 KiB
JavaScript
Raw Normal View History

2020-02-27 13:45:29 +01:00
/**
* State Manager
2020-06-30 18:41:58 +02:00
* Belongs to LocalCDN (since 2020-02-26)
* (Origin: Decentraleyes)
2020-02-27 13:45:29 +01:00
*
* @author Thomas Rientjes
* @since 2017-03-10
*
2020-06-30 18:41:58 +02:00
* @author nobody
* @since 2020-02-26
*
2020-02-27 13:45:29 +01:00
* @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';
2021-02-17 07:01:08 +01:00
2020-02-27 13:45:29 +01:00
/**
* State Manager
*/
var stateManager = {};
2021-02-17 07:01:08 +01:00
2020-02-27 13:45:29 +01:00
/**
* Public Methods
*/
stateManager.registerInjection = function (tabIdentifier, injection) {
let injectionIdentifier, registeredTab, injectionCount, missingCount;
2020-02-27 13:45:29 +01:00
injectionIdentifier = injection.source + injection.path + injection.version;
registeredTab = stateManager.tabs[tabIdentifier];
registeredTab.injections[injectionIdentifier] = injection;
2020-02-27 13:45:29 +01:00
injectionCount = Object.keys(registeredTab.injections).length || 0;
missingCount = registeredTab.missing || 0;
2020-02-27 13:45:29 +01:00
if (injectionCount > 0) {
chrome.browserAction.setTitle({
'tabId': tabIdentifier,
'title': `LocalCDN (${injectionCount})`
});
}
2021-03-13 06:55:20 +01:00
if (stateManager.showIconBadge === true) {
if (missingCount > 0 && stateManager.changeBadgeColorMissingResources) {
wrappers.setBadgeMissing(tabIdentifier, injectionCount);
} else {
wrappers.defaultBadge(tabIdentifier, injectionCount);
}
}
if (isNaN(storageManager.amountInjected)) {
2020-08-30 18:56:36 +02:00
storageManager.type.get(Setting.AMOUNT_INJECTED, function (items) {
storageManager.amountInjected = items.amountInjected;
2020-08-30 18:56:36 +02:00
storageManager.type.set({
[Setting.AMOUNT_INJECTED]: ++storageManager.amountInjected
2020-02-27 13:45:29 +01:00
});
});
} else {
2020-08-30 18:56:36 +02:00
chrome.storage.local.set({
[Setting.AMOUNT_INJECTED]: ++storageManager.amountInjected
2020-02-27 13:45:29 +01:00
});
}
2021-02-19 07:22:14 +01:00
2020-08-08 07:30:04 +02:00
if (stateManager.internalStatistics) {
stats.setStats(injection);
}
2020-02-27 13:45:29 +01:00
};
2020-10-17 07:09:30 +02:00
stateManager.addDomainToAllowlist = function (domain) {
2020-02-27 13:45:29 +01:00
return new Promise((resolve) => {
2020-10-17 07:09:30 +02:00
let allowlistedDomains = requestAnalyzer.allowlistedDomains;
2020-02-27 13:45:29 +01:00
2021-02-19 07:22:14 +01:00
allowlistedDomains[domain] = true;
2020-10-17 07:09:30 +02:00
storageManager.type.set({allowlistedDomains}, resolve);
2020-02-27 13:45:29 +01:00
});
};
2020-10-17 07:09:30 +02:00
stateManager.removeDomainFromAllowlist = function (domain) {
2020-02-27 13:45:29 +01:00
return new Promise((resolve) => {
2021-01-17 09:52:49 +01:00
let allowlistedDomains, wildcard;
allowlistedDomains = requestAnalyzer.allowlistedDomains;
wildcard = helpers.getWildcard(domain);
if (allowlistedDomains[domain]) {
delete allowlistedDomains[domain];
} else {
delete allowlistedDomains[wildcard];
}
2020-02-27 13:45:29 +01:00
2020-10-17 07:09:30 +02:00
storageManager.type.set({allowlistedDomains}, resolve);
2020-02-27 13:45:29 +01:00
});
};
stateManager.addDomainToManipulateDOMlist = function (domain) {
return new Promise((resolve) => {
let domainsManipulateDOM = requestAnalyzer.domainsManipulateDOM;
2021-02-19 07:22:14 +01:00
domainsManipulateDOM[domain] = true;
2020-08-30 18:56:36 +02:00
storageManager.type.set({domainsManipulateDOM}, resolve);
});
};
stateManager.removeDomainFromManipulateDOMlist = function (domain) {
return new Promise((resolve) => {
let domainsManipulateDOM = requestAnalyzer.domainsManipulateDOM;
2021-02-19 07:22:14 +01:00
delete domainsManipulateDOM[domain];
2020-08-30 18:56:36 +02:00
storageManager.type.set({domainsManipulateDOM}, resolve);
});
};
2020-02-27 13:45:29 +01:00
/**
* Private Methods
*/
stateManager._createTab = function (tab) {
let tabIdentifier, requestFilters;
tabIdentifier = tab.id;
stateManager.tabs[tabIdentifier] = {
'injections': {},
'missing': 0
2020-02-27 13:45:29 +01:00
};
requestFilters = {
'tabId': tabIdentifier,
'urls': stateManager.validHosts
};
chrome.webRequest.onBeforeRequest.addListener(function (requestDetails) {
2020-06-21 08:18:18 +02:00
tab = stateManager.tabs[tabIdentifier].details || {};
2020-02-27 13:45:29 +01:00
return interceptor.handleRequest(requestDetails, tabIdentifier, tab);
}, requestFilters, [WebRequest.BLOCKING]);
};
stateManager._removeTab = function (tabIdentifier) {
delete stateManager.tabs[tabIdentifier];
};
stateManager._updateTab = function (details) {
2020-10-17 07:09:30 +02:00
let tabDomain, domainIsAllowlisted, frameIdentifier, tabIdentifier;
2020-02-27 13:45:29 +01:00
tabDomain = helpers.extractDomainFromUrl(details.url, true);
2020-10-17 07:09:30 +02:00
domainIsAllowlisted = stateManager._domainIsListed(tabDomain);
2020-02-27 13:45:29 +01:00
frameIdentifier = details.frameId;
tabIdentifier = details.tabId;
if (frameIdentifier !== 0 || tabIdentifier === -1) {
return;
}
chrome.browserAction.setTitle({
'tabId': tabIdentifier,
'title': 'LocalCDN (0)'
});
2020-10-17 07:09:30 +02:00
if (domainIsAllowlisted) {
2020-02-27 13:45:29 +01:00
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.tabs[tabIdentifier].missing = 0;
2020-02-27 13:45:29 +01:00
}
};
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);
});
}
2021-02-19 07:22:14 +01:00
} else if (Setting.STRIP_METADATA in changes) {
2020-02-27 13:45:29 +01:00
requestSanitizer.disable();
if (changes.stripMetadata.newValue !== false) {
requestSanitizer.enable();
}
2021-02-19 07:22:14 +01:00
} else if (Setting.NEGATE_HTML_FILTER_LIST in changes) {
stateManager.getInvertOption = changes.negateHtmlFilterList.newValue;
2021-02-19 07:22:14 +01:00
} else if (Setting.SELECTED_ICON in changes) {
stateManager.selectedIcon = changes.selectedIcon.newValue;
2021-02-19 07:22:14 +01:00
} else if (Setting.INTERNAL_STATISTICS in changes) {
2020-08-08 07:30:04 +02:00
stateManager.internalStatistics = changes.internalStatistics.newValue;
2021-02-19 07:22:14 +01:00
} else if (Setting.INTERNAL_STATISTICS_DATA in changes) {
stats.data = changes.internalStatisticsData.newValue;
2021-02-21 19:56:13 +01:00
} else if (Setting.HIDE_DONATION_BUTTON in changes) {
stateManager.hideDonationButton = changes.hideDonationButton.newValue;
2021-03-13 06:55:20 +01:00
} else if (Setting.CHANGE_BADGE_COLOR_MISSING_RESOURCES in changes) {
stateManager.changeBadgeColorMissingResources = changes.changeBadgeColorMissingResources.newValue;
}
2020-02-27 13:45:29 +01:00
};
stateManager._clearBadgeText = function (tabIdentifier) {
wrappers.setBadgeText({
'tabId': tabIdentifier,
'text': ''
});
};
stateManager._removeIconBadgeFromTab = function (tab) {
stateManager._clearBadgeText(tab.id);
};
stateManager._domainIsListed = function (domain, listname) {
2020-02-27 13:45:29 +01:00
if (domain !== null) {
2020-10-17 07:09:30 +02:00
let allowlistRecord, isAllowlisted;
2020-02-27 13:45:29 +01:00
2021-02-17 07:01:08 +01:00
if (listname === 'manipulate-dom') {
2020-10-17 07:09:30 +02:00
allowlistRecord = requestAnalyzer.domainsManipulateDOM[domain];
isAllowlisted = Boolean(allowlistRecord);
} else {
2021-01-18 19:05:41 +01:00
allowlistRecord = helpers.checkAllowlisted(domain);
2020-10-17 07:09:30 +02:00
isAllowlisted = Boolean(allowlistRecord);
}
2020-10-17 07:09:30 +02:00
return isAllowlisted;
2020-02-27 13:45:29 +01:00
}
return false;
};
stateManager._setIconDisabled = function (tabIdentifier) {
wrappers.setIcon({
'path': stateManager.selectedIcon,
2020-02-27 13:45:29 +01:00
'tabId': tabIdentifier
}, 'Disabled');
2020-02-27 13:45:29 +01:00
};
2020-04-30 19:13:27 +02:00
2020-02-27 13:45:29 +01:00
/**
* Initializations
*/
stateManager.requests = {};
stateManager.tabs = {};
2020-06-30 18:44:50 +02:00
stateManager.getInvertOption = false;
2020-02-27 13:45:29 +01:00
stateManager.validHosts = [];
stateManager.selectedIcon = 'Default';
2020-08-08 07:30:04 +02:00
stateManager.internalStatistics = false;
2021-02-21 19:56:13 +01:00
stateManager.hideDonationButton = false;
2021-03-13 06:55:20 +01:00
stateManager.changeBadgeColorMissingResources = false;
2020-02-27 13:45:29 +01:00
for (let mapping in mappings.cdn) {
2020-02-27 13:45:29 +01:00
let supportedHost = Address.ANY_PROTOCOL + mapping + Address.ANY_PATH;
stateManager.validHosts.push(supportedHost);
}
chrome.tabs.query({}, function (tabs) {
tabs.forEach(stateManager._createTab);
});
2021-03-13 06:55:20 +01:00
storageManager.type.get([
Setting.SHOW_ICON_BADGE,
Setting.SELECTED_ICON,
Setting.CHANGE_BADGE_COLOR_MISSING_RESOURCES
], function (items) {
2020-02-27 13:45:29 +01:00
if (items.showIconBadge === undefined) {
items.showIconBadge = true;
}
if (items.selectedIcon === undefined) {
stateManager.selectedIcon = 'Default';
}
2020-02-27 13:45:29 +01:00
stateManager.showIconBadge = items.showIconBadge;
stateManager.selectedIcon = items.selectedIcon;
2021-03-13 06:55:20 +01:00
stateManager.changeBadgeColorMissingResources = items.changeBadgeColorMissingResources;
2020-02-27 13:45:29 +01:00
});
2020-08-08 07:30:04 +02:00
chrome.storage.local.get([Setting.INTERNAL_STATISTICS], function (items) {
stateManager.internalStatistics = items.internalStatistics;
});
2020-02-27 13:45:29 +01:00
/**
* 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]);
2020-02-27 13:45:29 +01:00
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);