equicord-companion/src/ast/vencord/hover/index.ts

Equicord Companion is a vscode extension to test Equicord patches & webpack finds right from the comfort of your IDE

import { AstParser } from "@ast/AstParser";
import { isStringLiteralLikeOrTemplateLiteralFragmentOrRegexLiteral } from "@ast/util";
import { runtimeHashMessageKey } from "@modules/intlHash";
import { outputChannel } from "@modules/logging";
import { intlRegex } from "@modules/patches";
import { sendAndGetData } from "@server/index";
import { PromiseProivderResult } from "@type/index";

// import mappings from "./mappings.json";
import { isRegularExpressionLiteral } from "typescript";
import { Hover, HoverProvider, MarkdownString, Position, TextDocument } from "vscode";

const i18nCache = {};

async function getI18nValue(hashedKey: string) {
    return (i18nCache[hashedKey] ??= (await sendAndGetData<"i18n">({
        type: "i18n",
        data: {
            hashedKey,
        },
    }))?.data.value ?? "ERROR FETCHING I18N VALUE");
}
export class I18nHover implements HoverProvider {
    async provideHover(document: TextDocument, position: Position): PromiseProivderResult<Hover> {
        try {
            const ast = new AstParser(document.getText());
            const offset = ast.offsetAt(position);
            const token = ast.getTokenAtOffset(offset);

            if (!token || !isStringLiteralLikeOrTemplateLiteralFragmentOrRegexLiteral(token))
                return null;
            if (!token.text.includes("#{intl::"))
                return null;

            const intls = [...token.text.matchAll(intlRegex)];

            if (intls.length === 0)
                return null;

            /**
             * index from the start of the string. this would be 0 `"|as"`
             */
            const stringStartingIndex = token.getStart(ast.sourceFile, true) - +isRegularExpressionLiteral(token);
            const stringIndex = offset - stringStartingIndex - 1;

            for (const { 0: { length }, index, 1: key } of intls) {
                if (stringIndex >= index && stringIndex <= index + length) {
                    const hashedKey = runtimeHashMessageKey(key);
                    const hasSpecialChars = !Number.isNaN(Number(hashedKey[0])) || hashedKey.includes("+") || hashedKey.includes("/");

                    const range = ast.makeRange({
                        pos: stringStartingIndex + index + 1,
                        end: stringStartingIndex + index + length + 1,
                    });

                    return {
                        range,
                        contents: [
                            new MarkdownString(hasSpecialChars ? `["${hashedKey}"]` : `.${hashedKey}`),
                            // FIXME: refactor
                            new MarkdownString(await getI18nValue(hashedKey)),
                        ],
                    };
                }
            }
        } catch (e) {
            outputChannel.error(e);
        }
    }
}