54 lines
2.2 KiB
JavaScript
54 lines
2.2 KiB
JavaScript
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;
|
|
});
|
|
}
|
|
}
|