NodeBB/src/posts/data.js

'use strict';

var async = require('async');

var db = require('../database');
var plugins = require('../plugins');

const intFields = ['uid', 'pid', 'tid', 'deleted'];

module.exports = function (Posts) {
	Posts.getPostsFields = function (pids, fields, callback) {
		if (!Array.isArray(pids) || !pids.length) {
			return callback(null, []);
		}

		var keys = pids.map(pid => 'post:' + pid);

		async.waterfall([
			function (next) {
				if (fields.length) {
					db.getObjectsFields(keys, fields, next);
				} else {
					db.getObjects(keys, next);
				}
			},
			function (posts, next) {
				plugins.fireHook('filter:post.getFields', { posts: posts, fields: fields }, next);
			},
			function (data, next) {
				data.posts.forEach(modifyPost);
				next(null, Array.isArray(data.posts) ? data.posts : null);
			},
		], callback);
	};

	Posts.getPostData = function (pid, callback) {
		Posts.getPostsFields([pid], [], function (err, posts) {
			callback(err, posts && posts.length ? posts[0] : null);
		});
	};

	Posts.getPostsData = function (pids, callback) {
		Posts.getPostsFields(pids, [], callback);
	};

	Posts.getPostField = function (pid, field, callback) {
		Posts.getPostFields(pid, [field], function (err, post) {
			callback(err, post ? post[field] : null);
		});
	};

	Posts.getPostFields = function (pid, fields, callback) {
		Posts.getPostsFields([pid], fields, function (err, posts) {
			callback(err, posts ? posts[0] : null);
		});
	};

	Posts.setPostField = function (pid, field, value, callback) {
		Posts.setPostFields(pid, { [field]: value }, callback);
	};

	Posts.setPostFields = function (pid, data, callback) {
		async.waterfall([
			function (next) {
				db.setObject('post:' + pid, data, next);
			},
			function (next) {
				data.pid = pid;
				plugins.fireHook('action:post.setFields', { data: data });
				next();
			},
		], callback);
	};
};

function modifyPost(post) {
	if (post) {
		intFields.forEach(field => db.parseIntField(post, field));
	}
}