Searching: Split out the search methods.

This splits out the search methods out into ones that process the search
result and ones that do not instead of toggling the return format using
a boolean.
pull/21833/head
Damir Jelić 2020-06-10 10:09:16 +02:00
parent 5dfe5ac135
commit 632f54d126
1 changed files with 42 additions and 38 deletions

View File

@ -19,7 +19,7 @@ import {MatrixClientPeg} from "./MatrixClientPeg";
const SEARCH_LIMIT = 10; const SEARCH_LIMIT = 10;
async function serverSideSearch(term, roomId = undefined, processResult = true) { async function serverSideSearch(term, roomId = undefined) {
const client = MatrixClientPeg.get(); const client = MatrixClientPeg.get();
const filter = { const filter = {
@ -45,16 +45,6 @@ async function serverSideSearch(term, roomId = undefined, processResult = true)
const response = await client.search({body: body}); const response = await client.search({body: body});
if (processResult) {
const searchResult = {
_query: body,
results: [],
highlights: [],
};
return client._processRoomEventsSearch(searchResult, response);
}
const result = { const result = {
response: response, response: response,
query: body, query: body,
@ -63,6 +53,19 @@ async function serverSideSearch(term, roomId = undefined, processResult = true)
return result; return result;
} }
async function serverSideSearchProcess(term, roomId = undefined) {
const client = MatrixClientPeg.get();
const result = await serverSideSearch(term, roomId);
const searchResult = {
_query: result.query,
results: [],
highlights: [],
};
return client._processRoomEventsSearch(searchResult, result.response);
}
function compareEvents(a, b) { function compareEvents(a, b) {
const aEvent = a.result; const aEvent = a.result;
const bEvent = b.result; const bEvent = b.result;
@ -78,8 +81,8 @@ async function combinedSearch(searchTerm) {
// Create two promises, one for the local search, one for the // Create two promises, one for the local search, one for the
// server-side search. // server-side search.
const serverSidePromise = serverSideSearch(searchTerm, undefined, false); const serverSidePromise = serverSideSearch(searchTerm, undefined);
const localPromise = localSearch(searchTerm, undefined, false); const localPromise = localSearch(searchTerm, undefined);
// Wait for both promises to resolve. // Wait for both promises to resolve.
await Promise.all([serverSidePromise, localPromise]); await Promise.all([serverSidePromise, localPromise]);
@ -121,6 +124,8 @@ async function combinedSearch(searchTerm) {
} }
async function localSearch(searchTerm, roomId = undefined, processResult = true) { async function localSearch(searchTerm, roomId = undefined, processResult = true) {
const eventIndex = EventIndexPeg.get();
const searchArgs = { const searchArgs = {
search_term: searchTerm, search_term: searchTerm,
before_limit: 1, before_limit: 1,
@ -134,30 +139,8 @@ async function localSearch(searchTerm, roomId = undefined, processResult = true)
searchArgs.room_id = roomId; searchArgs.room_id = roomId;
} }
const emptyResult = {
seshatQuery: searchArgs,
results: [],
highlights: [],
};
if (searchTerm === "") return emptyResult;
const eventIndex = EventIndexPeg.get();
const localResult = await eventIndex.search(searchArgs); const localResult = await eventIndex.search(searchArgs);
if (processResult) {
emptyResult.seshatQuery.next_batch = localResult.next_batch;
const response = {
search_categories: {
room_events: localResult,
},
};
return MatrixClientPeg.get()._processRoomEventsSearch(emptyResult, response);
}
searchArgs.next_batch = localResult.next_batch; searchArgs.next_batch = localResult.next_batch;
const result = { const result = {
@ -168,6 +151,27 @@ async function localSearch(searchTerm, roomId = undefined, processResult = true)
return result; return result;
} }
async function localSearchProcess(searchTerm, roomId = undefined) {
const emptyResult = {
results: [],
highlights: [],
};
if (searchTerm === "") return emptyResult;
const result = await localSearch(searchTerm, roomId);
emptyResult.seshatQuery = result.query;
const response = {
search_categories: {
room_events: result.response,
},
};
return MatrixClientPeg.get()._processRoomEventsSearch(emptyResult, response);
}
async function localPagination(searchResult) { async function localPagination(searchResult) {
const eventIndex = EventIndexPeg.get(); const eventIndex = EventIndexPeg.get();
@ -469,11 +473,11 @@ function eventIndexSearch(term, roomId = undefined) {
if (MatrixClientPeg.get().isRoomEncrypted(roomId)) { if (MatrixClientPeg.get().isRoomEncrypted(roomId)) {
// The search is for a single encrypted room, use our local // The search is for a single encrypted room, use our local
// search method. // search method.
searchPromise = localSearch(term, roomId); searchPromise = localSearchProcess(term, roomId);
} else { } else {
// The search is for a single non-encrypted room, use the // The search is for a single non-encrypted room, use the
// server-side search. // server-side search.
searchPromise = serverSideSearch(term, roomId); searchPromise = serverSideSearchProcess(term, roomId);
} }
} else { } else {
// Search across all rooms, combine a server side search and a // Search across all rooms, combine a server side search and a
@ -523,6 +527,6 @@ export function searchPagination(searchResult) {
export default function eventSearch(term, roomId = undefined) { export default function eventSearch(term, roomId = undefined) {
const eventIndex = EventIndexPeg.get(); const eventIndex = EventIndexPeg.get();
if (eventIndex === null) return serverSideSearch(term, roomId); if (eventIndex === null) return serverSideSearchProcess(term, roomId);
else return eventIndexSearch(term, roomId); else return eventIndexSearch(term, roomId);
} }