using HotelContracts.BindingModels; using HotelContracts.SearchModels; using HotelContracts.StoragesContracts; using HotelContracts.ViewModels; using HotelDataBaseImplement.Models; using Microsoft.EntityFrameworkCore; using Microsoft.IdentityModel.Tokens; using System; using System.Collections.Generic; using System.Linq; using System.Text; using System.Threading.Tasks; namespace HotelDataBaseImplement.Implements { public class RoomStorage : IRoomStorage { public List GetFullList() { using var context = new HotelDataBase(); return context.Rooms .Include(x => x.Dinners) .ThenInclude(x => x.Dinner) .ThenInclude(x => x.ConferenceBookingDinner) .ThenInclude(x => x.ConferenceBooking) .Include(x => x.MealPlanRooms) .ThenInclude(x => x.MealPlan) .ThenInclude(x => x.Participants) .ThenInclude(x => x.Participant) .Include(x => x.Administrator) .ToList() .Select(x => x.GetViewModel) .ToList(); } public List GetFilteredList(RoomSearchModel model) { if (!model.DateFrom.HasValue && !model.DateTo.HasValue && !model.Id.HasValue && !model.AdministratorId.HasValue) { return new(); } using var context = new HotelDataBase(); if (model.DateFrom.HasValue) { return context.Rooms .Include(x => x.Dinners) .ThenInclude(x => x.Dinner) .ThenInclude(x => x.ConferenceBookingDinner) .ThenInclude(x => x.ConferenceBooking) .Include(x => x.MealPlanRooms) .ThenInclude(x => x.MealPlan) .ThenInclude(x => x.Participants) .ThenInclude(x => x.Participant) .Include(x => x.Administrator) .Where(x => x.DateCreate >= model.DateFrom && x.DateCreate <= model.DateTo && x.AdministratorId == model.AdministratorId) .Select(x => x.GetViewModel) .ToList(); } else if (model.AdministratorId.HasValue) { return context.Rooms .Include(x => x.Dinners) .ThenInclude(x => x.Dinner) .ThenInclude(x => x.ConferenceBookingDinner) .ThenInclude(x => x.ConferenceBooking) .Include(x => x.MealPlanRooms) .ThenInclude(x => x.MealPlan) .ThenInclude(x => x.Participants) .ThenInclude(x => x.Participant) .Include(x => x.Administrator) .Where(x => x.AdministratorId == model.AdministratorId) .ToList() .Select(x => x.GetViewModel) .ToList(); } return context.Rooms .Include(x => x.Dinners) .ThenInclude(x => x.Dinner) .ThenInclude(x => x.ConferenceBookingDinner) .ThenInclude(x => x.ConferenceBooking) .Include(x => x.MealPlanRooms) .ThenInclude(x => x.MealPlan) .ThenInclude(x => x.Participants) .ThenInclude(x => x.Participant) .Include(x => x.Administrator) .Where(x => x.RoomNumber == model.RoomNumber) .ToList() .Select(x => x.GetViewModel) .ToList(); } public RoomViewModel? GetElement(RoomSearchModel model) { if (!model.Id.HasValue && model.RoomNumber != null) { return null; } using var context = new HotelDataBase(); var room = context.Rooms .Include(x => x.Dinners) .ThenInclude(x => x.Dinner) .ThenInclude(x => x.ConferenceBookingDinner) .ThenInclude(x => x.ConferenceBooking) .Include(x => x.MealPlanRooms) .ThenInclude(x => x.MealPlan) .ThenInclude(x => x.Participants) .ThenInclude(x => x.Participant) .Include(x => x.Administrator) .FirstOrDefault(x => (model.RoomNumber != null && x.RoomNumber == model.RoomNumber) || (model.Id.HasValue && x.Id == model.Id)); return room?.GetViewModel; } public RoomViewModel? Insert(RoomBindingModel model) { using var context = new HotelDataBase(); var newRoom = Room.Create(context, model); if (newRoom == null) { return null; } context.Rooms.Add(newRoom); context.SaveChanges(); return context.Rooms .Include(x => x.Dinners) .ThenInclude(x => x.Dinner) .ThenInclude(x => x.ConferenceBookingDinner) .ThenInclude(x => x.ConferenceBooking) .Include(x => x.MealPlanRooms) .ThenInclude(x => x.MealPlan) .ThenInclude(x => x.Participants) .ThenInclude(x => x.Participant) .Include(x => x.Administrator) .FirstOrDefault(x => x.Id == newRoom.Id) ?.GetViewModel; } public RoomViewModel? Update(RoomBindingModel model) { using var context = new HotelDataBase(); using var transaction = context.Database.BeginTransaction(); try { var elem = context.Rooms .Include(x => x.Dinners) .ThenInclude(x => x.Dinner) .ThenInclude(x => x.ConferenceBookingDinner) .ThenInclude(x => x.ConferenceBooking) .ThenInclude(x => x.Conference) .FirstOrDefault(rec => rec.Id == model.Id); if (elem == null) { return null; } elem.Update(model); context.SaveChanges(); if (model.RoomDinners != null) elem.UpdateDinners(context, model); transaction.Commit(); return elem.GetViewModel; } catch { transaction.Rollback(); throw; } } public RoomViewModel? Delete(RoomBindingModel model) { using var context = new HotelDataBase(); var element = context.Rooms .Include(x => x.Dinners) .ThenInclude(x => x.Dinner) .ThenInclude(x => x.ConferenceBookingDinner) .ThenInclude(x => x.ConferenceBooking) .ThenInclude(x => x.Conference) .FirstOrDefault(rec => rec.Id == model.Id); if (element != null) { context.Rooms.Remove(element); context.SaveChanges(); return element.GetViewModel; } return null; } } }