Sharkey/packages/backend/src/core/AntennaService.ts

231 lines
7.2 KiB
TypeScript
Raw Normal View History

2022-09-17 21:27:08 +03:00
import { Inject, Injectable } from '@nestjs/common';
import Redis from 'ioredis';
import type { Antenna } from '@/models/entities/Antenna.js';
import type { Note } from '@/models/entities/Note.js';
import type { User } from '@/models/entities/User.js';
import { NoteEntityService } from '@/core/entities/NoteEntityService.js';
import { AntennaEntityService } from '@/core/entities/AntennaEntityService.js';
2022-09-17 21:27:08 +03:00
import { IdService } from '@/core/IdService.js';
import { isUserRelated } from '@/misc/is-user-related.js';
import { GlobalEventService } from '@/core/GlobalEventService.js';
import { PushNotificationService } from '@/core/PushNotificationService.js';
2022-09-17 21:27:08 +03:00
import * as Acct from '@/misc/acct.js';
2023-03-10 07:22:37 +02:00
import type { Packed } from '@/misc/json-schema.js';
2022-09-17 21:27:08 +03:00
import { DI } from '@/di-symbols.js';
import type { MutingsRepository, NotesRepository, AntennaNotesRepository, AntennasRepository, UserListJoiningsRepository } from '@/models/index.js';
2022-12-04 03:16:03 +02:00
import { UtilityService } from '@/core/UtilityService.js';
import { bindThis } from '@/decorators.js';
import { StreamMessages } from '@/server/api/stream/types.js';
import type { OnApplicationShutdown } from '@nestjs/common';
2022-09-17 21:27:08 +03:00
@Injectable()
export class AntennaService implements OnApplicationShutdown {
2022-09-18 21:11:50 +03:00
private antennasFetched: boolean;
private antennas: Antenna[];
2022-09-17 21:27:08 +03:00
constructor(
@Inject(DI.redisSubscriber)
private redisSubscriber: Redis.Redis,
@Inject(DI.mutingsRepository)
private mutingsRepository: MutingsRepository,
@Inject(DI.notesRepository)
private notesRepository: NotesRepository,
@Inject(DI.antennaNotesRepository)
private antennaNotesRepository: AntennaNotesRepository,
@Inject(DI.antennasRepository)
private antennasRepository: AntennasRepository,
@Inject(DI.userListJoiningsRepository)
private userListJoiningsRepository: UserListJoiningsRepository,
private utilityService: UtilityService,
private idService: IdService,
2023-02-04 03:02:03 +02:00
private globalEventService: GlobalEventService,
private pushNotificationService: PushNotificationService,
private noteEntityService: NoteEntityService,
private antennaEntityService: AntennaEntityService,
2022-09-17 21:27:08 +03:00
) {
2022-09-18 21:11:50 +03:00
this.antennasFetched = false;
this.antennas = [];
2022-09-17 21:27:08 +03:00
this.redisSubscriber.on('message', this.onRedisMessage);
}
@bindThis
2022-09-17 21:27:08 +03:00
public onApplicationShutdown(signal?: string | undefined) {
this.redisSubscriber.off('message', this.onRedisMessage);
}
@bindThis
2022-09-24 00:45:44 +03:00
private async onRedisMessage(_: string, data: string): Promise<void> {
2022-09-17 21:27:08 +03:00
const obj = JSON.parse(data);
if (obj.channel === 'internal') {
const { type, body } = obj.message as StreamMessages['internal']['payload'];
2022-09-17 21:27:08 +03:00
switch (type) {
case 'antennaCreated':
2023-01-25 04:18:16 +02:00
this.antennas.push({
...body,
createdAt: new Date(body.createdAt),
lastUsedAt: new Date(body.lastUsedAt),
2023-01-25 04:18:16 +02:00
});
2022-09-17 21:27:08 +03:00
break;
case 'antennaUpdated':
2023-01-25 04:18:16 +02:00
this.antennas[this.antennas.findIndex(a => a.id === body.id)] = {
...body,
createdAt: new Date(body.createdAt),
lastUsedAt: new Date(body.lastUsedAt),
2023-01-25 04:18:16 +02:00
};
2022-09-17 21:27:08 +03:00
break;
case 'antennaDeleted':
2022-09-18 21:11:50 +03:00
this.antennas = this.antennas.filter(a => a.id !== body.id);
2022-09-17 21:27:08 +03:00
break;
default:
break;
}
}
}
@bindThis
2022-09-17 21:27:08 +03:00
public async addNoteToAntenna(antenna: Antenna, note: Note, noteUser: { id: User['id']; }): Promise<void> {
// 通知しない設定になっているか、自分自身の投稿なら既読にする
const read = !antenna.notify || (antenna.userId === noteUser.id);
this.antennaNotesRepository.insert({
id: this.idService.genId(),
antennaId: antenna.id,
noteId: note.id,
read: read,
});
2023-02-04 03:02:03 +02:00
this.globalEventService.publishAntennaStream(antenna.id, 'note', note);
2022-09-17 21:27:08 +03:00
if (!read) {
const mutings = await this.mutingsRepository.find({
where: {
muterId: antenna.userId,
},
select: ['muteeId'],
});
// Copy
const _note: Note = {
...note,
};
if (note.replyId != null) {
_note.reply = await this.notesRepository.findOneByOrFail({ id: note.replyId });
}
if (note.renoteId != null) {
_note.renote = await this.notesRepository.findOneByOrFail({ id: note.renoteId });
}
if (isUserRelated(_note, new Set<string>(mutings.map(x => x.muteeId)))) {
return;
}
// 2秒経っても既読にならなかったら通知
setTimeout(async () => {
const unread = await this.antennaNotesRepository.findOneBy({ antennaId: antenna.id, read: false });
if (unread) {
2023-02-04 03:02:03 +02:00
this.globalEventService.publishMainStream(antenna.userId, 'unreadAntenna', antenna);
this.pushNotificationService.pushNotification(antenna.userId, 'unreadAntennaNote', {
antenna: { id: antenna.id, name: antenna.name },
note: await this.noteEntityService.pack(note),
});
2022-09-17 21:27:08 +03:00
}
}, 2000);
}
}
// NOTE: フォローしているユーザーのノート、リストのユーザーのノート、グループのユーザーのノート指定はパフォーマンス上の理由で無効になっている
@bindThis
public async checkHitAntenna(antenna: Antenna, note: (Note | Packed<'Note'>), noteUser: { id: User['id']; username: string; host: string | null; }): Promise<boolean> {
2022-09-17 21:27:08 +03:00
if (note.visibility === 'specified') return false;
if (note.visibility === 'followers') return false;
2022-09-17 21:27:08 +03:00
if (!antenna.withReplies && note.replyId != null) return false;
if (antenna.src === 'home') {
// TODO
2022-09-17 21:27:08 +03:00
} else if (antenna.src === 'list') {
const listUsers = (await this.userListJoiningsRepository.findBy({
userListId: antenna.userListId!,
})).map(x => x.userId);
if (!listUsers.includes(note.userId)) return false;
} else if (antenna.src === 'users') {
const accts = antenna.users.map(x => {
const { username, host } = Acct.parse(x);
return this.utilityService.getFullApAccount(username, host).toLowerCase();
});
if (!accts.includes(this.utilityService.getFullApAccount(noteUser.username, noteUser.host).toLowerCase())) return false;
}
const keywords = antenna.keywords
// Clean up
.map(xs => xs.filter(x => x !== ''))
.filter(xs => xs.length > 0);
if (keywords.length > 0) {
if (note.text == null && note.cw == null) return false;
const _text = (note.text ?? '') + '\n' + (note.cw ?? '');
2022-09-17 21:27:08 +03:00
const matched = keywords.some(and =>
and.every(keyword =>
antenna.caseSensitive
? _text.includes(keyword)
: _text.toLowerCase().includes(keyword.toLowerCase()),
2022-09-17 21:27:08 +03:00
));
if (!matched) return false;
}
const excludeKeywords = antenna.excludeKeywords
// Clean up
.map(xs => xs.filter(x => x !== ''))
.filter(xs => xs.length > 0);
if (excludeKeywords.length > 0) {
if (note.text == null && note.cw == null) return false;
const _text = (note.text ?? '') + '\n' + (note.cw ?? '');
2022-09-17 21:27:08 +03:00
const matched = excludeKeywords.some(and =>
and.every(keyword =>
antenna.caseSensitive
? _text.includes(keyword)
: _text.toLowerCase().includes(keyword.toLowerCase()),
2022-09-17 21:27:08 +03:00
));
if (matched) return false;
}
if (antenna.withFile) {
if (note.fileIds && note.fileIds.length === 0) return false;
}
// TODO: eval expression
return true;
}
@bindThis
2022-09-17 21:27:08 +03:00
public async getAntennas() {
2022-09-18 21:11:50 +03:00
if (!this.antennasFetched) {
this.antennas = await this.antennasRepository.findBy({
isActive: true,
});
2022-09-18 21:11:50 +03:00
this.antennasFetched = true;
2022-09-17 21:27:08 +03:00
}
2022-09-18 21:11:50 +03:00
return this.antennas;
2022-09-17 21:27:08 +03:00
}
}