/* Copyright 2021 Tusky Contributors
 *
 * This file is a part of Tusky.
 *
 * This program is free software; you can redistribute it and/or modify it under the terms of the
 * GNU General Public License as published by the Free Software Foundation; either version 3 of the
 * License, or (at your option) any later version.
 *
 * Tusky is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even
 * the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
 * Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with Tusky; if not,
 * see <http://www.gnu.org/licenses>. */

package com.keylesspalace.tusky.components.search

import android.util.Log
import androidx.lifecycle.viewModelScope
import androidx.paging.Pager
import androidx.paging.PagingConfig
import androidx.paging.cachedIn
import com.keylesspalace.tusky.components.instanceinfo.InstanceInfoRepository.Companion.CAN_USE_QUOTE_ID
import com.keylesspalace.tusky.components.search.adapter.SearchNotestockPagingSourceFactory
import com.keylesspalace.tusky.components.search.adapter.SearchPagingSourceFactory
import com.keylesspalace.tusky.db.AccountEntity
import com.keylesspalace.tusky.db.AccountManager
import com.keylesspalace.tusky.entity.DeletedStatus
import com.keylesspalace.tusky.entity.Poll
import com.keylesspalace.tusky.entity.Status
import com.keylesspalace.tusky.network.MastodonApi
import com.keylesspalace.tusky.network.NotestockApi
import com.keylesspalace.tusky.usecase.TimelineCases
import com.keylesspalace.tusky.util.RxAwareViewModel
import com.keylesspalace.tusky.util.toViewData
import com.keylesspalace.tusky.viewdata.StatusViewData
import io.reactivex.rxjava3.android.schedulers.AndroidSchedulers
import io.reactivex.rxjava3.core.Single
import javax.inject.Inject

