Skip to content

Function parameters and return type inference with infer keyword #38182

Description

I saw a couple of similar issues, but it all provided very poor specification of the problem, so I decided to open my own issue.

Description

The further step in bringing nice fp declarations into TypeScript is adding function parameters and return type inference through infer keyword.

Lets look at some examples:

type F1 = <T extends number>(a: T, b: T) => T

// Now: [number, number]
// Propose: [1, 1]
type R1 = F1 extends ((a: 1, b: infer P) => infer R) ? [P, R] : never

Here we passed parameter a as 1 and expect:

  1. Type parameter T to be inferred as 1
  2. Type P to be inferred as T -> 1
  3. Type R to be inferred as T -> 1
  4. Result to be inferred as [P, R] -> [1, 1]

type F2 = <T1 extends AnyTuple, T2>(a: T1, b: T2, c: Reverse<T1>) => [T1, T2]  

// Now: [never, unknown, [AnyTuple, unknown]]
// Propose: [[3, 2, 1], unknown, [[1, 2, 3], unknown]]
type R2 = F2 extends (a: [1, 2, 3], b: infer P2, c: infer P1) => infer R ? [P1, P2, R] : never

Here we passed parameter a as [1, 2, 3] and expect:

  1. Type parameter T1 to be inferred as [1, 2, 3]
  2. Type parameter T2 to be replaced with unknown type because we provided no info to be able to infer it
  3. Type P2 to be inferred as T2 -> unknown
  4. Type P1 to be inferred as Reverse<T1> -> Reverse<[1, 2, 3]> -> [3, 2, 1]
  5. Type R to be inferred as [T1, T2] -> [[1, 2, 3], unknown]
  6. Result to be inferred as [P1, P2, R] -> `[[3, 2, 1], unknown, [[1, 2, 3], unknown]]

type F3 = <T extends AnyTuple>(a: T, b: T['length']) => typeof b

// Now: never
// Propose: 3
type R3 = F3 extends ((a: [1, 2, 3], b: infer LEN) => any) ? LEN : never

Here we expect:

  1. Type parameter T to be inferred as [1, 2, 3]
  2. Type LEN to be inferred as T['length'] -> [1, 2, 3]['length'] -> 3
  3. Result to be inferred as LEN -> 3

Case of wrapping type checkers:

enum string_literal { string_literal = '' }

type StringLiteral<T> = T extends string ? string extends T ? string_literal : T : string_literal

type F4 = <T>(a: StringLiteral<T>, b: T) => T extends '' ? [] : T

// Now: all cases never

// Proposed: ['foobar', 'foobar']
type R4_1 = F4 extends ((a: 'foobar', b: infer P) => infer R) ? [P, R] : never
// Proposed: ['', []]
type R4_2 = F4 extends ((a: '', b: infer P) => infer R) ? [P, R] : never
// Proposed: [string_literal, string]
type R4_3 = F4 extends ((a: string, b: infer P) => infer R) ? [P, R] : never

Use cases

In other similar issues I saw only use cases of function parameters inference, so I'll extend it with return type inference cases.

The first example is a call function:

declare function call<P extends AnyTuple, F extends ((...args: P) => unknown)>(
  params: P, func: F
): F extends (...args: P) => infer R ? R : never

// Got: [unknown, unknown]
// Expected: [1, 2]
const a = call([1, 2], <A, B>(a: A, b: B): [A, B] => [a, b])

Also it may be used to infer return types on each step of computing pipe and flow function versions with unlimited parameters count. I'm sure, there're much more cases where this feature may be applied, but I can't remember any more right now.

Util types used

type Empty = readonly []


type AnyTuple = ReadonlyArray<unknown> & { readonly 0: unknown } | Empty


type Tail<T extends AnyTuple> = ((...args: T) => any) extends (head: any, ...tail: infer R) => any

  ? R extends AnyTuple ? Readonly<R> : never : Empty


type Prepend<T extends AnyTuple, E> = ((e: E, ...args: T) => any) extends

  (...args: infer R) => any ? R extends AnyTuple ? Readonly<R> : never : never


type _ReverseWith<T extends AnyTuple, Result extends AnyTuple> = {
  0: _ReverseWith<Tail<T>, Prepend<Result, T[0]>>
  1: Readonly<Result>
}[T extends Empty ? 1 : 0]


type Reverse<T extends AnyTuple> = _ReverseWith<T, Empty>

Activity

  1. mscottnelson commented on May 3, 2020

    @mscottnelson

    This is a great write-up, thank you!

    I've been fighting with the infer typing for longer than I'd care to admit around this. Many of the js libraries out there provide something like an options object where the options are 1) highly constrained and 2) constrain the resulting properties of the returned function. I admit that I've been fighting with it because it feels like it ought to be possible with the tools available today AND it feels like it ought to be able to be incredibly succinct.

    But I still haven't ruled out that I'm just not smart enough to have figured it out already.

    Consider

    const options = [
      {name: "first", max: 500, label: "primary"},
      {name: "second", code: 2, label: "secondary"}
    ];
    

    ..with the expectation that by changing the configuration above, you can then have an initializer and painlessly infer something like the signature below, where "primary" is known by ts to be on the object.

    initiateFn(options);
    initialized.primary(386) // known to return {first: { min: 0, max: 500, value: number }, primary: /*alias to first*/}
    

    You know that options will contain all the correct constraints, but that you want all the types to be inferred so that changes to the code result in automatic type checking to check that all code will work for the given configuration.

    It can be especially frustrating when trying to infer a tuple or a tuple-like object. const sizes = [{size:1,label:"small"},{size:2,label:"medium"},{size:3,label:"large"}] as const; gets you so very close (as does the variant [obj as const, obj2 as const ...], but breaks down as soon as you want to change a parameter name. label:"small"|"medium"|"large" won't get inferred appropriately when trying to have an inferred map type, sizes.map(s=>{[s.label]: myObject}), with no good inference workarounds for constrained key-name sets.

  2. RyanCavanaugh commented on May 11, 2020

    @RyanCavanaugh
    Member
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    In DiscussionNot yet reached consensusSuggestionAn idea for TypeScript

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions