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(): T {
    val context = LocalContext.current
    return retain {
        val annotation =
            T::class.java.getAnnotation(RetainedEntryPoint::class.java)
                ?: error("${T::class} must be annotated with @RetainedEntryPoint")

        val entryPointClass = annotation.value.java
        val entryPoint = EntryPoints.get(context, entryPointClass) as RetainedViewModelEntryPoint<T>

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
}

CostaFot

class CustomRetainedViewModel @Inject constructor(
    private val helloRepository: HelloRepository
) : RetainedViewModel() {