using Microsoft.EntityFrameworkCore; using PlumbingRepairContracts.BindingModels; using PlumbingRepairContracts.SearchModels; using PlumbingRepairContracts.StoragesContracts; using PlumbingRepairContracts.ViewModels; using PlumbingRepairDatabaseImplement.Models; namespace PlumbingRepairDatabaseImplement.Implements { public class OrderStorage : IOrderStorage { public OrderViewModel? Delete(OrderBindingModel model) { using var context = new PlumbingRepairDatabase(); var element = context.Orders.Include(x => x.Work).Include(x => x.Client).Include(x => x.Implementer).FirstOrDefault(rec => rec.Id == model.Id); if (element != null) { context.Orders.Remove(element); context.SaveChanges(); return AttachNames(element.GetViewModel); } return null; } public OrderViewModel? GetElement(OrderSearchModel model) { if (!model.Id.HasValue) { return null; } using var context = new PlumbingRepairDatabase(); if (model.Id.HasValue) return AttachNames(context.Orders .Include(x => x.Work) .Include(x => x.Implementer) .Include(x => x.Client) .FirstOrDefault(x => model.Id.HasValue && x.Id == model.Id) ?.GetViewModel); else if (model.ImplementerId.HasValue && model.Status.HasValue) return AttachNames(context.Orders .Include(x => x.Implementer) .Include(x => x.Work) .Include(x => x.Client) .FirstOrDefault(x => model.ImplementerId.HasValue && x.ImplementerId == model.ImplementerId) ?.GetViewModel); return null; } public List GetFilteredList(OrderSearchModel model) { using var context = new PlumbingRepairDatabase(); if (!model.Id.HasValue && !model.DateFrom.HasValue && !model.ClientId.HasValue && !model.Status.HasValue) { return new(); } if (model.Id.HasValue) { return context.Orders .Include(x => x.Work) .Include(x => x.Implementer) .Include(x => x.Client) .Where(x => x.Id == model.Id) .Select(x => AttachNames(x.GetViewModel)).ToList(); } if (model.Status.HasValue) { return context.Orders .Include(x => x.Work) .Include(x => x.Implementer) .Include(x => x.Client) .Where(x => x.ImplementerId == model.ImplementerId && x.Status == model.Status) .Select(x => AttachNames(x.GetViewModel)).ToList(); } if (model.ClientId.HasValue) { return context.Orders .Include(x => x.Client) .Include(x => x.Work) .Include(x => x.Implementer) .Where(x => x.ClientId == model.ClientId) .Select(x => AttachNames(x.GetViewModel)).ToList(); } if (model.ImplementerId.HasValue) { return context.Orders .Include(x => x.Work) .Include(x => x.Implementer) .Include(x => x.Client) .Where(x => x.ImplementerId == model.ImplementerId) .Select(x => AttachNames(x.GetViewModel)).ToList(); } if (model.DateFrom != null && model.DateTo != null) { return context.Orders .Include(x => x.Work) .Include(x => x.Client) .Include(x => x.Implementer) .Where(x => (x.DateCreate >= model.DateFrom && x.DateCreate <= model.DateTo) && (!model.ClientId.HasValue || x.ClientId == model.ClientId)) .Select(x => x.GetViewModel) .ToList(); } return new(); } public List GetFullList() { using var context = new PlumbingRepairDatabase(); return context.Orders .Include(x => x.Work) .Include(x => x.Implementer) .Include(x => x.Client) .Select(x =>AttachNames(x.GetViewModel)) .ToList(); } public OrderViewModel? Insert(OrderBindingModel model) { var newOrder = Order.Create(model); if (newOrder == null) { return null; } using var context = new PlumbingRepairDatabase(); context.Orders.Add(newOrder); context.SaveChanges(); return AttachNames(context.Orders.Include(x => x.Work).Include(x => x.Implementer).Include(x => x.Client).FirstOrDefault(x => x.Id == newOrder.Id)?.GetViewModel); } public OrderViewModel? Update(OrderBindingModel model) { using var context = new PlumbingRepairDatabase(); var order = context.Orders.Include(x => x.Work).Include(x => x.Implementer).Include(x => x.Client).FirstOrDefault(x => x.Id == model.Id); if (order == null) { return null; } order.Update(model); context.SaveChanges(); return AttachNames(order.GetViewModel); } public static OrderViewModel? AttachNames(OrderViewModel? model) { if (model == null) { return null; } using var context = new PlumbingRepairDatabase(); foreach (var work in context.Works) { if(work.Id == model.WorkId) { model.WorkName = work.WorkName; break; } } foreach ( var client in context.Clients) { if (client.Id == model.ClientId) { model.ClientFIO = client.ClientFIO; model.ClientEmail = client.Email; return model; } } foreach (var implementer in context.Implementers) { if (implementer.Id == model.ImplementerId) { model.ImplementerFIO = implementer.ImplementerFIO; return model; } } return model; } } }