Создание и заполнение HeadwaiterStorage в HotelDataBaseImplement.Implemets

Hotel/HotelDataBaseImplement/Implemets/HeadwaiterStorage.cs

@@ -0,0 +1,127 @@
+using HotelContracts.BindingModels;
+using HotelContracts.SearchModels;
+using HotelContracts.StoragesContracts;
+using HotelContracts.ViewModels;
+using HotelDataBaseImplement.Models;
+using Microsoft.EntityFrameworkCore;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace HotelDataBaseImplement.Implemets
+{
+    public class HeadwaiterStorage : IHeadwaiterStorage
+    {
+        public HeadwaiterViewModel? Delete(HeadwaiterBindingModel model)
+        {
+            using var context = new HotelDataBase();
+
+            var element = context.Headwaiters.FirstOrDefault(rec => rec.Id == model.Id);
+
+            if (element != null)
+            {
+                context.Headwaiters.Remove(element);
+                context.SaveChanges();
+
+                return element.GetViewModel;
+            }
+
+            return null;
+        }
+
+        public HeadwaiterViewModel? GetElement(HeadwaiterSearchModel model)
+        {
+            using var context = new HotelDataBase();
+
+            if (model.Id.HasValue)
+                return context.Headwaiters
+                    .Include(x => x.ConferenceBookings)
+                    .Include(x => x.Dinners)
+                    .Include(x => x.Rooms)
+                    .FirstOrDefault(x => x.Id == model.Id)?
+                    .GetViewModel;
+
+            if (!string.IsNullOrEmpty(model.HeadwaiterLogin) && !string.IsNullOrEmpty(model.HeadwaiterPassword))
+                return context.Headwaiters
+                    .Include(x => x.ConferenceBookings)
+                    .Include(x => x.Dinners)
+                    .Include(x => x.Rooms)
+                    .FirstOrDefault(x => x.HeadwaiterLogin.Equals(model.HeadwaiterLogin) && x.HeadwaiterPassword.Equals(model.HeadwaiterPassword))?
+                    .GetViewModel;
+
+            if (!string.IsNullOrEmpty(model.HeadwaiterLogin))
+                return context.Headwaiters
+                    .Include(x => x.ConferenceBookings)
+                    .Include(x => x.Dinners)
+                    .Include(x => x.Rooms)
+                    .FirstOrDefault(x => x.HeadwaiterLogin.Equals(model.HeadwaiterLogin))?
+                    .GetViewModel;
+
+            return null;
+        }
+
+        public List<HeadwaiterViewModel> GetFilteredList(HeadwaiterSearchModel model)
+        {
+            if (string.IsNullOrEmpty(model.HeadwaiterFIO))
+            {
+                return new();
+            }
+
+            using var context = new HotelDataBase();
+
+            return context.Headwaiters
+                .Include(x => x.ConferenceBookings)
+                .Include(x => x.Dinners)
+                .Include(x => x.Rooms)
+                .Where(x => x.HeadwaiterLogin.Contains(model.HeadwaiterLogin) && x.HeadwaiterPassword == model.HeadwaiterPassword)
+                .Select(x => x.GetViewModel)
+                .ToList();
+        }
+
+        public List<HeadwaiterViewModel> GetFullList()
+        {
+            using var context = new HotelDataBase();
+
+            return context.Headwaiters
+                .Select(x => x.GetViewModel)
+                .ToList();
+        }
+
+        public HeadwaiterViewModel? Insert(HeadwaiterBindingModel model)
+        {
+            var newHeadwaiter = Headwaiter.Create(model);
+
+            if (newHeadwaiter == null)
+            {
+                return null;
+            }
+
+            using var context = new HotelDataBase();
+
+            context.Headwaiters.Add(newHeadwaiter);
+            context.SaveChanges();
+
+            return newHeadwaiter.GetViewModel;
+        }
+
+        public HeadwaiterViewModel? Update(HeadwaiterBindingModel model)
+        {
+            using var context = new HotelDataBase();
+
+            var headwaiter = context.Headwaiters
+                .FirstOrDefault(x => x.Id == model.Id);
+
+            if (headwaiter == null)
+            {
+                return null;
+            }
+
+            headwaiter.Update(model);
+            context.SaveChanges();
+
+            return headwaiter.GetViewModel;
+        }
+    }
+}