This commit is contained in:
ThatOneCalculator 2023-05-29 10:00:16 -07:00
parent d7d5915e3b
commit 316a09409c
No known key found for this signature in database
GPG Key ID: 8703CACD01000000
1 changed files with 55 additions and 57 deletions

View File

@ -15,7 +15,6 @@ const hasConfig =
config.meilisearch.port || config.meilisearch.port ||
config.meilisearch.apiKey); config.meilisearch.apiKey);
if (hasConfig) {
const host = hasConfig ? config.meilisearch.host ?? "localhost" : ""; const host = hasConfig ? config.meilisearch.host ?? "localhost" : "";
const port = hasConfig ? config.meilisearch.port ?? 7700 : 0; const port = hasConfig ? config.meilisearch.port ?? 7700 : 0;
const auth = hasConfig ? config.meilisearch.apiKey ?? "" : ""; const auth = hasConfig ? config.meilisearch.apiKey ?? "" : "";
@ -59,7 +58,6 @@ if (hasConfig) {
); );
logger.info("Connected to MeiliSearch"); logger.info("Connected to MeiliSearch");
}
export type MeilisearchNote = { export type MeilisearchNote = {
id: string; id: string;
@ -90,37 +88,37 @@ export default hasConfig
/// filter:following => show results only from users you follow /// filter:following => show results only from users you follow
/// filter:followers => show results only from followers /// filter:followers => show results only from followers
let constructedFilters: string[] = []; const constructedFilters: string[] = [];
let splitSearch = query.split(" "); const splitSearch = query.split(" ");
// Detect search operators and remove them from the actual query // Detect search operators and remove them from the actual query
let filteredSearchTerms = ( const filteredSearchTerms = (
await Promise.all( await Promise.all(
splitSearch.map(async (term) => { splitSearch.map(async (term) => {
if (term.startsWith("has:")) { if (term.startsWith("has:")) {
let fileType = term.slice(4); const fileType = term.slice(4);
constructedFilters.push(`mediaAttachment = "${fileType}"`); constructedFilters.push(`mediaAttachment = "${fileType}"`);
return null; return null;
} else if (term.startsWith("from:")) { } else if (term.startsWith("from:")) {
let user = term.slice(5); const user = term.slice(5);
constructedFilters.push(`userName = ${user}`); constructedFilters.push(`userName = ${user}`);
return null; return null;
} else if (term.startsWith("domain:")) { } else if (term.startsWith("domain:")) {
let domain = term.slice(7); const domain = term.slice(7);
constructedFilters.push(`userHost = ${domain}`); constructedFilters.push(`userHost = ${domain}`);
return null; return null;
} else if (term.startsWith("after:")) { } else if (term.startsWith("after:")) {
let timestamp = term.slice(6); const timestamp = term.slice(6);
// Try to parse the timestamp as JavaScript Date // Try to parse the timestamp as JavaScript Date
let date = Date.parse(timestamp); const date = Date.parse(timestamp);
if (isNaN(date)) return null; if (isNaN(date)) return null;
constructedFilters.push(`createdAt > ${date / 1000}`); constructedFilters.push(`createdAt > ${date / 1000}`);
return null; return null;
} else if (term.startsWith("before:")) { } else if (term.startsWith("before:")) {
let timestamp = term.slice(7); const timestamp = term.slice(7);
// Try to parse the timestamp as JavaScript Date // Try to parse the timestamp as JavaScript Date
let date = Date.parse(timestamp); const date = Date.parse(timestamp);
if (isNaN(date)) return null; if (isNaN(date)) return null;
constructedFilters.push(`createdAt < ${date / 1000}`); constructedFilters.push(`createdAt < ${date / 1000}`);
return null; return null;
@ -128,7 +126,7 @@ export default hasConfig
// Check if we got a context user // Check if we got a context user
if (userCtx) { if (userCtx) {
// Fetch user follows from DB // Fetch user follows from DB
let followedUsers = await Followings.find({ const followedUsers = await Followings.find({
where: { where: {
followerId: userCtx.id, followerId: userCtx.id,
}, },
@ -136,7 +134,7 @@ export default hasConfig
followeeId: true, followeeId: true,
}, },
}); });
let followIDs = followedUsers.map((user) => user.followeeId); const followIDs = followedUsers.map((user) => user.followeeId);
if (followIDs.length === 0) return null; if (followIDs.length === 0) return null;
@ -152,7 +150,7 @@ export default hasConfig
// Check if we got a context user // Check if we got a context user
if (userCtx) { if (userCtx) {
// Fetch users follows from DB // Fetch users follows from DB
let followedUsers = await Followings.find({ const followedUsers = await Followings.find({
where: { where: {
followeeId: userCtx.id, followeeId: userCtx.id,
}, },
@ -160,7 +158,7 @@ export default hasConfig
followerId: true, followerId: true,
}, },
}); });
let followIDs = followedUsers.map((user) => user.followerId); const followIDs = followedUsers.map((user) => user.followerId);
if (followIDs.length === 0) return null; if (followIDs.length === 0) return null;
@ -179,7 +177,7 @@ export default hasConfig
) )
).filter((term) => term !== null); ).filter((term) => term !== null);
let sortRules = []; const sortRules = [];
// An empty search term with defined filters means we have a placeholder search => https://www.meilisearch.com/docs/reference/api/search#placeholder-search // An empty search term with defined filters means we have a placeholder search => https://www.meilisearch.com/docs/reference/api/search#placeholder-search
// These have to be ordered manually, otherwise the *oldest* posts are returned first, which we don't want // These have to be ordered manually, otherwise the *oldest* posts are returned first, which we don't want
@ -205,9 +203,9 @@ export default hasConfig
ingestNotes = [ingestNotes]; ingestNotes = [ingestNotes];
} }
let indexingBatch: MeilisearchNote[] = []; const indexingBatch: MeilisearchNote[] = [];
for (let note of ingestNotes) { for (const note of ingestNotes) {
if (note.user === undefined) { if (note.user === undefined) {
note.user = await Users.findOne({ note.user = await Users.findOne({
where: { where: {
@ -255,8 +253,8 @@ export default hasConfig
); );
}, },
serverStats: async () => { serverStats: async () => {
let health: Health = await client.health(); const health: Health = await client.health();
let stats: Stats = await client.getStats(); const stats: Stats = await client.getStats();
return { return {
health: health.status, health: health.status,