session-android/libsession-util/src/main/java/network/loki/messenger/libsession_util/util/ExpiryMode.kt

package network.loki.messenger.libsession_util.util

import kotlin.time.Duration.Companion.seconds

sealed class ExpiryMode(val expirySeconds: Long) {
    object NONE: ExpiryMode(0)
    data class Legacy(private val seconds: Long = 0L): ExpiryMode(seconds)
    data class AfterSend(private val seconds: Long = 0L): ExpiryMode(seconds)
    data class AfterRead(private val seconds: Long = 0L): ExpiryMode(seconds)

    val duration get() = expirySeconds.seconds

    val expiryMillis get() = expirySeconds * 1000L

    fun coerceSendToRead(coerce: Boolean = true) = if (coerce && this is AfterSend) AfterRead(expirySeconds) else this
}

fun afterSend(seconds: Long) = seconds.takeIf { it > 0 }?.let(ExpiryMode::AfterSend) ?: ExpiryMode.NONE