"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.take = exports.take_ = void 0
// tracing: off
const A = __importStar(require("../../Collections/Immutable/Chunk"))
const Function_1 = require("../../Function")
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")
const empty_1 = require("./empty")
/**
 * Takes the specified number of elements from this stream.
 */
function take_(self, n) {
  if (n <= 0) {
    return empty_1.empty
  } else {
    return new definitions_1.Stream(
      M.map_(
        M.let_(
          M.bind_(
            M.bind_(M.do, "chunks", () => self.proc),
            "counterRef",
            () => T.toManaged(Ref.makeRef(0))
          ),
          "pull",
          ({ chunks, counterRef }) =>
            T.chain_(counterRef.get, (cnt) => {
              if (cnt >= n) {
                return Pull.end
              } else {
                return T.map_(
                  T.tap_(
                    T.let_(
                      T.bind_(T.do, "chunk", () => chunks),
                      "taken",
                      ({ chunk }) => {
                        if (A.size(chunk) <= n - cnt) {
                          return chunk
                        } else {
                          return A.take_(chunk, n - cnt)
                        }
                      }
                    ),
                    ({ taken }) => counterRef.set(cnt + A.size(taken))
                  ),
                  ({ taken }) => taken
                )
              }
            })
        ),
        ({ pull }) => pull
      )
    )
  }
}
exports.take_ = take_
/**
 * Takes the specified number of elements from this stream.
 */
function take(n) {
  return (self) => take_(self, n)
}
exports.take = take
//# sourceMappingURL=take.js.map
