using static System.Net.Mime.MediaTypeNames;
using iText.Kernel.Pdf;
using iText.Kernel.Geom;
using iText.IO.Image;
using iText.Layout.Element;
using Image = iText.Layout.Element.Image;
using iText.Layout.Properties;
using iText.Layout;
using iText.Kernel.Font;
using iText.IO.Font;
using iText.Kernel.Colors;

namespace PDFGenerateApi
{
    public class PdfOptHelper
    {


        /// <summary>
        /// 
        /// </summary>
        /// <param name="pagetitle"></param>
        /// <param name="headers"></param>
        /// <param name="majors"></param>
        /// <param name="dataModels"></param>
        /// <returns></returns>
        public static async Task<MemoryStream> CreatePdfAsync(string pagetitle, List<string> headers, List<string> majors, List<dataModel> dataModels, DateTime dtnow)
        {
            // 创建内存流
            using (var stream = new MemoryStream())
            {
                // 创建 PDF Writer
                PdfWriter writer = new PdfWriter(stream, new WriterProperties().SetCompressionLevel(CompressionConstants.BEST_COMPRESSION));
                PdfDocument pdf = new PdfDocument(writer);
                // 设置页面大小为横向 A4
                PageSize pageSize = PageSize.A4.Rotate();
                // 设置页面大小为 A4
                pdf.SetDefaultPageSize(pageSize);
                Document document = new Document(pdf);
                // 加载中文字体
                string fontsPath = System.IO.Path.Combine(Directory.GetCurrentDirectory(), "wwwroot", "fonts", "AlimamaDongFangDaKai-Regular.otf");
                //if (!File.Exists(fontsPath))
                //{
                //    bool aa = true;
                //    //  _logger.LogError($"Font file not found at path: {fontsPath}");
                //    //     return NotFound("Font file not found.");
                //}
                PdfFont font = PdfFontFactory.CreateFont(fontsPath, PdfEncodings.IDENTITY_H, PdfFontFactory.EmbeddingStrategy.PREFER_EMBEDDED);
                // 加载 logo 图像
                string logoPath = "https://i5.youzy.cn/util4y/media/files/20231026/7840b6f1f30c413eade15babc70be26e_%E5%BE%AE%E4%BF%A1%E5%9B%BE%E7%89%87_20230811154443.png"; // 确保 logo.png 放在 wwwroot 目录下
                ImageData imageData = ImageDataFactory.Create(logoPath);
                Image logo = new Image(imageData);
                logo.SetWidth(124);
                logo.SetHeight(40);
                // 将 logo 添加到文档中
                document.Add(logo);
                Paragraph title = CreateTitle(font, pagetitle);
                document.Add(title);
                // 添加标题下方内容
                Paragraph subtitle = CreateSubTitle(font, string.Format("最后修改时间:{0}", dtnow.ToString("yyyy-MM-dd HH:mm")));
                document.Add(subtitle);
              

                // 创建并行任务队列
                List<Task> tasks = new List<Task>();
                Parallel.For(0, dataModels.Count(), tableIndex =>
                {
                    lock (document)
                    {
                            Table universitytable = HeaderTable(font, 9, headers, new List<dataModel>() { dataModels[tableIndex] });
                            universitytable.SetMarginTop(20);
                            document.Add(universitytable);
                            Table majortable = MajorTable(font, 9, majors, dataModels[tableIndex].majorModels);
                            document.Add(majortable);
                        // 添加空白段落以增加表格间的间距
                        document.Add(new Paragraph("\n"));
                    }
                });

                // 等待所有并行任务完成
                await Task.WhenAll(tasks);
                // 关闭文档
                document.Close();
                // 返回 PDF 文件
                return stream;
            }

        }

        /// <summary>
        /// 标题
        /// </summary>
        /// <param name="font"></param>
        /// <param name="titletext"></param>
        /// <returns></returns>
        private static Paragraph CreateTitle(PdfFont font, string titletext)
        {
            Paragraph title = new Paragraph(titletext)
                  .SetTextAlignment(TextAlignment.CENTER)
                  .SetFont(font)
                  .SetBold()
                  .SetFontSize(16)
                  .SetMarginTop(20)
                  .SetMarginBottom(5);
            return title;
        }

