/*
 * =============================================================================
 *  VibeQuant (vquant) — AI-Powered Financial Intelligence Platform
 * -----------------------------------------------------------------------------
 *  File:      client/src/utils/pdfGenerator.ts
 *
 *  Author:    Simon-Pierre Boucher
 *  Contact:   contact@spboucher.ai
 *  Website:   https://www.spboucher.ai
 *  Demo:      https://www.vquant.ai
 *  License:   MIT (see LICENSE)
 *
 *  Copyright © 2026 Simon-Pierre Boucher. All rights reserved.
 * =============================================================================
 */

import jsPDF from 'jspdf';
import html2canvas from 'html2canvas';

interface PDFOptions {
  title: string;
  content: string;
  includeCharts?: boolean;
}

interface AdvancedPDFOptions {
  title: string;
  contentElement: HTMLElement;
  includeAllVisuals?: boolean;
}

// Helper function to draw the VQuant logo
function drawLogo(doc: jsPDF, x: number, y: number, size: number): void {
  // Draw gradient circle background
  doc.setFillColor(139, 92, 246); // Purple
  doc.circle(x, y, size, 'F');

  // Add lighter circle for depth
  doc.setFillColor(168, 85, 247);
  doc.circle(x - size * 0.15, y - size * 0.15, size * 0.6, 'F');

  // Draw "VQ" text
  doc.setTextColor(255, 255, 255);
  doc.setFontSize(size * 1.2);
  doc.setFont('helvetica', 'bold');
  const vqText = 'VQ';
  const textWidth = doc.getTextWidth(vqText);
  doc.text(vqText, x - textWidth / 2, y + size * 0.25);

  // Draw small chart lines for visual effect
  doc.setDrawColor(255, 255, 255);
  doc.setLineWidth(0.5);
  const chartY = y + size * 0.7;
  const chartPoints = [
    { x: x - size * 0.6, y: chartY },
    { x: x - size * 0.3, y: chartY - size * 0.2 },
    { x: x, y: chartY - size * 0.1 },
    { x: x + size * 0.3, y: chartY - size * 0.3 },
    { x: x + size * 0.6, y: chartY - size * 0.15 }
  ];

  for (let i = 0; i < chartPoints.length - 1; i++) {
    doc.line(
      chartPoints[i].x,
      chartPoints[i].y,
      chartPoints[i + 1].x,
      chartPoints[i + 1].y
    );
    doc.circle(chartPoints[i].x, chartPoints[i].y, 0.5, 'F');
  }
  doc.circle(chartPoints[chartPoints.length - 1].x, chartPoints[chartPoints.length - 1].y, 0.5, 'F');
}

