Initial commit :four_leaf_clover:

2 files changed, 70 insertions, 0 deletions

diff --git a/src/api/stream/home.ts b/src/api/stream/home.ts
new file mode 100644
index 0000000000..975bea4c60
--- /dev/null
+++ b/src/api/stream/home.ts
@@ -0,0 +1,10 @@
+import * as websocket from 'websocket';
+import * as redis from 'redis';
+
+export default function homeStream(request: websocket.request, connection: websocket.connection, subscriber: redis.RedisClient, user: any): void {
+	// Subscribe Home stream channel
+	subscriber.subscribe(`misskey:user-stream:${user._id}`);
+	subscriber.on('message', (_, data) => {
+		connection.send(data);
+	});
+}
diff --git a/src/api/stream/messaging.ts b/src/api/stream/messaging.ts
new file mode 100644
index 0000000000..4ec139b82b
--- /dev/null
+++ b/src/api/stream/messaging.ts
@@ -0,0 +1,60 @@
+import * as mongodb from 'mongodb';
+import * as websocket from 'websocket';
+import * as redis from 'redis';
+import Message from '../models/messaging-message';
+import { publishMessagingStream } from '../event';
+
+export default function messagingStream(request: websocket.request, connection: websocket.connection, subscriber: redis.RedisClient, user: any): void {
+	const otherparty = request.resourceURL.query.otherparty;
+
+	// Subscribe messaging stream
+	subscriber.subscribe(`misskey:messaging-stream:${user._id}-${otherparty}`);
+	subscriber.on('message', (_, data) => {
+		connection.send(data);
+	});
+
+	connection.on('message', async (data) => {
+		const msg = JSON.parse(data.utf8Data);
+
+		switch (msg.type) {
+			case 'read':
+					if (!msg.id) {
+						return;
+					}
+
+					const id = new mongodb.ObjectID(msg.id);
+
+					// Fetch message
+					// SELECT _id, user_id, is_read
+					const message = await Message.findOne({
+						_id: id,
+						recipient_id: user._id
+					}, {
+						fields: {
+							_id: true,
+							user_id: true,
+							is_read: true
+						}
+					});
+
+					if (message == null) {
+						return;
+					}
+
+					if (message.is_read) {
+						return;
+					}
+
+					// Update documents
+					await Message.update({
+						_id: id
+					}, {
+						$set: { is_read: true }
+					});
+
+					// Publish event
+					publishMessagingStream(message.user_id, user._id, 'read', id.toString());
+				break;
+		}
+	});
+}