120 lines
2.8 KiB
TypeScript
120 lines
2.8 KiB
TypeScript
import { EventEmitter } from "events";
|
|
import ExtendedRequest from "./extended-request";
|
|
|
|
export type Sources = "cookie" | "pathname" | "queryparams" | "header";
|
|
|
|
import { TCString, TCModel } from "@iabtcf/core";
|
|
|
|
const id = (function* id() {
|
|
let i = 0;
|
|
while (true) {
|
|
i++;
|
|
yield i;
|
|
}
|
|
})();
|
|
|
|
export class StolenDataEntry {
|
|
public isIAB = false;
|
|
public iab: TCModel | null = null;
|
|
public id: number;
|
|
|
|
constructor(
|
|
public request: ExtendedRequest,
|
|
public source: Sources,
|
|
public name: string,
|
|
public value: string
|
|
) {
|
|
try {
|
|
this.iab = TCString.decode(value);
|
|
// console.log(this.iab);
|
|
this.isIAB = true;
|
|
} catch (e) {}
|
|
this.id = id.next().value as number;
|
|
}
|
|
|
|
getPriority() {
|
|
let priority = 0;
|
|
priority += this.value.length;
|
|
const url = new URL(this.request.getOrigin());
|
|
if (this.value.includes(url.host) || this.value.includes(url.pathname)) {
|
|
priority += 100;
|
|
}
|
|
return priority;
|
|
}
|
|
}
|
|
|
|
export class RequestCluster extends EventEmitter {
|
|
public requests: ExtendedRequest[] = [];
|
|
constructor(public id: string) {
|
|
super();
|
|
}
|
|
add(request: ExtendedRequest) {
|
|
this.requests.push(request);
|
|
this.emit("change");
|
|
}
|
|
|
|
hasCookies() {
|
|
for (const request of this.requests) {
|
|
if (request.hasCookie()) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
getStolenData(filter: {
|
|
minValueLength: number;
|
|
cookiesOnly: boolean;
|
|
}): StolenDataEntry[] {
|
|
return this.requests
|
|
.map((request) => request.getAllStolenData())
|
|
.reduce((a, b) => a.concat(b), [])
|
|
.filter((entry) => {
|
|
return entry.value.length >= filter.minValueLength;
|
|
})
|
|
.filter((entry) => !filter.cookiesOnly || entry.source === "cookie")
|
|
.sort((entry1, entry2) =>
|
|
entry1.getPriority() > entry2.getPriority() ? -1 : 1
|
|
)
|
|
.filter((element, index, array) => {
|
|
// remove duplicate neighbours
|
|
if (index == 0) {
|
|
return true;
|
|
}
|
|
if (
|
|
element.name != array[index - 1].name ||
|
|
element.value != array[index - 1].value
|
|
) {
|
|
return true;
|
|
}
|
|
});
|
|
}
|
|
|
|
static sortCompare(a: RequestCluster, b: RequestCluster) {
|
|
if (a.hasCookies() == b.hasCookies()) {
|
|
if (a.id < b.id) {
|
|
return -1;
|
|
} else {
|
|
return 1;
|
|
}
|
|
} else {
|
|
if (a.hasCookies()) {
|
|
return -1;
|
|
} else {
|
|
return 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
async removeAllCookies() {
|
|
const cookies = await browser.cookies.getAll({ domain: this.id });
|
|
for (const cookie of cookies) {
|
|
console.log("removing cookie", cookie.name, "from", cookie.domain);
|
|
await browser.cookies.remove({
|
|
name: cookie.name,
|
|
url: `https://${cookie.domain}`,
|
|
});
|
|
}
|
|
}
|
|
}
|