// Helper function to add a professional title page
function addTitlePage(doc: jsPDF, title: string): void {
  const pageWidth = doc.internal.pageSize.getWidth();
  const pageHeight = doc.internal.pageSize.getHeight();
  const margin = 20;
  const maxWidth = pageWidth - (margin * 2);

  // Background gradient effect - plus subtil
  doc.setFillColor(250, 251, 255); // Ultra light indigo
  doc.rect(0, 0, pageWidth, pageHeight, 'F');

  // Top decorative bar avec dégradé simulé
  doc.setFillColor(99, 102, 241);
  doc.rect(0, 0, pageWidth, 10, 'F');
  doc.setFillColor(79, 82, 221, 0.8);
  doc.rect(0, 8, pageWidth, 2, 'F');

  // Draw large logo in center - position améliorée
  const logoSize = 28;
  const logoX = pageWidth / 2;
  const logoY = 75;
  drawLogo(doc, logoX, logoY, logoSize);

  // Main title avec meilleur espacement
  doc.setTextColor(55, 48, 163); // Indigo-700
  doc.setFontSize(42);
  doc.setFont('helvetica', 'bold');
  const mainTitle = 'VQuant';
  const titleWidth = doc.getTextWidth(mainTitle);
  doc.text(mainTitle, (pageWidth - titleWidth) / 2, 125);

  // Subtitle avec meilleure taille
  doc.setFontSize(13);
  doc.setFont('helvetica', 'normal');
  doc.setTextColor(99, 102, 241);
  const subtitle = 'Analyse Financière par Intelligence Artificielle';
  const subtitleWidth = doc.getTextWidth(subtitle);
  doc.text(subtitle, (pageWidth - subtitleWidth) / 2, 137);

  // Decorative line plus élégante
  doc.setDrawColor(139, 92, 246);
  doc.setLineWidth(0.8);
  const lineWidth = 70;
  doc.line((pageWidth - lineWidth) / 2, 148, (pageWidth + lineWidth) / 2, 148);

  // Report title
  doc.setFontSize(11);
  doc.setFont('helvetica', 'bold');
  doc.setTextColor(71, 85, 105);
  doc.text('RAPPORT D\'ANALYSE', pageWidth / 2, 165, { align: 'center' });

  // Title box avec shadow effect
  doc.setFontSize(12);
  doc.setFont('helvetica', 'normal');
  doc.setTextColor(15, 23, 42);
  const titleLines = doc.splitTextToSize(title, maxWidth - 30);
  const boxHeight = titleLines.length * 8 + 16;
  const boxY = 175;

  // Shadow
  doc.setFillColor(226, 232, 240);
  doc.roundedRect((pageWidth - maxWidth + 28) / 2 + 1, boxY + 1, maxWidth - 26, boxHeight, 4, 4, 'F');

  // Box principal
  doc.setFillColor(255, 255, 255);
  doc.setDrawColor(203, 213, 225);
  doc.setLineWidth(0.5);
  doc.roundedRect((pageWidth - maxWidth + 28) / 2, boxY, maxWidth - 26, boxHeight, 4, 4, 'FD');

  let textY = boxY + 10;
  titleLines.forEach((line: string) => {
    doc.text(line, pageWidth / 2, textY, { align: 'center' });
    textY += 8;
  });

  // Date and metadata avec meilleur style
  doc.setFontSize(10);
  doc.setFont('helvetica', 'italic');
  doc.setTextColor(100, 116, 139);
  const dateStr = new Date().toLocaleDateString('fr-FR', {
    weekday: 'long',
    year: 'numeric',
    month: 'long',
    day: 'numeric'
  });
  doc.text(`Généré le ${dateStr}`, pageWidth / 2, pageHeight - 45, { align: 'center' });

  // Footer decorative line
  doc.setDrawColor(203, 213, 225);
  doc.setLineWidth(0.3);
  doc.line(40, pageHeight - 30, pageWidth - 40, pageHeight - 30);

  // Footer decorative elements - plus élégants
  doc.setFillColor(139, 92, 246);
  doc.circle(35, pageHeight - 20, 1.5, 'F');
  doc.circle(pageWidth - 35, pageHeight - 20, 1.5, 'F');

  doc.setFontSize(9);
  doc.setFont('helvetica', 'normal');
  doc.setTextColor(148, 163, 184);
  doc.text('www.vibequant.com', pageWidth / 2, pageHeight - 18, { align: 'center' });

  // Watermark subtil
  doc.setFontSize(8);
  doc.setTextColor(203, 213, 225);
  doc.text('Confidentiel', pageWidth / 2, pageHeight - 10, { align: 'center' });

  // Add new page for content
  doc.addPage();
}

// Helper to capture element as image
async function captureElementAsImage(element: HTMLElement): Promise<string | null> {
  try {
    console.log('captureElementAsImage: Starting capture');
    console.log('Element:', element);
    console.log('Element dimensions:', {
      width: element.scrollWidth,
      height: element.scrollHeight,
      offsetWidth: element.offsetWidth,
      offsetHeight: element.offsetHeight
    });
    console.log('Element innerHTML length:', element.innerHTML.length);

    // Clone the element to avoid modifying the original
    const clone = element.cloneNode(true) as HTMLElement;

    // Remove any animations or transitions that might interfere
    clone.style.animation = 'none';
    clone.style.transition = 'none';

    // Temporarily add to DOM for measurement
    clone.style.position = 'absolute';
    clone.style.left = '-9999px';
    clone.style.visibility = 'hidden';
    clone.style.pointerEvents = 'none';
    clone.style.width = element.scrollWidth + 'px';
    document.body.appendChild(clone);

    // Wait for any images or charts to load
    await new Promise(resolve => setTimeout(resolve, 500));

    const canvas = await html2canvas(clone, {
      scale: 2, // Réduit à 2 pour équilibrer qualité et vitesse
      backgroundColor: '#ffffff',
      logging: true, // Activé pour déboguer
      useCORS: true,
      allowTaint: true,
      foreignObjectRendering: true,
      removeContainer: false,
      imageTimeout: 15000,
      // Amélioration de la qualité de rendu
      width: clone.scrollWidth,
      height: clone.scrollHeight,
      windowWidth: clone.scrollWidth,
      windowHeight: clone.scrollHeight,
      onclone: (clonedDoc) => {
        const style = clonedDoc.createElement('style');
        style.textContent = `
          * {
            animation: none !important;
            transition: none !important;
            -webkit-font-smoothing: antialiased;
            -moz-osx-font-smoothing: grayscale;
          }
          /* Améliorer le rendu des polices */
          body {
            font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;
          }
          /* Ensure charts and canvases are visible */
          canvas, svg {
            display: block !important;
            visibility: visible !important;
          }
          /* Fix potential display issues */
          .recharts-wrapper, .recharts-surface {
            display: block !important;
          }
        `;
        clonedDoc.head.appendChild(style);

        // Force render all canvas elements
        const canvases = clonedDoc.querySelectorAll('canvas');
        canvases.forEach((canvas) => {
          canvas.style.display = 'block';
          canvas.style.visibility = 'visible';
        });
      }
    });

    // Clean up
    document.body.removeChild(clone);

    console.log('Canvas created:', {
      width: canvas.width,
      height: canvas.height
    });

    const dataUrl = canvas.toDataURL('image/png', 1.0);
    console.log('DataURL created, length:', dataUrl.length);

    return dataUrl;
  } catch (error) {
    console.error('Error capturing element:', error);
    console.error('Error stack:', error);
    return null;
  }
}

