import autobind from 'autobind-decorator'; import * as websocket from 'websocket'; import { readNotification } from '../common/read-notification'; import call from '../call'; import Channel from './channel'; import channels from './channels'; import { EventEmitter } from 'events'; import { User } from '../../../models/entities/user'; import { Channel as ChannelModel } from '../../../models/entities/channel'; import { Users, Followings, Mutings, UserProfiles, ChannelFollowings } from '../../../models'; import { ApiError } from '../error'; import { AccessToken } from '../../../models/entities/access-token'; import { UserProfile } from '../../../models/entities/user-profile'; import { publishChannelStream, publishGroupMessagingStream, publishMessagingStream } from '../../../services/stream'; import { UserGroup } from '../../../models/entities/user-group'; import { readMention } from '../../../services/note/read-mention'; import { readSpecifiedNote } from '../../../services/note/read-specified-note'; /** * Main stream connection */ export default class Connection { public user?: User; public userProfile?: UserProfile; public following: Set = new Set(); public muting: Set = new Set(); public followingChannels: Set = new Set(); public token?: AccessToken; private wsConnection: websocket.connection; public subscriber: EventEmitter; private channels: Channel[] = []; private subscribingNotes: any = {}; private followingClock: ReturnType; private mutingClock: ReturnType; private followingChannelsClock: ReturnType; private userProfileClock: ReturnType; constructor( wsConnection: websocket.connection, subscriber: EventEmitter, user: User | null | undefined, token: AccessToken | null | undefined ) { this.wsConnection = wsConnection; this.subscriber = subscriber; if (user) this.user = user; if (token) this.token = token; this.wsConnection.on('message', this.onWsConnectionMessage); this.subscriber.on('broadcast', async ({ type, body }) => { this.onBroadcastMessage(type, body); }); // TODO: reidsでイベントをもらうようにし、ポーリングはしないようにする if (this.user) { this.updateFollowing(); this.followingClock = setInterval(this.updateFollowing, 5000); this.updateMuting(); this.mutingClock = setInterval(this.updateMuting, 5000); this.updateFollowingChannels(); this.followingChannelsClock = setInterval(this.updateFollowingChannels, 5000); this.updateUserProfile(); this.userProfileClock = setInterval(this.updateUserProfile, 5000); } } /** * クライアントからメッセージ受信時 */ @autobind private async onWsConnectionMessage(data: websocket.IMessage) { if (data.utf8Data == null) return; let obj: Record; try { obj = JSON.parse(data.utf8Data); } catch (e) { return; } const { type, body } = obj; switch (type) { case 'api': this.onApiRequest(body); break; case 'readNotification': this.onReadNotification(body); break; case 'readMention': this.onReadMention(body); break; case 'readSpecifiedNote': this.onReadSpecifiedNote(body); break; case 'subNote': this.onSubscribeNote(body); break; case 'sn': this.onSubscribeNote(body); break; // alias case 'unsubNote': this.onUnsubscribeNote(body); break; case 'un': this.onUnsubscribeNote(body); break; // alias case 'connect': this.onChannelConnectRequested(body); break; case 'disconnect': this.onChannelDisconnectRequested(body); break; case 'channel': this.onChannelMessageRequested(body); break; case 'ch': this.onChannelMessageRequested(body); break; // alias // 個々のチャンネルではなくルートレベルでこれらのメッセージを受け取る理由は、 // クライアントの事情を考慮したとき、入力フォームはノートチャンネルやメッセージのメインコンポーネントとは別 // なこともあるため、それらのコンポーネントがそれぞれ各チャンネルに接続するようにするのは面倒なため。 case 'typingOnChannel': this.typingOnChannel(body.channel); break; case 'typingOnMessaging': this.typingOnMessaging(body); break; } } @autobind private onBroadcastMessage(type: string, body: any) { this.sendMessageToWs(type, body); } /** * APIリクエスト要求時 */ @autobind private async onApiRequest(payload: any) { // 新鮮なデータを利用するためにユーザーをフェッチ const user = this.user ? await Users.findOne(this.user.id) : null; const endpoint = payload.endpoint || payload.ep; // alias // 呼び出し call(endpoint, user, this.token, payload.data).then(res => { this.sendMessageToWs(`api:${payload.id}`, { res }); }).catch((e: ApiError) => { this.sendMessageToWs(`api:${payload.id}`, { error: { message: e.message, code: e.code, id: e.id, kind: e.kind, ...(e.info ? { info: e.info } : {}) } }); }); } @autobind private onReadNotification(payload: any) { if (!payload.id) return; readNotification(this.user!.id, [payload.id]); } @autobind private onReadMention(payload: any) { if (!payload.id) return; if (this.user) { // TODO: ある程度まとめてreadMentionするようにする // 具体的には、この箇所ではキュー的な配列にread予定ノートを溜めておくに留めて、別の箇所で定期的にキューにあるノートを配列でreadMentionに渡すような実装にする readMention(this.user.id, [payload.id]); } } @autobind private onReadSpecifiedNote(payload: any) { if (!payload.id) return; if (this.user) { // TODO: ある程度まとめてreadSpecifiedNoteするようにする // 具体的には、この箇所ではキュー的な配列にread予定ノートを溜めておくに留めて、別の箇所で定期的にキューにあるノートを配列でreadSpecifiedNoteに渡すような実装にする readSpecifiedNote(this.user.id, [payload.id]); } } /** * 投稿購読要求時 */ @autobind private onSubscribeNote(payload: any) { if (!payload.id) return; if (this.subscribingNotes[payload.id] == null) { this.subscribingNotes[payload.id] = 0; } this.subscribingNotes[payload.id]++; if (this.subscribingNotes[payload.id] === 1) { this.subscriber.on(`noteStream:${payload.id}`, this.onNoteStreamMessage); } } /** * 投稿購読解除要求時 */ @autobind private onUnsubscribeNote(payload: any) { if (!payload.id) return; this.subscribingNotes[payload.id]--; if (this.subscribingNotes[payload.id] <= 0) { delete this.subscribingNotes[payload.id]; this.subscriber.off(`noteStream:${payload.id}`, this.onNoteStreamMessage); } } @autobind private async onNoteStreamMessage(data: any) { this.sendMessageToWs('noteUpdated', { id: data.body.id, type: data.type, body: data.body.body, }); } /** * チャンネル接続要求時 */ @autobind private onChannelConnectRequested(payload: any) { const { channel, id, params, pong } = payload; this.connectChannel(id, params, channel, pong); } /** * チャンネル切断要求時 */ @autobind private onChannelDisconnectRequested(payload: any) { const { id } = payload; this.disconnectChannel(id); } /** * クライアントにメッセージ送信 */ @autobind public sendMessageToWs(type: string, payload: any) { this.wsConnection.send(JSON.stringify({ type: type, body: payload })); } /** * チャンネルに接続 */ @autobind public connectChannel(id: string, params: any, channel: string, pong = false) { if ((channels as any)[channel].requireCredential && this.user == null) { return; } // 共有可能チャンネルに接続しようとしていて、かつそのチャンネルに既に接続していたら無意味なので無視 if ((channels as any)[channel].shouldShare && this.channels.some(c => c.chName === channel)) { return; } const ch: Channel = new (channels as any)[channel](id, this); this.channels.push(ch); ch.init(params); if (pong) { this.sendMessageToWs('connected', { id: id }); } } /** * チャンネルから切断 * @param id チャンネルコネクションID */ @autobind public disconnectChannel(id: string) { const channel = this.channels.find(c => c.id === id); if (channel) { if (channel.dispose) channel.dispose(); this.channels = this.channels.filter(c => c.id !== id); } } /** * チャンネルへメッセージ送信要求時 * @param data メッセージ */ @autobind private onChannelMessageRequested(data: any) { const channel = this.channels.find(c => c.id === data.id); if (channel != null && channel.onMessage != null) { channel.onMessage(data.type, data.body); } } @autobind private typingOnChannel(channel: ChannelModel['id']) { if (this.user) { publishChannelStream(channel, 'typing', this.user.id); } } @autobind private typingOnMessaging(param: { partner?: User['id']; group?: UserGroup['id']; }) { if (this.user) { if (param.partner) { publishMessagingStream(param.partner, this.user.id, 'typing', this.user.id); } else if (param.group) { publishGroupMessagingStream(param.group, 'typing', this.user.id); } } } @autobind private async updateFollowing() { const followings = await Followings.find({ where: { followerId: this.user!.id }, select: ['followeeId'] }); this.following = new Set(followings.map(x => x.followeeId)); } @autobind private async updateMuting() { const mutings = await Mutings.find({ where: { muterId: this.user!.id }, select: ['muteeId'] }); this.muting = new Set(mutings.map(x => x.muteeId)); } @autobind private async updateFollowingChannels() { const followings = await ChannelFollowings.find({ where: { followerId: this.user!.id }, select: ['followeeId'] }); this.followingChannels = new Set(followings.map(x => x.followeeId)); } @autobind private async updateUserProfile() { this.userProfile = await UserProfiles.findOne({ userId: this.user!.id }); } /** * ストリームが切れたとき */ @autobind public dispose() { for (const c of this.channels.filter(c => c.dispose)) { if (c.dispose) c.dispose(); } if (this.followingClock) clearInterval(this.followingClock); if (this.mutingClock) clearInterval(this.mutingClock); if (this.followingChannelsClock) clearInterval(this.followingChannelsClock); if (this.userProfileClock) clearInterval(this.userProfileClock); } }