Fixed 267U.pre2
This commit is contained in:
@@ -0,0 +1,366 @@
|
||||
import { makePaginationQuery } from "../../common/make-pagination-query.js";
|
||||
import { DriveFiles, Metas, NoteEdits, NoteFavorites, NoteReactions, Notes, UserNotePinings } from "../../../../models/index.js";
|
||||
import { generateVisibilityQuery } from "../../common/generate-visibility-query.js";
|
||||
import { generateMutedUserQuery } from "../../common/generate-muted-user-query.js";
|
||||
import { generateBlockedUserQuery } from "../../common/generate-block-query.js";
|
||||
import { getNote } from "../../common/getters.js";
|
||||
import createReaction from "../../../../services/note/reaction/create.js";
|
||||
import deleteReaction from "../../../../services/note/reaction/delete.js";
|
||||
import createNote, { extractMentionedUsers } from "../../../../services/note/create.js";
|
||||
import editNote from "../../../../services/note/edit.js";
|
||||
import deleteNote from "../../../../services/note/delete.js";
|
||||
import { genId } from "../../../../misc/gen-id.js";
|
||||
import { PaginationHelpers } from "./pagination.js";
|
||||
import { UserConverter } from "../converters/user.js";
|
||||
import { UserHelpers } from "./user.js";
|
||||
import { addPinned, removePinned } from "../../../../services/i/pin.js";
|
||||
import { NoteConverter } from "../converters/note.js";
|
||||
import { awaitAll } from "../../../../prelude/await-all.js";
|
||||
import { VisibilityConverter } from "../converters/visibility.js";
|
||||
import mfm from "mfm-js";
|
||||
import { FileConverter } from "../converters/file.js";
|
||||
import { MfmHelpers } from "./mfm.js";
|
||||
import { toArray, unique } from "../../../../prelude/array.js";
|
||||
import { MastoApiError } from "../middleware/catch-errors.js";
|
||||
import { Cache } from "../../../../misc/cache.js";
|
||||
import AsyncLock from "async-lock";
|
||||
import { IdentifiableError } from "../../../../misc/identifiable-error.js";
|
||||
import { IsNull } from "typeorm";
|
||||
import { getStubMastoContext } from "../index.js";
|
||||
export class NoteHelpers {
|
||||
static postIdempotencyCache = new Cache('postIdempotencyCache', 60 * 60);
|
||||
static postIdempotencyLocks = new AsyncLock();
|
||||
static async getDefaultReaction() {
|
||||
return Metas.createQueryBuilder().select('"defaultReaction"').execute().then((p)=>p[0].defaultReaction).then((p)=>{
|
||||
if (p != null) return p;
|
||||
throw new MastoApiError(500, "Failed to get default reaction");
|
||||
});
|
||||
}
|
||||
static async reactToNote(note, reaction, ctx) {
|
||||
const user = ctx.user;
|
||||
await createReaction(user, note, reaction).catch((e)=>{
|
||||
if (e instanceof IdentifiableError && e.id == '51c42bb4-931a-456b-bff7-e5a8a70dd298') return;
|
||||
throw e;
|
||||
});
|
||||
return getNote(note.id, user);
|
||||
}
|
||||
static async removeReactFromNote(note, ctx) {
|
||||
const user = ctx.user;
|
||||
await deleteReaction(user, note);
|
||||
return getNote(note.id, user);
|
||||
}
|
||||
static async reblogNote(note, ctx) {
|
||||
const user = ctx.user;
|
||||
const existingRenote = await Notes.findOneBy({
|
||||
userId: user.id,
|
||||
renoteId: note.id,
|
||||
text: IsNull()
|
||||
});
|
||||
if (existingRenote) return existingRenote;
|
||||
const data = {
|
||||
createdAt: new Date(),
|
||||
files: [],
|
||||
renote: note
|
||||
};
|
||||
return await createNote(user, data);
|
||||
}
|
||||
static async unreblogNote(note, ctx) {
|
||||
const user = ctx.user;
|
||||
return Notes.findBy({
|
||||
userId: user.id,
|
||||
renoteId: note.id
|
||||
}).then((p)=>p.map((n)=>deleteNote(user, n))).then((p)=>Promise.all(p)).then((_)=>getNote(note.id, user));
|
||||
}
|
||||
static async bookmarkNote(note, ctx) {
|
||||
const user = ctx.user;
|
||||
const bookmarked = await NoteFavorites.exist({
|
||||
where: {
|
||||
noteId: note.id,
|
||||
userId: user.id
|
||||
}
|
||||
});
|
||||
if (!bookmarked) {
|
||||
await NoteFavorites.insert({
|
||||
id: genId(),
|
||||
createdAt: new Date(),
|
||||
noteId: note.id,
|
||||
userId: user.id
|
||||
});
|
||||
}
|
||||
return note;
|
||||
}
|
||||
static async unbookmarkNote(note, ctx) {
|
||||
const user = ctx.user;
|
||||
return NoteFavorites.findOneBy({
|
||||
noteId: note.id,
|
||||
userId: user.id
|
||||
}).then((p)=>p !== null ? NoteFavorites.delete(p.id) : null).then((_)=>note);
|
||||
}
|
||||
static async pinNote(note, ctx) {
|
||||
const user = ctx.user;
|
||||
const pinned = await UserNotePinings.exist({
|
||||
where: {
|
||||
userId: user.id,
|
||||
noteId: note.id
|
||||
}
|
||||
});
|
||||
if (!pinned) {
|
||||
await addPinned(user, note.id);
|
||||
}
|
||||
return note;
|
||||
}
|
||||
static async unpinNote(note, ctx) {
|
||||
const user = ctx.user;
|
||||
const pinned = await UserNotePinings.exist({
|
||||
where: {
|
||||
userId: user.id,
|
||||
noteId: note.id
|
||||
}
|
||||
});
|
||||
if (pinned) {
|
||||
await removePinned(user, note.id);
|
||||
}
|
||||
return note;
|
||||
}
|
||||
static async deleteNote(note, ctx) {
|
||||
const user = ctx.user;
|
||||
if (user.id !== note.userId) throw new MastoApiError(404);
|
||||
const status = await NoteConverter.encode(note, ctx);
|
||||
await deleteNote(user, note);
|
||||
status.content = undefined;
|
||||
return status;
|
||||
}
|
||||
static async getNoteFavoritedBy(note, maxId, sinceId, minId, limit = 40, ctx) {
|
||||
if (limit > 80) limit = 80;
|
||||
const query = PaginationHelpers.makePaginationQuery(NoteReactions.createQueryBuilder("reaction"), sinceId, maxId, minId).andWhere("reaction.noteId = :noteId", {
|
||||
noteId: note.id
|
||||
}).innerJoinAndSelect("reaction.user", "user");
|
||||
return PaginationHelpers.execQueryLinkPagination(query, limit, minId !== undefined, ctx).then((reactions)=>{
|
||||
return reactions.map((p)=>p.user).filter((p)=>p);
|
||||
});
|
||||
}
|
||||
static async getNoteEditHistory(note, ctx) {
|
||||
const user = Promise.resolve(note.user ?? await UserHelpers.getUserCached(note.userId, ctx));
|
||||
const account = user.then((p)=>UserConverter.encode(p, ctx));
|
||||
const edits = await NoteEdits.find({
|
||||
where: {
|
||||
noteId: note.id
|
||||
},
|
||||
order: {
|
||||
id: "ASC"
|
||||
}
|
||||
});
|
||||
const history = [];
|
||||
const curr = {
|
||||
id: note.id,
|
||||
noteId: note.id,
|
||||
note: note,
|
||||
text: note.text,
|
||||
cw: note.cw,
|
||||
fileIds: note.fileIds,
|
||||
updatedAt: note.updatedAt ?? note.createdAt
|
||||
};
|
||||
edits.push(curr);
|
||||
let lastDate = note.createdAt;
|
||||
for (const edit of edits){
|
||||
const files = DriveFiles.packMany(edit.fileIds);
|
||||
const item = {
|
||||
account: account,
|
||||
content: MfmHelpers.toHtml(mfm.parse(edit.text ?? ''), JSON.parse(note.mentionedRemoteUsers), note.userHost).then((p)=>p ?? ''),
|
||||
created_at: lastDate.toISOString(),
|
||||
emojis: [],
|
||||
sensitive: files.then((files)=>files.length > 0 ? files.some((f)=>f.isSensitive) : false),
|
||||
spoiler_text: edit.cw ?? '',
|
||||
poll: null,
|
||||
media_attachments: files.then((files)=>files.length > 0 ? files.map((f)=>FileConverter.encode(f)) : [])
|
||||
};
|
||||
lastDate = edit.updatedAt;
|
||||
history.push(awaitAll(item));
|
||||
}
|
||||
return Promise.all(history);
|
||||
}
|
||||
static getNoteSource(note) {
|
||||
return {
|
||||
id: note.id,
|
||||
text: note.text ?? '',
|
||||
spoiler_text: note.cw ?? '',
|
||||
content_type: 'text/x.misskeymarkdown'
|
||||
};
|
||||
}
|
||||
static async getNoteRebloggedBy(note, maxId, sinceId, minId, limit = 40, ctx) {
|
||||
if (limit > 80) limit = 80;
|
||||
const user = ctx.user;
|
||||
const query = PaginationHelpers.makePaginationQuery(Notes.createQueryBuilder("note"), sinceId, maxId, minId).andWhere("note.renoteId = :noteId", {
|
||||
noteId: note.id
|
||||
}).andWhere("note.text IS NULL") // We don't want to count quotes as renotes
|
||||
.andWhere('note.hasPoll = FALSE').andWhere("note.fileIds = '{}'").innerJoinAndSelect("note.user", "user");
|
||||
generateVisibilityQuery(query, user);
|
||||
return PaginationHelpers.execQueryLinkPagination(query, limit, minId !== undefined, ctx).then((renotes)=>{
|
||||
return renotes.map((p)=>p.user).filter((p)=>p);
|
||||
});
|
||||
}
|
||||
static async getNoteDescendants(note, limit = 10, depth = 2, ctx) {
|
||||
const user = ctx.user;
|
||||
const noteId = typeof note === "string" ? note : note.id;
|
||||
const query = makePaginationQuery(Notes.createQueryBuilder("note")).andWhere("note.id IN (SELECT id FROM note_replies(:noteId, :depth, :limit))", {
|
||||
noteId,
|
||||
depth,
|
||||
limit
|
||||
});
|
||||
generateVisibilityQuery(query, user);
|
||||
if (user) {
|
||||
generateMutedUserQuery(query, user);
|
||||
generateBlockedUserQuery(query, user);
|
||||
}
|
||||
return query.getMany().then((p)=>p.reverse());
|
||||
}
|
||||
static async getNoteAncestors(rootNote, limit = 10, ctx) {
|
||||
const user = ctx.user;
|
||||
const notes = new Array;
|
||||
for(let i = 0; i < limit; i++){
|
||||
const currentNote = notes.at(-1) ?? rootNote;
|
||||
if (!currentNote.replyId) break;
|
||||
const nextNote = await getNote(currentNote.replyId, user).catch((e)=>{
|
||||
if (e.id === "9725d0ce-ba28-4dde-95a7-2cbb2c15de24") return null;
|
||||
throw e;
|
||||
});
|
||||
if (nextNote && await Notes.isVisibleForMe(nextNote, user?.id ?? null)) notes.push(nextNote);
|
||||
else break;
|
||||
}
|
||||
return notes.reverse();
|
||||
}
|
||||
static async createNote(request, ctx) {
|
||||
const user = ctx.user;
|
||||
const files = request.media_ids && request.media_ids.length > 0 ? DriveFiles.findByIds(request.media_ids) : [];
|
||||
const reply = request.in_reply_to_id ? await getNote(request.in_reply_to_id, user) : undefined;
|
||||
const renote = request.quote_id ? await getNote(request.quote_id, user) : undefined;
|
||||
const visibility = request.visibility ?? UserHelpers.getDefaultNoteVisibility(ctx);
|
||||
const data = {
|
||||
createdAt: new Date(),
|
||||
files: files,
|
||||
poll: request.poll ? {
|
||||
choices: request.poll.options,
|
||||
multiple: request.poll.multiple,
|
||||
expiresAt: request.poll.expires_in && request.poll.expires_in > 0 ? new Date(new Date().getTime() + request.poll.expires_in * 1000) : null
|
||||
} : undefined,
|
||||
text: request.text,
|
||||
reply: reply,
|
||||
renote: renote,
|
||||
cw: request.spoiler_text,
|
||||
visibility: visibility,
|
||||
visibleUsers: Promise.resolve(visibility).then((p)=>p === 'specified' ? this.extractMentions(request.text ?? '', ctx) : undefined)
|
||||
};
|
||||
return createNote(user, await awaitAll(data));
|
||||
}
|
||||
static async editNote(request, note, ctx) {
|
||||
const user = ctx.user;
|
||||
const files = request.media_ids && request.media_ids.length > 0 ? DriveFiles.findByIds(request.media_ids) : [];
|
||||
const data = {
|
||||
files: files,
|
||||
poll: request.poll ? {
|
||||
choices: request.poll.options,
|
||||
multiple: request.poll.multiple,
|
||||
expiresAt: request.poll.expires_in && request.poll.expires_in > 0 ? new Date(new Date().getTime() + request.poll.expires_in * 1000) : null
|
||||
} : null,
|
||||
text: request.text,
|
||||
cw: request.spoiler_text
|
||||
};
|
||||
return editNote(user, note, await awaitAll(data));
|
||||
}
|
||||
static async extractMentions(text, ctx) {
|
||||
const user = ctx.user;
|
||||
return extractMentionedUsers(user, mfm.parse(text));
|
||||
}
|
||||
static normalizeComposeOptions(body) {
|
||||
const result = {};
|
||||
if (body.status != null && body.status.trim().length > 0) result.text = body.status;
|
||||
if (body.spoiler_text != null && body.spoiler_text.trim().length > 0) result.spoiler_text = body.spoiler_text;
|
||||
if (body.visibility != null) result.visibility = VisibilityConverter.decode(body.visibility);
|
||||
if (body.language != null) result.language = body.language;
|
||||
if (body.scheduled_at != null) result.scheduled_at = new Date(Date.parse(body.scheduled_at));
|
||||
if (body.in_reply_to_id) result.in_reply_to_id = body.in_reply_to_id;
|
||||
if (body.quoted_status_id ?? body.quote_id) result.quote_id = body.quoted_status_id ?? body.quote_id;
|
||||
if (body.media_ids) result.media_ids = body.media_ids && body.media_ids.length > 0 ? toArray(body.media_ids) : undefined;
|
||||
if (body.poll) {
|
||||
result.poll = {
|
||||
expires_in: parseInt(body.poll.expires_in, 10),
|
||||
options: body.poll.options,
|
||||
multiple: !!body.poll.multiple
|
||||
};
|
||||
}
|
||||
result.sensitive = !!body.sensitive;
|
||||
return result;
|
||||
}
|
||||
static normalizeEditOptions(body) {
|
||||
const result = {};
|
||||
if (body.status != null && body.status.trim().length > 0) result.text = body.status;
|
||||
if (body.spoiler_text != null && body.spoiler_text.trim().length > 0) result.spoiler_text = body.spoiler_text;
|
||||
if (body.language != null) result.language = body.language;
|
||||
if (body.media_ids) result.media_ids = body.media_ids && body.media_ids.length > 0 ? toArray(body.media_ids) : undefined;
|
||||
if (body.poll) {
|
||||
result.poll = {
|
||||
expires_in: parseInt(body.poll.expires_in, 10),
|
||||
options: body.poll.options,
|
||||
multiple: !!body.poll.multiple
|
||||
};
|
||||
}
|
||||
result.sensitive = !!body.sensitive;
|
||||
return result;
|
||||
}
|
||||
static async getNoteOr404(id, ctx) {
|
||||
const user = ctx.user;
|
||||
return getNote(id, user).catch((_)=>{
|
||||
throw new MastoApiError(404);
|
||||
});
|
||||
}
|
||||
static async getConversationFromEvent(noteId, user) {
|
||||
const ctx = getStubMastoContext(user);
|
||||
const note = await getNote(noteId, ctx.user);
|
||||
const conversationId = note.threadId ?? note.id;
|
||||
const userIds = unique([
|
||||
note.userId
|
||||
].concat(note.visibleUserIds).filter((p)=>p != ctx.user.id));
|
||||
const users = userIds.map((id)=>UserHelpers.getUserCached(id, ctx).catch((_)=>null));
|
||||
const accounts = Promise.all(users).then((u)=>UserConverter.encodeMany(u.filter((u)=>u), ctx));
|
||||
const res = {
|
||||
id: conversationId,
|
||||
accounts: accounts.then((u)=>u.length > 0 ? u : UserConverter.encodeMany([
|
||||
ctx.user
|
||||
], ctx)),
|
||||
last_status: NoteConverter.encode(note, ctx),
|
||||
unread: true
|
||||
};
|
||||
return awaitAll(res);
|
||||
}
|
||||
static fixupEventNote(note) {
|
||||
note.createdAt = note.createdAt ? new Date(note.createdAt) : note.createdAt;
|
||||
note.updatedAt = note.updatedAt ? new Date(note.updatedAt) : note.updatedAt;
|
||||
note.reply = null;
|
||||
note.renote = null;
|
||||
note.user = null;
|
||||
return note;
|
||||
}
|
||||
static getIdempotencyKey(ctx) {
|
||||
const headers = ctx.headers;
|
||||
const user = ctx.user;
|
||||
if (headers["idempotency-key"] === undefined || headers["idempotency-key"] === null) return null;
|
||||
return `${user.id}-${Array.isArray(headers["idempotency-key"]) ? headers["idempotency-key"].at(-1) : headers["idempotency-key"]}`;
|
||||
}
|
||||
static async getFromIdempotencyCache(key) {
|
||||
return this.postIdempotencyLocks.acquire(key, async ()=>{
|
||||
if (await this.postIdempotencyCache.get(key) !== undefined) {
|
||||
let i = 5;
|
||||
while((await this.postIdempotencyCache.get(key))?.status === undefined){
|
||||
if (++i > 5) throw new Error('Post is duplicate but unable to resolve original');
|
||||
await new Promise((resolve)=>{
|
||||
setTimeout(resolve, 500);
|
||||
});
|
||||
}
|
||||
return (await this.postIdempotencyCache.get(key))?.status;
|
||||
} else {
|
||||
await this.postIdempotencyCache.set(key, {});
|
||||
return undefined;
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user