bbc/flashheart

View on GitHub
docconfig/template/fixtures/documents/collector.js

Summary

Maintainability
D
2 days
Test Coverage
"use strict";
/**
 @fileOverview An object and array collector
 @module ink/collector
 */

var probe = require( "ink-probe" );
var sys = require( "lodash" );
var dcl = require( "dcl" );

/**
 * A collector
 * @constructor
 */
var CollectorBase = dcl( Destroyable, {
    declaredClass : "CollectorBase",
    constructor   : function ( obj ) {
        var that = this;
        if ( obj && !sys.isObject( obj ) ) {
            throw new TypeError( "Collectors require an initial object or array passed to the constructor" );
        }
        /**
         * The collection that being managed
         * @type {object|array}
         */
        this.heap = obj || {};
        // mixin the probe
        probe.mixTo( this, this.heap );
        /**
         * Get the size of the collection
         * @name length
         * @type {number}
         * @memberOf module:documents/collector~CollectorBase#
         */
        Object.defineProperty( this, "length", {
                get : function () {
                    return sys.size( that.heap );
                }
            }
        );
        /**
         * Creates an array of shuffled array values, using a version of the Fisher-Yates shuffle.
         * See http://en.wikipedia.org/wiki/Fisher-Yates_shuffle.
         * @function
         * @memberOf module:documents/collector~CollectorBase#
         * @returns {array}
         */
        this.shuffle = sys.bind( sys.shuffle, this, this.heap );

    },
    /**
     * Adds an item to the collection
     * @param {*} key The key to use for the item being added.
     * @param {*} item The item to add to the collection. The item is not iterated so that you could add bundled items to the collection
     */
    add           : function ( key, item ) {
        this.heap[key] = item;
    },
    /**
     * Iterate over each item in the collection, or a subset that matches a query. This supports two signatures:
     * `.each(query, function)` and `.each(function)`. If you pass in a query, only the items that match the query
     * are iterated over.
     * @param {object=} query A query to evaluate
     * @param {function(val, key)} iterator Function to execute against each item in the collection
     * @param {object=} thisobj The value of `this`
     */
    each          : function ( query, iterator, thisobj ) {
        if ( sys.isPlainObject( query ) ) {
            thisobj = thisobj || this;
            sys.each( this.find( query ), iterator, thisobj );
        } else {
            thisobj = iterator || this;
            sys.each( this.heap, query, thisobj );
        }
    },
    /**
     * Returns the collection as an array. If it is already an array, it just returns that.
     * @return {array}
     */
    toArray       : function () {
        return sys.toArray( this.heap );
    },
    /**
     * Supports conversion to a JSON string or for passing over the wire
     * @return {object}
     * @returns {Object|array}
     */
    toJSON        : function () {
        return this.heap;
    },
    /**
     * Maps the contents to an array by iterating over it and transforming it. You supply the iterator. Supports two signatures:
     * `.map(query, function)` and `.map(function)`. If you pass in a query, only the items that match the query
     * are iterated over.
     * @param {object=} query A query to evaluate
     * @param {function(val, key)} iterator Function to execute against each item in the collection
     * @param {object=} thisobj The value of `this`
     */
    map           : function ( query, iterator, thisobj ) {
        if ( sys.isPlainObject( query ) ) {
            thisobj = thisobj || this;
            return sys.map( this.find( query ), iterator, thisobj );
        } else {
            thisobj = iterator || this;
            return sys.map( this.heap, query, thisobj );
        }
    },
    /**
     * Reduces a collection to a value which is the accumulated result of running each element in the collection through the
     * callback, where each successive callback execution consumes the return value of the previous execution. If accumulator
     * is not passed, the first element of the collection will be used as the initial accumulator value.
     * are iterated over.
     * @param {object=} query A query to evaluate
     * @param {function(result, val, key)} iterator The function that will be executed in each item in the collection
     * @param {*=} accumulator Initial value of the accumulator.
     * @param {object=} thisobj The value of `this`
     * @return {*}
     */
    reduce        : function ( query, iterator, accumulator, thisobj ) {
        if ( sys.isPlainObject( query ) ) {
            thisobj = thisobj || this;
            return sys.reduce( this.find( query ), iterator, accumulator, thisobj );
        } else {
            thisobj = accumulator || this;
            return  sys.reduce( this.heap, query, iterator, thisobj );
        }
    },
    /**
     * Creates an object composed of keys returned from running each element
     * of the collection through the given callback. The corresponding value of each key
     * is the number of times the key was returned by the callback.
     * @param {object=} query A query to evaluate. If you pass in a query, only the items that match the query
     * are iterated over.
     * @param  {function(value, key, collection)} iterator
     * @param {object=} thisobj The value of `this`
     * @return {object}
     */
    countBy       : function ( query, iterator, thisobj ) {
        if ( sys.isPlainObject( query ) ) {
            thisobj = thisobj || this;
            return sys.countBy( this.find( query ), iterator, thisobj );
        } else {
            thisobj = iterator || this;
            return sys.countBy( this.heap, query, thisobj );
        }
    },
    /**
     * Creates an object composed of keys returned from running each element of the collection through the callback.
     * The corresponding value of each key is an array of elements passed to callback that returned the key.
     * The callback is invoked with three arguments: (value, index|key, collection).
     * @param {object=} query A query to evaluate . If you pass in a query, only the items that match the query
     * are iterated over.
     * @param {function(value, key, collection)} iterator
     * @param {object=} thisobj The value of `this`
     * @return {object}
     */
    groupBy       : function ( query, iterator, thisobj ) {
        if ( sys.isPlainObject( query ) ) {
            thisobj = thisobj || this;
            return sys.groupBy( this.find( query ), iterator, thisobj );
        } else {
            thisobj = iterator || this;
            return sys.groupBy( this.heap, query, thisobj );
        }
    },
    /**
     * Reduce the collection to a single value. Supports two signatures:
     * `.pluck(query, function)` and `.pluck(function)`
     * @param {object=} query The query to evaluate. If you pass in a query, only the items that match the query
     * are iterated over.
     * @param {string} property The property that will be 'plucked' from the contents of the collection
     * @return {*}
     */
    pluck         : function ( query, property ) {
        if ( arguments.length === 2 ) {
            return sys.map( this.find( query ), function ( record ) {
                return probe.get( record, property );
            } );
        } else {
            return sys.map( this.heap, function ( record ) {
                return probe.get( record, query );
            } );
        }
    },
    /**
     * Returns a sorted copy of the collection.
     * @param {object=} query The query to evaluate. If you pass in a query, only the items that match the query
     * are iterated over.
     * @param {function(value, key)} iterator
     * @param {object=} thisobj The value of `this`
     * @return {array}
     */
    sortBy        : function ( query, iterator, thisobj ) {
        if ( sys.isPlainObject( query ) ) {
            thisobj = thisobj || this;
            return sys.sortBy( this.find( query ), iterator, thisobj );
        } else {
            thisobj = iterator || this;
            return sys.sortBy( this.heap, query, thisobj );
        }
    },
    /**
     * Retrieves the maximum value of an array. If callback is passed,
     * it will be executed for each value in the array to generate the criterion by which the value is ranked.
     * @param {object=} query A query to evaluate . If you pass in a query, only the items that match the query
     * are iterated over.
     * @param {function(value, key, collection)} iterator
     * @param {object=} thisobj The value of `this`
     * @return {number}
     */
    max           : function ( query, iterator, thisobj ) {
        if ( sys.isPlainObject( query ) ) {
            thisobj = thisobj || this;
            return sys.max( this.find( query ), iterator, thisobj );
        } else {
            thisobj = iterator || this;
            return sys.max( this.heap, query, thisobj );
        }
    },
    /**
     * Retrieves the minimum value of an array. If callback is passed,
     * it will be executed for each value in the array to generate the criterion by which the value is ranked.
     * @param {object=} query A query to evaluate . If you pass in a query, only the items that match the query
     * are iterated over.
     * @param {function(value, key, collection)} iterator
     * @param {object=} thisobj The value of `this`
     * @return {number}
     */
    min           : function ( query, iterator, thisobj ) {
        if ( sys.isPlainObject( query ) ) {
            thisobj = thisobj || this;
            return sys.min( this.find( query ), iterator, thisobj );
        } else {
            thisobj = iterator || this;
            return sys.min( this.heap, query, thisobj );
        }
    },
    /**
     * Destructor called when the object is destroyed.
     */
    destroy       : function () {
        this.heap = null;
    }
} );

