ArgProvider.kt

class LoginScreenArg(val id: String)

class BuySellScreenArg(val ppRefId: String,
                       val isModifyFlow: Boolean,
                       val orderId: String?)

inline fun <reified T> SavedStateHandle.getArgs(): T {
    return when (T::class) {
        LoginScreenArg::class -> {
            LoginScreenArg(id = this.get<String>("id")!!) as T
        }
        BuySellScreenArg::class -> {
            BuySellScreenArg(
                ppRefId = get<String>("ppRefId")!!,
                isModifyFlow = get<Boolean>("isModifyFlow")!!,
                orderId = get<String>("orderId")
            ) as T
        }
        else -> throw IllegalArgumentException("Invalid type requested")
    }
}