whenAsReceiver.kt

fun <T, R> T.also(block: () -> R): R {
    return null!!
}

fun foo(b: Boolean, a: Int) {
    val x = when (b) {
        true -> a
        else -> null
    }?.also {
        1
    }
}