"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
    if (mod && mod.__esModule) return mod;
    var result = {};
    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
    __setModuleDefault(result, mod);
    return result;
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.collectWhile = exports.collectWhile_ = exports.collectWhileMap = exports.collectWhileMap_ = void 0;
// ets_tracing: off
const A = __importStar(require("../../Collections/Immutable/Chunk"));
const Function_1 = require("../../Function");
const O = __importStar(require("../../Option"));
const T = __importStar(require("../_internal/effect"));
const M = __importStar(require("../_internal/managed"));
const Ref = __importStar(require("../_internal/ref"));
const Pull = __importStar(require("../Pull"));
const definitions_1 = require("./definitions");
/**
 * Transforms all elements of the stream for as long as the specified partial function is defined.
 */
function collectWhileMap_(self, f) {
    return new definitions_1.Stream((M.map_(M.let_(M.bind_(M.bind_(M.do, "chunks", () => self.proc), "doneRef", () => T.toManaged(Ref.makeRef(false))), "pull", ({ chunks, doneRef }) => T.chain_(doneRef.get, (done) => {
        if (done) {
            return Pull.end;
        }
        else {
            return (T.chain_(T.bind_(T.do, "chunk", () => chunks), ({ chunk }) => {
                const remaining = A.collectWhile_(chunk, f);
                return T.as_(T.when_(doneRef.set(true), () => A.size(remaining) < A.size(chunk)), remaining);
            }));
        }
    })), ({ pull }) => pull)));
}
exports.collectWhileMap_ = collectWhileMap_;
/**
 * Transforms all elements of the stream for as long as the specified partial function is defined.
 */
function collectWhileMap(f) {
    return (self) => collectWhileMap_(self, f);
}
exports.collectWhileMap = collectWhileMap;
function collectWhile_(self, f) {
    return new definitions_1.Stream((M.map_(M.let_(M.bind_(M.bind_(M.do, "chunks", () => self.proc), "doneRef", () => T.toManaged(Ref.makeRef(false))), "pull", ({ chunks, doneRef }) => T.chain_(doneRef.get, (done) => {
        if (done) {
            return Pull.end;
        }
        else {
            return (T.chain_(T.bind_(T.do, "chunk", () => chunks), ({ chunk }) => {
                const remaining = A.collectWhile_(chunk, O.fromPredicate(f));
                return T.as_(T.when_(doneRef.set(true), () => A.size(remaining) < A.size(chunk)), remaining);
            }));
        }
    })), ({ pull }) => pull)));
}
exports.collectWhile_ = collectWhile_;
function collectWhile(f) {
    return (self) => collectWhile_(self, f);
}
exports.collectWhile = collectWhile;
//# sourceMappingURL=collectWhileMap.js.map