82 lines
2.1 KiB
TypeScript
82 lines
2.1 KiB
TypeScript
/*
|
||
* SPDX-FileCopyrightText: syuilo and misskey-project
|
||
* SPDX-License-Identifier: AGPL-3.0-only
|
||
*/
|
||
|
||
import { EventEmitter } from 'eventemitter3';
|
||
import * as Misskey from 'misskey-js';
|
||
import type { Channels, StreamEvents, IStream, IChannelConnection } from 'misskey-js';
|
||
|
||
type AnyOf<T extends Record<any, any>> = T[keyof T];
|
||
type OmitFirst<T extends any[]> = T extends [any, ...infer R] ? R : never;
|
||
|
||
/**
|
||
* Websocket無効化時に使うStreamのモック（なにもしない）
|
||
*/
|
||
export class StreamMock extends EventEmitter<StreamEvents> implements IStream {
|
||
public readonly state = 'initializing';
|
||
|
||
constructor(...args: ConstructorParameters<typeof Misskey.Stream>) {
|
||
super();
|
||
// do nothing
|
||
}
|
||
|
||
public useChannel<C extends keyof Channels>(channel: C, params?: Channels[C]['params'], name?: string): ChannelConnectionMock<Channels[C]> {
|
||
return new ChannelConnectionMock(this, channel, name);
|
||
}
|
||
|
||
public removeSharedConnection(connection: any): void {
|
||
// do nothing
|
||
}
|
||
|
||
public removeSharedConnectionPool(pool: any): void {
|
||
// do nothing
|
||
}
|
||
|
||
public disconnectToChannel(): void {
|
||
// do nothing
|
||
}
|
||
|
||
public send(typeOrPayload: string): void;
|
||
public send(typeOrPayload: string, payload: any): void;
|
||
public send(typeOrPayload: Record<string, any> | any[]): void;
|
||
public send(typeOrPayload: string | Record<string, any> | any[], payload?: any): void {
|
||
// do nothing
|
||
}
|
||
|
||
public ping(): void {
|
||
// do nothing
|
||
}
|
||
|
||
public heartbeat(): void {
|
||
// do nothing
|
||
}
|
||
|
||
public close(): void {
|
||
// do nothing
|
||
}
|
||
}
|
||
|
||
class ChannelConnectionMock<Channel extends AnyOf<Channels> = any> extends EventEmitter<Channel['events']> implements IChannelConnection<Channel> {
|
||
public id = '';
|
||
public name?: string; // for debug
|
||
public inCount = 0; // for debug
|
||
public outCount = 0; // for debug
|
||
public channel: string;
|
||
|
||
constructor(stream: IStream, ...args: OmitFirst<ConstructorParameters<typeof Misskey.ChannelConnection<Channel>>>) {
|
||
super();
|
||
|
||
this.channel = args[0];
|
||
this.name = args[1];
|
||
}
|
||
|
||
public send<T extends keyof Channel['receives']>(type: T, body: Channel['receives'][T]): void {
|
||
// do nothing
|
||
}
|
||
|
||
public dispose(): void {
|
||
// do nothing
|
||
}
|
||
}
|