235 lines
10 KiB
JavaScript
235 lines
10 KiB
JavaScript
import { Brackets } from "typeorm";
|
|
import { sqlLikeEscape } from "../../../misc/sql-like-escape.js";
|
|
import { sqlRegexEscape } from "../../../misc/sql-regex-escape.js";
|
|
import { Followings, NoteFavorites, NoteReactions, Users } from "../../../models/index.js";
|
|
const filters = {
|
|
"from": fromFilter,
|
|
"-from": fromFilterInverse,
|
|
"mention": mentionFilter,
|
|
"-mention": mentionFilterInverse,
|
|
"reply": replyFilter,
|
|
"-reply": replyFilterInverse,
|
|
"to": replyFilter,
|
|
"-to": replyFilterInverse,
|
|
"before": beforeFilter,
|
|
"until": beforeFilter,
|
|
"after": afterFilter,
|
|
"since": afterFilter,
|
|
"instance": instanceFilter,
|
|
"-instance": instanceFilterInverse,
|
|
"domain": instanceFilter,
|
|
"-domain": instanceFilterInverse,
|
|
"host": instanceFilter,
|
|
"-host": instanceFilterInverse,
|
|
"filter": miscFilter,
|
|
"-filter": miscFilterInverse,
|
|
"in": inFilter,
|
|
"-in": inFilterInverse,
|
|
"has": attachmentFilter
|
|
};
|
|
export function generateFtsQuery(query, q) {
|
|
const components = q.trim().split(" ");
|
|
const terms = [];
|
|
let finalTerms = [];
|
|
let counter = 0;
|
|
let caseSensitive = false;
|
|
let matchWords = false;
|
|
for (const component of components){
|
|
const split = component.split(":");
|
|
if (split.length > 1 && filters[split[0]] !== undefined) filters[split[0]](query, split.slice(1).join(":"), counter++);
|
|
else if (split.length > 1 && (split[0] === "search" || split[0] === "match")) matchWords = split[1] === 'word' || split[1] === 'words';
|
|
else if (split.length > 1 && split[0] === "case") caseSensitive = split[1] === 'sensitive';
|
|
else terms.push(component);
|
|
}
|
|
let idx = 0;
|
|
let state = 'idle';
|
|
for(let i = 0; i < terms.length; i++){
|
|
if (state === 'idle') {
|
|
if (terms[i].startsWith('"') && terms[i].endsWith('"') || terms[i].startsWith('(') && terms[i].endsWith(')')) {
|
|
finalTerms.push(trimStartAndEnd(terms[i]));
|
|
} else if (terms[i].startsWith('"')) {
|
|
idx = i;
|
|
state = 'quote';
|
|
} else if (terms[i].startsWith('(')) {
|
|
idx = i;
|
|
state = 'parenthesis';
|
|
} else {
|
|
finalTerms.push(terms[i]);
|
|
}
|
|
} else if (state === 'quote' && terms[i].endsWith('"')) {
|
|
finalTerms.push(extractToken(terms, idx, i));
|
|
state = 'idle';
|
|
} else if (state === 'parenthesis' && terms[i].endsWith(')')) {
|
|
query.andWhere(new Brackets((qb)=>{
|
|
for (const term of extractToken(terms, idx, i).split(' OR ')){
|
|
const id = counter++;
|
|
appendSearchQuery(term, 'or', query, qb, id, term.startsWith('-'), matchWords, caseSensitive);
|
|
}
|
|
}));
|
|
state = 'idle';
|
|
}
|
|
}
|
|
if (state != "idle") {
|
|
finalTerms.push(...extractToken(terms, idx, terms.length - 1, false).substring(1).split(' '));
|
|
}
|
|
for (const term of finalTerms){
|
|
const id = counter++;
|
|
appendSearchQuery(term, 'and', query, query, id, term.startsWith('-'), matchWords, caseSensitive);
|
|
}
|
|
}
|
|
function fromFilter(query, filter, id) {
|
|
const userQuery = generateUserSubquery(filter, id);
|
|
query.andWhere(`note.userId = (${userQuery.getQuery()})`);
|
|
query.setParameters(userQuery.getParameters());
|
|
}
|
|
function fromFilterInverse(query, filter, id) {
|
|
const userQuery = generateUserSubquery(filter, id);
|
|
query.andWhere(`note.userId <> (${userQuery.getQuery()})`);
|
|
query.setParameters(userQuery.getParameters());
|
|
}
|
|
function mentionFilter(query, filter, id) {
|
|
const userQuery = generateUserSubquery(filter, id);
|
|
query.addCommonTableExpression(userQuery.getQuery(), `cte_${id}`, {
|
|
materialized: true
|
|
});
|
|
query.andWhere(`note.mentions @> array[(SELECT * FROM cte_${id})]::varchar[]`);
|
|
query.setParameters(userQuery.getParameters());
|
|
}
|
|
function mentionFilterInverse(query, filter, id) {
|
|
const userQuery = generateUserSubquery(filter, id);
|
|
query.addCommonTableExpression(userQuery.getQuery(), `cte_${id}`, {
|
|
materialized: true
|
|
});
|
|
query.andWhere(`NOT (note.mentions @> array[(SELECT * FROM cte_${id})]::varchar[])`);
|
|
query.setParameters(userQuery.getParameters());
|
|
}
|
|
function replyFilter(query, filter, id) {
|
|
const userQuery = generateUserSubquery(filter, id);
|
|
query.andWhere(`note.replyUserId = (${userQuery.getQuery()})`);
|
|
query.setParameters(userQuery.getParameters());
|
|
}
|
|
function replyFilterInverse(query, filter, id) {
|
|
const userQuery = generateUserSubquery(filter, id);
|
|
query.andWhere(`note.replyUserId <> (${userQuery.getQuery()})`);
|
|
query.setParameters(userQuery.getParameters());
|
|
}
|
|
function beforeFilter(query, filter) {
|
|
query.andWhere('note.createdAt < :before', {
|
|
before: filter
|
|
});
|
|
}
|
|
function afterFilter(query, filter) {
|
|
query.andWhere('note.createdAt > :after', {
|
|
after: filter
|
|
});
|
|
}
|
|
function instanceFilter(query, filter, id) {
|
|
if (filter === 'local') {
|
|
query.andWhere(`note.userHost IS NULL`);
|
|
} else {
|
|
query.andWhere(`note.userHost = :instance_${id}`);
|
|
query.setParameter(`instance_${id}`, filter);
|
|
}
|
|
}
|
|
function instanceFilterInverse(query, filter, id) {
|
|
if (filter === 'local') {
|
|
query.andWhere(`note.userHost IS NOT NULL`);
|
|
} else {
|
|
query.andWhere(`note.userHost <> :instance_${id}`);
|
|
query.setParameter(`instance_${id}`, filter);
|
|
}
|
|
}
|
|
function miscFilter(query, filter) {
|
|
let subQuery = null;
|
|
if (filter === 'followers') {
|
|
subQuery = Followings.createQueryBuilder('following').select('following.followerId').where('following.followeeId = :meId');
|
|
} else if (filter === 'following') {
|
|
subQuery = Followings.createQueryBuilder('following').select('following.followeeId').where('following.followerId = :meId');
|
|
} else if (filter === 'replies' || filter === 'reply') {
|
|
query.andWhere('note.replyId IS NOT NULL');
|
|
} else if (filter === 'boosts' || filter === 'boost' || filter === 'renotes' || filter === 'renote') {
|
|
query.andWhere('note.renoteId IS NOT NULL');
|
|
}
|
|
if (subQuery !== null) query.andWhere(`note.userId IN (${subQuery.getQuery()})`);
|
|
}
|
|
function miscFilterInverse(query, filter) {
|
|
let subQuery = null;
|
|
if (filter === 'followers') {
|
|
subQuery = Followings.createQueryBuilder('following').select('following.followerId').where('following.followeeId = :meId');
|
|
} else if (filter === 'following') {
|
|
subQuery = Followings.createQueryBuilder('following').select('following.followeeId').where('following.followerId = :meId');
|
|
} else if (filter === 'replies' || filter === 'reply') {
|
|
query.andWhere('note.replyId IS NULL');
|
|
} else if (filter === 'boosts' || filter === 'boost' || filter === 'renotes' || filter === 'renote') {
|
|
query.andWhere('note.renoteId IS NULL');
|
|
}
|
|
if (subQuery !== null) query.andWhere(`note.userId NOT IN (${subQuery.getQuery()})`);
|
|
}
|
|
function inFilter(query, filter) {
|
|
let subQuery = null;
|
|
if (filter === 'bookmarks') {
|
|
subQuery = NoteFavorites.createQueryBuilder('bookmark').select('bookmark.noteId').where('bookmark.userId = :meId');
|
|
} else if (filter === 'favorites' || filter === 'favourites' || filter === 'reactions' || filter === 'likes') {
|
|
subQuery = NoteReactions.createQueryBuilder('react').select('react.noteId').where('react.userId = :meId');
|
|
}
|
|
if (subQuery !== null) query.andWhere(`note.id IN (${subQuery.getQuery()})`);
|
|
}
|
|
function inFilterInverse(query, filter) {
|
|
let subQuery = null;
|
|
if (filter === 'bookmarks') {
|
|
subQuery = NoteFavorites.createQueryBuilder('bookmark').select('bookmark.noteId').where('bookmark.userId = :meId');
|
|
} else if (filter === 'favorites' || filter === 'favourites' || filter === 'reactions' || filter === 'likes') {
|
|
subQuery = NoteReactions.createQueryBuilder('react').select('react.noteId').where('react.userId = :meId');
|
|
}
|
|
if (subQuery !== null) query.andWhere(`note.id NOT IN (${subQuery.getQuery()})`);
|
|
}
|
|
function attachmentFilter(query, filter) {
|
|
switch(filter){
|
|
case 'image':
|
|
query.andWhere(`note."attachedFileTypes"::varchar ILIKE '%image/%'`);
|
|
break;
|
|
case 'video':
|
|
query.andWhere(`note."attachedFileTypes"::varchar ILIKE '%video/%'`);
|
|
break;
|
|
case 'audio':
|
|
query.andWhere(`note."attachedFileTypes"::varchar ILIKE '%audio/%'`);
|
|
break;
|
|
case 'file':
|
|
query.andWhere(`note."attachedFileTypes" <> '{}'`);
|
|
query.andWhere(`NOT (note."attachedFileTypes"::varchar ILIKE '%image/%')`);
|
|
query.andWhere(`NOT (note."attachedFileTypes"::varchar ILIKE '%video/%')`);
|
|
query.andWhere(`NOT (note."attachedFileTypes"::varchar ILIKE '%audio/%')`);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
function generateUserSubquery(filter, id) {
|
|
if (filter.startsWith('@')) filter = filter.substring(1);
|
|
const split = filter.split('@');
|
|
const query = Users.createQueryBuilder('user').select('user.id').where(`user.usernameLower = :user_${id}`).andWhere(`user.host ${split[1] !== undefined ? `= :host_${id}` : 'IS NULL'}`);
|
|
query.setParameter(`user_${id}`, split[0].toLowerCase());
|
|
if (split[1] !== undefined) query.setParameter(`host_${id}`, split[1].toLowerCase());
|
|
return query;
|
|
}
|
|
function extractToken(array, start, end, trim = true) {
|
|
const slice = array.slice(start, end + 1).join(" ");
|
|
return trim ? trimStartAndEnd(slice) : slice;
|
|
}
|
|
function trimStartAndEnd(str) {
|
|
return str.substring(1, str.length - 1);
|
|
}
|
|
function appendSearchQuery(term, mode, query, qb, id, negate, matchWords, caseSensitive) {
|
|
const sql = `note.text ${getSearchMatchOperator(negate, matchWords, caseSensitive)} :q_${id}`;
|
|
if (mode === 'and') qb.andWhere(sql);
|
|
else if (mode === 'or') qb.orWhere(sql);
|
|
query.setParameter(`q_${id}`, escapeSqlSearchParam(term.substring(negate ? 1 : 0), matchWords));
|
|
}
|
|
function getSearchMatchOperator(negate, matchWords, caseSensitive) {
|
|
const negatePrefix = matchWords ? '!' : 'NOT ';
|
|
return `${negate ? negatePrefix : ''}${matchWords ? caseSensitive ? '~' : '~*' : caseSensitive ? 'LIKE' : 'ILIKE'}`;
|
|
}
|
|
function escapeSqlSearchParam(param, matchWords) {
|
|
return matchWords ? `\\y${sqlRegexEscape(param)}\\y` : `%${sqlLikeEscape(param)}%`;
|
|
}
|