using TravelCompanyContracts.BindingModels; using TravelCompanyContracts.SearchModels; using TravelCompanyContracts.StoragesContracts; using TravelCompanyContracts.ViewModels; using TravelCompanyDatabaseImplement.Models; namespace TravelCompanyDatabaseImplement.Implements { public class MessageInfoStorage : IMessageInfoStorage { public List GetFullList() { using var context = new TravelCompanyDataBase(); return context.MessageInfos .Select(x => x.GetViewModel) .ToList(); } public List GetFilteredList(MessageInfoSearchModel model) { if (!model.ClientId.HasValue) { return new(); } using var context = new TravelCompanyDataBase(); return context.MessageInfos .Where(x => x.ClientId.HasValue && x.ClientId == model.ClientId) .Select(x => x.GetViewModel) .ToList(); } public MessageInfoViewModel? GetElement(MessageInfoSearchModel model) { if (string.IsNullOrEmpty(model.MessageId)) { return new(); } using var context = new TravelCompanyDataBase(); return context.MessageInfos .FirstOrDefault(x => model.MessageId.Equals(x.MessageId)) ?.GetViewModel; } public MessageInfoViewModel? Insert(MessageInfoBindingModel model) { var newMessage = MessageInfo.Create(model); if (newMessage == null) { return null; } using var context = new TravelCompanyDataBase(); context.MessageInfos.Add(newMessage); context.SaveChanges(); return newMessage.GetViewModel; } } }