using Microsoft.EntityFrameworkCore; using System; using System.Collections.Generic; using System.Linq; using System.Text; using System.Threading.Tasks; using TravelCompanyContracts.BindingModels; using TravelCompanyContracts.SearchModels; using TravelCompanyContracts.StoragesContracts; using TravelCompanyContracts.ViewModels; using TravelCompanyDatabaseImplement.Models; namespace TravelCompanyDatabaseImplement.Implements { public class OrderStorage : IOrderStorage { public List GetFullList() { using var context = new TravelCompanyDatabase(); return context.Orders.Include(x => x.Client) .Select(x => AccessTravelStorage(x.GetViewModel, context)) .ToList(); } public List GetFilteredList(OrderSearchModel model) { if (!model.ClientId.HasValue && !model.DateFrom.HasValue && !model.DateTo.HasValue) { return new(); } if (model.DateFrom.HasValue && model.DateTo.HasValue) { using var context = new TravelCompanyDatabase(); return context.Orders.Include(x => x.Client) .Where(x => x.DateCreate >= model.DateFrom && x.DateCreate <= model.DateTo) .Select(x => AccessTravelStorage(x.GetViewModel, context)) .ToList(); } if (model.ClientId.HasValue) { using var context = new TravelCompanyDatabase(); return context.Orders.Include(x => x.Client) .Where(x => x.ClientId == model.ClientId) .Select(x => AccessTravelStorage(x.GetViewModel, context)) .ToList(); } else { using var context = new TravelCompanyDatabase(); return context.Orders.Include(x => x.Client) .Where(x => x.Id == model.Id) .Select(x => AccessTravelStorage(x.GetViewModel, context)) .ToList(); } } public OrderViewModel? GetElement(OrderSearchModel model) { if (!model.Id.HasValue) { return new(); } using var context = new TravelCompanyDatabase(); return AccessTravelStorage(context.Orders.Include(x => x.Client) .FirstOrDefault(x => model.Id.HasValue && x.Id == model.Id) ?.GetViewModel, context); } public OrderViewModel? Insert(OrderBindingModel model) { using var context = new TravelCompanyDatabase(); var newOrder = Order.Create(model, context); if (newOrder == null) { return null; } context.Orders.Add(newOrder); context.SaveChanges(); return AccessTravelStorage(newOrder.GetViewModel, context); } public OrderViewModel? Update(OrderBindingModel model) { using var context = new TravelCompanyDatabase(); var order = context.Orders.FirstOrDefault(x => x.Id == model.Id); if (order == null) { return null; } order.Update(model); context.SaveChanges(); return AccessTravelStorage(order.GetViewModel, context); } public OrderViewModel? Delete(OrderBindingModel model) { using var context = new TravelCompanyDatabase(); var element = context.Orders.FirstOrDefault(rec => rec.Id == model.Id); if (element != null) { context.Orders.Remove(element); context.SaveChanges(); return AccessTravelStorage(element.GetViewModel, context); } return null; } static OrderViewModel AccessTravelStorage(OrderViewModel model, TravelCompanyDatabase context) { if (model == null) return model; string? travelName = context.Travels.FirstOrDefault(x => x.Id == model.TravelId)?.TravelName; if (travelName != null) model.TravelName = travelName; return model; } } }