feat: rewrite getRawPost to async/await

promisify SocketPosts

src/socket.io/index.js

@@ -113,7 +113,7 @@ function onMessage(socket, payload) {
 		return null;
 	}, Namespaces);
 
-	if (!methodToCall) {
+	if (!methodToCall || typeof methodToCall !== 'function') {
 		if (process.env.NODE_ENV === 'development') {
 			winston.warn('[socket.io] Unrecognized message: ' + eventName);
 		}
@@ -146,17 +146,19 @@ function onMessage(socket, payload) {
 			}
 		},
 		function (next) {
-			let callbackCalled = false;
+			async function tryAsyncFunc(done) {
-			function nextOnce(err, res) {
+				try {
-				if (callbackCalled) { return; }
+					const result = await methodToCall(socket, params);
-				callbackCalled = true;
+					done(null, result);
-				next(err, res);
+				} catch (err) {
+					done(err);
+				}
 			}
-			const returned = methodToCall(socket, params, nextOnce);
+
-			if (returned && typeof returned.then === 'function') {
+			if (methodToCall.constructor && methodToCall.constructor.name === 'AsyncFunction') {
-				returned.then((payload) => {
+				tryAsyncFunc(next);
-					nextOnce(null, payload);
+			} else {
-				}, next);
+				methodToCall(socket, params, next);
 			}
 		},
 	], function (err, result) {

src/socket.io/posts.js

@@ -70,35 +70,19 @@ function postReply(socket, data, callback) {
 	], callback);
 }
 
-SocketPosts.getRawPost = function (socket, pid, callback) {
+SocketPosts.getRawPost = async function (socket, pid) {
-	async.waterfall([
+	const canRead = await privileges.posts.can('topics:read', pid, socket.uid);
-		function (next) {
+	if (!canRead) {
-			privileges.posts.can('topics:read', pid, socket.uid, next);
+		throw new Error('[[error:no-privileges]]');
-		},
+	}
-		function (canRead, next) {
-			if (!canRead) {
-				return next(new Error('[[error:no-privileges]]'));
-			}
-			posts.getPostFields(pid, ['content', 'deleted'], next);
-		},
-		function (postData, next) {
-			if (postData.deleted) {
-				return next(new Error('[[error:no-post]]'));
-			}
-			next(null, postData);
-		},
-		function (postData, next) {
-			plugins.fireHook('filter:post.getRawPost', Object.assign(postData, {
-				pid: pid,
-			}), next);
-		},
-	], function (err, postData) {
-		if (err) {
-			return callback(err);
-		}
 
-		callback(null, postData.content);
+	const postData = await posts.getPostFields(pid, ['content', 'deleted']);
-	});
+	if (postData.deleted) {
+		throw new Error('[[error:no-post]]');
+	}
+	postData.pid = pid;
+	const result = await plugins.fireHook('filter:post.getRawPost', { uid: socket.uid, postData: postData });
+	return result.postData.content;
 };
 
 SocketPosts.getTimestampByIndex = function (socket, data, callback) {
@@ -269,3 +253,5 @@ function acceptOrReject(method, socket, data, callback) {
 		},
 	], callback);
 }
+
+require('../promisify')(SocketPosts);