138 lines
3.8 KiB
TypeScript
138 lines
3.8 KiB
TypeScript
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,
|
|
);
|
|
}
|
|
}
|