'use strict'; const db = require('../database'); const plugins = require('../plugins'); const utils = require('../utils'); const intFields = [ 'uid', 'pid', 'tid', 'deleted', 'timestamp', 'upvotes', 'downvotes', 'deleterUid', 'edited', ]; module.exports = function (Posts) { Posts.getPostsFields = async function (pids, fields) { if (!Array.isArray(pids) || !pids.length) { return []; } const keys = pids.map(pid => 'post:' + pid); let postData; if (fields.length) { postData = await db.getObjectsFields(keys, fields); } else { postData = await db.getObjects(keys); } const result = await plugins.fireHook('filter:post.getFields', { pids: pids, posts: postData, fields: fields, }); result.posts.forEach(post => modifyPost(post, fields)); return Array.isArray(result.posts) ? result.posts : null; }; Posts.getPostData = async function (pid) { const posts = await Posts.getPostsFields([pid], []); return posts && posts.length ? posts[0] : null; }; Posts.getPostsData = async function (pids) { return await Posts.getPostsFields(pids, []); }; Posts.getPostField = async function (pid, field) { const post = await Posts.getPostFields(pid, [field]); return post ? post[field] : null; }; Posts.getPostFields = async function (pid, fields) { const posts = await Posts.getPostsFields([pid], fields); return posts ? posts[0] : null; }; Posts.setPostField = async function (pid, field, value) { await Posts.setPostFields(pid, { [field]: value }); }; Posts.setPostFields = async function (pid, data) { await db.setObject('post:' + pid, data); data.pid = pid; plugins.fireHook('action:post.setFields', { data: data }); }; }; function modifyPost(post, fields) { if (post) { db.parseIntFields(post, intFields, fields); if (post.hasOwnProperty('upvotes') && post.hasOwnProperty('downvotes')) { post.votes = post.upvotes - post.downvotes; } if (post.hasOwnProperty('timestamp')) { post.timestampISO = utils.toISOString(post.timestamp); } if (post.hasOwnProperty('edited')) { post.editedISO = post.edited !== 0 ? utils.toISOString(post.edited) : ''; } } }