// New function to generate PDF from full HTML content
export async function generateAdvancedPDF(options: AdvancedPDFOptions): Promise<void> {
  const { title, contentElement, includeAllVisuals = true } = options;

  console.log('generateAdvancedPDF: Starting PDF generation');
  console.log('Title:', title);
  console.log('Content element:', contentElement);
  console.log('Include visuals:', includeAllVisuals);

  // Create new PDF document
  const pdf = new jsPDF({
    orientation: 'portrait',
    unit: 'mm',
    format: 'a4',
  });

  const pageWidth = pdf.internal.pageSize.getWidth();
  const pageHeight = pdf.internal.pageSize.getHeight();
  const margin = 15;
  const contentWidth = pageWidth - (margin * 2);

  console.log('PDF page dimensions:', { pageWidth, pageHeight, margin, contentWidth });

  // Add title page first
  console.log('Adding title page...');
  addTitlePage(pdf, title);

  let yPosition = margin;

  // Add VQuant header with logo - amélioré
  // Dégradé simulé pour l'en-tête
  pdf.setFillColor(139, 92, 246);
  pdf.rect(0, 0, pageWidth, 40, 'F');
  pdf.setFillColor(119, 72, 226, 0.9);
  pdf.rect(0, 38, pageWidth, 2, 'F');

  // Draw small logo in header
  drawLogo(pdf, 20, 20, 9);

  pdf.setTextColor(255, 255, 255);
  pdf.setFontSize(22);
  pdf.setFont('helvetica', 'bold');
  pdf.text('VQuant', 34, 18);

  pdf.setFontSize(10);
  pdf.setFont('helvetica', 'normal');
  pdf.setTextColor(240, 240, 255);
  pdf.text('Financial Intelligence Platform', 34, 27);

  // Add date in header - style amélioré
  pdf.setFontSize(8);
  pdf.setFont('helvetica', 'italic');
  pdf.setTextColor(240, 240, 255);
  const date = new Date().toLocaleDateString('fr-FR', {
    year: 'numeric',
    month: 'long',
    day: 'numeric',
  });
  pdf.text(date, pageWidth - margin - 40, 22);

  yPosition = 52;

  // Add title avec meilleur style
  pdf.setTextColor(15, 23, 42);
  pdf.setFontSize(18);
  pdf.setFont('helvetica', 'bold');
  const titleLines = pdf.splitTextToSize(title, contentWidth);
  pdf.text(titleLines, margin, yPosition);
  yPosition += (titleLines.length * 8) + 10;

  // Add horizontal line - plus élégante
  pdf.setDrawColor(139, 92, 246);
  pdf.setLineWidth(0.8);
  pdf.line(margin, yPosition, pageWidth - margin, yPosition);
  yPosition += 12;

  try {
    // Capture the entire content element directly
    console.log('Starting PDF capture...');
    const imgData = await captureElementAsImage(contentElement);

    if (!imgData) {
      console.error('Failed to capture content - imgData is null');
      throw new Error('Failed to capture content');
    }

    console.log('Content captured successfully');
    const img = new Image();
    img.src = imgData;
    await new Promise((resolve) => { img.onload = resolve; });

    const imgWidth = contentWidth;
    const imgHeight = (img.height * imgWidth) / img.width;

    console.log(`Image dimensions: ${imgWidth}x${imgHeight}`);

    // Split into multiple pages if needed
    const pageContentHeight = pageHeight - yPosition - 15;
    let currentHeight = 0;

    while (currentHeight < imgHeight) {
      const heightToAdd = Math.min(imgHeight - currentHeight, pageContentHeight);

      if (currentHeight > 0) {
        pdf.addPage();
        yPosition = margin;
      }

      pdf.addImage(
        imgData,
        'PNG',
        margin,
        yPosition - currentHeight,
        imgWidth,
        imgHeight,
        undefined,
        'FAST'
      );

      currentHeight += pageContentHeight;

      if (currentHeight < imgHeight) {
        yPosition = margin;
      }
    }
  } catch (error) {
    console.error('Error capturing content:', error);

    // Fallback: Add error message with details
    pdf.setFontSize(11);
    pdf.setTextColor(255, 0, 0);
    pdf.text('Error: Could not capture content for PDF', margin, yPosition);
    yPosition += 10;

    pdf.setFontSize(9);
    pdf.setTextColor(100, 100, 100);
    const errorMsg = error instanceof Error ? error.message : 'Unknown error';
    const errorLines = pdf.splitTextToSize(`Details: ${errorMsg}`, contentWidth);
    pdf.text(errorLines, margin, yPosition);

    throw error; // Re-throw to be caught by caller
  }

  // Add footer on each page - amélioré
  const pageCount = pdf.internal.pages.length - 1;
  for (let i = 1; i <= pageCount; i++) {
    pdf.setPage(i);

    // Footer line avec dégradé
    pdf.setDrawColor(203, 213, 225);
    pdf.setLineWidth(0.5);
    pdf.line(margin, pageHeight - 15, pageWidth - margin, pageHeight - 15);

    // Footer dots décoratifs
    pdf.setFillColor(139, 92, 246);
    pdf.circle(margin + 2, pageHeight - 15, 0.8, 'F');
    pdf.circle(pageWidth - margin - 2, pageHeight - 15, 0.8, 'F');

    // Footer text - meilleur style
    pdf.setFontSize(7.5);
    pdf.setFont('helvetica', 'normal');
    pdf.setTextColor(100, 116, 139);
    pdf.text('Généré par VQuant • www.vibequant.com', margin, pageHeight - 9);

    pdf.setFont('helvetica', 'bold');
    pdf.setTextColor(99, 102, 241);
    pdf.text(`Page ${i} / ${pageCount}`, pageWidth - margin - 18, pageHeight - 9);
  }

  // Save PDF
  const filename = `VQuant_Report_${new Date().toISOString().split('T')[0]}.pdf`;
  pdf.save(filename);
}

