"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.throttleShapeM = exports.throttleShapeM_ = void 0
// tracing: off
const CL = __importStar(require("../../Clock"))
const Tp = __importStar(require("../../Collections/Immutable/Tuple"))
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 definitions_1 = require("./definitions")
/**
 * Delays the chunks of this stream according to the given bandwidth parameters using the token bucket
 * algorithm. Allows for burst in the processing of elements by allowing the token bucket to accumulate
 * tokens up to a `units + burst` threshold. The weight of each chunk is determined by the `costFn`
 * effectful function.
 */
function throttleShapeM_(self, costFn, units, duration, burst = 0) {
  return new definitions_1.Stream(
    M.map_(
      M.let_(
        M.bind_(
          M.bind_(
            M.bind_(M.do, "chunks", () => self.proc),
            "currentTime",
            () => T.toManaged(CL.currentTime)
          ),
          "bucket",
          ({ currentTime }) => T.toManaged(Ref.makeRef(Tp.tuple(units, currentTime)))
        ),
        "pull",
        ({ bucket, chunks }) =>
          T.map_(
            T.tap_(
              T.bind_(
                T.bind_(
                  T.bind_(
                    T.bind_(T.do, "chunk", () => chunks),
                    "weight",
                    ({ chunk }) => T.mapError_(costFn(chunk), O.some)
                  ),
                  "current",
                  () => CL.currentTime
                ),
                "delay",
                ({ current, weight }) =>
                  Ref.modify_(bucket, ({ tuple: [tokens, timestamp] }) => {
                    const elapsed = current - timestamp
                    const cycles = elapsed / duration
                    const available = (() => {
                      const sum = tokens + cycles * units
                      const max = units + burst < 0 ? Number.MAX_VALUE : units + burst
                      return sum < 0 ? max : Math.min(sum, max)
                    })()
                    const remaining = available - weight
                    const waitCycles = remaining >= 0 ? 0 : -remaining / units
                    const delay = waitCycles * duration
                    return Tp.tuple(delay, Tp.tuple(remaining, current))
                  })
              ),
              ({ delay }) => T.when_(CL.sleep(delay), () => delay > 0)
            ),
            ({ chunk }) => chunk
          )
      ),
      ({ pull }) => pull
    )
  )
}
exports.throttleShapeM_ = throttleShapeM_
/**
 * Delays the chunks of this stream according to the given bandwidth parameters using the token bucket
 * algorithm. Allows for burst in the processing of elements by allowing the token bucket to accumulate
 * tokens up to a `units + burst` threshold. The weight of each chunk is determined by the `costFn`
 * effectful function.
 *
 * @dataFirst throttleShapeM_
 */
function throttleShapeM(costFn, units, duration, burst = 0) {
  return (self) => throttleShapeM_(self, costFn, units, duration, burst)
}
exports.throttleShapeM = throttleShapeM
//# sourceMappingURL=throttleShapeM.js.map
