Typescript 使用 Compiler API 推断类型参数

Question

使用此代码 (playground)：

declare class Test<P = unknown, R = unknown> {
    test(p: P): R;
}

declare class M extends Test {
    test(q: number): boolean;
}

// these lines are not in real code. This is an example of what TS
// could infer from my code and what I would like to find using Compiler API
type Sgn<M> = M extends Test<infer P, infer R> ? [P, R] : never;
type sgn = Sgn<M>; // [number, boolean]

typescript 可以从类声明中的方法 test 推断出类 M（P = number，R = boolean）的隐式类型参数。

我想用Compiler API 做同样的事情。我有program.typeChecker，我被困在这里。如何获得隐式类型参数？

Answer 1

extends Test 中的 Test 类型 M 没有隐式类型为 Test<number, boolean>。 M 类将 test 的声明覆盖为 test(q: number): boolean。如果您在extends Test 的M 中获得ExpressionWithTypeArguemnts 的类型，它将是Test<unknown, unknown>。

从M获取类型

如果你知道Test 的结构是什么并且你有...

declare class M extends Test {
    test(q: number): boolean;
}

...如果你想找到test的参数类型和返回类型，那么你可以这样做：

const mDecl = sourceFile.statements.find(child =>
    ts.isClassDeclaration(child) && child.name?.escapedText === "M"
)! as ts.ClassDeclaration;
const testMethod = mDecl.members.find(member =>
    ts.isMethodDeclaration(member)
    && ts.isIdentifier(member.name)
    && member.name.escapedText === "test"
)! as ts.MethodDeclaration;

// number type
typeChecker.getTypeAtLocation(testMethod.parameters[0]);
// boolean type
typeChecker.getReturnTypeOfSignature(typeChecker.getSignatureFromDeclaration(testMethod));

给定Test，在M中找到P和R

如果您有Test，并且想在M 中找到P 和R，并且您不知道Test 的类型可能是什么样的，那么您不能使用前面描述的方法。

为此，你必须手动遍历Test的类型，然后以同样的方式遍历M，看看Test中的类型参数在M中使用的是什么类型。这样做有点复杂，而且在这里发布的工作/代码太多。

基本上，没有简单的方法来构建类型，然后与编译器 API 进行比较。有一些建议可以让这更容易（例如Type Builder API Proposal 和Type Relationship API Proposal）。

也就是说，一种解决方法是通过ts.createSourceFile 创建一个虚拟文件并在其中编写一些代码（例如您的Sgn 和sgn 声明）然后使用该文件上的类型检查器来解决类型。