package world.respect.lib.dataloadstate.ext

import kotlinx.coroutines.flow.Flow
import kotlinx.coroutines.flow.combine
import world.respect.lib.dataloadstate.DataErrorResult
import world.respect.lib.dataloadstate.DataLoadMetaInfo
import world.respect.lib.dataloadstate.DataReadyState
import world.respect.lib.dataloadstate.DataLoadState
import world.respect.lib.dataloadstate.DataLoadingState
import world.respect.lib.dataloadstate.NoDataLoadedState

fun <T: Any> DataLoadState<T>.combineWithRemote(
    remote: DataLoadState<*>
): DataLoadState<T> {
    return this.copyLoadState(
        metaInfo = metaInfo.copy(
            url = remote.metaInfo.url
        ),
        localState = this,
        remoteState = remote
    )
}

fun <T: Any> DataLoadState<T>.combineWithRemoteIfNotNull(
    remote: DataLoadState<*>?
): DataLoadState<T> {
    return if(remote != null) {
        this.combineWithRemote(remote)
    }else {
        this
    }
}

fun <T: Any> Flow<DataLoadState<T>>.combineWithRemote(
    remoteFlow: Flow<DataLoadState<*>>,
): Flow<DataLoadState<T>> =combine(remoteFlow) { local, remote ->
    local.combineWithRemote(remote)
}

fun <T: Any> DataLoadState<T>.copyLoadState(
    metaInfo: DataLoadMetaInfo = this.metaInfo,
    localState: DataLoadState<T>? = this.localState,
    remoteState: DataLoadState<*>? = this.remoteState,
) : DataLoadState<T> {
    return when(this) {
        is DataReadyState -> copy(
            metaInfo = metaInfo,
            localState = localState,
            remoteState = remoteState,
        )
        is DataLoadingState -> copy(
            metaInfo = metaInfo,
            localState = localState,
            remoteState = remoteState,
        )
        is DataErrorResult -> copy(
            metaInfo = metaInfo,
            localState = localState,
            remoteState = remoteState,
        )
        is NoDataLoadedState -> copy(
            metaInfo = metaInfo,
            localState = localState,
            remoteState = remoteState,
        )
    }
}

fun <T: Any> DataLoadState<T>.dataOrNull(): T? {
    return (this as? DataReadyState)?.data ?: (this as? DataLoadingState)?.partialData
}


/**
 *  A DataReadyState may be provided when local data is available, however update checks are still
 *  going on in the background. In the case of an edit screen, we may want to show local data to
 *  a user but not allow editing until the remote data is check is done (if possible).
 *
 *  @return true if the DataLoadState is DataReadyState and there are no pending remote updates.
 */
fun DataLoadState<*>.isReadyAndSettled(): Boolean {
    return this is DataReadyState && this.remoteState !is DataLoadingState
}

fun DataLoadState<*>.isLoadedOrNotModified(): Boolean {
    return this is DataReadyState ||
            (this is NoDataLoadedState && reason == NoDataLoadedState.Reason.NOT_MODIFIED)
}

fun <T: Any, R: Any> DataLoadState<T>.map(
    transform: (T) -> R
): DataLoadState<R> {
    return when(this) {
        is DataReadyState -> {
            DataReadyState(
                data = transform(data),
                metaInfo = metaInfo,
                localState = localState?.map(transform),
                remoteState = remoteState,
            )
        }
        is DataLoadingState -> {
            DataLoadingState(
                metaInfo = metaInfo,
                localState = localState?.map(transform),
                remoteState = remoteState,
            )
        }
        is DataErrorResult -> {
            DataErrorResult(
                error = error,
                metaInfo = metaInfo,
                localState = localState?.map(transform),
                remoteState = remoteState,
            )
        }
        is NoDataLoadedState -> {
            NoDataLoadedState(
                reason = reason,
                metaInfo = metaInfo,
                localState = localState?.map(transform),
                remoteState = remoteState,
            )
        }
    }
}

/**
 * Sometimes a list http endpoint is used, even though there is only going to be one result at most
 * e.g. Person when searched for a specific guid or username.
 *
 * @receiver a DataLoadState
 * @return If the DataLoadState is DataReadyState, and the list has at least one item, then a
 *         DataReadyState with the first item.
 *         If the DataLoadState is DataReadyState and the list is empty, then a NoDataLoadedState
 *         Otherwise, the original DataLoadState
 *
 */
fun <T: Any> DataLoadState<List<T>>.firstOrNotLoaded(): DataLoadState<T> {
    return when {
        this is DataReadyState && this.data.isNotEmpty() -> {
            DataReadyState(
                data = data.first(),
                metaInfo = metaInfo,
                localState = localState?.firstOrNotLoaded(),
                remoteState = remoteState,
            )
        }

        this is DataReadyState -> {
            NoDataLoadedState(
                reason = NoDataLoadedState.Reason.NOT_FOUND,
                metaInfo = metaInfo,
                localState = localState?.firstOrNotLoaded(),
                remoteState = remoteState,
            )
        }

        this is DataLoadingState -> {
            DataLoadingState(
                metaInfo = metaInfo,
                localState = localState?.firstOrNotLoaded(),
                remoteState = remoteState,
            )
        }

        this is DataErrorResult -> {
            DataErrorResult(
                error = error,
                metaInfo = metaInfo,
                localState = localState?.firstOrNotLoaded(),
                remoteState = remoteState,
            )
        }

        this is NoDataLoadedState -> {
            NoDataLoadedState(
                reason = reason,
                metaInfo = metaInfo,
                localState = localState?.firstOrNotLoaded(),
                remoteState = remoteState,
            )
        }

        else -> {
            throw IllegalStateException("Unknown type: not really possible")
        }
    }
}

/**
 * REST API HTTP server endpoints SHOULD use
 */
fun DataLoadState<*>.lastModifiedForHttpResponseHeader(): Long? {
    return when {
        metaInfo.lastStored > 0 -> {
            metaInfo.lastStored
        }

        metaInfo.lastModified > 0 -> {
            metaInfo.lastModified
        }

        else -> null
    }
}

