Sharkey/src/server/api/stream/index.ts

232 lines
5.7 KiB
TypeScript
Raw Normal View History

import autobind from 'autobind-decorator';
import * as websocket from 'websocket';
import User, { IUser } from '../../../models/user';
import readNotification from '../common/read-notification';
import call from '../call';
import { IApp } from '../../../models/app';
import readNote from '../../../services/note/read';
import Channel from './channel';
import channels from './channels';
import { EventEmitter } from 'events';
/**
* Main stream connection
*/
export default class Connection {
public user?: IUser;
public app: IApp;
private wsConnection: websocket.connection;
public subscriber: EventEmitter;
private channels: Channel[] = [];
private subscribingNotes: any = {};
2018-10-07 18:58:10 +03:00
public sendMessageToWsOverride: any = null; // 後方互換性のため
constructor(
wsConnection: websocket.connection,
subscriber: EventEmitter,
user: IUser,
app: IApp
) {
this.wsConnection = wsConnection;
this.user = user;
this.app = app;
this.subscriber = subscriber;
this.wsConnection.on('message', this.onWsConnectionMessage);
}
/**
*
*/
@autobind
private async onWsConnectionMessage(data: websocket.IMessage) {
const { type, body } = JSON.parse(data.utf8Data);
switch (type) {
case 'api': this.onApiRequest(body); break;
case 'alive': this.onAlive(); break;
case 'readNotification': this.onReadNotification(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;
2018-10-09 21:24:09 +03:00
case 'ch': this.onChannelMessageRequested(body); break; // alias
}
}
/**
* APIリクエスト要求時
*/
@autobind
private async onApiRequest(payload: any) {
// 新鮮なデータを利用するためにユーザーをフェッチ
const user = this.user ? await User.findOne({ _id: this.user._id }) : null;
const endpoint = payload.endpoint || payload.ep; // alias
// 呼び出し
call(endpoint, user, this.app, payload.data).then(res => {
this.sendMessageToWs(`api:${payload.id}`, { res });
}).catch(e => {
this.sendMessageToWs(`api:${payload.id}`, { e });
});
}
@autobind
private onAlive() {
// Update lastUsedAt
User.update({ _id: this.user._id }, {
$set: {
'lastUsedAt': new Date()
}
});
}
@autobind
private onReadNotification(payload: any) {
if (!payload.id) return;
readNotification(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);
}
if (payload.read) {
readNote(this.user._id, payload.id);
}
}
/**
* 稿
*/
@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) {
2018-10-07 18:58:10 +03:00
if (this.sendMessageToWsOverride) return this.sendMessageToWsOverride(type, payload); // 後方互換性のため
this.wsConnection.send(JSON.stringify({
type: type,
body: payload
}));
}
/**
*
*/
@autobind
public connectChannel(id: string, params: any, channel: string, pong = false) {
2018-11-10 19:22:34 +02:00
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
});
}
}
/**
*
2018-10-07 11:06:28 +03:00
* @param id ID
*/
@autobind
2018-10-07 11:19:52 +03:00
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);
}
}
2018-10-07 11:06:28 +03:00
/**
*
* @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
public dispose() {
this.channels.forEach(c => {
if (c.dispose) c.dispose();
});
}
}