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,53 @@
+import { db } from "../../db/postgre.js";
+import { UserProfile } from "../entities/user-profile.js";
+import mfm from "mfm-js";
+import { extractMentions } from "../../misc/extract-mentions.js";
+import { resolveMentionToUserAndProfile } from "../../remote/resolve-user.js";
+import { unique } from "../../prelude/array.js";
+import config from "../../config/index.js";
+import { Mutex, Semaphore } from "async-mutex";
+const queue = new Semaphore(5);
+export const UserProfileRepository = db.getRepository(UserProfile).extend({
+    // We must never await this without promiseEarlyReturn, otherwise giant webring-style profile mention trees will cause the queue to stop working
+    async updateMentions (id, limiter = new RecursionLimiter()) {
+        const profile = await this.findOneBy({
+            userId: id
+        });
+        if (!profile) return;
+        const tokens = [];
+        if (profile.description) tokens.push(...mfm.parse(profile.description));
+        if (profile.fields.length > 0) tokens.push(...profile.fields.map((p)=>mfm.parse(p.value).concat(mfm.parse(p.name))).flat());
+        return queue.runExclusive(async ()=>{
+            const partial = {
+                mentions: await populateMentions(tokens, profile.userHost, limiter)
+            };
+            return UserProfileRepository.update(profile.userId, partial);
+        });
+    }
+});
+async function populateMentions(tokens, objectHost, limiter) {
+    const mentions = extractMentions(tokens);
+    const resolved = await Promise.all(mentions.map((m)=>resolveMentionToUserAndProfile(m.username, m.host, objectHost, limiter)));
+    const remote = resolved.filter((p)=>p && p.data.host !== config.domain && (p.data.host !== null || objectHost !== null)).map((p)=>p);
+    const res = remote.map((m)=>{
+        return {
+            uri: m.user.uri,
+            url: m.profile?.url ?? undefined,
+            username: m.data.username,
+            host: m.data.host
+        };
+    });
+    return unique(res);
+}
+export class RecursionLimiter {
+    counter;
+    mutex = new Mutex();
+    constructor(count = 10){
+        this.counter = count;
+    }
+    shouldContinue() {
+        return this.mutex.runExclusive(()=>{
+            return this.counter-- > 0;
+        });
+    }
+}