/**
 * An object based collector
 * @extends module:documents/collector~CollectorBase
 * @constructor
 */
var OCollector = dcl( CollectorBase, {
    /**
     * Get a record by key
     * @param {*} key The key of the record to get
     * @return {*}
     */
    key : function ( key ) {
        return this.heap[key];
    }
} );

//noinspection JSCommentMatchesSignature
/**
 An array based collector
 @extends module:documents/collector~CollectorBase
 @constructor
 */
var ACollector = dcl( CollectorBase, {
        constructor : function ( obj ) {
            if ( obj && !sys.isArray( obj ) ) {
                throw new TypeError( "Collectors require an array passed to the constructor" );
            }
            this.heap = obj || [];
            /**
             * Creates an array of array elements not present in the other arrays using strict equality for comparisons, i.e. ===.
             * @returns {array}
             */
            this.difference = sys.bind( sys.difference, this, this.heap );
            /**
             * This method gets all but the first values of array
             * @param {number=} n The numer of items to return
             * @returns {*}
             */
            this.tail = sys.bind( sys.tail, this, this.heap );
            /**
             * Gets the first n values of the array
             * @param {number=} n The numer of items to return
             * @returns {*}
             */
            this.head = sys.bind( sys.head, this, this.heap );
        },
        /**
         * Adds to the top of the collection
         * @param {*} item The item to add to the collection. Only one item at a time can be added
         */
        add         : function ( item ) {
            this.heap.unshift( item );
        },
        /**
         * Add to the bottom of the list
         * @param {*} item The item to add to the collection.  Only one item at a time can be added
         */
        append      : function ( item ) {
            this.heap.push( item );
        },
        /**
         * Add an item to the top of the list. This is identical to `add`, but is provided for stack semantics
         * @param {*} item The item to add to the collection. Only one item at a time can be added
         */
        push        : function ( item ) {
            this.add( item );
        },
        /**
         * Modifies the collection with all falsey values of array removed. The values false, null, 0, "", undefined and NaN are all falsey.
         */
        compact     : function () {
            this.heap = sys.compact( this.heap );
        },
        /**
         * Creates an array of elements from the specified indexes, or keys, of the collection. Indexes may be specified as
         * individual arguments or as arrays of indexes
         * @param {indexes} args The indexes to use
         */
        at          : function () {
            var arr = sys.toArray( arguments );
            arr.unshift( this.heap );
            return sys.at.apply( this, arr );
        },
        /**
         * Flattens a nested array (the nesting can be to any depth). If isShallow is truthy, array will only be flattened a single level.
         * If callback is passed, each element of array is passed through a callback before flattening.
         * @param {object=} query A query to evaluate . If you pass in a query, only the items that match the query
         * are iterated over.
         * @param {function(value, key, collection)} iterator,
         * @param {object=} thisobj The value of `this`
         * @return {number}
         */
        flatten     : function ( query, iterator, thisobj ) {
            if ( sys.isPlainObject( query ) ) {
                thisobj = thisobj || this;
                return sys.flatten( this.find( query ), iterator, thisobj );
            } else {
                thisobj = iterator || this;
                return sys.flatten( this.heap, query, thisobj );
            }
        },
        /**
         * Gets an items by its index
         * @param {number} key The index to get
         * @return {*}
         */
        index       : function ( index ) {
            return this.heap[ index ];
        }
    }
);

