All files flatMap.ts

100% Statements 12/12
100% Branches 0/0
100% Functions 2/2
100% Lines 10/10
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 2313x 13x               5x 5x 2x 6x 6x 12x 24x           13x  
import { Collection, asIterable } from './utils'
import Sequence from './Sequence'
 
/**
 * Return a Sequence that contains the elements of flattened collections created from the input collection elements.
 * @param collection A collection to use as input.
 * @param fn A function that produces an element of the new Sequence using an element of the old collection.
 */
function flatMap<T, U>(collection: Collection<T>, fn: (value: T, index: number) => Collection<U>) {
  const iterable = asIterable(collection)
  return new Sequence(function* () {
    let index = 0
    for(const value of iterable) {
      const result = asIterable(fn(value, index++))
      for(const innerValue of result) {
        yield innerValue
      }
    }
  })
}
 
export default flatMap