SafeApiCall.kt

suspend fun <D, E> safeApiCall(
    apiCall: suspend () -> D,
    onError: suspend (StatusCode) -> E? = { null }
): DataResult<D, E> {
    return withContext(Dispatchers.IO) {
        try {
            DataResult.Success(apiCall.invoke())
        } catch (e: IOException) {
            when (val appError = e.mapToAppError()) {
                is AppError.NetworkError -> when (val error = onError(appError.statusCode)) {
                    null -> throw e
                    else -> DataResult.Error(error)
                }

                else -> throw e
            }
        }
    }
}