import { EntityRepository, Repository, In } from 'typeorm'; import { Note } from '../entities/note'; import { User } from '../entities/user'; import { unique, concat } from '../../prelude/array'; import { nyaize } from '../../misc/nyaize'; import { Emojis, Users, Apps, PollVotes, DriveFiles, NoteReactions, Followings, Polls } from '..'; import { ensure } from '../../prelude/ensure'; import { SchemaType } from '../../misc/schema'; import { awaitAll } from '../../prelude/await-all'; export type PackedNote = SchemaType; @EntityRepository(Note) export class NoteRepository extends Repository { public validateCw(x: string) { return x.trim().length <= 100; } private async hideNote(packedNote: PackedNote, meId: User['id'] | null) { let hide = false; // visibility が specified かつ自分が指定されていなかったら非表示 if (packedNote.visibility === 'specified') { if (meId == null) { hide = true; } else if (meId === packedNote.userId) { hide = false; } else { // 指定されているかどうか const specified = packedNote.visibleUserIds!.some((id: any) => meId === id); if (specified) { hide = false; } else { hide = true; } } } // visibility が followers かつ自分が投稿者のフォロワーでなかったら非表示 if (packedNote.visibility == 'followers') { if (meId == null) { hide = true; } else if (meId === packedNote.userId) { hide = false; } else if (packedNote.reply && (meId === (packedNote.reply as PackedNote).userId)) { // 自分の投稿に対するリプライ hide = false; } else if (packedNote.mentions && packedNote.mentions.some(id => meId === id)) { // 自分へのメンション hide = false; } else { // フォロワーかどうか const following = await Followings.findOne({ followeeId: packedNote.userId, followerId: meId }); if (following == null) { hide = true; } else { hide = false; } } } if (hide) { packedNote.visibleUserIds = undefined; packedNote.fileIds = []; packedNote.files = []; packedNote.text = null; packedNote.poll = undefined; packedNote.cw = null; packedNote.geo = undefined; packedNote.isHidden = true; } } public async pack( src: Note['id'] | Note, me?: User['id'] | User | null | undefined, options?: { detail?: boolean; skipHide?: boolean; } ): Promise { const opts = Object.assign({ detail: true, skipHide: false }, options); const meId = me ? typeof me === 'string' ? me : me.id : null; const note = typeof src === 'object' ? src : await this.findOne(src).then(ensure); const host = note.userHost; async function populatePoll() { const poll = await Polls.findOne(note.id).then(ensure); const choices = poll.choices.map(c => ({ text: c, votes: poll.votes[poll.choices.indexOf(c)], isVoted: false })); if (poll.multiple) { const votes = await PollVotes.find({ 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.findOne({ userId: meId!, noteId: note.id }); if (vote) { choices[vote.choice].isVoted = true; } } return { multiple: poll.multiple, expiresAt: poll.expiresAt, choices }; } async function populateMyReaction() { const reaction = await NoteReactions.findOne({ userId: meId!, noteId: note.id, }); if (reaction) { return reaction.reaction; } return undefined; } let text = note.text; if (note.name) { text = `【${note.name}】\n${note.text}`; } const reactionEmojis = unique(concat([note.emojis, Object.keys(note.reactions)])); const packed = await awaitAll({ id: note.id, createdAt: note.createdAt.toISOString(), app: note.appId ? Apps.pack(note.appId) : undefined, userId: note.userId, user: Users.pack(note.user || note.userId, meId), text: text, cw: note.cw, visibility: note.visibility, localOnly: note.localOnly || undefined, visibleUserIds: note.visibility === 'specified' ? note.visibleUserIds : undefined, viaMobile: note.viaMobile || undefined, renoteCount: note.renoteCount, repliesCount: note.repliesCount, reactions: note.reactions, tags: note.tags.length > 0 ? note.tags : undefined, emojis: reactionEmojis.length > 0 ? Emojis.find({ name: In(reactionEmojis), host: host }) : [], fileIds: note.fileIds, files: DriveFiles.packMany(note.fileIds), replyId: note.replyId, renoteId: note.renoteId, mentions: note.mentions.length > 0 ? note.mentions : undefined, uri: note.uri || undefined, ...(opts.detail ? { reply: note.replyId ? this.pack(note.replyId, meId, { detail: false }) : undefined, renote: note.renoteId ? this.pack(note.renoteId, meId, { detail: true }) : undefined, poll: note.hasPoll ? populatePoll() : undefined, ...(meId ? { myReaction: populateMyReaction() } : {}) } : {}) }); if (packed.user.isCat && packed.text) { packed.text = nyaize(packed.text); } if (!opts.skipHide) { await this.hideNote(packed, meId); } return packed; } public packMany( notes: (Note['id'] | Note)[], me?: User['id'] | User | null | undefined, options?: { detail?: boolean; skipHide?: boolean; } ) { return Promise.all(notes.map(n => this.pack(n, me, options))); } } export const packedNoteSchema = { type: 'object' as const, optional: false as const, nullable: false as const, properties: { id: { type: 'string' as const, optional: false as const, nullable: false as const, format: 'id', description: 'The unique identifier for this Note.', example: 'xxxxxxxxxx', }, createdAt: { type: 'string' as const, optional: false as const, nullable: false as const, format: 'date-time', description: 'The date that the Note was created on Misskey.' }, text: { type: 'string' as const, optional: false as const, nullable: true as const, }, cw: { type: 'string' as const, optional: true as const, nullable: true as const, }, userId: { type: 'string' as const, optional: false as const, nullable: false as const, format: 'id', }, user: { type: 'object' as const, ref: 'User', optional: false as const, nullable: false as const, }, replyId: { type: 'string' as const, optional: true as const, nullable: true as const, format: 'id', example: 'xxxxxxxxxx', }, renoteId: { type: 'string' as const, optional: true as const, nullable: true as const, format: 'id', example: 'xxxxxxxxxx', }, reply: { type: 'object' as const, optional: true as const, nullable: true as const, ref: 'Note' }, renote: { type: 'object' as const, optional: true as const, nullable: true as const, ref: 'Note' }, viaMobile: { type: 'boolean' as const, optional: true as const, nullable: false as const, }, isHidden: { type: 'boolean' as const, optional: true as const, nullable: false as const, }, visibility: { type: 'string' as const, optional: false as const, nullable: false as const, }, mentions: { type: 'array' as const, optional: true as const, nullable: false as const, items: { type: 'string' as const, optional: false as const, nullable: false as const, format: 'id' } }, visibleUserIds: { type: 'array' as const, optional: true as const, nullable: false as const, items: { type: 'string' as const, optional: false as const, nullable: false as const, format: 'id' } }, fileIds: { type: 'array' as const, optional: true as const, nullable: false as const, items: { type: 'string' as const, optional: false as const, nullable: false as const, format: 'id' } }, files: { type: 'array' as const, optional: true as const, nullable: false as const, items: { type: 'object' as const, optional: false as const, nullable: false as const, ref: 'DriveFile' } }, tags: { type: 'array' as const, optional: true as const, nullable: false as const, items: { type: 'string' as const, optional: false as const, nullable: false as const, } }, poll: { type: 'object' as const, optional: true as const, nullable: true as const, }, geo: { type: 'object' as const, optional: true as const, nullable: true as const, }, }, };