//引入工具
|
import PizZip from 'pizzip';
|
import Docxtemplater from 'docxtemplater';
|
import JSZipUtils from 'jszip-utils';
|
import { saveAs } from 'file-saver';
|
import ImageModule from 'docxtemplater-image-module-free'; // 新增：图片处理模
|
|
// 加载 .docx 模板文件
|
function loadFile(url, callback) {
|
JSZipUtils.getBinaryContent(url, callback);
|
}
|
|
// 下载生成的文档
|
export function download(file, name) {
|
|
}
|
// 辅助函数：将 base64 图片转为 Uint8Array（图片模块需要此格式）
|
function base64ToUint8Array(base64) {
|
// 去掉 base64 前缀（如 "data:image/png;base64,"）
|
const base64WithoutPrefix = base64.replace(/^data:image\/\w+;base64,/, '');
|
const binaryString = atob(base64WithoutPrefix);
|
const length = binaryString.length;
|
const uint8Array = new Uint8Array(length);
|
for (let i = 0; i < length; i++) {
|
uint8Array[i] = binaryString.charCodeAt(i);
|
}
|
return uint8Array;
|
}
|
|
|
// 生成并下载 Word 文档(templatePath是word文档模版地址,data是对应的数据)
|
export function generateWordDocument(templatePath, data, name) {
|
loadFile(templatePath, function (error, content) {
|
if (error) {
|
throw error
|
return;
|
}
|
|
try {
|
// 新版创建解析器的正确方式（无需 import AngularParser）
|
const parser = (tag, _variable) => {
|
return {
|
get(scope) {
|
if (scope[tag] !== undefined) {
|
return scope[tag] || '';
|
}
|
// 处理特殊变量（如 $first）
|
if (tag.startsWith('$')) {
|
const varName = tag.slice(1);
|
return scope[varName] || '';
|
}
|
return '';
|
}
|
};
|
};
|
// 1. 配置图片处理规则（核心）
|
const imageModule = new ImageModule({
|
// 获取图片：根据传入的图片数据（base64）转为模块需要的格式
|
getImage: function (tagValue, tagName) {
|
// tagValue：data 中对应图片键的值（必须是 base64 字符串）
|
// tagName：图片占位符的键名（如 "avatar"）
|
return base64ToUint8Array(tagValue);
|
},
|
// 设置图片尺寸：返回 [宽度, 高度]（单位：px），可动态调整
|
getSize: function (tagValue, tagName) {
|
|
// 示例：根据不同的图片键，返回不同尺寸
|
switch (tagName) {
|
case 'avatar': // 头像：200x200
|
return [200, 200];
|
case 'coverImg': // 封面图：600x300
|
return [600, 300];
|
default: // 默认尺寸：400x300
|
return [80, 100];
|
}
|
}
|
});
|
// 加载模板文件内容到 PizZip
|
const zip = new PizZip(content);
|
const doc = new Docxtemplater(zip, {
|
paragraphLoop: true,
|
linebreaks: true,
|
modules: [imageModule] // 关键：注入图片模块
|
});
|
|
// const parser = new AngularParser();
|
// doc.setOptions({ parser });
|
|
|
// 设置模板中的占位符数据
|
doc.setData(data);
|
|
// 渲染文档
|
doc.render();
|
|
// 生成最终的文档 Blob
|
const fileWord = doc.getZip().generate({
|
type: 'blob',
|
mimeType: 'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
|
});
|
|
saveAs(fileWord, name);
|
|
// // 返回生成的文档 Blob
|
// resolve(fileWord);
|
} catch (error) {
|
console.error('Error rendering document:', error);
|
throw error
|
}
|
});
|
|
}
|
