使用Node.js自动生成带动态图表的Word文档

mkdir generate-word-docx
cd generate-word-docx
npm init -y

npm install http fs path pizzip docxtemplater open-docxtemplater-image-module canvas echarts

const http = require('http');
const fs = require('fs');
const path = require('path');
const PizZip = require('pizzip');
const Docxtemplater = require('docxtemplater');
const ImageModule = require('open-docxtemplater-image-module');
const { createCanvas } = require('canvas');
const echarts = require('echarts');

// 定义替换数据
const data = {
  name1: 'John Doe',
  name2: 'Jane Smith',
  // 在这里添加柱状图数据
  image: './chart.png' // 替换为你的图片 URL 地址
};

// 定义模板文件路径
const templatePath = 'template.docx';

// 创建 HTTP 服务器
const server = http.createServer((req, res) => {
  // 填充 Word 文档模板并生成新的 Word 文档
  generateWord(templatePath, data, (err, outputPath) => {
    if (err) {
      res.writeHead(500, { 'Content-Type': 'text/plain' });
      res.end('Internal Server Error');
      return;
    }

    // 将生成的 Word 文档作为响应发送给客户端
    const fileStream = fs.createReadStream(outputPath);
    res.setHeader('Content-disposition', 'attachment; filename=output.docx');
    res.setHeader('Content-type', 'application/vnd.openxmlformats-officedocument.wordprocessingml.document');
    fileStream.pipe(res);
  });
});

// 监听端口
const port = 3000;
server.listen(port, () => {
  console.log(`Server is running on port ${port}`);
});

// 填充 Word 文档模板并生成新的 Word 文档
function generateWord(inputPath, data, callback) {
  try {
    // 读取模板文件内容
    const content = fs.readFileSync(inputPath, 'binary');

    // 初始化 PizZip
    const zip = new PizZip(content);

    // 初始化 Docxtemplater
    const doc = new Docxtemplater();

    // 图片替换配置
    const opts = {
      centered: false,
      getImage: function (tagValue, tagName) {
        // 直接返回图片地址
        return fs.readFileSync(path.join(__dirname, tagValue));
      },
      getSize: function (img, tagValue, tagName) {
        // 设置图片大小，这里使用默认大小 [250, 250]
        return [250, 250];
      }
    };

    // 添加图片模块
    doc.attachModule(new ImageModule(opts));

    // 加载模板文件
    doc.loadZip(zip);

    // 设置替换数据
    doc.setData(data);

    // 渲染数据
    doc.render();

    // 生成新的 Word 文档
    const outputPath = 'output.docx';
    const buf = doc.getZip().generate({ type: 'nodebuffer' });

    // 写入新的 Word 文档
    fs.writeFileSync(outputPath, buf);

    // 调用回调函数，传递生成的 Word 文档路径
    callback(null, outputPath);
  } catch (error) {
    // 如果发生错误，则调用回调函数，并传递错误对象
    callback(error);
  }
}

// 使用 ECharts 生成柱状图图片
function generateChartImage() {
  // 创建画布
  const canvas = createCanvas(800, 600);
  const ctx = canvas.getContext('2d');

  // 将 ECharts 图表绘制到画布上
  echarts.setCanvasCreator(() => canvas);
  const chart = echarts.init(canvas, null, { renderer: 'canvas' });
  chart.setOption({
    title: {
      text: '柱状图示例'
    },
    tooltip: {},
    xAxis: {
      data: ['A', 'B', 'C', 'D', 'E']
    },
    yAxis: {},
    series: [{
      name: '销量',
      type: 'bar',
      data: [5, 20, 36, 10, 10]
    }]
  });

  // 将画布转换为 PNG 图片并保存到文件
  const buffer = canvas.toBuffer('image/png');
  fs.writeFileSync('chart.png', buffer);
}

// 调用函数生成柱状图图片
generateChartImage();

bashCopy code
node your-script-name.js