Destination.kt

abstract class Destination(
    ...
) {

    private var parcelableArguments: HashMap<String, Parcelable> = hashMapOf()

    protected fun saveParcelableArgument(key: String, argument: Parcelable) {
        parcelableArguments[key] = argument
    }
@Suppress("UNCHECKED_CAST")
protected fun <T> NavBackStackEntry.extract(key: String): T? {
    val argument = extendedArguments.firstOrNull { it.name == key }?.argument
    return (when (argument?.type) {
        NavType.IntType -> {
            arguments?.getInt(key) ?: argument.defaultValue
        }
        NavType.FloatType -> {
            arguments?.getFloat(key) ?: argument.defaultValue
        }
        NavType.LongType -> {
            arguments?.getLong(key) ?: argument.defaultValue
        }
        NavType.BoolType -> {
            arguments?.getBoolean(key) ?: argument.defaultValue
        }
        NavType.StringType -> {
            arguments?.getString(key) ?: argument.defaultValue
        }
        else -> {
            parcelableArguments[key]
        }
    }) as? T
}
    @Suppress("UNCHECKED_CAST")
    @Composable
    fun <T: DestinationArguments> rememberArguments(navBackStackEntry: NavBackStackEntry): T {
        parcelableArguments = rememberSaveable {
            parcelableArguments
        }
        return remember {
            getArguments(navBackStackEntry) ?: EmptyArguments
        } as T
    }

    protected open fun getArguments(navBackStackEntry: NavBackStackEntry): DestinationArguments? {
        return null
    }
}