geninq/dist/functions/take.js

A JavaScript version of the Linq library for Generators

"use strict";
var __await = (this && this.__await) || function (v) { return this instanceof __await ? (this.v = v, this) : new __await(v); }
var __asyncValues = (this && this.__asyncValues) || function (o) {
    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
    var m = o[Symbol.asyncIterator], i;
    return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
    function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
    function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
};
var __asyncGenerator = (this && this.__asyncGenerator) || function (thisArg, _arguments, generator) {
    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
    var g = generator.apply(thisArg, _arguments || []), i, q = [];
    return i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i;
    function verb(n) { if (g[n]) i[n] = function (v) { return new Promise(function (a, b) { q.push([n, v, a, b]) > 1 || resume(n, v); }); }; }
    function resume(n, v) { try { step(g[n](v)); } catch (e) { settle(q[0][3], e); } }
    function step(r) { r.value instanceof __await ? Promise.resolve(r.value.v).then(fulfill, reject) : settle(q[0][2], r); }
    function fulfill(value) { resume("next", value); }
    function reject(value) { resume("throw", value); }
    function settle(f, v) { if (f(v), q.shift(), q.length) resume(q[0][0], q[0][1]); }
};
Object.defineProperty(exports, "__esModule", { value: true });
const utils_1 = require("./utils");
require("./count");
(0, utils_1.Build)("take", function* (count, at) {
    let items = this;
    if (at === "last") {
        const array = this.array();
        count = array.length - count;
        items = array;
    }
    let index = 0;
    for (const item of items) {
        if (!at && index < count) {
            yield item;
        }
        else if (at === "last" && index >= count) {
            yield item;
        }
        index++;
    }
}, function (count, at) {
    return __asyncGenerator(this, arguments, function* () {
        var e_1, _a;
        let items = this;
        if (at === "last") {
            const array = yield __await(this.array());
            count = array.length - count;
            items = array;
        }
        let index = 0;
        try {
            for (var items_1 = __asyncValues(items), items_1_1; items_1_1 = yield __await(items_1.next()), !items_1_1.done;) {
                const item = items_1_1.value;
                if (!at && index < count) {
                    yield yield __await(item);
                }
                else if (at === "last" && index >= count) {
                    yield yield __await(item);
                }
                index++;
            }
        }
        catch (e_1_1) { e_1 = { error: e_1_1 }; }
        finally {
            try {
                if (items_1_1 && !items_1_1.done && (_a = items_1.return)) yield __await(_a.call(items_1));
            }
            finally { if (e_1) throw e_1.error; }
        }
    });
});