10ce7bf3c4
* chore: add JsonValue type * refactor: kill any from Connection.ts * refactor: fix StreamEventEmitter contains undefined instead of null * refactor: kill any from channels * docs(changelog): Fix: Steaming APIが不正なデータを受けた場合の動作が不安定である問題 * fix license header * fix lints
80 lines
2.1 KiB
TypeScript
80 lines
2.1 KiB
TypeScript
/*
|
|
* SPDX-FileCopyrightText: syuilo and misskey-project
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
import { Injectable } from '@nestjs/common';
|
|
import { NoteEntityService } from '@/core/entities/NoteEntityService.js';
|
|
import { bindThis } from '@/decorators.js';
|
|
import type { GlobalEvents } from '@/core/GlobalEventService.js';
|
|
import type { JsonObject } from '@/misc/json-value.js';
|
|
import Channel, { type MiChannelService } from '../channel.js';
|
|
|
|
class AntennaChannel extends Channel {
|
|
public readonly chName = 'antenna';
|
|
public static shouldShare = false;
|
|
public static requireCredential = true as const;
|
|
public static kind = 'read:account';
|
|
private antennaId: string;
|
|
|
|
constructor(
|
|
private noteEntityService: NoteEntityService,
|
|
|
|
id: string,
|
|
connection: Channel['connection'],
|
|
) {
|
|
super(id, connection);
|
|
//this.onEvent = this.onEvent.bind(this);
|
|
}
|
|
|
|
@bindThis
|
|
public async init(params: JsonObject) {
|
|
if (typeof params.antennaId !== 'string') return;
|
|
this.antennaId = params.antennaId;
|
|
|
|
// Subscribe stream
|
|
this.subscriber.on(`antennaStream:${this.antennaId}`, this.onEvent);
|
|
}
|
|
|
|
@bindThis
|
|
private async onEvent(data: GlobalEvents['antenna']['payload']) {
|
|
if (data.type === 'note') {
|
|
const note = await this.noteEntityService.pack(data.body.id, this.user, { detail: true });
|
|
|
|
if (this.isNoteMutedOrBlocked(note)) return;
|
|
|
|
this.connection.cacheNote(note);
|
|
|
|
this.send('note', note);
|
|
} else {
|
|
this.send(data.type, data.body);
|
|
}
|
|
}
|
|
|
|
@bindThis
|
|
public dispose() {
|
|
// Unsubscribe events
|
|
this.subscriber.off(`antennaStream:${this.antennaId}`, this.onEvent);
|
|
}
|
|
}
|
|
|
|
@Injectable()
|
|
export class AntennaChannelService implements MiChannelService<true> {
|
|
public readonly shouldShare = AntennaChannel.shouldShare;
|
|
public readonly requireCredential = AntennaChannel.requireCredential;
|
|
public readonly kind = AntennaChannel.kind;
|
|
|
|
constructor(
|
|
private noteEntityService: NoteEntityService,
|
|
) {
|
|
}
|
|
|
|
@bindThis
|
|
public create(id: string, connection: Channel['connection']): AntennaChannel {
|
|
return new AntennaChannel(
|
|
this.noteEntityService,
|
|
id,
|
|
connection,
|
|
);
|
|
}
|
|
}
|