https://github.com/Microsoft/TypeScript
Raw File
Tip revision: 2e859a2650cd8154bc4d4acbb689b42d1a9949fe authored by Anders Hejlsberg on 17 February 2022, 19:48:23 UTC
Merge branch 'main' into fix44572
Tip revision: 2e859a2
requireInTs.ts
/* @internal */
namespace ts.codefix {
    const fixId = "requireInTs";
    const errorCodes = [Diagnostics.require_call_may_be_converted_to_an_import.code];
    registerCodeFix({
        errorCodes,
        getCodeActions(context) {
            const info = getInfo(context.sourceFile, context.program, context.span.start);
            if (!info) {
                return undefined;
            }
            const changes = textChanges.ChangeTracker.with(context, t => doChange(t, context.sourceFile, info));
            return [createCodeFixAction(fixId, changes, Diagnostics.Convert_require_to_import, fixId, Diagnostics.Convert_all_require_to_import)];
        },
        fixIds: [fixId],
        getAllCodeActions: context => codeFixAll(context, errorCodes, (changes, diag) => {
            const info = getInfo(diag.file, context.program, diag.start);
            if (info) {
                doChange(changes, context.sourceFile, info);
            }
        }),
    });

    function doChange(changes: textChanges.ChangeTracker, sourceFile: SourceFile, info: Info) {
        const { allowSyntheticDefaults, defaultImportName, namedImports, statement, required } = info;
        changes.replaceNode(sourceFile, statement, defaultImportName && !allowSyntheticDefaults
            ? factory.createImportEqualsDeclaration(/*decorators*/ undefined, /*modifiers*/ undefined, /*isTypeOnly*/ false, defaultImportName, factory.createExternalModuleReference(required))
            : factory.createImportDeclaration(/*decorators*/ undefined, /*modifiers*/ undefined, factory.createImportClause(/*isTypeOnly*/ false, defaultImportName, namedImports), required, /*assertClause*/ undefined));
    }

    interface Info {
        readonly allowSyntheticDefaults: boolean;
        readonly defaultImportName: Identifier | undefined;
        readonly namedImports: NamedImports | undefined;
        readonly statement: VariableStatement;
        readonly required: StringLiteralLike;
    }

    function getInfo(sourceFile: SourceFile, program: Program, pos: number): Info | undefined {
        const { parent } = getTokenAtPosition(sourceFile, pos);
        if (!isRequireCall(parent, /*checkArgumentIsStringLiteralLike*/ true)) {
            throw Debug.failBadSyntaxKind(parent);
        }

        const decl = cast(parent.parent, isVariableDeclaration);
        const defaultImportName = tryCast(decl.name, isIdentifier);
        const namedImports = isObjectBindingPattern(decl.name) ? tryCreateNamedImportsFromObjectBindingPattern(decl.name) : undefined;
        if (defaultImportName || namedImports) {
            return {
                allowSyntheticDefaults: getAllowSyntheticDefaultImports(program.getCompilerOptions()),
                defaultImportName,
                namedImports,
                statement: cast(decl.parent.parent, isVariableStatement),
                required: first(parent.arguments)
            };
        }
    }

    function tryCreateNamedImportsFromObjectBindingPattern(node: ObjectBindingPattern): NamedImports | undefined {
        const importSpecifiers: ImportSpecifier[] = [];
        for (const element of node.elements) {
            if (!isIdentifier(element.name) || element.initializer) {
                return undefined;
            }
            importSpecifiers.push(factory.createImportSpecifier(/*isTypeOnly*/ false, tryCast(element.propertyName, isIdentifier), element.name));
        }

        if (importSpecifiers.length) {
            return factory.createNamedImports(importSpecifiers);
        }
    }
}
back to top