@Composable
fun CodeEditor(value: TextFieldValue, onValueChange: (TextFieldValue) -> Unit) {
    TextField(
        textStyle = currentTextStyle().copy(
            fontFamily = FontFamily.Monospace,
            fontSize = 16.sp
        ),
        value = value,
        onValueChange = onValueChange,
        visualTransformation = CodeVisualTransformation()
    )
}

class CodeVisualTransformation : VisualTransformation {
    override fun filter(text: AnnotatedString): TransformedText {

        val transformedText = text.text
            .split("\n")
            .mapIndexed { i, t -> String.format("%2d %s", i + 1, t) }
            .joinToString("\n")

        val gutterSize = 3 // "%2d "

        val codeOffsetTranslator = object : OffsetMap {
            override fun originalToTransformed(offset: Int): Int =
                gutterSize + offset + text.text.take(offset).count { it == '\n' } * gutterSize

            override fun transformedToOriginal(offset: Int): Int =
                offset - gutterSize - transformedText.take(offset).count { it == '\n' } * gutterSize
        }

        val lineNumStyle = SpanStyle(color = Color.LightGray, fontSize = 12.sp)

        val transformedTextStyled = with(AnnotatedString.Builder(transformedText)) {

            """^.{2} """.toRegex(RegexOption.MULTILINE).findAll(transformedText).forEach {
                addStyle(lineNumStyle, it.range.first, it.range.last)
            }

            toAnnotatedString()
        }

        return TransformedText(transformedTextStyled, codeOffsetTranslator)
    }
}