package com.aikes.util;

import lombok.extern.slf4j.Slf4j;
import org.openqa.selenium.By;
import org.openqa.selenium.Pdf;
import org.openqa.selenium.PrintsPage;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;
import org.openqa.selenium.print.PrintOptions;
import org.openqa.selenium.support.ui.ExpectedConditions;
import org.openqa.selenium.support.ui.WebDriverWait;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.io.File;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.time.Duration;
import java.util.Base64;
import java.util.Collections;

/**
 * 調(diào)用vue頁面獲取PDF文件
 */
@Slf4j
@Service
public class WebAccessServerUtil {

    @Value("${dllUrl:D:/DLL}")
    private String dllUrl;

    @Value("${webAddress:http://127.0.0.1:8080/mcpc/ui/dist/index.html}")
    private String mWebAddress;

    @Value("${chromePath:C:\\Program Files\\Google\\Chrome\\Application\\chrome.exe}")
    private String chromePath;

    public boolean getPdfFromWeb(String cPath,String cpk_notification,String cpk_emgency){

        log.info("根目錄：" + dllUrl);
        log.info("Web端訪問地址：" + mWebAddress);
        log.info("谷歌瀏覽器路徑：" + chromePath);

        // 完全禁用WebDriverManager自動下載，手動指定驅(qū)動路徑（避免版本沖突）
        System.setProperty("wdm.enabled", "false");
        // 手動指定本地ChromeDriver路徑（替換為你的實(shí)際路徑）
        //驅(qū)動版本需要和服務(wù)器瀏覽器版本對應(yīng)   138.0.7204.158
        System.setProperty("webdriver.chrome.driver", dllUrl + File.separator + "chromedriver" + File.separator + "chromedriver.exe");
        System.setProperty("webdriver.chrome.bin", chromePath);

        // 配置瀏覽器選項(xiàng)（重點(diǎn)修復(fù)）
        ChromeOptions options = new ChromeOptions();
        options.addArguments("--headless=new"); // 新版headless模式
        options.addArguments("--disable-gpu");
        options.addArguments("--no-sandbox");
        options.addArguments("--disable-dev-shm-usage");
        options.addArguments("--remote-allow-origins=*"); // 允許跨域WebSocket
        options.setExperimentalOption("excludeSwitches", Collections.singletonList("enable-automation"));

        WebDriver driver = new ChromeDriver(options);
        WebDriverWait wait = new WebDriverWait(driver, Duration.ofSeconds(15)); // 最大等待10秒

        try {

            String vuePageUrl = mWebAddress + "#/generatePdfTest?pk_notification=" + cpk_notification + "&pk_emgency=" + cpk_emgency;
            driver.get(vuePageUrl);

            // 智能等待：直到頁面設(shè)置true
            wait.until(ExpectedConditions.attributeToBe(
                    By.tagName("body"), "data-pdf-ready", "true"
            ));

            PrintOptions printOptions = new PrintOptions();
            printOptions.setBackground(true);
            printOptions.setPageRanges("1-5");

            // 替換原PDF字節(jié)處理代碼
            Pdf pdf = ((PrintsPage) driver).print(printOptions);
            String base64Pdf = pdf.getContent(); // 獲取Base64編碼的PDF內(nèi)容
            byte[] pdfBytes = Base64.getDecoder().decode(base64Pdf); // 解碼為原始二進(jìn)制

            Files.write(Paths.get(cPath), pdfBytes);
            log.info("調(diào)用Web端生成PDF文件成功: " + cPath);
            return true;
        } catch (Exception e) {
            log.error("獲取Web端PDF文件失敗:{}", e.getMessage(), e);
            return false;
        } finally {
            try {
                driver.quit();
            } catch (Exception ignore) {
            }
        }
    }
}

    {
      path: '/generatePdfTest',
      name: 'GeneratePDFTest',
      component: (resolve) =>
        require(['@/components/notificationRecord/MedicalNotificationPdf'], resolve),
    }

<template>
  <div class="pdf-container">
    <h1>{{ title }}</h1>
    <div v-for="item in items" :key="item.id" class="item">
      <p><strong>{{ item.label }}:</strong> {{ item.value }}</p>
    </div>
  </div>
</template>

<script>
  export default {
    data() {
      return {
        title: '測試通知記錄書',
        items: []
      }
    },
    mounted() {
      try {
        // 檢查路由是否可用（避免this.$route未定義的錯誤）
        if (!this.$route) {
          throw new Error("Vue Router未正確配置");
        }
        const recordId = this.$route.query.id || 'default';

        // 模擬數(shù)據(jù)加載
        setTimeout(() => {
          this.items = [
            { id: 1, label: '患者姓名', value: '張三' },
            { id: 2, label: '病歷號', value: recordId },
            { id: 3, label: '診斷結(jié)果', value: 'stemi' },
            { id: 4, label: '診斷日期', value: '2025-08-12' },
            { id: 5, label: '診斷醫(yī)生', value: '李四' }
          ];

          // 確認(rèn)body元素存在后再設(shè)置屬性
          if (document.body) {
            document.body.setAttribute('data-pdf-ready', 'true');
            console.log("PDF渲染標(biāo)記已設(shè)置"); // 用于瀏覽器控制臺調(diào)試
          } else {
            console.error("body元素不存在，無法設(shè)置標(biāo)記");
          }
        }, 500);
      } catch (error) {
        console.error("組件加載錯誤：", error);
      }
    }
  }
</script>

<style scoped>
  /* 樣式保持不變 */
  .pdf-container {
    width: 210mm;
    margin: 0 auto;
    padding: 20px;
    font-family: "SimSun";
  }
  .item {
    margin: 10px 0;
  }
</style>