/*
 * @flow
 * Copyright (c) 2018 Michael Wiencek
 *
 * This source code is licensed under the MIT license. A copy can be found
 * in the file named "LICENSE" at the root directory of this distribution.
 */

import {EMPTY_ITERATOR, TAKE} from './constants';
import makeTerable from './Terable';
import type {IterableExt, Terable} from './types';

export default function take(count: number) {
  return function <T>(iterable: IterableExt<T>): Terable<T> {
    if (count <= 0) {
      return (EMPTY_ITERATOR: Terable<T>);
    }
    return makeTerable({type: TAKE, arg: count, source: iterable});
  };
}