import { VencordAstParser } from "@ast/vencord";

import { CodeLens, CodeLensProvider, TextDocument } from "vscode";

export class PatchCodeLensProvider implements CodeLensProvider {
    provideCodeLenses(document: TextDocument) {
        const lenses: CodeLens[] = [];
        const file = new VencordAstParser(document);
        const patches = file.getPatches();

        for (const { range, ...data } of patches) {
            lenses.push(new CodeLens(range, {
                title: "View Module",
                command: "equicord-companion.extractSearch",
                arguments: [data.find, data.findType],
                tooltip: "View Module",
            }));
            lenses.push(new CodeLens(range, {
                title: "Diff Module",
                command: "equicord-companion.diffModuleSearch",
                arguments: [data.find, data.findType],
                tooltip: "Diff Module",
            }));
            lenses.push(new CodeLens(range, {
                title: "Test Patch",
                command: "equicord-companion.testPatch",
                arguments: [data],
                tooltip: "Test Patch",
            }));
            lenses.push(new CodeLens(range, {
                title: "Open in Patch Helper",
                tooltip: "Opens the patch in patch helper, if another patch from this file is aready open, it will be replaced",
                command: "equicord-companion.openPatchHelper",
                arguments: [document, data],
            }));
        }
        return lenses;
    }
}