96 lines
2.6 KiB
JavaScript
96 lines
2.6 KiB
JavaScript
import { database } from '../_database/database'
|
|
import { store } from '../_store/store'
|
|
import { search } from '../_api/search'
|
|
import { SEARCH_RESULTS_LIMIT } from '../_static/autosuggest'
|
|
import { concat } from '../_utils/arrays'
|
|
import uniqBy from 'lodash-es/uniqBy'
|
|
import { scheduleIdleTask } from '../_utils/scheduleIdleTask'
|
|
import { RequestThrottler } from '../_utils/RequestThrottler'
|
|
|
|
const DATABASE_SEARCH_RESULTS_LIMIT = 30
|
|
|
|
function byUsername (a, b) {
|
|
const usernameA = a.acct.toLowerCase()
|
|
const usernameB = b.acct.toLowerCase()
|
|
|
|
return usernameA < usernameB ? -1 : usernameA === usernameB ? 0 : 1
|
|
}
|
|
|
|
function byAccountId (a) {
|
|
return a.id
|
|
}
|
|
|
|
export function doAccountSearch (searchText) {
|
|
let canceled = false
|
|
let localResults
|
|
let remoteResults
|
|
const { currentInstance, accessToken } = store.get()
|
|
const requestThrottler = new RequestThrottler(doSearchAccountsRemotely)
|
|
|
|
async function searchAccountsLocally () {
|
|
localResults = await database.searchAccountsByUsername(
|
|
currentInstance, searchText.substring(1), DATABASE_SEARCH_RESULTS_LIMIT)
|
|
}
|
|
|
|
async function searchAccountsRemotely () {
|
|
remoteResults = await requestThrottler.request()
|
|
}
|
|
|
|
async function doSearchAccountsRemotely (signal) {
|
|
return (await search(
|
|
currentInstance, accessToken, searchText, false, SEARCH_RESULTS_LIMIT, false, signal
|
|
)).accounts
|
|
}
|
|
|
|
function mergeAndTruncateResults () {
|
|
// Always include local results; they are more likely to be relevant
|
|
// because the user has seen their content before. Otherwise, sort by username.
|
|
let results = (localResults || [])
|
|
.slice()
|
|
.sort(byUsername)
|
|
.slice(0, SEARCH_RESULTS_LIMIT)
|
|
|
|
if (results.length < SEARCH_RESULTS_LIMIT) {
|
|
const topRemoteResults = (remoteResults || [])
|
|
.sort(byUsername)
|
|
.slice(0, SEARCH_RESULTS_LIMIT - results.length)
|
|
results = concat(results, topRemoteResults)
|
|
results = uniqBy(results, byAccountId)
|
|
}
|
|
|
|
return results
|
|
}
|
|
|
|
function onNewResults () {
|
|
if (canceled) {
|
|
return
|
|
}
|
|
const results = mergeAndTruncateResults()
|
|
store.setForCurrentAutosuggest({
|
|
autosuggestType: 'account',
|
|
autosuggestSelected: 0,
|
|
autosuggestSearchResults: results
|
|
})
|
|
}
|
|
|
|
function onError (err) {
|
|
console.warn('ignored autosuggest error', err)
|
|
}
|
|
|
|
scheduleIdleTask(() => {
|
|
if (canceled) {
|
|
return
|
|
}
|
|
// run the two searches in parallel
|
|
searchAccountsLocally().then(onNewResults).catch(onError)
|
|
searchAccountsRemotely().then(onNewResults).catch(onError)
|
|
})
|
|
|
|
return {
|
|
cancel: () => {
|
|
canceled = true
|
|
requestThrottler.cancel()
|
|
}
|
|
}
|
|
}
|