demo.kt

file addLine "override fun get(bundle: Bundle, key: String): "
addVariableType(file, propertyInfo)
file addPhrase "? {"
tabs++
when (propertyInfo.composeArgumentType) {
    ComposeArgumentType.PARCELABLE -> file addLine "return bundle.getParcelable(key)"
    ComposeArgumentType.PARCELABLE_ARRAY -> file addLine "return bundle.getParcelableArrayList(key)"
    ComposeArgumentType.SERIALIZABLE -> {
        file addLine "return bundle.getSerializable(key) as? "
        addVariableType(file, propertyInfo)
    }
}
tabs--
file addLine "}"

file addLine "override fun put(bundle: Bundle, key: String, value: "
addVariableType(file, propertyInfo)
file addPhrase ") {"
tabs++
when (propertyInfo.composeArgumentType) {
    ComposeArgumentType.PARCELABLE -> file addLine "bundle.putParcelable(key, value)"
    ComposeArgumentType.PARCELABLE_ARRAY -> file addLine "bundle.putParcelableArrayList(key, value)"
    ComposeArgumentType.SERIALIZABLE -> file addLine "bundle.putSerializable(key, value)"
}
tabs--
file addLine "}"