import { Inject, Injectable } from '@nestjs/common'; import type { UserListJoiningsRepository, UserListsRepository } from '@/models/index.js'; import type { User } from '@/models/entities/User.js'; import { isUserRelated } from '@/misc/is-user-related.js'; import type { Packed } from '@/misc/json-schema.js'; import { NoteEntityService } from '@/core/entities/NoteEntityService.js'; import { DI } from '@/di-symbols.js'; import { bindThis } from '@/decorators.js'; import Channel from '../channel.js'; class UserListChannel extends Channel { public readonly chName = 'userList'; public static shouldShare = false; public static requireCredential = false; private listId: string; public listUsers: User['id'][] = []; private listUsersClock: NodeJS.Timer; constructor( private userListsRepository: UserListsRepository, private userListJoiningsRepository: UserListJoiningsRepository, private noteEntityService: NoteEntityService, id: string, connection: Channel['connection'], ) { super(id, connection); //this.updateListUsers = this.updateListUsers.bind(this); //this.onNote = this.onNote.bind(this); } @bindThis public async init(params: any) { this.listId = params.listId as string; // Check existence and owner const list = await this.userListsRepository.findOneBy({ id: this.listId, userId: this.user!.id, }); if (!list) return; // Subscribe stream this.subscriber.on(`userListStream:${this.listId}`, this.send); this.subscriber.on('notesStream', this.onNote); this.updateListUsers(); this.listUsersClock = setInterval(this.updateListUsers, 5000); } @bindThis private async updateListUsers() { const users = await this.userListJoiningsRepository.find({ where: { userListId: this.listId, }, select: ['userId'], }); this.listUsers = users.map(x => x.userId); } @bindThis private async onNote(note: Packed<'Note'>) { if (!this.listUsers.includes(note.userId)) return; if (['followers', 'specified'].includes(note.visibility)) { note = await this.noteEntityService.pack(note.id, this.user, { detail: true, }); if (note.isHidden) { return; } } else { // リプライなら再pack if (note.replyId != null) { note.reply = await this.noteEntityService.pack(note.replyId, this.user, { detail: true, }); } // Renoteなら再pack if (note.renoteId != null) { note.renote = await this.noteEntityService.pack(note.renoteId, this.user, { detail: true, }); } } // 流れてきたNoteがミュートしているユーザーが関わるものだったら無視する if (isUserRelated(note, this.userIdsWhoMeMuting)) return; // 流れてきたNoteがブロックされているユーザーが関わるものだったら無視する if (isUserRelated(note, this.userIdsWhoBlockingMe)) return; if (note.renote && !note.text && isUserRelated(note, this.userIdsWhoMeMutingRenotes)) return; this.send('note', note); } @bindThis public dispose() { // Unsubscribe events this.subscriber.off(`userListStream:${this.listId}`, this.send); this.subscriber.off('notesStream', this.onNote); clearInterval(this.listUsersClock); } } @Injectable() export class UserListChannelService { public readonly shouldShare = UserListChannel.shouldShare; public readonly requireCredential = UserListChannel.requireCredential; constructor( @Inject(DI.userListsRepository) private userListsRepository: UserListsRepository, @Inject(DI.userListJoiningsRepository) private userListJoiningsRepository: UserListJoiningsRepository, private noteEntityService: NoteEntityService, ) { } @bindThis public create(id: string, connection: Channel['connection']): UserListChannel { return new UserListChannel( this.userListsRepository, this.userListJoiningsRepository, this.noteEntityService, id, connection, ); } }