Check/MagicCarpetProject/MagicCarpetDatabase/Implementations/SaleStorageContract.cs

using AutoMapper;
using MagicCarpetContracts.DataModels;
using MagicCarpetContracts.Exceptions;
using MagicCarpetContracts.Mapper;
using MagicCarpetContracts.Resources;
using MagicCarpetContracts.StoragesContracts;
using MagicCarpetDatabase.Models;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Localization;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace MagicCarpetDatabase.Implementations;

internal class SaleStorageContract(MagicCarpetDbContext dbContext, IStringLocalizer<Messages> localizer) : ISaleStorageContract
{
    private readonly MagicCarpetDbContext _dbContext = dbContext;
    private readonly IStringLocalizer<Messages> _localizer = localizer;

    public List<SaleDataModel> GetList(DateTime? startDate = null, DateTime? endDate = null, string? employeeId = null, string? clientId = null, string? tourId = null)
    {
        try
        {
            var query = _dbContext.Sales
                .Include(r => r.Employee)
                .Include(r => r.Client)
                .Include(r => r.SaleTours)!
                .ThenInclude(d => d.Tour)
                .AsQueryable();
            if (startDate.HasValue)
                query = query.Where(x => x.SaleDate >= DateTime.SpecifyKind(startDate ?? DateTime.UtcNow, DateTimeKind.Utc));
            if (endDate.HasValue)
                query = query.Where(x => x.SaleDate <= DateTime.SpecifyKind(endDate ?? DateTime.UtcNow, DateTimeKind.Utc));
            if (employeeId != null)
                query = query.Where(x => x.EmployeeId == employeeId);
            if (clientId != null)
                query = query.Where(x => x.ClientId == clientId);
            if (tourId != null)
                query = query.Where(x => x.SaleTours!.Any(y => y.TourId == tourId));
            var s = query.ToList();
            return [.. query.Select(x => CustomMapper.MapObject<SaleDataModel>(x))];
        }
        catch (Exception ex)
        {
            _dbContext.ChangeTracker.Clear();
            throw new StorageException(ex, _localizer);
        }
    }

    public async Task<List<SaleDataModel>> GetListAsync(DateTime startDate, DateTime endDate, CancellationToken ct)
    {
        try
        {
            return [.. await _dbContext.Sales
                .Include(x => x.Employee)
                .Include(x => x.Client)
                .Include(x => x.SaleTours)!
                    .ThenInclude(sc => sc.Tour)
                .Where(x => x.SaleDate >= DateTime.SpecifyKind(startDate, DateTimeKind.Utc)
                         && x.SaleDate < DateTime.SpecifyKind(endDate, DateTimeKind.Utc))
                .Select(x => CustomMapper.MapObject < SaleDataModel >(x))
                .ToListAsync(ct)];
        }
        catch (Exception ex)
        {
            _dbContext.ChangeTracker.Clear();
            throw new StorageException(ex, _localizer);
        }
    }

    public SaleDataModel? GetElementById(string id)
    {
        try
        {
            return CustomMapper.MapObjectWithNull<SaleDataModel>(GetSaleById(id));
        }
        catch (Exception ex)
        {
            _dbContext.ChangeTracker.Clear();
            throw new StorageException(ex, _localizer);
        }
    }

    public void AddElement(SaleDataModel saleDataModel)
    {
        try
        {
            var sale = MapToEntity(saleDataModel);
            _dbContext.Sales.Add(sale);
            _dbContext.SaveChanges();
        }
        catch (Exception ex)
        {
            _dbContext.ChangeTracker.Clear();
            throw new StorageException(ex, _localizer);
        }
    }

    public void DelElement(string id)
    {
        try
        {
            var element = GetSaleById(id) ?? throw new ElementNotFoundException(id, _localizer);
            if (element.IsCancel)
            {
                throw new ElementDeletedException(id, _localizer);
            }
            element.IsCancel = true;
            _dbContext.SaveChanges();
        }
        catch (Exception ex) when (ex is ElementDeletedException || ex is ElementNotFoundException)
        {
            _dbContext.ChangeTracker.Clear();
            throw;
        }
        catch (Exception ex)
        {
            _dbContext.ChangeTracker.Clear();
            throw new StorageException(ex, _localizer);
        }
    }

    private Sale MapToEntity(SaleDataModel dataModel)
    {
        var sale = CustomMapper.MapObject<Sale>(dataModel);

        sale.IsCancel = false;
        sale.SaleTours = dataModel.Tours?
            .Select(p => new SaleTour
            {
                TourId = p.TourId,
                Count = p.Count,
                Price = p.Price,
                SaleId = sale.Id
            })
            .ToList();

        return sale;
    }

    private Sale? GetSaleById(string id) => _dbContext.Sales.Include(x => x.Client).Include(x => x.Employee).Include(x => x.SaleTours)!.ThenInclude(x => x.Tour).FirstOrDefault(x => x.Id == id);
}