export async function generatePDF(options: PDFOptions): Promise<void> {
  const { title, content, includeCharts = true } = options;

  // Create new PDF document
  const pdf = new jsPDF({
    orientation: 'portrait',
    unit: 'mm',
    format: 'a4',
  });

  const pageWidth = pdf.internal.pageSize.getWidth();
  const pageHeight = pdf.internal.pageSize.getHeight();
  const margin = 20;
  const contentWidth = pageWidth - (margin * 2);

  // Add title page first
  addTitlePage(pdf, title);

  let yPosition = margin;

  // Add VQuant header with logo
  pdf.setFillColor(59, 130, 246); // Blue
  pdf.rect(0, 0, pageWidth, 30, 'F');

  // Draw small logo in header
  drawLogo(pdf, 25, 15, 7);

  pdf.setTextColor(255, 255, 255);
  pdf.setFontSize(24);
  pdf.setFont('helvetica', 'bold');
  pdf.text('VQuant', 35, 15);

  pdf.setFontSize(12);
  pdf.setFont('helvetica', 'normal');
  pdf.text('Financial Intelligence Platform', 35, 23);

  yPosition = 40;

  // Add title
  pdf.setTextColor(0, 0, 0);
  pdf.setFontSize(18);
  pdf.setFont('helvetica', 'bold');
  const titleLines = pdf.splitTextToSize(title, contentWidth);
  pdf.text(titleLines, margin, yPosition);
  yPosition += (titleLines.length * 8) + 10;

  // Add date
  pdf.setFontSize(10);
  pdf.setFont('helvetica', 'normal');
  pdf.setTextColor(100, 100, 100);
  const date = new Date().toLocaleDateString('en-US', {
    year: 'numeric',
    month: 'long',
    day: 'numeric',
  });
  pdf.text(`Generated on ${date}`, margin, yPosition);
  yPosition += 15;

  // Add horizontal line
  pdf.setDrawColor(200, 200, 200);
  pdf.line(margin, yPosition, pageWidth - margin, yPosition);
  yPosition += 10;

  // Process and add content
  pdf.setFontSize(11);
  pdf.setTextColor(0, 0, 0);
  pdf.setFont('helvetica', 'normal');

  // Split content into sections and paragraphs
  const sections = content.split(/\n\n+/);

  for (const section of sections) {
    if (!section.trim()) continue;

    // Check if section is a heading (starts with #)
    if (section.trim().startsWith('#')) {
      const heading = section.trim().replace(/^#+\s*/, '');

      // Check if new page needed
      if (yPosition > pageHeight - 30) {
        pdf.addPage();
        yPosition = margin;
      }

      pdf.setFontSize(14);
      pdf.setFont('helvetica', 'bold');
      pdf.setTextColor(59, 130, 246);
      const headingLines = pdf.splitTextToSize(heading, contentWidth);
      pdf.text(headingLines, margin, yPosition);
      yPosition += (headingLines.length * 7) + 5;

      pdf.setFontSize(11);
      pdf.setFont('helvetica', 'normal');
      pdf.setTextColor(0, 0, 0);
      continue;
    }

    // Regular paragraph
    const lines = pdf.splitTextToSize(section.trim(), contentWidth);

    for (const line of lines) {
      // Check if new page needed
      if (yPosition > pageHeight - 30) {
        pdf.addPage();
        yPosition = margin;
      }

      pdf.text(line, margin, yPosition);
      yPosition += 6;
    }

    yPosition += 4; // Add space between paragraphs
  }

  // Capture charts if requested
  if (includeCharts) {
    try {
      // Find all chart containers
      const chartElements = document.querySelectorAll('[data-chart]');

      for (let i = 0; i < chartElements.length; i++) {
        const element = chartElements[i] as HTMLElement;

        // Add new page for chart
        pdf.addPage();
        yPosition = margin;

        // Add chart title if available
        const chartTitle = element.getAttribute('data-chart-title');
        if (chartTitle) {
          pdf.setFontSize(14);
          pdf.setFont('helvetica', 'bold');
          pdf.text(chartTitle, margin, yPosition);
          yPosition += 10;
        }

        // Capture chart as image
        const canvas = await html2canvas(element, {
          scale: 2,
          backgroundColor: '#ffffff',
          logging: false,
        });

        const imgData = canvas.toDataURL('image/png');
        const imgWidth = contentWidth;
        const imgHeight = (canvas.height * imgWidth) / canvas.width;

        // Check if image fits on current page
        if (yPosition + imgHeight > pageHeight - margin) {
          pdf.addPage();
          yPosition = margin;
        }

        pdf.addImage(imgData, 'PNG', margin, yPosition, imgWidth, imgHeight);
        yPosition += imgHeight + 10;
      }
    } catch (error) {
      console.error('Error capturing charts:', error);
    }
  }

  // Add footer on each page
  const pageCount = pdf.internal.pages.length - 1; // Subtract 1 for the internal page array
  for (let i = 1; i <= pageCount; i++) {
    pdf.setPage(i);

    // Footer line
    pdf.setDrawColor(200, 200, 200);
    pdf.line(margin, pageHeight - 15, pageWidth - margin, pageHeight - 15);

    // Footer text
    pdf.setFontSize(9);
    pdf.setTextColor(100, 100, 100);
    pdf.setFont('helvetica', 'normal');
    pdf.text('Generated by VQuant', margin, pageHeight - 10);
    pdf.text(`Page ${i} of ${pageCount}`, pageWidth - margin - 20, pageHeight - 10);
  }

  // Save PDF
  const filename = `VQuant_Analysis_${new Date().toISOString().split('T')[0]}.pdf`;
  pdf.save(filename);
}

// Helper function to clean markdown formatting for PDF
export function cleanMarkdownForPDF(markdown: string): string {
  return markdown
    // Remove code blocks
    .replace(/```[\s\S]*?```/g, '')
    // Remove inline code
    .replace(/`([^`]+)`/g, '$1')
    // Remove bold/italic
    .replace(/\*\*([^*]+)\*\*/g, '$1')
    .replace(/\*([^*]+)\*/g, '$1')
    // Remove links but keep text
    .replace(/\[([^\]]+)\]\([^)]+\)/g, '$1')
    // Clean up extra whitespace
    .replace(/\n{3,}/g, '\n\n')
    .trim();
}