fun main() = singleWindowApplication {
    var background by remember { mutableStateOf(Color.WHITE) }
    val coroutineScope = rememberCoroutineScope()
    JPopupContextMenu(
        background = background,
        onContextMenuCreated = {
            coroutineScope.launch {
                delay(2000)
                background = Color(Random.nextInt())
            }
        }
    ) {
        Box(Modifier.fillMaxSize(), contentAlignment = Alignment.Center) {
            TextField(rememberTextFieldState("abc"), lineLimits = TextFieldLineLimits.SingleLine)
        }
    }
}

@OptIn(ExperimentalFoundationApi::class)
@Composable
private fun WindowScope.JPopupContextMenu(
    background: java.awt.Color,
    onContextMenuCreated: () -> Unit,
    content: @Composable () -> Unit
) {
    CompositionLocalProvider(
        LocalTextContextMenu provides JPopupTextMenu(window) { textManager, items ->
            onContextMenuCreated()

            fun swingMenuItem(text: String, action: TextContextMenu.Action) = JMenuItem(text).also {
                it.background = background
                it.addActionListener { action.execute() }
            }
            JPopupMenu().apply {
                textManager.cut?.also {
                    add(swingMenuItem("Cut", it))
                }
                textManager.copy?.also {
                    add(swingMenuItem("Copy", it))
                }
                textManager.paste?.also {
                    add(swingMenuItem("Paste", it))
                }
                textManager.selectAll?.also {
                    add(swingMenuItem("Select All", it))
                }
            }
        },
        content = content
    )
}