"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.branchAfter = exports.branchAfter_ = void 0;
// ets_tracing: off
const CK = __importStar(require("../../../../Collections/Immutable/Chunk"));
const CH = __importStar(require("../../Channel"));
const C = __importStar(require("../core"));
const Empty = __importStar(require("./empty"));
const FromChunk = __importStar(require("./fromChunk"));
/**
 * Reads the first n values from the stream and uses them to choose the pipeline that will be
 * used for the remainder of the stream.
 */
function branchAfter_(self, n, f) {
    const collecting = (buf) => CH.readWithCause((chunk) => {
        const newBuf = CK.concat_(buf, chunk);
        if (CK.size(newBuf) >= n) {
            const { tuple: [is, is1] } = CK.splitAt_(newBuf, n);
            const pipeline = f(is);
            return CH.zipRight_(pipeline(FromChunk.fromChunk(is1)).channel, emitting(pipeline));
        }
        else {
            return collecting(newBuf);
        }
    }, (_) => CH.failCause(_), (_) => {
        if (CK.isEmpty(buf)) {
            return CH.unit;
        }
        else {
            const pipeline = f(buf);
            return pipeline(Empty.empty).channel;
        }
    });
    const emitting = (pipeline) => CH.readWithCause((chunk) => CH.zipRight_(pipeline(FromChunk.fromChunk(chunk)).channel, emitting(pipeline)), (_) => CH.failCause(_), (_) => CH.unit);
    return new C.Stream(self.channel[">>>"](collecting(CK.empty())));
}
exports.branchAfter_ = branchAfter_;
/**
 * Reads the first n values from the stream and uses them to choose the pipeline that will be
 * used for the remainder of the stream.
 *
 * @ets_data_first branchAfter_
 */
function branchAfter(n, f) {
    return (self) => branchAfter_(self, n, f);
}
exports.branchAfter = branchAfter;
//# sourceMappingURL=branchAfter.js.map