/**
 Collect an object
 @param {array|object} obj What to collect
 @return {ACollector|OCollector}
 */
exports.collect = function ( obj ) {
    if ( sys.isArray( obj ) ) {
        return new ACollector( obj );
    } else {
        return new OCollector( obj );
    }
};

exports.array = function ( obj ) {
    return new ACollector( obj );
};

exports.object = function ( obj ) {
    return new OCollector( obj );
};

/**
 Returns true if all items match the query. Aliases as `all`
 @function

 @param {object} qu The query to execute
 @returns {boolean}
 @name every
 @memberOf module:documents/collector~CollectorBase#
 */


/**
 Returns true if any of the items match the query. Aliases as `any`
 @function

 @param {object} qu The query to execute
 @returns {boolean}
 @memberOf module:documents/collector~CollectorBase#
 @name some
 */


/**
 Returns the set of unique records that match a query

 @param {object} qu The query to execute.
 @return {array}
 @memberOf module:documents/collector~CollectorBase#
 @name unique
 @method
 **/

/**
 Returns true if all items match the query. Aliases as `every`
 @function

 @param {object} qu The query to execute
 @returns {boolean}
 @name all
 @memberOf module:documents/collector~CollectorBase#
 */


/**
 Returns true if any of the items match the query. Aliases as `all`
 @function

 @param {object} qu The query to execute
 @returns {boolean}
 @memberOf module:documents/collector~CollectorBase#
 @name any
 */


/**
 Remove all items in the object/array that match the query

 @param {object} qu The query to execute. See {@link module:ink/probe.queryOperators} for the operators you can use.
 @return {object|array} The array or object as appropriate without the records.
 @memberOf module:documents/collector~CollectorBase#
 @name remove
 @method
 **/

/**
 Returns the first record that matches the query and returns its key or index depending on whether `obj` is an object or array respectively.
 Aliased as `seekKey`.

 @param {object} qu The query to execute.
 @returns {object}
 @memberOf module:documents/collector~CollectorBase#
 @name findOneKey
 @method
 */


/**
 Returns the first record that matches the query. Aliased as `seek`.

 @param {object} qu The query to execute.
 @returns {object}
 @memberOf module:documents/collector~CollectorBase#
 @name findOne
 @method
 */


/**
 Find all records that match a query and returns the keys for those items. This is similar to {@link module:ink/probe.find} but instead of returning
 records, returns the keys. If `obj` is an object it will return the hash key. If 'obj' is an array, it will return the index

 @param {object} qu The query to execute.
 @returns {array}
 @memberOf module:documents/collector~CollectorBase#
 @name findKeys
 @method
 */


/**
 Find all records that match a query

 @param {object} qu The query to execute.
 @returns {array} The results
 @memberOf module:documents/collector~CollectorBase#
 @name find
 @method
 **/

/**
 Updates all records in obj that match the query. See {@link module:ink/probe.updateOperators} for the operators that are supported.

 @param {object} qu The query which will be used to identify the records to updated
 @param {object} setDocument The update operator. See {@link module:ink/probe.updateOperators}
 @memberOf module:documents/collector~CollectorBase#
 @name update
 @method
 */