import androidx.compose.foundation.ExperimentalFoundationApi
import androidx.compose.foundation.layout.Arrangement
import androidx.compose.foundation.layout.Column
import androidx.compose.foundation.layout.padding
import androidx.compose.foundation.text.input.rememberTextFieldState
import androidx.compose.material.OutlinedTextField
import androidx.compose.material.Text
import androidx.compose.runtime.Composable
import androidx.compose.runtime.LaunchedEffect
import androidx.compose.ui.ExperimentalComposeUiApi
import androidx.compose.ui.ImageComposeScene
import androidx.compose.ui.Modifier
import androidx.compose.ui.awt.ComposePanel
import androidx.compose.ui.semantics.SemanticsNode
import androidx.compose.ui.semantics.SemanticsOwner
import androidx.compose.ui.semantics.SemanticsProperties
import androidx.compose.ui.semantics.getOrNull
import androidx.compose.ui.text.AnnotatedString
import androidx.compose.ui.unit.dp
import androidx.compose.ui.window.singleWindowApplication
import java.awt.BorderLayout
import javax.swing.JFrame
import javax.swing.SwingUtilities

@OptIn(ExperimentalFoundationApi::class, ExperimentalComposeUiApi::class)
fun main() = singleWindowApplication {
    SemanticsApp()

    LaunchedEffect(Unit) {
        val strings = mutableListOf<AnnotatedString>()
        window.semanticsOwners.forEach {
            it.rootSemanticsNode.collectTextRecursive(strings)
        }
        println(strings.joinToString("\n"))
    }
}

@OptIn(ExperimentalFoundationApi::class, ExperimentalComposeUiApi::class)
fun main_panel() = SwingUtilities.invokeLater {
    val window = JFrame()
    val composePanel = ComposePanel()
    composePanel.setContent {
        SemanticsApp()
    }

    window.contentPane.add(composePanel, BorderLayout.CENTER)
    window.pack()
    window.isVisible = true

    println(composePanel.semanticsOwners.collectText().joinToString("\n"))
}

@OptIn(ExperimentalFoundationApi::class, ExperimentalComposeUiApi::class)
fun main_image_compose_scene() {
    val imageComposeScene = ImageComposeScene(800, 600) {
        SemanticsApp()
    }
    imageComposeScene.render(0L)

    println(imageComposeScene.semanticsOwners.collectText().joinToString("\n"))
}


@Composable
fun SemanticsApp() {
    Column(
        verticalArrangement = Arrangement.spacedBy(10.dp),
        modifier = Modifier.padding(64.dp)
    ) {
        Text("Hello")
        OutlinedTextField(rememberTextFieldState("World"))
    }
}

private fun Collection<SemanticsOwner>.collectText(): List<AnnotatedString> {
    val result = mutableListOf<AnnotatedString>()
    forEach {
        it.rootSemanticsNode.collectTextRecursive(result)
    }
    return result
}

private fun SemanticsNode.collectTextRecursive(result: MutableList<AnnotatedString>) {
    result.addAll(config.getOrNull(SemanticsProperties.Text) ?: emptyList())
    config.getOrNull(SemanticsProperties.EditableText)?.let {
        result.add(it)
    }
    for (child in children) {
        child.collectTextRecursive(result)
    }
}