using System;
using System.Collections.Generic;
using System.Linq;
using Chernobyl.Extensions;

namespace Chernobyl.Generation
{
    /// <summary>
    /// Methods for building patterns.
    /// </summary>
    public static class Pattern
    {
        /// <summary>
        /// Creates a function which repeats the output of a factory the specified number of times.
        /// </summary>
        public static Func<IEnumerable<TResult>> AddRepeater<TResult>(this Func<TResult> factory, int count)
        {
            count.Throw(nameof(count)).IfLessThan(0);
            return () => factory.AsEnumerable().Take(count);
        }
    }
}