class SearchViewModel @Inject constructor(
    mastodonApi: MastodonApi,
    notestockApi: NotestockApi,
    private val timelineCases: TimelineCases,
    private val accountManager: AccountManager,
) : RxAwareViewModel() {

    var currentQuery: String = ""

    var activeAccount: AccountEntity?
        get() = accountManager.activeAccount
        set(value) {
            accountManager.activeAccount = value
        }

    val mediaPreviewEnabled = activeAccount?.mediaPreviewEnabled ?: false
    val alwaysShowSensitiveMedia = activeAccount?.alwaysShowSensitiveMedia ?: false
    val alwaysOpenSpoiler = activeAccount?.alwaysOpenSpoiler ?: false
    val quoteEnabled = activeAccount?.domain in CAN_USE_QUOTE_ID

    private val loadedStatuses: MutableList<StatusViewData.Concrete> = mutableListOf()
    private val loadedNotestockStatuses: MutableList<StatusViewData.Concrete> = mutableListOf()

    private val statusesPagingSourceFactory = SearchPagingSourceFactory(mastodonApi, SearchType.Status, loadedStatuses) {
        it.statuses.map { status ->
            status.toViewData(
                isShowingContent = alwaysShowSensitiveMedia || !status.actionableStatus.sensitive,
                isExpanded = alwaysOpenSpoiler,
                isCollapsed = true
            )
        }.apply {
            loadedStatuses.addAll(this)
        }
    }
    private val accountsPagingSourceFactory = SearchPagingSourceFactory(mastodonApi, SearchType.Account) {
        it.accounts
    }
    private val hashtagsPagingSourceFactory = SearchPagingSourceFactory(mastodonApi, SearchType.Hashtag) {
        it.hashtags
    }
    private val notestockStatusesPagingSourceFactory = SearchNotestockPagingSourceFactory(notestockApi) {
        it.statuses.map { status ->
            status.toViewData(
                isShowingContent = alwaysShowSensitiveMedia || !status.actionableStatus.sensitive,
                isExpanded = alwaysOpenSpoiler,
                isCollapsed = true
            )
        }
            .apply {
                loadedNotestockStatuses.addAll(this)
            }
    }

    val statusesFlow = Pager(
        config = PagingConfig(pageSize = DEFAULT_LOAD_SIZE, initialLoadSize = DEFAULT_LOAD_SIZE),
        pagingSourceFactory = statusesPagingSourceFactory
    ).flow
        .cachedIn(viewModelScope)

    val accountsFlow = Pager(
        config = PagingConfig(pageSize = DEFAULT_LOAD_SIZE, initialLoadSize = DEFAULT_LOAD_SIZE),
        pagingSourceFactory = accountsPagingSourceFactory
    ).flow
        .cachedIn(viewModelScope)

    val hashtagsFlow = Pager(
        config = PagingConfig(pageSize = DEFAULT_LOAD_SIZE, initialLoadSize = DEFAULT_LOAD_SIZE),
        pagingSourceFactory = hashtagsPagingSourceFactory
    ).flow
        .cachedIn(viewModelScope)

    val notestockStatusesFlow = Pager(
        config = PagingConfig(pageSize = DEFAULT_LOAD_SIZE, initialLoadSize = DEFAULT_LOAD_SIZE),
        pagingSourceFactory = notestockStatusesPagingSourceFactory
    ).flow
        .cachedIn(viewModelScope)

    fun search(query: String) {
        loadedStatuses.clear()
        statusesPagingSourceFactory.newSearch(query)
        accountsPagingSourceFactory.newSearch(query)
        hashtagsPagingSourceFactory.newSearch(query)
        notestockStatusesPagingSourceFactory.newSearch(query)
    }

    fun removeItem(statusViewData: StatusViewData.Concrete) {
        timelineCases.delete(statusViewData.id)
            .subscribe(
                {
                    if (loadedStatuses.remove(statusViewData))
                        statusesPagingSourceFactory.invalidate()
                },
                { err ->
                    Log.d(TAG, "Failed to delete status", err)
                }
            )
            .autoDispose()
    }

    fun removeNotestockItem(statusViewData: StatusViewData.Concrete) {
        if (loadedNotestockStatuses.remove(statusViewData)) {
            notestockStatusesPagingSourceFactory.invalidate()
        }
    }

    fun expandedChange(statusViewData: StatusViewData.Concrete, expanded: Boolean) {
        val idx = loadedStatuses.indexOf(statusViewData)
        if (idx >= 0) {
            loadedStatuses[idx] = statusViewData.copy(isExpanded = expanded)
            statusesPagingSourceFactory.invalidate()
        }
    }

    fun expandedNotestockChange(statusViewData: StatusViewData.Concrete, expanded: Boolean) {
        val idx = loadedNotestockStatuses.indexOf(statusViewData)
        if (idx >= 0) {
            loadedNotestockStatuses[idx] = statusViewData.copy(isExpanded = expanded)
            notestockStatusesPagingSourceFactory.invalidate()
        }
    }

    fun reblog(statusViewData: StatusViewData.Concrete, reblog: Boolean) {
        timelineCases.reblog(statusViewData.id, reblog)
            .observeOn(AndroidSchedulers.mainThread())
            .subscribe(
                { setRebloggedForStatus(statusViewData, reblog) },
                { t -> Log.d(TAG, "Failed to reblog status ${statusViewData.id}", t) }
            )
            .autoDispose()
    }

    private fun setRebloggedForStatus(statusViewData: StatusViewData.Concrete, reblog: Boolean) {
        statusViewData.status.reblogged = reblog
        statusViewData.status.reblog?.reblogged = reblog
        statusesPagingSourceFactory.invalidate()
    }

    fun contentHiddenChange(statusViewData: StatusViewData.Concrete, isShowing: Boolean) {
        val idx = loadedStatuses.indexOf(statusViewData)
        if (idx >= 0) {
            loadedStatuses[idx] = statusViewData.copy(isShowingContent = isShowing)
            statusesPagingSourceFactory.invalidate()
        }
    }

    fun contentHiddenNotestockChange(statusViewData: StatusViewData.Concrete, isShowing: Boolean) {
        val idx = loadedNotestockStatuses.indexOf(statusViewData)
        if (idx >= 0) {
            loadedNotestockStatuses[idx] = statusViewData.copy(isShowingContent = isShowing)
            notestockStatusesPagingSourceFactory.invalidate()
        }
    }

    fun collapsedChange(statusViewData: StatusViewData.Concrete, collapsed: Boolean) {
        val idx = loadedStatuses.indexOf(statusViewData)
        if (idx >= 0) {
            loadedStatuses[idx] = statusViewData.copy(isCollapsed = collapsed)
            statusesPagingSourceFactory.invalidate()
        }
    }

    fun collapsedNotestockChange(statusViewData: StatusViewData.Concrete, collapsed: Boolean) {
        val idx = loadedNotestockStatuses.indexOf(statusViewData)
        if (idx >= 0) {
            loadedNotestockStatuses[idx] = statusViewData.copy(isCollapsed = collapsed)
            notestockStatusesPagingSourceFactory.invalidate()
        }
    }

    fun voteInPoll(statusViewData: StatusViewData.Concrete, choices: MutableList<Int>) {
        val votedPoll = statusViewData.status.actionableStatus.poll!!.votedCopy(choices)
        updateStatus(statusViewData, votedPoll)
        timelineCases.voteInPoll(statusViewData.id, votedPoll.id, choices)
            .observeOn(AndroidSchedulers.mainThread())
            .subscribe(
                { newPoll -> updateStatus(statusViewData, newPoll) },
                { t -> Log.d(TAG, "Failed to vote in poll: ${statusViewData.id}", t) }
            )
            .autoDispose()
    }

    private fun updateStatus(statusViewData: StatusViewData.Concrete, newPoll: Poll) {
        val idx = loadedStatuses.indexOf(statusViewData)
        if (idx >= 0) {
            val newStatus = statusViewData.status.copy(poll = newPoll)
            loadedStatuses[idx] = statusViewData.copy(status = newStatus)
            statusesPagingSourceFactory.invalidate()
        }
    }

    fun favorite(statusViewData: StatusViewData.Concrete, isFavorited: Boolean) {
        statusViewData.status.favourited = isFavorited
        statusesPagingSourceFactory.invalidate()
        timelineCases.favourite(statusViewData.id, isFavorited)
            .onErrorReturnItem(statusViewData.status)
            .subscribe()
            .autoDispose()
    }

    fun bookmark(statusViewData: StatusViewData.Concrete, isBookmarked: Boolean) {
        statusViewData.status.bookmarked = isBookmarked
        statusesPagingSourceFactory.invalidate()
        timelineCases.bookmark(statusViewData.id, isBookmarked)
            .onErrorReturnItem(statusViewData.status)
            .subscribe()
            .autoDispose()
    }

    fun getAllAccountsOrderedByActive(): List<AccountEntity> {
        return accountManager.getAllAccountsOrderedByActive()
    }

    fun muteAccount(accountId: String, notifications: Boolean, duration: Int?) {
        timelineCases.mute(accountId, notifications, duration)
    }

    fun pinAccount(status: Status, isPin: Boolean) {
        timelineCases.pin(status.id, isPin)
    }

    fun blockAccount(accountId: String) {
        timelineCases.block(accountId)
    }

    fun deleteStatus(id: String): Single<DeletedStatus> {
        return timelineCases.delete(id)
    }

    fun muteConversation(statusViewData: StatusViewData.Concrete, mute: Boolean) {
        val idx = loadedStatuses.indexOf(statusViewData)
        if (idx >= 0) {
            val newStatus = statusViewData.status.copy(muted = mute)
            loadedStatuses[idx] = statusViewData.copy(status = newStatus)
            statusesPagingSourceFactory.invalidate()
        }
        timelineCases.muteConversation(statusViewData.id, mute)
            .onErrorReturnItem(statusViewData.status)
            .subscribe()
            .autoDispose()
    }

    companion object {
        private const val TAG = "SearchViewModel"
        private const val DEFAULT_LOAD_SIZE = 20
    }
}