f

internal/shared/pdf/html_pdf_generator.go

@@ -10,6 +10,12 @@ import (
 	"go.uber.org/zap"
 )
 
+// A4 纵向 @ 96dpi，与 CSS @page / 报告页 max-width 对齐
+const (
+	htmlPDFViewportWidth  = 794
+	htmlPDFViewportHeight = 1123
+)
+
 // HTMLPDFGenerator 使用 headless Chrome 将 HTML 页面渲染为 PDF
 type HTMLPDFGenerator struct {
 	logger *zap.Logger
@@ -26,41 +32,54 @@ func NewHTMLPDFGenerator(logger *zap.Logger) *HTMLPDFGenerator {
 }
 
 // GenerateFromURL 使用 headless Chrome 打开指定 URL，并导出为 PDF 字节流
-// 这里固定使用 A4 纵向纸张，开启背景打印
+// 流程：A4 视口渲染 → 等待报告 DOM → 滚动触发展开 → 切换 print 媒体 → PrintToPDF（边距由 CSS @page 控制）
 func (g *HTMLPDFGenerator) GenerateFromURL(ctx context.Context, url string) ([]byte, error) {
 	if ctx == nil {
 		ctx = context.Background()
 	}
 
-	// 整个生成过程增加超时时间，避免长时间卡死
-	timeoutCtx, cancel := context.WithTimeout(ctx, 60*time.Second)
+	timeoutCtx, cancel := context.WithTimeout(ctx, 90*time.Second)
 	defer cancel()
 
-	// 创建 Chrome 上下文（使用系统默认的 headless Chrome/Chromium）
-	chromeCtx, cancelChrome := chromedp.NewContext(timeoutCtx)
+	opts := append(chromedp.DefaultExecAllocatorOptions[:],
+		chromedp.Flag("headless", true),
+		chromedp.Flag("disable-gpu", true),
+		chromedp.Flag("no-sandbox", true),
+		chromedp.Flag("disable-dev-shm-usage", true),
+	)
+	allocCtx, allocCancel := chromedp.NewExecAllocator(timeoutCtx, opts...)
+	defer allocCancel()
+
+	chromeCtx, cancelChrome := chromedp.NewContext(allocCtx)
 	defer cancelChrome()
 
 	var pdfBuf []byte
 
 	tasks := chromedp.Tasks{
+		chromedp.EmulateViewport(htmlPDFViewportWidth, htmlPDFViewportHeight),
 		chromedp.Navigate(url),
-		// 等待页面主体和报告容器就绪，确保数据渲染完成
 		chromedp.WaitReady("body", chromedp.ByQuery),
 		chromedp.WaitVisible(".page", chromedp.ByQuery),
+		waitReportReadyForPDF(),
+		waitReportPosterImages(),
+		scrollPageForPrintLayout(),
+		chromedp.ActionFunc(func(ctx context.Context) error {
+			return chromedp.Evaluate(`document.body.classList.add('pdf-export')`, nil).Do(ctx)
+		}),
+		chromedp.ActionFunc(func(ctx context.Context) error {
+			time.Sleep(150 * time.Millisecond)
+			return nil
+		}),
 		chromedp.ActionFunc(func(ctx context.Context) error {
 			g.logger.Info("开始通过 headless Chrome 生成企业报告 PDF", zap.String("url", url))
-			var (
-				buf []byte
-				err error
-			)
-			buf, _, err = page.PrintToPDF().
+			buf, _, err := page.PrintToPDF().
 				WithPrintBackground(true).
-				WithPaperWidth(8.27).  // A4 宽度（英寸 -> 约 210mm）
-				WithPaperHeight(11.69). // A4 高度（英寸 -> 约 297mm）
-				WithMarginTop(0.4).
-				WithMarginBottom(0.4).
-				WithMarginLeft(0.4).
-				WithMarginRight(0.4).
+				WithPaperWidth(8.27).
+				WithPaperHeight(11.69).
+				WithMarginTop(0.39).
+				WithMarginBottom(0.39).
+				WithMarginLeft(0.47).
+				WithMarginRight(0.47).
 				Do(ctx)
 			if err == nil {
 				pdfBuf = buf
@@ -86,3 +105,87 @@ func (g *HTMLPDFGenerator) GenerateFromURL(ctx context.Context, url string) ([]b
 	return pdfBuf, nil
 }
 
+// waitReportReadyForPDF 等待企业报告脚本渲染完成（qiye.html 设置 data-qygl-report-ready）
+func waitReportReadyForPDF() chromedp.Action {
+	return chromedp.ActionFunc(func(ctx context.Context) error {
+		deadline := time.Now().Add(45 * time.Second)
+		for {
+			if ctx.Err() != nil {
+				return ctx.Err()
+			}
+			var ready string
+			_ = chromedp.Evaluate(`document.body && document.body.dataset.qyglReportReady || ''`, &ready).Do(ctx)
+			if ready == "1" {
+				return nil
+			}
+			var sectionCount float64
+			_ = chromedp.Evaluate(`document.querySelectorAll('#reportSections .section-card, .report-sections .section-block').length`, &sectionCount).Do(ctx)
+			if sectionCount > 0 {
+				return nil
+			}
+			if time.Now().After(deadline) {
+				return fmt.Errorf("等待报告内容渲染超时")
+			}
+			time.Sleep(200 * time.Millisecond)
+		}
+	})
+}
+
+func waitReportPosterImages() chromedp.Action {
+	return chromedp.ActionFunc(func(ctx context.Context) error {
+		deadline := time.Now().Add(20 * time.Second)
+		for {
+			if ctx.Err() != nil {
+				return ctx.Err()
+			}
+			var pending float64
+			js := `(function(){
+				var imgs = document.querySelectorAll('img.header-poster, img.footer-poster');
+				if (!imgs.length) return 0;
+				for (var i = 0; i < imgs.length; i++) {
+					var img = imgs[i];
+					if (!img.complete || img.naturalWidth === 0) return 1;
+				}
+				return 0;
+			})()`
+			if err := chromedp.Evaluate(js, &pending).Do(ctx); err != nil {
+				return err
+			}
+			if pending == 0 {
+				return nil
+			}
+			if time.Now().After(deadline) {
+				return fmt.Errorf("等待报告海报图片加载超时")
+			}
+			time.Sleep(150 * time.Millisecond)
+		}
+	})
+}
+
+func scrollPageForPrintLayout() chromedp.Action {
+	return chromedp.ActionFunc(func(ctx context.Context) error {
+		var height float64
+		if err := chromedp.Evaluate(`Math.max(document.body.scrollHeight, document.documentElement.scrollHeight)`, &height).Do(ctx); err != nil {
+			return err
+		}
+		step := float64(900)
+		if height < step {
+			step = height
+		}
+		if step <= 0 {
+			return nil
+		}
+		for y := float64(0); y <= height; y += step {
+			js := fmt.Sprintf(`window.scrollTo(0, %f);`, y)
+			if err := chromedp.Evaluate(js, nil).Do(ctx); err != nil {
+				return err
+			}
+			time.Sleep(120 * time.Millisecond)
+		}
+		if err := chromedp.Evaluate(`window.scrollTo(0, 0);`, nil).Do(ctx); err != nil {
+			return err
+		}
+		time.Sleep(300 * time.Millisecond)
+		return nil
+	})
+}