287 lines
11 KiB
Kotlin
287 lines
11 KiB
Kotlin
/* 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
|
|
}
|
|
}
|