1 files changed, 103 insertions, 0 deletions
diff --git a/src/api/serializers/post.ts b/src/api/serializers/post.ts
new file mode 100644
index 0000000000..a17aa9035b
--- /dev/null
+++ b/src/api/serializers/post.ts
@@ -0,0 +1,103 @@
+'use strict';
+
+/**
+ * Module dependencies
+ */
+import * as mongo from 'mongodb';
+import Post from '../models/post';
+import Like from '../models/like';
+import serializeUser from './user';
+import serializeDriveFile from './drive-file';
+const deepcopy = require('deepcopy');
+
+/**
+ * Serialize a post
+ *
+ * @param {Object} post
+ * @param {Object} me?
+ * @param {Object} options?
+ * @return {Promise<Object>}
+ */
+const self = (
+	post: any,
+	me?: any,
+	options?: {
+		serializeReplyTo: boolean,
+		serializeRepost: boolean,
+		includeIsLiked: boolean
+	}
+) => new Promise<Object>(async (resolve, reject) => {
+	const opts = options || {
+		serializeReplyTo: true,
+		serializeRepost: true,
+		includeIsLiked: true
+	};
+
+	let _post: any;
+
+	// Populate the post if 'post' is ID
+	if (mongo.ObjectID.prototype.isPrototypeOf(post)) {
+		_post = await Post.findOne({
+			_id: post
+		});
+	} else if (typeof post === 'string') {
+		_post = await Post.findOne({
+			_id: new mongo.ObjectID(post)
+		});
+	} else {
+		_post = deepcopy(post);
+	}
+
+	const id = _post._id;
+
+	// Rename _id to id
+	_post.id = _post._id;
+	delete _post._id;
+
+	delete _post.mentions;
+
+	// Populate user
+	_post.user = await serializeUser(_post.user_id, me);
+
+	if (_post.media_ids) {
+		// Populate media
+		_post.media = await Promise.all(_post.media_ids.map(async fileId =>
+			await serializeDriveFile(fileId)
+		));
+	}
+
+	if (_post.reply_to_id && opts.serializeReplyTo) {
+		// Populate reply to post
+		_post.reply_to = await self(_post.reply_to_id, me, {
+			serializeReplyTo: false,
+			serializeRepost: false,
+			includeIsLiked: false
+		});
+	}
+
+	if (_post.repost_id && opts.serializeRepost) {
+		// Populate repost
+		_post.repost = await self(_post.repost_id, me, {
+			serializeReplyTo: _post.text == null,
+			serializeRepost: _post.text == null,
+			includeIsLiked: _post.text == null
+		});
+	}
+
+	// Check if it is liked
+	if (me && opts.includeIsLiked) {
+		const liked = await Like
+			.count({
+				user_id: me._id,
+				post_id: id
+			}, {
+				limit: 1
+			});
+
+		_post.is_liked = liked === 1;
+	}
+
+	resolve(_post);
+});
+
+export default self;