typescript - Como escrever um método flatten tipado para um array

Estou tentando escrever algo em TypeScript semelhante a Array.prototype.flat().

No entanto, quero ser capaz de achatar uma lista aninhada de profundidade arbitrária; e também quero restringir todos os elementos que não são arrays a um único tipo.

Algo como:

interface NestedList<T> extends Array<T | NestedList<T>> {}

Eu consegui isso até agora:

export const flat = <T>(ls: NestedList<T>): T[] => {
   const reducer = (acc: T[], it: T | NestedList<T>): T[] => acc.concat(
      Array.isArray(it) ? it.reduce(reducer, []) : it
   );

   return ls.reduce(reducer, []);
};

No entanto, a inferência de tipo não parece estar funcionando:

   it('works with type inference', () => {
      interface Foo<T> { v: T; }

      const data: { [_: string]: { [_: string]: Foo<number> } } = {
         bar: { x: { v: 1 } },
         moo: { y: { v: 2 }, z: { v: 3 } }
      };

      const nested: Foo<number>[][] = Object.values(data).map(val => Object.values(val));
      const list2: Foo<number>[] = flat(nested);
      expect(list2).toEqual([{ v: 1 }, { v: 2 }, { v: 3 }]);
   });

Ele recebe um erro de tipo nesta linha:

const list2: Foo<number>[] = flat(nested);

Erro :

Type '(Foo<number>[] | ConcatArray<Foo<number>[]>)[]' is not assignable to type 'Foo<number>[]'.
  Type 'Foo<number>[] | ConcatArray<Foo<number>[]>' is not assignable to type 'Foo<number>'.
    Property 'v' is missing in type 'Foo<number>[]' but required in type 'Foo<number>'

O que estou perdendo aqui?

Resposta #1

Eu tropecei no mesmo problema que você. Esta é a solução que eu encontrei:

function flat<U>(arr: U[][]): U[] {
    if ((arr as unknown as U[]).every((val) => !Array.isArray(val))) {
        return (arr as unknown as U[]).slice();
    }
    return arr
        .reduce((acc, val) => acc
            .concat(Array.isArray(val) ? flat((val as unknown as U[][])) : val), []);
}

console.log(flat([[1,2],[[[3]]],4])