ISEbd-21.Fedotov.I.A.Diner/Diner/DinerDataBaseImplement/Implements/OrderStorage.cs

using DinerContracts.BindingModels;
using DinerContracts.SearchModels;
using DinerContracts.StoragesContracts;
using DinerContracts.ViewModels;
using DinerDataBaseImplement.Models;
using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace DinerDataBaseImplement.Implements
{
    public class OrderStorage : IOrderStorage
    {
        public OrderViewModel? Delete(OrderBindingModel model)
        {
            using var context = new DinerDataBase();
            var element = context.Orders.FirstOrDefault(x => x.ID == model.ID);
            if (element != null)
            {
                context.Orders.Remove(element);
                context.SaveChanges();
                return GetViewModel(element);
            }
            return null;
        }

        public OrderViewModel? GetElement(OrderSearchModel model)
        {
            if (!model.ID.HasValue) { 
                return null; 
            }
            using var context = new DinerDataBase();
            return context.Orders
                .FirstOrDefault(x => model.ID.HasValue && x.ID == model.ID)?.GetViewModel;
        }

        public List<OrderViewModel> GetFilteredList(OrderSearchModel model)
        {
            using var context = new DinerDataBase();
            if (!model.ID.HasValue && (model.DateFrom == null || model.DateTo == null)) { 
                return new(); 
            }
            else {
                return context.Orders
                    .Include(x => x.Snack)
                    .Where(x => x.DateCreate >= model.DateFrom && x.DateCreate <= model.DateTo)
                    .Select(x => x.GetViewModel)
                    .ToList();
            }
        }

        public List<OrderViewModel> GetFullList()
        {
            using var context = new DinerDataBase();
            return context.Orders.Select(GetViewModel).ToList();
        }
        public OrderViewModel? Insert(OrderBindingModel model)
        {
            using var context = new DinerDataBase();
            var newOrder = Order.Create(model);
            if (newOrder == null) { 
                return null; 
            }
            context.Orders.Add(newOrder);
            context.SaveChanges();
            return GetViewModel(newOrder);
        }
        public OrderViewModel? Update(OrderBindingModel model)
        {
            using var context = new DinerDataBase();
            var order = context.Orders.FirstOrDefault(x => x.ID == model.ID);
            if (order == null) { return null; }
            order.Update(model);
            context.SaveChanges();
            return GetViewModel(order);
        }
        private OrderViewModel GetViewModel(Order order)
        {
            using var context = new DinerDataBase();
            var viewModel = order.GetViewModel;
            foreach (var pr in context.Products) { 
                if (pr.ID == order.ProductID) { 
                    viewModel.ProductName = pr.ProductName; 
                    break;
                }
            }
            return viewModel;
        }
    }
}