NodeBB/src/pagination.js

'use strict';

const qs = require('querystring');
const _ = require('lodash');

const pagination = module.exports;

pagination.create = function (currentPage, pageCount, queryObj) {
	if (pageCount <= 1) {
		return {
			prev: { page: 1, active: currentPage > 1 },
			next: { page: 1, active: currentPage < pageCount },
			first: { page: 1, active: currentPage === 1 },
			last: { page: 1, active: currentPage === pageCount },
			rel: [],
			pages: [],
			currentPage: 1,
			pageCount: 1,
		};
	}
	pageCount = parseInt(pageCount, 10);
	let pagesToShow = [1, 2, pageCount - 1, pageCount];

	currentPage = parseInt(currentPage, 10) || 1;
	const previous = Math.max(1, currentPage - 1);
	const next = Math.min(pageCount, currentPage + 1);

	let startPage = Math.max(1, currentPage - 2);
	if (startPage > pageCount - 5) {
		startPage -= 2 - (pageCount - currentPage);
	}
	let i;
	for (i = 0; i < 5; i += 1) {
		pagesToShow.push(startPage + i);
	}

	pagesToShow = _.uniq(pagesToShow).filter(page => page > 0 && page <= pageCount).sort((a, b) => a - b);

	queryObj = { ...(queryObj || {}) };

	delete queryObj._;

	const pages = pagesToShow.map((page) => {
		queryObj.page = page;
		return { page: page, active: page === currentPage, qs: qs.stringify(queryObj) };
	});

	for (i = pages.length - 1; i > 0; i -= 1) {
		if (pages[i].page - 2 === pages[i - 1].page) {
			pages.splice(i, 0, { page: pages[i].page - 1, active: false, qs: qs.stringify(queryObj) });
		} else if (pages[i].page - 1 !== pages[i - 1].page) {
			pages.splice(i, 0, { separator: true });
		}
	}

	const data = { rel: [], pages: pages, currentPage: currentPage, pageCount: pageCount };
	queryObj.page = previous;
	data.prev = { page: previous, active: currentPage > 1, qs: qs.stringify(queryObj) };
	queryObj.page = next;
	data.next = { page: next, active: currentPage < pageCount, qs: qs.stringify(queryObj) };

	queryObj.page = 1;
	data.first = { page: 1, active: currentPage === 1, qs: qs.stringify(queryObj) };
	queryObj.page = pageCount;
	data.last = { page: pageCount, active: currentPage === pageCount, qs: qs.stringify(queryObj) };

	if (currentPage < pageCount) {
		data.rel.push({
			rel: 'next',
			href: `?${qs.stringify({ ...queryObj, page: next })}`,
		});
	}

	if (currentPage > 1) {
		data.rel.push({
			rel: 'prev',
			href: `?${qs.stringify({ ...queryObj, page: previous })}`,
		});
	}
	return data;
};