Skip to content

Commit

Permalink
feat: add tag filter to getSortedTopics
Browse files Browse the repository at this point in the history
  • Loading branch information
barisusakli committed Feb 17, 2021
1 parent 5286f20 commit 9ce6f8a
Show file tree
Hide file tree
Showing 4 changed files with 32 additions and 8 deletions.
1 change: 1 addition & 0 deletions public/src/modules/topicList.js
Original file line number Diff line number Diff line change
Expand Up @@ -209,6 +209,7 @@ define('topicList', [
sort: tplToSort[templateName],
count: config.topicsPerPage,
cid: query.cid,
tags: query.tags,
query: query,
term: ajaxify.data.selectedTerm && ajaxify.data.selectedTerm.term,
filter: ajaxify.data.selectedFilter.filter,
Expand Down
3 changes: 2 additions & 1 deletion src/controllers/recent.js
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ recentController.get = async function (req, res, next) {
recentController.getData = async function (req, url, sort) {
const page = parseInt(req.query.page, 10) || 1;
let term = helpers.terms[req.query.term];
const { cid } = req.query;
const { cid, tags } = req.query;
const filter = req.query.filter || '';

if (!term && req.query.term) {
Expand All @@ -45,6 +45,7 @@ recentController.getData = async function (req, url, sort) {

const data = await topics.getSortedTopics({
cids: cid,
tags: tags,
uid: req.uid,
start: start,
stop: stop,
Expand Down
1 change: 1 addition & 0 deletions src/socket.io/topics/infinitescroll.js
Original file line number Diff line number Diff line change
Expand Up @@ -78,6 +78,7 @@ module.exports = function (SocketTopics) {
return await topics.getUnreadTopics(params);
}
params.cids = data.cid;
params.tags = data.tags;
params.sort = data.sort;
params.term = data.term;
return await topics.getSortedTopics(params);
Expand Down
35 changes: 28 additions & 7 deletions src/topics/sorted.js
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,10 @@ module.exports = function (Topics) {
if (params.hasOwnProperty('cids') && params.cids && !Array.isArray(params.cids)) {
params.cids = [params.cids];
}
params.tags = params.tags || [];
if (params.tags && !Array.isArray(params.tags)) {
params.tags = [params.tags];
}
data.tids = await getTids(params);
data.tids = await sortTids(data.tids, params);
data.tids = await filterTids(data.tids.slice(0, meta.config.recentMaxTopics), params);
Expand All @@ -48,14 +52,36 @@ module.exports = function (Topics) {
tids = await db.getSortedSetRevRange(`uid:${params.uid}:followed_tids`, 0, -1);
} else if (params.cids) {
tids = await getCidTids(params);
} else if (params.tags.length) {
tids = await getTagTids(params);
} else {
tids = await db.getSortedSetRevRange(`topics:${params.sort}`, 0, meta.config.recentMaxTopics - 1);
}

return tids;
}

async function getTagTids(params) {
const sets = [
`topics:${params.sort}`,
...params.tags.map(tag => `tag:${tag}:topics`),
];
return await db.getSortedSetRevIntersect({
sets: sets,
start: 0,
stop: meta.config.recentMaxTopics - 1,
weights: sets.map((s, index) => (index ? 0 : 1)),
});
}

async function getCidTids(params) {
if (params.tags.length) {
return _.intersection(...await Promise.all(params.tags.map(async (tag) => {
const sets = params.cids.map(cid => `cid:${cid}:tag:${tag}:topics`);
return await db.getSortedSetRevRange(sets, 0, -1);
})));
}

const sets = [];
const pinnedSets = [];
params.cids.forEach((cid) => {
Expand All @@ -73,7 +99,7 @@ module.exports = function (Topics) {
}

async function sortTids(tids, params) {
if (params.term === 'alltime' && !params.cids && params.filter !== 'watched' && !params.floatPinned) {
if (params.term === 'alltime' && !params.cids && !params.tags.length && params.filter !== 'watched' && !params.floatPinned) {
return tids;
}
const topicData = await Topics.getTopicsFields(tids, ['tid', 'lastposttime', 'upvotes', 'downvotes', 'postcount', 'pinned']);
Expand All @@ -94,12 +120,7 @@ module.exports = function (Topics) {
}

function floatPinned(topicData, sortFn) {
topicData.sort((a, b) => {
if (a.pinned !== b.pinned) {
return b.pinned - a.pinned;
}
return sortFn(a, b);
});
topicData.sort((a, b) => (a.pinned !== b.pinned ? b.pinned - a.pinned : sortFn(a, b)));
}

function sortRecent(a, b) {
Expand Down

0 comments on commit 9ce6f8a

Please sign in to comment.