A composable that allow particular block to be rerendered manually

rerenderable.kt

@Composable
fun <T> rerenderable(
  beforeRerender: () -> Unit = {},
  block: @Composable (rerender: () -> Unit) -> T,
): () -> Unit {
  var id by remember { mutableStateOf(0) }
  val rerender = {
    beforeRerender()
    id += 1
  }

  key(id) { block(rerender) }

  return rerender
}

// Example
@Composable
fun SomeComponent() {
  val values = ...

  rerenderable(beforeRerender = { values = INITIAL_VALUES }) { rerender ->
    MyFormLayout(
      values = formValues,
      onValuesChange = { ... },
      onSubmit = {
        doSomething()
        rerender()
    )
  }
}