CostaFot

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

    },

CostaFot

@Composable
fun GameScreen(
   gameViewModel: GameViewModel = viewModel()
) {
   // ...
}

CostaFot

@Composable
fun FirstScreen() {
    val customRetainedViewModel = retain {
        CustomRetainedViewModel()
    }
    val state by customRetainedViewModel.state.collectAsStateWithLifecycle()
    // .... 
}

CostaFot

@Composable
fun FirstScreen() {
    val customRetainedViewModel = rememberRetainedViewModel<CustomRetainedViewModel>()
    val state by customRetainedViewModel.state.collectAsStateWithLifecycle()
    // .... rest of the owl
}

CostaFot

@Composable
inline fun <reified T : RetainedViewModel> rememberRetainedViewModel(noinline factory: (Context) -> T): T {
    val context = LocalContext.current
    return retain { factory(context) }
}

// ..compose layer
val customRetainedViewModel = rememberRetainedViewModel { context ->
    EntryPoints.get(context, CustomRetainedViewModelEntryPoint::class.java).customRetainedViewModel()
}

CostaFot

@EntryPoint
@InstallIn(ActivityComponent::class)
interface CustomRetainedViewModelEntryPoint : RetainedViewModelEntryPoint<CustomRetainedViewModel>

@RetainedEntryPoint(CustomRetainedViewModelEntryPoint::class)
class CustomRetainedViewModel @Inject constructor(
    // ...
) : RetainedViewModel() {
    // ...
}

CostaFot

@Target(AnnotationTarget.CLASS)
@Retention(AnnotationRetention.RUNTIME)
annotation class RetainedEntryPoint(
    val value: KClass<out Any>,
)

CostaFot

interface RetainedViewModelEntryPoint<T : RetainedViewModel> {
    fun create(): T
}

@EntryPoint
@InstallIn(ActivityComponent::class)
interface CustomRetainedViewModelEntryPoint : RetainedViewModelEntryPoint<CustomRetainedViewModel>

CostaFot

@EntryPoint
@InstallIn(ActivityComponent::class)
interface CustomRetainedViewModelEntryPoint {
    fun customRetainedViewModel(): CustomRetainedViewModel
}
@Composable 
fun rememberCustomRetainedViewModel(): CustomRetainedViewModel {
    val context = LocalContext.current
    return retain {
        val entryPoint = EntryPoints.get(context, CustomRetainedViewModelEntryPoint::class.java)

CostaFot

interface RetainedViewModelEntryPoint<T : RetainedViewModel> {
    fun create(): T
}