TypeScriptToLua/src/lualib/ArrayFrom.ts at project-references · TypeScriptToLua/TypeScriptToLua

36 lines (32 loc) · 1.06 KB

/** @noSelfInFile */

import { __TS__Iterator } from "./Iterator";

function arrayLikeStep(this: ArrayLike<unknown>, index: number): LuaMultiReturn<[number, unknown] | []> {
    index += 1;
    if (index > this.length) return $multi();
    return $multi(index, this[index]);
}

const arrayLikeIterator: (
    this: void,
    arr: ArrayLike<unknown> | Iterable<unknown>
) => LuaIterable<LuaMultiReturn<[number, unknown]>> = (arr => {
    if (typeof arr.length === "number") return $multi(arrayLikeStep, arr, 0);
    return __TS__Iterator(arr);
}) as any;

export function __TS__ArrayFrom(
    this: void,
    arrayLike: ArrayLike<unknown> | Iterable<unknown>,
    mapFn?: (this: unknown, element: unknown, index: number) => unknown,
    thisArg?: unknown
): unknown[] {
    const result = [];
    if (mapFn === undefined) {
        for (const [, v] of arrayLikeIterator(arrayLike)) {
            result.push(v);
        }
    } else {
        for (const [i, v] of arrayLikeIterator(arrayLike)) {
            result.push(mapFn.call(thisArg, v, i - 1));
        }
    }
    return result;
}

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Expand file tree

Search code, repositories, users, issues, pull requests...

Uh oh!

FilesExpand file tree

ArrayFrom.ts

Latest commit

History

ArrayFrom.ts

File metadata and controls

Expand file tree