RetainDecorator.kt

RetainDecorator.kt

class RetainDecorator<T : Any>(
    retainedValuesStoreRegistry: RetainedValuesStoreRegistry,
) : NavEntryDecorator<T>(
    decorate = { entry ->
        logDebug { "Retaining ${entry.contentKey}" }
        retainedValuesStoreRegistry.LocalRetainedValuesStoreProvider(entry.contentKey) {
            entry.Content()
        }

    },
    onPop = { contentKey ->
        logDebug { "Popping $contentKey" }
        retainedValuesStoreRegistry.clearChild(contentKey)
    }
)


@Composable
fun rememberRetainDecorator(): RetainDecorator<Any> {
    val retainedValuesStoreRegistry = retainRetainedValuesStoreRegistry()

    return remember {
        RetainDecorator(
            retainedValuesStoreRegistry = retainedValuesStoreRegistry,
        )
    }
}