        /// <summary>
        /// 创建副标
        /// </summary>
        /// <param name="font"></param>
        /// <param name="titletext"></param>
        /// <returns></returns>
        private static Paragraph CreateSubTitle(PdfFont font, string titletext)
        {
            Paragraph title = new Paragraph(titletext)
                    .SetTextAlignment(TextAlignment.CENTER)
                     .SetFont(font)
                     .SetFontColor(ColorConstants.GRAY)  // 设置字体颜色为灰色
                    .SetFontSize(12)
                    .SetMarginBottom(20);
            return title;
        }


        /// <summary>
        /// table表头
        /// </summary>
        /// <param name="tb"></param>
        /// <param name="headers"></param>
        /// <returns></returns>
        private static Table HeaderTable(PdfFont font, int tb, List<string> headers, List<dataModel> dataList)
        {
            float[] columnWidths = { 1, 1, 2, 3, 2, 2, 2, 1, 1 };
            Table table = new Table(columnWidths);
            table.SetWidth(UnitValue.CreatePercentValue(100));


            int h = 0;
            foreach (var header in headers)
            {
                h++;
                if (h == 4)
                {
                    table.AddHeaderCell(new Cell().Add(new Paragraph(header).SetWidth(120).SetFont(font).SetBold().SetTextAlignment(TextAlignment.CENTER)));
                }
                else
                {
                    table.AddHeaderCell(new Cell().Add(new Paragraph(header).SetWidth(50).SetFont(font).SetBold().SetTextAlignment(TextAlignment.CENTER)));
                }
            }
            dataList.ForEach(a =>
            {
                table.AddCell(new Cell().Add(new Paragraph(a.id.ToString()).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                table.AddCell(new Cell().Add(new Paragraph(a.probability).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                table.AddCell(new Cell().Add(new Paragraph(a.universityCode).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                table.AddCell(new Cell().Add(new Paragraph(a.universityName).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                table.AddCell(new Cell().Add(new Paragraph(a.planCount).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                table.AddCell(new Cell().Add(new Paragraph(a.history).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                a.years.ForEach(a =>
                {
                    table.AddCell(new Cell().Add(new Paragraph(a).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                });
            });


            return table;
        }


        private static Table MajorTable(PdfFont font, int tb, List<string> headers, List<dataMajorModel> dataList)
        {
            float[] columnWidths = { 1, 1, 2, 3, 2, 2, 2, 1, 1 };
            Table table = new Table(columnWidths);
            table.SetWidth(UnitValue.CreatePercentValue(100));
            int h = 0;
            foreach (var header in headers)
            {
                h++;
                if (h == 4)
                {
                    table.AddCell(new Cell().Add(new Paragraph(header).SetWidth(85).SetFont(font).SetBold().SetTextAlignment(TextAlignment.CENTER)));
                }
                else
                {
                    table.AddCell(new Cell().Add(new Paragraph(header).SetFont(font).SetBold().SetTextAlignment(TextAlignment.CENTER)));
                }
            }
            dataList.ForEach(a =>
            {
                table.AddCell(new Cell().Add(new Paragraph(a.id.ToString()).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                table.AddCell(new Cell().Add(new Paragraph(a.probability).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                table.AddCell(new Cell().Add(new Paragraph(a.majorCode).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                table.AddCell(new Cell().Add(new Paragraph(a.mjaorName).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                table.AddCell(new Cell().Add(new Paragraph(a.planCount).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                table.AddCell(new Cell().Add(new Paragraph(a.history).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                a.years.ForEach(a =>
                {
                    table.AddCell(new Cell().Add(new Paragraph(a).SetFont(font).SetTextAlignment(TextAlignment.CENTER)));
                });
            });
            return table;
        }



        public class dataModel
        {
            public int id { get; set; }

            /// <summary>
            /// 
            /// </summary>
            public string probability { get; set; }

            /// <summary>
            /// 
            /// </summary>
            public string universityCode { get; set; }

            /// <summary>
            /// 
            /// </summary>
            public string universityName { get; set; }


            public string planCount { get; set; }


            public string history { get; set; }

            public List<string> years { get; set; }

            /// <summary>
            /// 
            /// </summary>
            public List<dataMajorModel> majorModels { get; set; }
        }


        public class dataMajorModel
        {
            public int id { get; set; }

            /// <summary>
            /// 
            /// </summary>
            public string probability { get; set; }

            /// <summary>
            /// 
            /// </summary>
            public string majorCode { get; set; }

            /// <summary>
            /// 
            /// </summary>
            public string mjaorName { get; set; }


            public string planCount { get; set; }


            public string history { get; set; }

            public List<string> years { get; set; }


        }


    }
}