1
0
mirror of https://github.com/hotomoe/hotomoe synced 2024-12-05 10:18:09 +09:00
hotomoe/src/stream.ts

128 lines
4.3 KiB
TypeScript
Raw Normal View History

2016-12-29 07:49:51 +09:00
import * as mongo from 'mongodb';
import redis from './db/redis';
2018-07-30 07:20:27 +09:00
import Xev from 'xev';
import { IMeta } from './models/meta';
import fetchMeta from './misc/fetch-meta';
2017-11-16 23:46:36 +09:00
2018-07-30 07:20:27 +09:00
type ID = string | mongo.ObjectID;
2017-03-20 13:54:59 +09:00
2018-09-12 02:48:19 +09:00
class Publisher {
private ev: Xev;
private meta: IMeta;
2018-04-25 18:04:16 +09:00
2018-09-12 02:48:19 +09:00
constructor() {
// Redisがインストールされてないときはプロセス間通信を使う
if (redis == null) {
this.ev = new Xev();
}
2017-05-24 20:50:17 +09:00
2018-09-12 02:48:19 +09:00
setInterval(async () => {
this.meta = await fetchMeta();
2018-09-12 02:48:19 +09:00
}, 5000);
}
2017-11-14 00:54:16 +09:00
public fetchMeta = async () => {
2018-09-12 02:48:19 +09:00
if (this.meta != null) return this.meta;
2018-03-07 11:40:40 +09:00
this.meta = await fetchMeta();
2018-09-12 02:48:19 +09:00
return this.meta;
}
2018-03-07 17:48:32 +09:00
2018-09-12 02:48:19 +09:00
private publish = (channel: string, type: string, value?: any): void => {
const message = type == null ? value : value == null ?
2018-10-08 01:56:36 +09:00
{ type: type, body: null } :
2018-09-12 02:48:19 +09:00
{ type: type, body: value };
if (this.ev) {
this.ev.emit(channel, message);
} else {
redis.publish('misskey', JSON.stringify({
channel: channel,
message: message
}));
}
2018-09-12 02:48:19 +09:00
}
2018-07-11 09:36:30 +09:00
public publishMainStream = (userId: ID, type: string, value?: any): void => {
this.publish(`mainStream:${userId}`, type, typeof value === 'undefined' ? null : value);
2018-09-12 02:48:19 +09:00
}
2018-09-12 02:48:19 +09:00
public publishDriveStream = (userId: ID, type: string, value?: any): void => {
this.publish(`driveStream:${userId}`, type, typeof value === 'undefined' ? null : value);
2018-09-12 02:48:19 +09:00
}
2017-05-24 20:50:17 +09:00
public publishNoteStream = (noteId: ID, type: string, value: any): void => {
this.publish(`noteStream:${noteId}`, type, {
id: noteId,
body: value
});
2018-09-12 02:48:19 +09:00
}
2016-12-29 07:49:51 +09:00
2018-09-12 02:48:19 +09:00
public publishUserListStream = (listId: ID, type: string, value?: any): void => {
this.publish(`userListStream:${listId}`, type, typeof value === 'undefined' ? null : value);
2018-09-12 02:48:19 +09:00
}
2016-12-29 07:49:51 +09:00
2018-09-12 02:48:19 +09:00
public publishMessagingStream = (userId: ID, otherpartyId: ID, type: string, value?: any): void => {
this.publish(`messagingStream:${userId}-${otherpartyId}`, type, typeof value === 'undefined' ? null : value);
2018-09-12 02:48:19 +09:00
}
public publishMessagingIndexStream = (userId: ID, type: string, value?: any): void => {
this.publish(`messagingIndexStream:${userId}`, type, typeof value === 'undefined' ? null : value);
2018-09-12 02:48:19 +09:00
}
public publishReversiStream = (userId: ID, type: string, value?: any): void => {
this.publish(`reversiStream:${userId}`, type, typeof value === 'undefined' ? null : value);
2018-09-12 02:48:19 +09:00
}
public publishReversiGameStream = (gameId: ID, type: string, value?: any): void => {
this.publish(`reversiGameStream:${gameId}`, type, typeof value === 'undefined' ? null : value);
}
public publishHomeTimelineStream = (userId: ID, note: any): void => {
this.publish(`homeTimeline:${userId}`, null, note);
2018-09-12 02:48:19 +09:00
}
2016-12-29 07:49:51 +09:00
2018-09-12 02:48:19 +09:00
public publishLocalTimelineStream = async (note: any): Promise<void> => {
const meta = await this.fetchMeta();
2018-09-12 02:48:19 +09:00
if (meta.disableLocalTimeline) return;
this.publish('localTimeline', null, note);
2018-09-12 02:48:19 +09:00
}
public publishHybridTimelineStream = async (userId: ID, note: any): Promise<void> => {
const meta = await this.fetchMeta();
2018-09-12 02:48:19 +09:00
if (meta.disableLocalTimeline) return;
this.publish(userId ? `hybridTimeline:${userId}` : 'hybridTimeline', null, note);
2018-09-12 02:48:19 +09:00
}
public publishGlobalTimelineStream = (note: any): void => {
this.publish('globalTimeline', null, note);
2018-09-12 02:48:19 +09:00
}
public publishHashtagStream = (note: any): void => {
this.publish('hashtag', null, note);
}
2018-11-03 11:38:00 +09:00
public publishApLogStream = (log: any): void => {
this.publish('apLog', null, log);
}
2018-07-30 07:20:27 +09:00
}
2018-09-12 02:48:19 +09:00
const publisher = new Publisher();
export default publisher;
export const publishMainStream = publisher.publishMainStream;
2018-09-12 02:48:19 +09:00
export const publishDriveStream = publisher.publishDriveStream;
export const publishNoteStream = publisher.publishNoteStream;
export const publishUserListStream = publisher.publishUserListStream;
export const publishMessagingStream = publisher.publishMessagingStream;
export const publishMessagingIndexStream = publisher.publishMessagingIndexStream;
export const publishReversiStream = publisher.publishReversiStream;
export const publishReversiGameStream = publisher.publishReversiGameStream;
export const publishHomeTimelineStream = publisher.publishHomeTimelineStream;
2018-09-12 02:48:19 +09:00
export const publishLocalTimelineStream = publisher.publishLocalTimelineStream;
export const publishHybridTimelineStream = publisher.publishHybridTimelineStream;
export const publishGlobalTimelineStream = publisher.publishGlobalTimelineStream;
export const publishHashtagStream = publisher.publishHashtagStream;
2018-11-03 11:38:00 +09:00
export const publishApLogStream = publisher.publishApLogStream;