Ghost/core/server/models/base/pagination.js

// # Pagination
//
// Extends Bookshelf.Model with a `fetchPage` method. Handles everything to do with paginated requests.
var _          = require('lodash'),
    Promise    = require('bluebird'),
    baseUtils  = require('./utils'),

    defaults,
    paginationUtils,
    pagination;

/**
 * ### Default pagination values
 * These are overridden via `options` passed to each function
 * @typedef {Object} defaults
 * @default
 * @property {Number} `page` \- page in set to display (default: 1)
 * @property {Number|String} `limit` \- no. results per page (default: 15)
 */
defaults = {
    page: 1,
    limit: 15
};

/**
 * ## Pagination Utils
 * @api private
 * @type {{parseOptions: Function, query: Function, formatResponse: Function}}
 */
paginationUtils = {
    /**
     * ### Parse Options
     * Take the given options and ensure they are valid pagination options, else use the defaults
     * @param {options} options
     * @returns {options} options sanitised for pagination
     */
    parseOptions: function parseOptions(options) {
        options = _.defaults(options || {}, defaults);

        if (options.limit !== 'all') {
            options.limit = parseInt(options.limit, 10) || defaults.limit;
        }

        options.page = parseInt(options.page, 10) || defaults.page;

        return options;
    },
    /**
     * ### Query
     * Apply the necessary parameters to paginate the query
     * @param {Bookshelf.Model|Bookshelf.Collection} itemCollection
     * @param {options} options
     */
    addLimitAndOffset: function addLimitAndOffset(itemCollection, options) {
        if (_.isNumber(options.limit)) {
            itemCollection
                .query('limit', options.limit)
                .query('offset', options.limit * (options.page - 1));
        }
    },

    /**
     * ### Format Response
     * Takes the no. items returned and original options and calculates all of the pagination meta data
     * @param {Number} totalItems
     * @param {options} options
     * @returns {pagination} pagination metadata
     */
    formatResponse: function formatResponse(totalItems, options) {
        var calcPages = Math.ceil(totalItems / options.limit) || 0,
            pagination = {
                page:  options.page || defaults.page,
                limit: options.limit,
                pages: calcPages === 0 ? 1 : calcPages,
                total: totalItems,
                next:  null,
                prev:  null
            };

        if (pagination.pages > 1) {
            if (pagination.page === 1) {
                pagination.next = pagination.page + 1;
            } else if (pagination.page === pagination.pages) {
                pagination.prev = pagination.page - 1;
            } else {
                pagination.next = pagination.page + 1;
                pagination.prev = pagination.page - 1;
            }
        }

        return pagination;
    }
};

// ## Object Definitions

/**
 * ### Pagination Object
 * @typedef {Object} pagination
 * @property {Number} `page` \- page in set to display
 * @property {Number|String} `limit` \- no. results per page, or 'all'
 * @property {Number} `pages` \- total no. pages in the full set
 * @property {Number} `total` \- total no. items in the full set
 * @property {Number|null} `next` \- next page
 * @property {Number|null} `prev` \- previous page
 */

/**
 * ### Fetch Page Options
 * @typedef {Object} options
 * @property {Number} `page` \- page in set to display
 * @property {Number|String} `limit` \- no. results per page, or 'all'
 * @property {Object} `order` \- set of order by params and directions
 */

/**
 * ### Fetch Page Response
 * @typedef {Object} paginatedResult
 * @property {Array} `collection` \- set of results
 * @property {pagination} pagination \- pagination metadata
 */

/**
 * ## Pagination
 * Extends `bookshelf.Model` with `fetchPage`
 * @param {Bookshelf} bookshelf \- the instance to plug into
 */
pagination = function pagination(bookshelf) {
    // Extend updates the first object passed to it, no need for an assignment
    _.extend(bookshelf.Model.prototype, {
        /**
         * ### Fetch page
         * A `fetch` extension to get a paginated set of items from a collection
         * @param {options} options
         * @returns {paginatedResult} set of results + pagination metadata
         */
        fetchPage: function fetchPage(options) {
            // Setup pagination options
            options = paginationUtils.parseOptions(options);

            // Get the table name and idAttribute for this model
            var tableName = _.result(this.constructor.prototype, 'tableName'),
                idAttribute = _.result(this.constructor.prototype, 'idAttribute'),
            // Create a new collection for running `this` query, ensuring we're using collection, rather than model
                collection = this.constructor.collection(),
                countPromise,
                collectionPromise,
                self = this;

            // #### Pre count clauses
            // Add any where or join clauses which need to be included with the aggregate query

            // Clone the base query & set up a promise to get the count of total items in the full set
            // Due to lack of support for count distinct, this is pretty complex.
            countPromise = this.query().clone().select(
                bookshelf.knex.raw('count(distinct ' + tableName + '.' + idAttribute + ') as aggregate')
            );

            // Clone the base query into our collection
            collection._knex = this.query().clone();

            // #### Post count clauses
            // Add any where or join clauses which need to NOT be included with the aggregate query

            // Setup the pagination parameters so that we return the correct items from the set
            paginationUtils.addLimitAndOffset(collection, options);

            // Apply ordering options if they are present
            if (options.order && !_.isEmpty(options.order)) {
                _.forOwn(options.order, function (direction, property) {
                    collection.query('orderBy', tableName + '.' + property, direction);
                });
            }

            if (options.groups && !_.isEmpty(options.groups)) {
                _.each(options.groups, function (group) {
                    collection.query('groupBy', group);
                });
            }

            // Apply count options if they are present
            baseUtils.collectionQuery.count(collection, options);

            this.resetQuery();
            if (this.relatedData) {
                collection.relatedData = this.relatedData;
            }

            // ensure that our model (self) gets the correct events fired upon it
            collection
                .on('fetching', function (collection, columns, options) {
                    return self.triggerThen('fetching:collection', collection, columns, options);
                })
                .on('fetched', function (collection, resp, options) {
                    return self.triggerThen('fetched:collection', collection, resp, options);
                });

            // Setup the promise to do a fetch on our collection, running the specified query
            // @TODO: ensure option handling is done using an explicit pick elsewhere
            collectionPromise = collection.fetch(_.omit(options, ['page', 'limit']));

            // Resolve the two promises
            return Promise.join(collectionPromise, countPromise).then(function formatResponse(results) {
                // Format the collection & count result into `{collection: [], pagination: {}}`
                return {
                    collection: results[0],
                    pagination: paginationUtils.formatResponse(results[1][0] ? results[1][0].aggregate : 0, options)
                };
            });
        }
    });
};

/**
 * ## Export pagination plugin
 * @api public
 */
module.exports = pagination;