Fixed 267U.pre2

2026-07-26 18:25:37 +09:00
parent 50bfaeafdf
commit 317d00a284
1286 changed files with 80222 additions and 1 deletions
@@ -0,0 +1,234 @@
+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)}%`;
+}