blob: 6be65898d4aefbb3631c9f77f1cc3b40830972d6 [file] [log] [blame]
import {compareLoose} from 'semver';
import actionApi from '../common/actionApi.js';
import {getSemVerExtVersion} from '../common/extUtils.js';
import * as commonPb from './api_proto/common_pb.js';
import {KillSwitchServicePromiseClient} from './api_proto/kill_switch_grpc_web_pb.js';
import * as ksPb from './api_proto/kill_switch_pb.js';
const host =
(PRODUCTION ? 'https://twpt-grpc-web.avm99963.com/' :
'http://localhost:8081');
const KILLSWITCH_BADGE_OPTIONS = {
'iconTitleI18nKey': 'actionbadge_killswitch_enabled',
'badgeText': '!',
'bgColor': '#B71C1C',
};
export default class KillSwitchMechanism {
constructor() {
this.client = new KillSwitchServicePromiseClient(host, null, null);
}
setBadge(anyKillSwitchEnabled) {
if (anyKillSwitchEnabled) {
actionApi.setBadgeBackgroundColor(
{color: KILLSWITCH_BADGE_OPTIONS.bgColor});
actionApi.setBadgeText({text: KILLSWITCH_BADGE_OPTIONS.badgeText});
let title =
chrome.i18n.getMessage(KILLSWITCH_BADGE_OPTIONS.iconTitleI18nKey);
actionApi.setTitle({title});
} else {
actionApi.setBadgeText({text: ''});
actionApi.setTitle({title: ''});
}
}
getCurrentBrowser() {
// #!if browser_target == 'gecko'
return commonPb.Environment.Browser.BROWSER_GECKO;
// #!else
return commonPb.Environment.Browser.BROWSER_CHROMIUM;
// #!endif
}
updateKillSwitchStatus() {
let request = new ksPb.GetKillSwitchOverviewRequest();
request.setWithNonactiveKillSwitches(false);
this.client.getKillSwitchOverview(request)
.then(res => {
let killSwitches = res.getKillSwitchesList();
let currentVersion = getSemVerExtVersion();
if (currentVersion === '0' || currentVersion === null) {
currentVersion = '0.0.0';
}
let forceDisabledFeaturesSet = new Set();
for (let killSwitch of killSwitches) {
// If it isn't active, this kill switch is not applicable.
if (!killSwitch.getActive()) continue;
// If min_version is set and is greater than the current version,
// this kill switch is not applicable.
if (killSwitch.getMinVersion() != '' &&
compareLoose(killSwitch.getMinVersion(), currentVersion) == 1)
continue;
// If max_version is set and is less than the current version, this
// kill switch is not applicable.
if (killSwitch.getMaxVersion() != '' &&
compareLoose(killSwitch.getMaxVersion(), currentVersion) == -1)
continue;
let browsers = killSwitch.getBrowsersList();
let currentBrowser = this.getCurrentBrowser();
// If this browser isn't included as part of the kill switch, the
// kill switch is not applicable.
if (!browsers.includes(currentBrowser)) continue;
console.warn(
'Kill switch ' + killSwitch.getId() + ' will be applied!');
let featureCodename = killSwitch.getFeature()?.getCodename?.();
if (featureCodename) forceDisabledFeaturesSet.add(featureCodename);
}
let forceDisabledFeatures = Array.from(forceDisabledFeaturesSet);
chrome.storage.sync.set(
{_forceDisabledFeatures: forceDisabledFeatures}, () => {
let anyKillSwitchEnabled = forceDisabledFeatures.length > 0;
this.setBadge(anyKillSwitchEnabled);
});
})
.catch(err => {
console.error(
'[killSwitch] Can\'t retrieve kill switch status: ', err);
});
}
}