using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using MongoDB.Bson;
using MongoDB.Driver;

namespace BuildFeed.Model
{
    public partial class BuildRepository
    {
        public Task<TypeOfSource[]> SelectAllSources(int limit = -1, int skip = 0)
            => Task.Run(() => Enum.GetValues(typeof(TypeOfSource)) as TypeOfSource[]);

        public Task<long> SelectAllSourcesCount() => Task.Run(() => Enum.GetValues(typeof(TypeOfSource)).LongLength);

        public async Task<List<Build>> SelectSource(TypeOfSource source, int limit = -1, int skip = 0)
        {
            var query = _buildCollection.Find(new BsonDocument(nameof(Build.SourceType), source))
                .Sort(sortByOrder)
                .Skip(skip);

            if (limit > 0)
            {
                query = query.Limit(limit);
            }

            return await query.ToListAsync();
        }

        public async Task<long> SelectSourceCount(TypeOfSource source)
            => await _buildCollection.CountAsync(new BsonDocument(nameof(Build.SourceType), source));
    }
}