You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
2.0 KiB
JavaScript
66 lines
2.0 KiB
JavaScript
'use strict';
|
|
|
|
const _ = require('lodash');
|
|
|
|
const db = require('../database');
|
|
const categories = require('../categories');
|
|
|
|
module.exports = function (User) {
|
|
User.setCategoryWatchState = async function (uid, cids, state) {
|
|
if (!(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 = Array.isArray(cids) ? cids : [cids];
|
|
const exists = await categories.exists(cids);
|
|
if (exists.includes(false)) {
|
|
throw new Error('[[error:no-category]]');
|
|
}
|
|
await db.sortedSetsAdd(cids.map(cid => 'cid:' + cid + ':uid:watch:state'), state, uid);
|
|
};
|
|
|
|
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 [];
|
|
}
|
|
return await User.getCategoriesByStates(uid, [categories.watchStates.ignoring]);
|
|
};
|
|
|
|
User.getWatchedCategories = async function (uid) {
|
|
if (!(parseInt(uid, 10) > 0)) {
|
|
return [];
|
|
}
|
|
return await User.getCategoriesByStates(uid, [categories.watchStates.watching]);
|
|
};
|
|
|
|
User.getCategoriesByStates = async function (uid, states) {
|
|
if (!(parseInt(uid, 10) > 0)) {
|
|
return await categories.getAllCidsFromSet('categories:cid');
|
|
}
|
|
const cids = await categories.getAllCidsFromSet('categories:cid');
|
|
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);
|
|
};
|
|
};
|