rentgen/memory.ts

111 lines
3.6 KiB
TypeScript
Raw Normal View History

2022-01-27 22:30:57 +01:00
import ExtendedRequest from './extended-request';
2022-07-09 15:51:34 +02:00
import { getshorthost } from './util';
2022-01-27 22:30:57 +01:00
import { RequestCluster } from './request-cluster';
2022-05-02 17:04:56 +02:00
import { SaferEmitter } from './safer-emitter';
2021-10-03 09:03:56 +02:00
function setDomainsCount(counter: number, tabId: number) {
2022-04-22 13:00:02 +02:00
browser.browserAction.setBadgeText({ text: counter < 0 ? '0' : counter.toString(), tabId });
browser.browserAction.setTitle({
2022-04-15 10:34:29 +02:00
title: 'Rentgen',
tabId,
});
}
2022-05-02 17:04:56 +02:00
export default class Memory extends SaferEmitter {
2022-01-27 22:30:57 +01:00
origin_to_history = {} as Record<string, Record<string, RequestCluster>>;
async register(request: ExtendedRequest) {
await request.init();
if (!request.isThirdParty()) {
return;
}
if (!this.origin_to_history[request.origin]) {
this.origin_to_history[request.origin] = {};
}
const shorthost = getshorthost(new URL(request.url).host);
if (!this.origin_to_history[request.origin][shorthost]) {
const cluster = new RequestCluster(shorthost);
this.origin_to_history[request.origin][shorthost] = cluster;
}
this.origin_to_history[request.origin][shorthost].add(request);
2022-05-02 17:04:56 +02:00
this.emit('change', shorthost);
2022-04-15 10:34:29 +02:00
Object.values(this.getClustersForOrigin(request.origin)).some((cluster) =>
cluster.hasCookies()
)
? browser.browserAction.setBadgeBackgroundColor({ color: '#ff726b' })
: browser.browserAction.setBadgeBackgroundColor({ color: '#ffb900' });
if (request.tabId >= 0) {
setDomainsCount(
Object.values(this.getClustersForOrigin(request.origin)).length,
request.tabId
);
}
2021-10-03 09:03:56 +02:00
}
2022-01-27 22:30:57 +01:00
constructor() {
super();
2021-11-26 20:58:31 +01:00
2022-01-27 22:30:57 +01:00
browser.webRequest.onBeforeRequest.addListener(
async (request) => {
new ExtendedRequest(request);
},
{ urls: ['<all_urls>'] },
['requestBody']
);
browser.webRequest.onBeforeSendHeaders.addListener(
async (request) => {
const extendedRequest = ExtendedRequest.by_id[request.requestId].addHeaders(
request.requestHeaders || []
);
2022-01-27 22:30:57 +01:00
this.register(extendedRequest);
},
{ urls: ['<all_urls>'] },
['requestHeaders']
);
}
2021-10-03 09:03:56 +02:00
2022-05-02 17:04:56 +02:00
emit(eventName: string, data = 'any'): boolean {
setTimeout(() => super.emit(eventName, data), 0);
2022-07-09 15:51:34 +02:00
return true;
}
2022-01-27 22:30:57 +01:00
getClustersForOrigin(origin: string): Record<string, RequestCluster> {
return this.origin_to_history[origin] || {};
}
2021-11-06 21:48:25 +01:00
2022-01-27 22:30:57 +01:00
async removeCookiesFor(origin: string, shorthost?: string): Promise<void> {
if (shorthost) {
const cookies = await browser.cookies.getAll({ domain: shorthost });
for (const cookie of cookies) {
await browser.cookies.remove({
name: cookie.name,
url: `https://${cookie.domain}`,
});
}
} else {
const clusters = this.getClustersForOrigin(origin);
2021-11-07 09:17:19 +01:00
2022-01-27 22:30:57 +01:00
await Promise.all(
Object.values(clusters)
.filter((cluster) => !shorthost || cluster.id === shorthost)
.map((cluster) => this.removeCookiesFor(origin, cluster.id))
);
}
2021-11-07 09:17:19 +01:00
}
2021-11-06 21:48:25 +01:00
2022-01-27 22:30:57 +01:00
async removeRequestsFor(origin: string) {
this.origin_to_history[origin] = {};
}
2021-10-03 09:03:56 +02:00
}
2021-11-07 13:57:24 +01:00
export function init() {
2022-01-27 22:30:57 +01:00
const memory = new Memory();
2021-10-03 09:03:56 +02:00
2022-01-27 22:30:57 +01:00
(window as any).memory = memory;
2021-11-07 13:57:24 +01:00
}
2021-11-07 17:23:48 +01:00
export function getMemory(): Memory {
return (browser.extension.getBackgroundPage().window as any).memory as Memory;
2021-11-07 17:23:48 +01:00
}