// WITH_STDLIB
// FULL_JDK
// TARGET_BACKEND: JVM

import kotlin.coroutines.*
import kotlin.coroutines.intrinsics.*

fun blackhole(vararg a: Any?) {}

val spilledVariables = mutableSetOf<Pair<String, String>>()

var c: Continuation<Unit>? = null

suspend fun saveSpilledVariables() = suspendCoroutineUninterceptedOrReturn<Unit> { continuation ->
    spilledVariables.clear()
    for (field in continuation.javaClass.declaredFields) {
        if (field.name != "label" && (field.name.length != 3 || field.name[1] != '$')) continue
        field.isAccessible = true
        val fieldValue = when (val obj = field.get(continuation)) {
            is Array<*> -> obj.joinToString(prefix = "[", postfix = "]")
            else -> obj
        }
        spilledVariables += field.name to "$fieldValue"
    }
    c = continuation
    COROUTINE_SUSPENDED
}

suspend fun test(check: Boolean) {
    if (check) {
        val a = "a1"
        saveSpilledVariables()
        blackhole(a)
    } else {
        val a = "a2"
        val b = "b2"
        saveSpilledVariables()
        blackhole(a, b)
    }
    // No variable visible - cleanup
    saveSpilledVariables()
}

fun builder(c: suspend () -> Unit) {
    c.startCoroutine(Continuation(EmptyCoroutineContext) {
        it.getOrThrow()
    })
}

fun box(): String {
    builder {
        test(true)
    }

    if (spilledVariables != setOf("label" to "1", "Z$0" to "true", "L$0" to "a1", "L$1" to "null"))
        return "FAIL 1: $spilledVariables"
    c?.resume(Unit)
    if (spilledVariables != setOf("label" to "3", "Z$0" to "true", "L$0" to "null", "L$1" to "null"))
        return "FAIL 2: $spilledVariables"
    c?.resume(Unit)
    if (spilledVariables != setOf("label" to "3", "Z$0" to "true", "L$0" to "null", "L$1" to "null"))
        return "FAIL 3: $spilledVariables"

    builder {
        test(false)
    }

    if (spilledVariables != setOf("label" to "2", "Z$0" to "false", "L$0" to "a2", "L$1" to "b2")) return "FAIL 4: $spilledVariables"
    c?.resume(Unit)
    if (spilledVariables != setOf("label" to "3", "Z$0" to "false", "L$0" to "null", "L$1" to "null")) return "FAIL 5: $spilledVariables"
    c?.resume(Unit)
    if (spilledVariables != setOf("label" to "3", "Z$0" to "false", "L$0" to "null", "L$1" to "null")) return "FAIL 6: $spilledVariables"

    return "OK"
}