在 TypeScript 中手动柯里化添加函数

Question

我正在尝试创建一个可选柯里化的类型安全添加函数

    const add:
      | ((x: number) => (y: number) => number)
      | ((x: number, y: number) => number) = (x: number, y?: number) => {
      if (y !== undefined) {
        return x + y;
      } else {
        return (y: number) => x + y;
      }
    };

我希望能够运行这个功能与

add(1, 5)

或

add(1)(5)

但是目前第二次调用在编译时抛出两个类型错误。错误是

Type '(x: number, y?: number | undefined) => number | ((y: number) => number)' is not assignable to type '((x: number) => (y: number) => number) | ((x: number, y: number) => number)'.
  Type '(x: number, y?: number | undefined) => number | ((y: number) => number)' is not assignable to type '(x: number) => (y: number) => number'.
    Type 'number | ((y: number) => number)' is not assignable to type '(y: number) => number'.
      Type 'number' is not assignable to type '(y: number) => number'.

在函数定义处和

Expected 2 arguments, but got 1.

在函数调用时。

我已经看到 return 和 any 类型的此问题的解决方案，但我一直无法找到类型安全的解决方案。我目前想做的事情可以用 TypeScript 实现吗？

Answer 1

是的，使用 function overloads，您可以使用此签名构造一个 add() 函数：

function add (x: number): (y: number) => number
function add (x: number, y: number): number
function add (x: number, y?: number): number | ((y: number) => number) {
    return y === undefined ? (y: number): number => x + y : x + y
}

前两个签名表示有效调用，最后一个签名是包含两个有效调用签名的联合签名。

在 TypeScript 中手动柯里化添加函数

Manually curry an add function in TypeScript

currying

typescript