NodeBB/src/user/categories.js

'use strict';

const _ = require('lodash');

const db = require('../database');
const meta = require('../meta');
const categories = require('../categories');
const activitypub = require('../activitypub');
const plugins = require('../plugins');
const utils = require('../utils');

module.exports = function (User) {
	User.setCategoryWatchState = async function (uid, cids, state) {
		if (utils.isNumber(uid) && parseInt(uid, 10) <= 0) {
			return;
		}

		const isStateValid = Object.values(categories.watchStates).includes(parseInt(state, 10));
		if (!isStateValid) {
			throw new Error('[[error:invalid-watch-state]]');
		}

		cids = new Set(Array.isArray(cids) ? cids : [cids]);
		cids.delete(-1); // cannot watch cid -1
		cids.delete('-1');
		cids = Array.from(cids);

		const exists = await categories.exists(cids);
		if (exists.includes(false)) {
			throw new Error('[[error:no-category]]');
		}

		const apiMethod = state >= categories.watchStates.tracking ? activitypub.out.follow : activitypub.out.undo.follow;
		const follows = cids.filter(cid => !utils.isNumber(cid)).map(cid => apiMethod('uid', uid, cid)); // returns promises

		await Promise.all([
			db.sortedSetsAdd(cids.map(cid => `cid:${cid}:uid:watch:state`), state, uid),
			...follows,
		]);
	};

	User.getCategoryWatchState = async function (uid) {
		if (!(parseInt(uid, 10) > 0)) {
			return {};
		}

		const cids = await categories.getAllCidsFromSet('categories:cid');
		const states = await categories.getWatchState(cids, uid);
		return _.zipObject(cids, states);
	};

	User.getIgnoredCategories = async function (uid) {
		if (!(parseInt(uid, 10) > 0)) {
			return [];
		}
		const cids = await User.getCategoriesByStates(uid, [categories.watchStates.ignoring]);
		const result = await plugins.hooks.fire('filter:user.getIgnoredCategories', {
			uid: uid,
			cids: cids,
		});
		return result.cids;
	};

	User.getWatchedCategories = async function (uid) {
		if (!(parseInt(uid, 10) > 0)) {
			return [];
		}
		let cids = await User.getCategoriesByStates(uid, [categories.watchStates.watching]);
		const categoryData = await categories.getCategoriesFields(cids, ['disabled']);
		cids = cids.filter((cid, index) => categoryData[index] && !categoryData[index].disabled);
		const result = await plugins.hooks.fire('filter:user.getWatchedCategories', {
			uid: uid,
			cids: cids,
		});
		return result.cids;
	};

	User.getCategoriesByStates = async function (uid, states) {
		const [localCids, remoteCids] = await Promise.all([
			categories.getAllCidsFromSet('categories:cid'),
			meta.config.activitypubEnabled ? db.getObjectValues('handle:cid') : [],
		]);
		const cids = localCids.concat(remoteCids);
		if (!(parseInt(uid, 10) > 0)) {
			return cids;
		}
		const userState = await categories.getWatchState(cids, uid);
		return cids.filter((cid, index) => states.includes(userState[index]));
	};

	User.ignoreCategory = async function (uid, cid) {
		await User.setCategoryWatchState(uid, cid, categories.watchStates.ignoring);
	};

	User.watchCategory = async function (uid, cid) {
		await User.setCategoryWatchState(uid, cid, categories.watchStates.watching);
	};
};