124 lines
4.7 KiB
Kotlin
124 lines
4.7 KiB
Kotlin
/*
|
|
* Twidere - Twitter client for Android
|
|
*
|
|
* Copyright (C) 2012-2014 Mariotaku Lee <mariotaku.lee@gmail.com>
|
|
*
|
|
* 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.
|
|
*
|
|
* This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package org.mariotaku.twidere.loader
|
|
|
|
import android.content.Context
|
|
import android.database.sqlite.SQLiteDatabase
|
|
import android.support.annotation.WorkerThread
|
|
import org.mariotaku.microblog.library.MicroBlog
|
|
import org.mariotaku.microblog.library.MicroBlogException
|
|
import org.mariotaku.microblog.library.twitter.model.Paging
|
|
import org.mariotaku.microblog.library.twitter.model.SearchQuery
|
|
import org.mariotaku.microblog.library.twitter.model.Status
|
|
import org.mariotaku.microblog.library.twitter.model.UniversalSearchQuery
|
|
import org.mariotaku.twidere.annotation.AccountType
|
|
import org.mariotaku.twidere.extension.model.official
|
|
import org.mariotaku.twidere.model.AccountDetails
|
|
import org.mariotaku.twidere.model.ParcelableStatus
|
|
import org.mariotaku.twidere.model.UserKey
|
|
import org.mariotaku.twidere.util.InternalTwitterContentUtils
|
|
|
|
open class TweetSearchLoader(
|
|
context: Context,
|
|
accountKey: UserKey?,
|
|
private val query: String?,
|
|
sinceId: String?,
|
|
maxId: String?,
|
|
page: Int,
|
|
adapterData: List<ParcelableStatus>?,
|
|
savedStatusesArgs: Array<String>?,
|
|
tabPosition: Int,
|
|
fromUser: Boolean,
|
|
override val isGapEnabled: Boolean,
|
|
loadingMore: Boolean
|
|
) : MicroBlogAPIStatusesLoader(context, accountKey, sinceId, maxId, page, adapterData, savedStatusesArgs,
|
|
tabPosition, fromUser, loadingMore) {
|
|
|
|
@Throws(MicroBlogException::class)
|
|
override fun getStatuses(microBlog: MicroBlog,
|
|
details: AccountDetails,
|
|
paging: Paging): List<Status> {
|
|
if (query == null) throw MicroBlogException("Empty query")
|
|
val queryText = processQuery(details, query)
|
|
when (details.type) {
|
|
AccountType.TWITTER -> {
|
|
if (details.extras?.official ?: false) {
|
|
val universalQuery = UniversalSearchQuery(queryText)
|
|
universalQuery.setModules(UniversalSearchQuery.Module.TWEET)
|
|
universalQuery.setResultType(UniversalSearchQuery.ResultType.RECENT)
|
|
universalQuery.setPaging(paging)
|
|
val searchResult = microBlog.universalSearch(universalQuery)
|
|
return searchResult.modules.mapNotNull { it.status?.data }
|
|
}
|
|
|
|
val searchQuery = SearchQuery(queryText)
|
|
searchQuery.paging(paging)
|
|
return microBlog.search(searchQuery)
|
|
}
|
|
AccountType.STATUSNET -> {
|
|
return microBlog.searchStatuses(queryText, paging)
|
|
}
|
|
AccountType.FANFOU -> {
|
|
return microBlog.searchPublicTimeline(queryText, paging)
|
|
}
|
|
}
|
|
throw MicroBlogException("Not implemented")
|
|
}
|
|
|
|
protected open fun processQuery(details: AccountDetails, query: String): String {
|
|
if (details.type == AccountType.TWITTER) {
|
|
if (details.extras?.official ?: false) {
|
|
return smQuery(query)
|
|
}
|
|
return "$query exclude:retweets"
|
|
}
|
|
return query
|
|
}
|
|
|
|
protected fun smQuery(query: String): String {
|
|
var universalQueryText = query
|
|
if (maxId != null) {
|
|
universalQueryText += " max_id:$maxId"
|
|
}
|
|
if (sinceId != null) {
|
|
universalQueryText += " since_id:$sinceId"
|
|
}
|
|
return universalQueryText
|
|
}
|
|
|
|
@WorkerThread
|
|
override fun shouldFilterStatus(database: SQLiteDatabase, status: ParcelableStatus): Boolean {
|
|
return InternalTwitterContentUtils.isFiltered(database, status, true)
|
|
}
|
|
|
|
override fun processPaging(details: AccountDetails, loadItemLimit: Int, paging: Paging) {
|
|
if (details.type == AccountType.STATUSNET) {
|
|
paging.setRpp(loadItemLimit)
|
|
val page = page
|
|
if (page > 0) {
|
|
paging.setPage(page)
|
|
}
|
|
} else {
|
|
super.processPaging(details, loadItemLimit, paging)
|
|
}
|
|
}
|
|
|
|
}
|