package exh.eh

import kotlinx.coroutines.delay
import kotlin.time.Duration
import kotlin.time.Duration.Companion.milliseconds
import kotlin.time.Duration.Companion.seconds

class EHentaiThrottleManager(
    private val max: Duration = THROTTLE_MAX,
    private val inc: Duration = THROTTLE_INC
) {
    private var lastThrottleTime = Duration.ZERO
    var throttleTime = Duration.ZERO
        private set

    suspend fun throttle() {
        // Throttle requests if necessary
        val now = System.currentTimeMillis().milliseconds
        val timeDiff = now - lastThrottleTime
        if (timeDiff < throttleTime) {
            delay(throttleTime - timeDiff)
        }

        if (throttleTime < max) {
            throttleTime += inc
        }

        lastThrottleTime = System.currentTimeMillis().milliseconds
    }

    fun resetThrottle() {
        lastThrottleTime = Duration.ZERO
        throttleTime = Duration.ZERO
    }

    companion object {
        val THROTTLE_MAX = 5.5.seconds
        val THROTTLE_INC = 20.milliseconds
    }
}