performance-test/downLoadServer/scripts/prepareReportTemplates.mjs

/**
 * 将健康/异常报告模板写入 templates/
 * 用法：node scripts/prepareReportTemplates.mjs
 */
import path from "path";
import fs from "fs";
import { fileURLToPath } from "url";
import PizZip from "pizzip";
import {
  HEALTH_IMAGE_MARKERS,
  patchAnomalyTemplateXml,
  patchTemplateXml,
} from "../src/server/reportService/docxReportBuilder.js";

const __dirname = path.dirname(fileURLToPath(import.meta.url));
const root = path.join(__dirname, "..");
const templatesDir = path.join(root, "templates");
const publicHealthV2 = path.join(
  root,
  "src",
  "public",
  "file",
  "风电机组健康评估_风场级自动报告模板_V2.0.docx",
);
const publicHealthV1 = path.join(
  root,
  "..",
  "public",
  "files",
  "风电机组健康评估_风场级自动报告模板_V1.0.docx",
);
const publicHealth = fs.existsSync(publicHealthV2)
  ? publicHealthV2
  : publicHealthV1;
const publicAnomaly = path.join(
  root,
  "src",
  "public",
  "file",
  "异常检测数据分析报告模板（大唐版）.docx",
);

function writeTemplate(zip, targetName) {
  fs.mkdirSync(templatesDir, { recursive: true });
  const outPath = path.join(templatesDir, targetName);
  if (fs.existsSync(outPath)) {
    try {
      fs.chmodSync(outPath, 0o644);
    } catch (_e) {
      // ignore
    }
    fs.unlinkSync(outPath);
  }
  fs.writeFileSync(outPath, zip.generate({ type: "nodebuffer" }), {
    mode: 0o644,
  });
  return outPath;
}

function patchAndWrite(sourcePath, targetName, patchFn) {
  if (!fs.existsSync(sourcePath)) {
    throw new Error(`源模板不存在: ${sourcePath}`);
  }
  const content = fs.readFileSync(sourcePath, "binary");
  const zip = new PizZip(content);
  const documentFile = zip.file("word/document.xml");
  if (!documentFile) throw new Error("模板缺少 word/document.xml");
  zip.file("word/document.xml", patchFn(documentFile.asText()));
  return writeTemplate(zip, targetName);
}

fs.mkdirSync(templatesDir, { recursive: true });

if (fs.existsSync(publicHealth)) {
  const healthOut = patchAndWrite(
    publicHealth,
    "health-report-template.docx",
    (xml) => patchTemplateXml(xml, HEALTH_IMAGE_MARKERS),
  );
  console.log("健康模板已写入并插入图片占位:", healthOut);
} else {
  console.warn("未找到健康模板，跳过:", publicHealth);
}

if (fs.existsSync(publicAnomaly)) {
  const anomalyOut = patchAndWrite(
    publicAnomaly,
    "anomaly-report-template.docx",
    (xml) => patchAnomalyTemplateXml(xml),
  );
  console.log("异常模板已写入并补齐 image 占位:", anomalyOut);
} else {
  console.warn("未找到异常模板，跳过:", publicAnomaly);
}

console.log("完成。重启 downLoadServer 后即可生成报告。");