public FlightOptionsViewModel GetFlightsForRequest(int id)
{
FlightOptionsViewModel vm = new FlightOptionsViewModel();
List <FlightItem> flightDataList = _mapper.Map <List <FlightInfo>, List <FlightItem> >(_unitofWork.FlightRepository.GetFlightsForRequest(id));
vm.OnwardFlightItems = flightDataList.FindAll(item => item.FlightDirection == "Onward");
vm.ReturnFlightItems = flightDataList.FindAll(item => item.FlightDirection == "Return");
return(vm);
}
public void AddFlights([FromBody] FlightOptionsViewModel flightData)
{
List <FlightInfo> _onwardflightItems = _mapper.Map <List <FlightItem>, List <FlightInfo> >(flightData.OnwardFlightItems);
List <FlightInfo> _returnflightItems = _mapper.Map <List <FlightItem>, List <FlightInfo> >(flightData.ReturnFlightItems);
_unitofWork.FlightRepository.AddOnwardFlightOptions(_onwardflightItems);
_unitofWork.FlightRepository.AddReturnFlightOptions(_returnflightItems);
_unitofWork.Complete();
}
public void UpdateFlights([FromBody] FlightOptionsViewModel flightData)
{
List <FlightItem> flightItems = new List <FlightItem>();
flightItems.AddRange(flightData.OnwardFlightItems);
flightItems.AddRange(flightData.ReturnFlightItems);
List <FlightInfo> flightDataList = (_unitofWork.FlightRepository.GetFlightsForRequest(flightItems.First().RequestInfoId));
foreach (var item in flightItems)
{
var refItem = flightDataList.FirstOrDefault(i => i.Id == item.Id);
if (refItem != null)
{
refItem.FlightFrom = item.FlightFrom;
refItem.FlightTo = item.FlightTo;
refItem.FlightName = item.FlightName;
refItem.FlightItemId = item.FlightItemId;
}
}
_unitofWork.Complete();
}
