ShiftScheduler/Server/Controllers/ShiftController.cs

using Microsoft.AspNetCore.Mvc;
using ShiftScheduler.Services;
using ShiftScheduler.Shared.Models;

namespace ShiftScheduler.Server.Controllers
{
    [ApiController]
    [Route("api/[controller]")]
    public class ShiftController : ControllerBase
    {
        private readonly ShiftService _shiftService;
        private readonly IcsExportService _icsService;
        private readonly PdfExportService _pdfExportService;
        private readonly TransportService _transportService;
        private readonly ShiftEnrichmentService _enrichmentService;

        public ShiftController(
            ShiftService shiftService, 
            IcsExportService icsService, 
            PdfExportService pdfExportService, 
            TransportService transportService,
            ShiftEnrichmentService enrichmentService)
        {
            _shiftService = shiftService;
            _icsService = icsService;
            _pdfExportService = pdfExportService;
            _transportService = transportService;
            _enrichmentService = enrichmentService;
        }

        [HttpGet("shifts")]
        public IActionResult GetShifts()
        {
            return Ok(_shiftService.GetShifts());
        }

        [HttpPost("transport_connection")]
        public async Task<IActionResult> GetTransportConnection([FromBody] TransportConnectionRequest request)
        {
            var connection = await _transportService.GetConnectionAsync(request.ArrivalTime, request.EndStation);
            return Ok(connection);
        }

        [HttpPost("shift_transport")]
        public async Task<IActionResult> GetShiftTransport([FromBody] ShiftTransportRequest request)
        {
            var shift = _shiftService.GetShifts().FirstOrDefault(s => s.Name == request.ShiftName);
            if (shift == null)
            {
                return NotFound($"Shift '{request.ShiftName}' not found");
            }

            TransportConnection? morningTransport = null;
            TransportConnection? afternoonTransport = null;

            // Get transport for morning shift if it has morning time
            if (!string.IsNullOrEmpty(shift.MorningTime))
            {
                var morningStartTime = ParseShiftTime(request.Date, shift.MorningTime);
                if (morningStartTime.HasValue)
                {
                    morningTransport = await _transportService.GetConnectionAsync(morningStartTime.Value);
                }
            }

            // Get transport for afternoon shift if it has afternoon time
            if (!string.IsNullOrEmpty(shift.AfternoonTime))
            {
                var afternoonStartTime = ParseShiftTime(request.Date, shift.AfternoonTime);
                if (afternoonStartTime.HasValue)
                {
                    afternoonTransport = await _transportService.GetConnectionAsync(afternoonStartTime.Value);
                }
            }

            var shiftWithTransport = new ShiftWithTransport
            {
                Date = request.Date,
                Shift = shift,
                MorningTransport = morningTransport,
                AfternoonTransport = afternoonTransport
            };

            return Ok(shiftWithTransport);
        }

        private static DateTime? ParseShiftTime(DateTime date, string timeRange)
        {
            try
            {
                var times = timeRange.Split('-');
                if (times.Length > 0 && TimeSpan.TryParse(times[0], out var startTime))
                {
                    return date.Add(startTime);
                }
            }
            catch
            {
                // Ignore parsing errors
            }
            return null;
        }

        [HttpPost("export_ics")]
        public async Task<IActionResult> ExportIcs([FromBody] Dictionary<DateTime, string> schedule)
        {
            var shifts = _shiftService.GetShifts();
            var shiftTransports = await _enrichmentService.EnrichShiftsWithTransportAsync(shifts, schedule);
            var ics = _icsService.GenerateIcs(schedule, shifts, shiftTransports);
            return File(System.Text.Encoding.UTF8.GetBytes(ics), "text/calendar", "schedule.ics");
        }

        [HttpPost("export_ics_with_transport")]
        public IActionResult ExportIcsWithTransport([FromBody] List<ShiftWithTransport> shiftsWithTransport)
        {
            var schedule = shiftsWithTransport.ToDictionary(s => s.Date, s => s.Shift.Name);
            var shifts = shiftsWithTransport.Select(s => s.Shift).DistinctBy(s => s.Name).ToList();
            var shiftTransports = shiftsWithTransport.Select(s => new ShiftTransport
            {
                ShiftName = s.Shift.Name,
                MorningTransport = s.MorningTransport,
                AfternoonTransport = s.AfternoonTransport
            }).DistinctBy(s => s.ShiftName).ToList();
            
            var ics = _icsService.GenerateIcs(schedule, shifts, shiftTransports);
            return File(System.Text.Encoding.UTF8.GetBytes(ics), "text/calendar", "schedule.ics");
        }

         [HttpPost("export_pdf")]
        public async Task<IActionResult> ExportPdf([FromBody] Dictionary<DateTime, string> schedule)
        {
            var shifts = _shiftService.GetShifts();
            var shiftTransports = await _enrichmentService.EnrichShiftsWithTransportAsync(shifts, schedule);
            var pdf = _pdfExportService.GenerateMonthlySchedulePdf(schedule, shifts, shiftTransports);
            return File(pdf, "application/pdf", "schedule.pdf");
        }

        [HttpPost("export_pdf_with_transport")]
        public IActionResult ExportPdfWithTransport([FromBody] List<ShiftWithTransport> shiftsWithTransport)
        {
            var schedule = shiftsWithTransport.ToDictionary(s => s.Date, s => s.Shift.Name);
            var shifts = shiftsWithTransport.Select(s => s.Shift).DistinctBy(s => s.Name).ToList();
            var shiftTransports = shiftsWithTransport.Select(s => new ShiftTransport
            {
                ShiftName = s.Shift.Name,
                MorningTransport = s.MorningTransport,
                AfternoonTransport = s.AfternoonTransport
            }).DistinctBy(s => s.ShiftName).ToList();
            
            var pdf = _pdfExportService.GenerateMonthlySchedulePdf(schedule, shifts, shiftTransports);
            return File(pdf, "application/pdf", "schedule.pdf");
        }
    }

    public class TransportConnectionRequest
    {
        public DateTime ArrivalTime { get; set; }
        public string? EndStation { get; set; }
    }

    public class ShiftTransportRequest
    {
        public string ShiftName { get; set; } = string.Empty;
        public DateTime Date { get; set; }
    }
}