112 lines
4.2 KiB
C#
112 lines
4.2 KiB
C#
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<OrderViewModel> GetFullList()
|
|
{
|
|
using var context = new TravelCompanyDatabase();
|
|
return context.Orders.Include(x => x.Client)
|
|
.Select(x => AccessTravelStorage(x.GetViewModel, context))
|
|
.ToList();
|
|
}
|
|
public List<OrderViewModel> GetFilteredList(OrderSearchModel model)
|
|
{
|
|
if (!model.Id.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.Id == model.Id)
|
|
.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.DateCreate >= model.DateFrom && x.DateCreate<=model.DateTo)
|
|
.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;
|
|
}
|
|
}
|
|
}
|