在对象中强制执行函数签名

Question

我正在寻找一种方法来定义具有两个函数的对象，其中一个函数总是将另一个函数的返回类型作为参数。例如

type DoActionArgType = { contactId: string };

type Args<T> = 
{
  doAction: (arg: DoActionArgType) => string; // <-- returns string
  onEnd?: (arg: string) => void; // <-- takes string as the argument
}

现在我希望它适用于任意返回类型和承诺，这很好，除非我不将类型添加到函数 doAction 函数参数中。

export type $Unwrap<T> = T extends Promise<infer U>
    ? U
    : T extends (...args: any) => Promise<infer U>
    ? U
    : T extends (...args: any) => infer U
    ? U
    : T;

type DoActionArgType = { contactId: string };

type Args<T> = 
{
  doAction: (arg: DoActionArgType) => Promise<T>;
  onEnd?: (arg: $Unwrap<T>) => void;
}

function test<T>(arg: Args<T>) {}

const doAction = async ({ contactId }) => 1

test({
  // doAction: async ({ contactId }: DoActionArgType) => 1, // Works
  // doAction: async () => 1, // Works
  // doAction, // Works
  doAction: async ({ contactId }) => 1, // <--- Does not work
  onEnd: (arg) => {
    const a:number = arg
  }
})

我假设打字稿不理解函数签名匹配？因为contactId 实际上输入了string，只是onEnd 的arg 输入了unknown。有没有办法在不将DoActionArgType 类型添加到doAction 函数的情况下完成这项工作？

Typescript Playground

Answer 1

你需要做一个帽子戏法：

type DoActionArgType = { contactId: string };

type PromiseUnwrap<P> =
  P extends Promise<infer Value>
  ? Value
  : P

const test = <T,>(arg: {
  doAction: (a_arg: DoActionArgType) => T;
  onEnd: <U extends T>(b_arg: PromiseUnwrap<U>) => void;
}) => {
  // ...
};

test({
  doAction: async ({ contactId }) => 1,
  onEnd: (arg) => {
    arg // number
    arg.toExponential // ok
  }
})

Playground

PromiseUnwrap - 获得 Promise 返回类型。

你可能会问，为什么这个功能不起作用：

const test = <T,>(
  arg: {
    doAction: (a_arg: number) => T;
    onEnd: (b_arg: T) => void
  }) => {
  // ...
};

Here你可以找到一个很好的答案。

如果你想了解更多关于 typescript 中的回调，你可以阅读我的article。第一个例子是你的