hrms-api-backend/BMA.EHR.Application/Repositories/MetaData/HolidayRepository.cs

using BMA.EHR.Application.Common.Interfaces;
using BMA.EHR.Domain.Models.MetaData;
using Microsoft.AspNetCore.Http;
using Microsoft.EntityFrameworkCore;

namespace BMA.EHR.Application.Repositories.MetaData
{
    public class HolidayRepository : GenericRepository<Guid, Holiday>
    {
        #region " Fields "

        private readonly IApplicationDBContext _dbContext;
        private readonly IHttpContextAccessor _httpContextAccessor;

        #endregion

        #region " Costructor and Destructor "

        public HolidayRepository(IApplicationDBContext dbContext, IHttpContextAccessor httpContextAccessor) : base(dbContext, httpContextAccessor)
        {
            _dbContext = dbContext;
            _httpContextAccessor = httpContextAccessor;
        }

        #endregion

        #region " Methods "

        public async Task<int> GetHolidayCountAsync(DateTime startDate, DateTime endDate, string category = "NORMAL")
        {
            var data = await _dbContext.Set<Holiday>().AsQueryable()
                                .Where(x => x.Category == category)
                                .Where(x => x.HolidayDate.Date >= startDate && x.HolidayDate.Date <= endDate)
                                .CountAsync();

            return data;
        }

        public int GetWeekEndCount(DateTime startDate, DateTime endDate)
        {
            var dates = new List<DateTime>();

            for (DateTime i = startDate; i < endDate; i = i.AddDays(1))
            {
                dates.Add(i);
            }

            var count = dates.Where(d => d.DayOfWeek == DayOfWeek.Saturday || d.DayOfWeek == DayOfWeek.Sunday).Count();

            return count;
        }

        #endregion
    }
}