flatMapArray.js 871 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38
"use strict";

/**
 * Copyright (c) 2013-present, Facebook, Inc.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 *
 * 
 * @typechecks
 */
var push = Array.prototype.push;
/**
 * Applies a function to every item in an array and concatenates the resulting
 * arrays into a single flat array.
 *
 * @param {array} array
 * @param {function} fn
 * @return {array}
 */

function flatMapArray(array, fn) {
  var ret = [];

  for (var ii = 0; ii < array.length; ii++) {
    var result = fn.call(array, array[ii], ii);

    if (Array.isArray(result)) {
      push.apply(ret, result);
    } else if (result != null) {
      throw new TypeError('flatMapArray: Callback must return an array or null, ' + 'received "' + result + '" instead');
    }
  }

  return ret;
}

module.exports = flatMapArray;