diff --git a/src/server/api/stream/index.ts b/src/server/api/stream/index.ts index 743d77b2a3..ef6397fcd9 100644 --- a/src/server/api/stream/index.ts +++ b/src/server/api/stream/index.ts @@ -24,6 +24,7 @@ export default class Connection { public subscriber: Xev; private channels: Channel[] = []; private subscribingNotes: any = {}; + public sendMessageToWsOverride: any = null; // 後方互換性のため constructor( wsConnection: websocket.connection, @@ -164,6 +165,7 @@ export default class Connection { */ @autobind public sendMessageToWs(type: string, payload: any) { + if (this.sendMessageToWsOverride) return this.sendMessageToWsOverride(type, payload); // 後方互換性のため this.wsConnection.send(JSON.stringify({ type: type, body: payload diff --git a/src/server/api/streaming.ts b/src/server/api/streaming.ts index b7793eb284..8bfa5574f1 100644 --- a/src/server/api/streaming.ts +++ b/src/server/api/streaming.ts @@ -25,7 +25,7 @@ module.exports = (server: http.Server) => { // 後方互換性のため if (request.resourceURL.pathname !== '/streaming') { - main.sendMessageToWs = (type: string, payload: any) => { + main.sendMessageToWsOverride = (type: string, payload: any) => { if (type == 'channel') { type = payload.type; payload = payload.body; @@ -35,13 +35,12 @@ module.exports = (server: http.Server) => { body: payload })); }; - if (request.resourceURL.pathname === '/') { - main.connectChannel(Math.random().toString(), null, - request.resourceURL.pathname === '/' ? channels.homeTimeline : - request.resourceURL.pathname === '/local-timeline' ? channels.localTimeline : - request.resourceURL.pathname === '/hybrid-timeline' ? channels.hybridTimeline : - request.resourceURL.pathname === '/global-timeline' ? channels.globalTimeline : null); - } + + main.connectChannel(Math.random().toString(), null, + request.resourceURL.pathname === '/' ? channels.homeTimeline : + request.resourceURL.pathname === '/local-timeline' ? channels.localTimeline : + request.resourceURL.pathname === '/hybrid-timeline' ? channels.hybridTimeline : + request.resourceURL.pathname === '/global-timeline' ? channels.globalTimeline : null); } connection.once('close', () => {