Skip to content

Commit

Permalink
feat: codefix for for await of (#50623)
Browse files Browse the repository at this point in the history
Jack-Works authored Sep 29, 2022

Verified

This commit was signed with the committer’s verified signature.
TotalTechGeek Jesse Mitchell
1 parent ecf50e8 commit 42f9143
Showing 4 changed files with 44 additions and 1 deletion.
18 changes: 17 additions & 1 deletion src/compiler/checker.ts
Original file line number Diff line number Diff line change
@@ -642,6 +642,11 @@ namespace ts {
getOptionalType: () => optionalType,
getPromiseType: () => getGlobalPromiseType(/*reportErrors*/ false),
getPromiseLikeType: () => getGlobalPromiseLikeType(/*reportErrors*/ false),
getAsyncIterableType: () => {
const type = getGlobalAsyncIterableType(/*reportErrors*/ false);
if (type === emptyGenericType) return undefined;
return type;
},
isSymbolAccessible,
isArrayType,
isTupleType,
@@ -39429,7 +39434,18 @@ namespace ts {
const message = allowAsyncIterables
? Diagnostics.Type_0_must_have_a_Symbol_asyncIterator_method_that_returns_an_async_iterator
: Diagnostics.Type_0_must_have_a_Symbol_iterator_method_that_returns_an_iterator;
return errorAndMaybeSuggestAwait(errorNode, !!getAwaitedTypeOfPromise(type), message, typeToString(type));
const suggestAwait =
// for (const x of Promise<...>) or [...Promise<...>]
!!getAwaitedTypeOfPromise(type)
// for (const x of AsyncIterable<...>)
|| (
!allowAsyncIterables &&
isForOfStatement(errorNode.parent) &&
errorNode.parent.expression === errorNode &&
getGlobalAsyncIterableType(/** reportErrors */ false) !== emptyGenericType &&
isTypeAssignableTo(type, getGlobalAsyncIterableType(/** reportErrors */ false)
));
return errorAndMaybeSuggestAwait(errorNode, suggestAwait, message, typeToString(type));
}

/**
1 change: 1 addition & 0 deletions src/compiler/types.ts
Original file line number Diff line number Diff line change
@@ -4760,6 +4760,7 @@ namespace ts {
/* @internal */ createPromiseType(type: Type): Type;
/* @internal */ getPromiseType(): Type;
/* @internal */ getPromiseLikeType(): Type;
/* @internal */ getAsyncIterableType(): Type | undefined;

/* @internal */ isTypeAssignableTo(source: Type, target: Type): boolean;
/* @internal */ createAnonymousType(symbol: Symbol | undefined, members: SymbolTable, callSignatures: Signature[], constructSignatures: Signature[], indexInfos: IndexInfo[]): Type;
9 changes: 9 additions & 0 deletions src/services/codefixes/addMissingAwait.ts
Original file line number Diff line number Diff line change
@@ -226,6 +226,15 @@ namespace ts.codefix {
}

function makeChange(changeTracker: textChanges.ChangeTracker, errorCode: number, sourceFile: SourceFile, checker: TypeChecker, insertionSite: Expression, fixedDeclarations?: Set<number>) {
if (isForOfStatement(insertionSite.parent) && !insertionSite.parent.awaitModifier) {
const exprType = checker.getTypeAtLocation(insertionSite);
const asyncIter = checker.getAsyncIterableType();
if (asyncIter && checker.isTypeAssignableTo(exprType, asyncIter)) {
const forOf = insertionSite.parent;
changeTracker.replaceNode(sourceFile, forOf, factory.updateForOfStatement(forOf, factory.createToken(SyntaxKind.AwaitKeyword), forOf.initializer, forOf.expression, forOf.statement));
return;
}
}
if (isBinaryExpression(insertionSite)) {
for (const side of [insertionSite.left, insertionSite.right]) {
if (fixedDeclarations && isIdentifier(side)) {
17 changes: 17 additions & 0 deletions tests/cases/fourslash/codeFixAddMissingAwait_forAwaitOf.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
/// <reference path="fourslash.ts" />
// @lib: es2020
// @target: es2020
////async function* g() {}
////async function fn() {
//// for (const { } of g()) { }
////}

verify.codeFix({
description: ts.Diagnostics.Add_await.message,
index: 0,
newFileContent:
`async function* g() {}
async function fn() {
for await (const { } of g()) { }
}`
});

0 comments on commit 42f9143

Please sign in to comment.