Fixed 267U.pre2
This commit is contained in:
@@ -0,0 +1,312 @@
|
||||
import { In } from "typeorm";
|
||||
import * as mfm from "mfm-js";
|
||||
import { Note } from "../entities/note.js";
|
||||
import { Users, PollVotes, DriveFiles, NoteReactions, Followings, Polls, Channels, Notes, Blockings, UserGroups } from "../index.js";
|
||||
import { nyaize } from "../../misc/nyaize.js";
|
||||
import { awaitAll } from "../../prelude/await-all.js";
|
||||
import { convertLegacyReaction, convertLegacyReactions, decodeReaction } from "../../misc/reaction-lib.js";
|
||||
import { aggregateNoteEmojis, populateEmojis, prefetchEmojis } from "../../misc/populate-emojis.js";
|
||||
import { db } from "../../db/postgre.js";
|
||||
import { IdentifiableError } from "../../misc/identifiable-error.js";
|
||||
import { isFiltered } from "../../misc/is-filtered.js";
|
||||
export async function populatePoll(note, meId) {
|
||||
const poll = await Polls.findOneByOrFail({
|
||||
noteId: note.id
|
||||
});
|
||||
const choices = poll.choices.map((c)=>({
|
||||
text: c,
|
||||
votes: poll.votes[poll.choices.indexOf(c)],
|
||||
isVoted: false
|
||||
}));
|
||||
if (meId) {
|
||||
if (poll.multiple) {
|
||||
const votes = await PollVotes.findBy({
|
||||
userId: meId,
|
||||
noteId: note.id
|
||||
});
|
||||
const myChoices = votes.map((v)=>v.choice);
|
||||
for (const myChoice of myChoices){
|
||||
choices[myChoice].isVoted = true;
|
||||
}
|
||||
} else {
|
||||
const vote = await PollVotes.findOneBy({
|
||||
userId: meId,
|
||||
noteId: note.id
|
||||
});
|
||||
if (vote) {
|
||||
choices[vote.choice].isVoted = true;
|
||||
}
|
||||
}
|
||||
}
|
||||
return {
|
||||
multiple: poll.multiple,
|
||||
expiresAt: poll.expiresAt,
|
||||
choices
|
||||
};
|
||||
}
|
||||
async function populateMyReaction(note, meId, _hint_) {
|
||||
if (_hint_?.myReactions) {
|
||||
const reaction = _hint_.myReactions.get(note.id);
|
||||
if (reaction) {
|
||||
return convertLegacyReaction(reaction.reaction);
|
||||
} else if (reaction === null) {
|
||||
return undefined;
|
||||
}
|
||||
// 実装上抜けがあるだけかもしれないので、「ヒントに含まれてなかったら(=undefinedなら)return」のようにはしない
|
||||
}
|
||||
const reaction = await NoteReactions.findOneBy({
|
||||
userId: meId,
|
||||
noteId: note.id
|
||||
});
|
||||
if (reaction) {
|
||||
return convertLegacyReaction(reaction.reaction);
|
||||
}
|
||||
return undefined;
|
||||
}
|
||||
async function populateIsRenoted(note, meId, _hint_) {
|
||||
return _hint_?.myRenotes ? _hint_.myRenotes.get(note.id) ? true : undefined : Notes.exist({
|
||||
where: {
|
||||
renoteId: note.id,
|
||||
userId: meId
|
||||
}
|
||||
}).then((res)=>res ? true : undefined);
|
||||
}
|
||||
export const NoteRepository = db.getRepository(Note).extend({
|
||||
async isVisibleForMe (note, meId) {
|
||||
if (meId != null && meId !== note.userId) {
|
||||
const blocked = await Blockings.count({
|
||||
where: [
|
||||
{
|
||||
blockeeId: meId,
|
||||
blockerId: note.userId,
|
||||
groupId: null
|
||||
},
|
||||
...note.groupId ? [
|
||||
{
|
||||
blockeeId: meId,
|
||||
groupId: note.groupId
|
||||
}
|
||||
] : []
|
||||
],
|
||||
take: 1
|
||||
});
|
||||
if (blocked !== 0) {
|
||||
return false;
|
||||
}
|
||||
const minorBadgeBlocked = await Users.createQueryBuilder("author").where("author.id = :authorId", {
|
||||
authorId: note.userId
|
||||
}).andWhere("'E' = ANY(author.\"minorBadges\")").andWhere(`EXISTS (` + `SELECT 1 FROM "user" viewer ` + `WHERE viewer.id = :meId ` + `AND viewer."isAdmin" = FALSE ` + `AND viewer."isModerator" = FALSE ` + `AND ('K' = ANY(viewer."minorBadges") OR 'T' = ANY(viewer."minorBadges"))` + `)`, {
|
||||
meId
|
||||
}).getCount();
|
||||
if (minorBadgeBlocked !== 0) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
// This code must always be synchronized with the checks in generateVisibilityQuery.
|
||||
// visibility が specified かつ自分が指定されていなかったら非表示
|
||||
if (note.visibility === "specified") {
|
||||
if (meId == null) {
|
||||
return false;
|
||||
} else if (meId === note.userId) {
|
||||
return true;
|
||||
} else {
|
||||
// 指定されているかどうか
|
||||
return note.visibleUserIds.some((id)=>meId === id);
|
||||
}
|
||||
}
|
||||
// visibility が followers かつ自分が投稿者のフォロワーでなかったら非表示
|
||||
if (note.visibility === "followers") {
|
||||
if (meId == null) {
|
||||
return false;
|
||||
} else if (meId === note.userId) {
|
||||
return true;
|
||||
} else if (note.reply && meId === note.reply.userId) {
|
||||
// 自分の投稿に対するリプライ
|
||||
return true;
|
||||
} else if (note.mentions?.some((id)=>meId === id)) {
|
||||
// 自分へのメンション
|
||||
return true;
|
||||
} else {
|
||||
// フォロワーかどうか
|
||||
const [following, user] = await Promise.all([
|
||||
Followings.count({
|
||||
where: {
|
||||
followeeId: note.userId,
|
||||
followerId: meId
|
||||
},
|
||||
take: 1
|
||||
}),
|
||||
Users.findOneByOrFail({
|
||||
id: meId
|
||||
})
|
||||
]);
|
||||
/* If we know the following, everyhting is fine.
|
||||
|
||||
But if we do not know the following, it might be that both the
|
||||
author of the note and the author of the like are remote users,
|
||||
in which case we can never know the following. Instead we have
|
||||
to assume that the users are following each other.
|
||||
*/ return following > 0 || note.userHost != null && user.host != null;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
},
|
||||
async pack (src, me, options, userCache = Users.getFreshPackedUserCache()) {
|
||||
const opts = Object.assign({
|
||||
detail: true
|
||||
}, options);
|
||||
const meId = me ? me.id : null;
|
||||
const note = typeof src === "object" ? src : await this.findOneByOrFail({
|
||||
id: src
|
||||
});
|
||||
const host = note.userHost;
|
||||
if (!opts.allowAdservice && !note._prId_ && note.tags.includes("adservice") && note.userId !== meId) {
|
||||
throw new IdentifiableError("9725d0ce-ba28-4dde-95a7-2cbb2c15de24", "No such note.");
|
||||
}
|
||||
if (!await this.isVisibleForMe(note, meId)) {
|
||||
throw new IdentifiableError("9725d0ce-ba28-4dde-95a7-2cbb2c15de24", "No such note.");
|
||||
}
|
||||
let text = note.text;
|
||||
if (note.name && (note.url ?? note.uri)) {
|
||||
text = `【${note.name}】\n${(note.text || "").trim()}\n\n${note.url ?? note.uri}`;
|
||||
}
|
||||
const channel = note.channelId ? note.channel ? note.channel : await Channels.findOneBy({
|
||||
id: note.channelId
|
||||
}) : null;
|
||||
const reactionEmojiNames = Object.keys(note.reactions).filter((x)=>x?.startsWith(":")).map((x)=>decodeReaction(x).reaction).map((x)=>x.replace(/:/g, ""));
|
||||
const noteEmoji = populateEmojis(note.emojis.concat(reactionEmojiNames), host);
|
||||
const reactionEmoji = populateEmojis(reactionEmojiNames, host);
|
||||
const packed = await awaitAll({
|
||||
id: note.id,
|
||||
createdAt: note.createdAt.toISOString(),
|
||||
userId: note.userId,
|
||||
user: Users.packCached(note.user ?? note.userId, userCache, me, {
|
||||
detail: false
|
||||
}),
|
||||
groupId: note.groupId,
|
||||
group: note.groupId ? UserGroups.pack(note.group ?? note.groupId) : null,
|
||||
text: text,
|
||||
cw: note.cw,
|
||||
visibility: note.visibility,
|
||||
localOnly: note.localOnly || undefined,
|
||||
visibleUserIds: note.visibility === "specified" ? note.visibleUserIds : undefined,
|
||||
renoteCount: note.renoteCount,
|
||||
repliesCount: note.repliesCount,
|
||||
viewCount: note.viewCount,
|
||||
reactions: convertLegacyReactions(note.reactions),
|
||||
reactionEmojis: reactionEmoji,
|
||||
emojis: noteEmoji,
|
||||
tags: note.tags.length > 0 ? note.tags : undefined,
|
||||
fileIds: note.fileIds,
|
||||
files: DriveFiles.packMany(note.fileIds),
|
||||
replyId: note.replyId,
|
||||
renoteId: note.renoteId,
|
||||
channelId: note.channelId || undefined,
|
||||
channel: channel ? {
|
||||
id: channel.id,
|
||||
name: channel.name
|
||||
} : undefined,
|
||||
mentions: note.mentions.length > 0 ? note.mentions : undefined,
|
||||
uri: note.uri || undefined,
|
||||
url: note.url || undefined,
|
||||
updatedAt: note.updatedAt?.toISOString() || undefined,
|
||||
poll: note.hasPoll ? populatePoll(note, meId) : undefined,
|
||||
quoteAuthorization: note.quoteAuthorization || undefined,
|
||||
canBite: false,
|
||||
...meId ? {
|
||||
myReaction: populateMyReaction(note, meId, options?._hint_),
|
||||
isRenoted: populateIsRenoted(note, meId, options?._hint_),
|
||||
isFiltered: isFiltered(note, me)
|
||||
} : {},
|
||||
...opts.detail ? {
|
||||
reply: note.replyId ? this.tryPack(note.reply || note.replyId, me, {
|
||||
detail: false,
|
||||
_hint_: options?._hint_
|
||||
}, userCache) : undefined,
|
||||
renote: note.renoteId ? this.pack(note.renote || note.renoteId, me, {
|
||||
detail: true,
|
||||
_hint_: options?._hint_
|
||||
}, userCache) : undefined
|
||||
} : {}
|
||||
});
|
||||
if (packed.user.isCat && packed.user.speakAsCat && packed.text) {
|
||||
const tokens = packed.text ? mfm.parse(packed.text) : [];
|
||||
function nyaizeNode(node) {
|
||||
if (node.type === "quote") return;
|
||||
if (node.type === "text") node.props.text = nyaize(node.props.text);
|
||||
if (node.children) {
|
||||
for (const child of node.children){
|
||||
nyaizeNode(child);
|
||||
}
|
||||
}
|
||||
}
|
||||
for (const node of tokens)nyaizeNode(node);
|
||||
packed.text = mfm.toString(tokens);
|
||||
}
|
||||
if (me) {
|
||||
if (packed.user.canBite === "anyone") {
|
||||
packed.canBite = true;
|
||||
} else if (packed.user.canBite === "followers") {
|
||||
const isFollowing = await Followings.exist({
|
||||
where: {
|
||||
followerId: me.id,
|
||||
followeeId: packed.userId
|
||||
},
|
||||
take: 1
|
||||
});
|
||||
packed.canBite = isFollowing;
|
||||
} else {
|
||||
packed.canBite = false;
|
||||
}
|
||||
}
|
||||
if (note._prId_) {
|
||||
packed._prId_ = note._prId_;
|
||||
}
|
||||
return packed;
|
||||
},
|
||||
async tryPack (src, me, options, userCache = Users.getFreshPackedUserCache()) {
|
||||
try {
|
||||
return await this.pack(src, me, options, userCache);
|
||||
} catch {
|
||||
return undefined;
|
||||
}
|
||||
},
|
||||
async packMany (notes, me, options, userCache = Users.getFreshPackedUserCache()) {
|
||||
if (notes.length === 0) return [];
|
||||
const meId = me ? me.id : null;
|
||||
const myReactionsMap = new Map();
|
||||
const myRenotesMap = new Map();
|
||||
if (meId) {
|
||||
const renoteIds = notes.filter((n)=>n.renoteId != null).map((n)=>n.renoteId);
|
||||
const targets = [
|
||||
...notes.map((n)=>n.id),
|
||||
...renoteIds
|
||||
];
|
||||
const myReactions = await NoteReactions.findBy({
|
||||
userId: meId,
|
||||
noteId: In(targets)
|
||||
});
|
||||
const myRenotes = await Notes.createQueryBuilder('note').select('note.renoteId').where('note.userId = :meId', {
|
||||
meId
|
||||
}).andWhere('note.renoteId IN (:...targets)', {
|
||||
targets
|
||||
}).andWhere('note.text IS NULL').andWhere('note.hasPoll = FALSE').andWhere(`note.fileIds = '{}'`).getMany();
|
||||
for (const target of targets){
|
||||
myReactionsMap.set(target, myReactions.find((reaction)=>reaction.noteId === target) || null);
|
||||
myRenotesMap.set(target, !!myRenotes.find((p)=>p.renoteId == target));
|
||||
}
|
||||
}
|
||||
await prefetchEmojis(aggregateNoteEmojis(notes));
|
||||
const promises = await Promise.allSettled(notes.map((n)=>this.pack(n, me, {
|
||||
...options,
|
||||
_hint_: {
|
||||
myReactions: myReactionsMap,
|
||||
myRenotes: myRenotesMap
|
||||
}
|
||||
}, userCache)));
|
||||
// filter out rejected promises, only keep fulfilled values
|
||||
return promises.flatMap((result)=>result.status === "fulfilled" ? [
|
||||
result.value
|
||||
] : []);
|
||||
}
|
||||
});
|
||||
Reference in New Issue
Block a user