1
Fork 0
tildes-reextended/source/content-scripts/features/anonymize-usernames.ts

77 lines
2.1 KiB
TypeScript
Raw Normal View History

import {hashSha256, log, querySelectorAll} from "../../utilities/exports.js";
import {
type AnonymizeUsernamesData,
ReplacementType,
} from "../../storage/exports.js";
export async function runAnonymizeUsernamesFeature(
data: AnonymizeUsernamesData,
) {
const count = await anonymizeUsernames(data);
log(`Anonymize Usernames: Initialized for ${count} user links.`);
}
async function anonymizeUsernames(
data: AnonymizeUsernamesData,
): Promise<number> {
const usernameElements = querySelectorAll<HTMLElement>(
2023-06-23 10:52:03 +00:00
".link-user:not(.trx-anonymized)",
);
const replacements = await generateReplacements(usernameElements, data);
for (const element of usernameElements) {
let username = usernameFromElement(element);
2023-06-23 10:52:03 +00:00
const isMention = username.startsWith("@");
if (isMention) {
username = username.slice(1);
}
const replacement = replacements[username];
element.textContent = isMention ? `@${replacement}` : `${replacement}`;
2023-06-23 10:52:03 +00:00
element.classList.add("trx-anonymized");
element.dataset.trxUsername = username;
}
return usernameElements.length;
}
async function generateReplacements(
elements: HTMLElement[],
data: AnonymizeUsernamesData,
): Promise<Record<string, string>> {
const usernames = new Set(
2023-06-23 10:52:03 +00:00
elements.map((element) => usernameFromElement(element).replace(/@/g, "")),
);
const replacements: Record<string, string> = {};
for (const [index, username] of Array.from(usernames).entries()) {
switch (data.replacementType) {
case ReplacementType.Hashed: {
const hash = await hashSha256(username);
replacements[username] = hash.slice(0, 10).toUpperCase();
break;
}
case ReplacementType.Numerical: {
replacements[username] = `Anonymous ${index + 1}`;
break;
}
default: {
throw new Error(
`Unknown ReplacementType in AnonymizeUsernamesData: ${JSON.stringify(
data,
)}`,
);
}
}
}
return replacements;
}
function usernameFromElement(element: HTMLElement): string {
return (element.textContent ?? "<unknown>").trim().toLowerCase();
}