import {

arrayFrom,

cast,

CodeActionCommand,

CodeFixAction,

CodeFixAllContext,

CodeFixContext,

CodeFixContextBase,

CodeFixRegistration,

CombinedCodeActions,

computeSuggestionDiagnostics,

contains,

createMultiMap,

Debug,

Diagnostic,

DiagnosticOrDiagnosticAndArguments,

diagnosticToString,

DiagnosticWithLocation,

FileTextChanges,

flatMap,

isString,

map,

TextChange,

textChanges,

} from "./_namespaces/ts";

const errorCodeToFixes = createMultiMap<string, CodeFixRegistration>();

const fixIdToRegistration = new Map<string, CodeFixRegistration>();

export function createCodeFixActionWithoutFixAll(fixName: string, changes: FileTextChanges[], description: DiagnosticOrDiagnosticAndArguments) {

return createCodeFixActionWorker(fixName, diagnosticToString(description), changes, /*fixId*/ undefined, /*fixAllDescription*/ undefined);

export function createCodeFixAction(fixName: string, changes: FileTextChanges[], description: DiagnosticOrDiagnosticAndArguments, fixId: {}, fixAllDescription: DiagnosticOrDiagnosticAndArguments, command?: CodeActionCommand): CodeFixAction {

return createCodeFixActionWorker(fixName, diagnosticToString(description), changes, fixId, diagnosticToString(fixAllDescription), command);

export function createCodeFixActionMaybeFixAll(fixName: string, changes: FileTextChanges[], description: DiagnosticOrDiagnosticAndArguments, fixId?: {}, fixAllDescription?: DiagnosticOrDiagnosticAndArguments, command?: CodeActionCommand) {

return createCodeFixActionWorker(fixName, diagnosticToString(description), changes, fixId, fixAllDescription && diagnosticToString(fixAllDescription), command);

function createCodeFixActionWorker(fixName: string, description: string, changes: FileTextChanges[], fixId?: {}, fixAllDescription?: string, command?: CodeActionCommand): CodeFixAction {

return { fixName, description, changes, fixId, fixAllDescription, commands: command ? [command] : undefined };

export function registerCodeFix(reg: CodeFixRegistration) {

let errorCodeToFixesArray: readonly string[] | undefined;

export function getSupportedErrorCodes(): readonly string[] {

return errorCodeToFixesArray ??= arrayFrom(errorCodeToFixes.keys());

function removeFixIdIfFixAllUnavailable(registration: CodeFixRegistration, diagnostics: Diagnostic[]) {

export function getFixes(context: CodeFixContext): readonly CodeFixAction[] {

export function getAllFixes(context: CodeFixAllContext): CombinedCodeActions {

// Currently fixId is always a string.

return fixIdToRegistration.get(cast(context.fixId, isString))!.getAllCodeActions!(context);

export function createCombinedCodeActions(changes: FileTextChanges[], commands?: CodeActionCommand[]): CombinedCodeActions {

return { changes, commands };

export function createFileTextChanges(fileName: string, textChanges: TextChange[]): FileTextChanges {

return { fileName, textChanges };

export function codeFixAll(

export function eachDiagnostic(context: CodeFixAllContext, errorCodes: readonly number[], cb: (diag: DiagnosticWithLocation) => void): void {

function getDiagnostics({ program, sourceFile, cancellationToken }: CodeFixContextBase) {