ShiftScheduler/Services/PdfExportService.cs

using QuestPDF.Fluent;
using QuestPDF.Helpers;
using QuestPDF.Infrastructure;
using ShiftScheduler.Shared.Models;
using System.Globalization;

namespace ShiftScheduler.Services
{
    public class PdfExportService
    {
        static PdfExportService()
        {
            QuestPDF.Settings.License = LicenseType.Community; 
        }

        public byte[] GenerateMonthlySchedulePdf(Dictionary<DateTime, string> schedule, List<Shift> shifts, List<ShiftTransport>? shiftTransports = null)
        {
            var year = schedule.First().Key.Year;
            var month = schedule.First().Key.Month;
            var daysInMonth = DateTime.DaysInMonth(year, month);
            var monthDates = Enumerable.Range(1, daysInMonth)
                .Select(day => new DateTime(year, month, day))
                .ToList();

            var document = Document.Create(container =>
            {
                container.Page(page =>
                {
                    page.Margin(30);
                    page.Size(PageSizes.A4);
                    page.DefaultTextStyle(x => x.FontSize(12));

                    page.Header()
                        .Text($"Shift Schedule for {CultureInfo.CurrentCulture.DateTimeFormat.GetMonthName(month)} {year}")
                        .FontSize(18).Bold().AlignCenter();

                    page.Content().Table(table =>
                    {
                        table.ColumnsDefinition(columns =>
                        {
                            columns.RelativeColumn(2); // Date
                            columns.RelativeColumn(2); // Shift Name
                            columns.RelativeColumn(2); // Times
                            columns.RelativeColumn(3); // Transport
                            columns.RelativeColumn(1); // Icon
                        });

                        // Header
                        table.Header(header =>
                        {
                            header.Cell().Text("Date").Bold();
                            header.Cell().Text("Shift").Bold();
                            header.Cell().Text("Times").Bold();
                            header.Cell().Text("Transport").Bold();
                            header.Cell().Text("Icon").Bold();
                        });

                        foreach (var date in monthDates)
                        {
                            var dateText = date.ToString("dd.MM.yyyy (dddd)", CultureInfo.InvariantCulture);
                            table.Cell().Text(dateText);

                            var shiftName = schedule.ContainsKey(date) ? schedule[date] : null;
                            if (shiftName != null)
                            {
                                var shift = shifts.Single(s => s.Name.Equals(shiftName));
                                table.Cell().Text(shift.Name);
                                table.Cell().Text($"{shift.MorningTime} - {shift.AfternoonTime}");
                                
                                // Add transport information
                                var transport = shiftTransports?.FirstOrDefault(t => t.ShiftName == shift.Name);
                                var transportInfo = GetTransportSummary(transport);
                                table.Cell().Text(transportInfo);

                                if (!string.IsNullOrWhiteSpace(shift.Icon))
                                {
                                    try
                                    {
                                        table.Cell().Image(shift.Icon, ImageScaling.FitArea);
                                    }
                                    catch
                                    {
                                        table.Cell().Text("[icon missing]");
                                    }
                                }
                                else
                                {
                                    table.Cell().Text("-");
                                }
                            }
                            else
                            {
                                table.Cell().Text("-");
                                table.Cell().Text("-");
                                table.Cell().Text("-");
                                table.Cell().Text("-");
                            }
                        }
                    });
                });
            });

            return document.GeneratePdf();
        }

        private string GetTransportSummary(ShiftTransport? transport)
        {
            var transportLines = new List<string>();

            if (transport?.MorningTransport != null && !string.IsNullOrEmpty(transport.MorningTransport.DepartureTime))
            {
                var morningInfo = FormatTransportConnection(transport.MorningTransport, "Morning");
                transportLines.Add(morningInfo);
            }

            if (transport?.AfternoonTransport != null && !string.IsNullOrEmpty(transport.AfternoonTransport.DepartureTime))
            {
                var afternoonInfo = FormatTransportConnection(transport.AfternoonTransport, "Afternoon");
                transportLines.Add(afternoonInfo);
            }

            return transportLines.Count > 0 ? string.Join("\n", transportLines) : "-";
        }

        private string FormatTransportConnection(TransportConnection transport, string timeOfDay)
        {
            var departure = DateTime.TryParse(transport.DepartureTime, out var dep) ? dep.ToString("HH:mm") : transport.DepartureTime;
            var arrival = DateTime.TryParse(transport.ArrivalTime, out var arr) ? arr.ToString("HH:mm") : transport.ArrivalTime;
            
            var mainJourney = transport.Sections?.FirstOrDefault()?.Journey;
            var trainInfo = mainJourney != null ? $"{mainJourney.Category} {mainJourney.Number}" : "Train";

            return $"{timeOfDay}: {trainInfo} {departure}→{arrival}";
        }
    }
}