index.js 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381
  1. const fs = require('fs')
  2. const path = require('path')
  3. const compressing = require('compressing')
  4. const rimrif = require('rimraf')
  5. const shell = require('shelljs');
  6. const crypto = require('crypto');
  7. // 文档模板生成
  8. const PizZip = require("pizzip");
  9. const Docxtemplater = require("docxtemplater");
  10. // 文档转换
  11. import { Chromiumly } from "chromiumly";
  12. // Chromiumly.configure({ endpoint: "http://8.140.98.43/docs" });
  13. Chromiumly.configure({ endpoint: "http://123.57.204.89/docs" });
  14. const { LibreOffice } = require("chromiumly");
  15. // const { PDFEngines } = require("chromiumly");
  16. const tempDir = path.join(__dirname , "temp");
  17. if(!fs.existsSync(tempDir)){fs.mkdirSync(tempDir)};
  18. const OSS = require("ali-oss");
  19. const ALI_OSS_BUCKET = process.env.ALI_OSS_BUCKET || "hep-textbook"
  20. const ALI_OSS_ACCESS_KEY_ID = process.env.ALI_OSS_ACCESS_KEY_ID || "LTAI5t6AbTiAvXmeoVdJZhL3"
  21. const ALI_OSS_ACCESS_KEY_SECRET = process.env.ALI_OSS_ACCESS_KEY_SECRET || "KLtQRdIW69KLP7jnzHNUf7eKmdptxH"
  22. const bwipjs = require("bwip-js")
  23. export async function toBarCode(text){
  24. return new Promise(resolve=>{
  25. bwipjs.toBuffer({
  26. bcid:"code128",
  27. text:text,
  28. scale:1.5,
  29. height:3,
  30. includetext:false,
  31. textalign:"center"
  32. },(err,png)=>{
  33. if(err){
  34. console.error(err)
  35. resolve(null)
  36. }else{
  37. resolve(png)
  38. }
  39. })
  40. })
  41. }
  42. export async function uploadFileToOSS(filePath){
  43. let client = new OSS({
  44. // yourRegion填写Bucket所在地域。以华东1(杭州)为例,yourRegion填写为oss-cn-hangzhou。
  45. region: "oss-cn-beijing",
  46. accessKeyId: ALI_OSS_ACCESS_KEY_ID,
  47. accessKeySecret: ALI_OSS_ACCESS_KEY_SECRET,
  48. // 填写Bucket名称。
  49. bucket: ALI_OSS_BUCKET || "hep-textbook",
  50. });
  51. let now = new Date();
  52. let fileName = getFileName(filePath);
  53. let fileKey = `export/report/${fileName}`;
  54. const r1 = await client?.put(fileKey, filePath);
  55. console.log('put success: %j', r1);
  56. return r1
  57. }
  58. export function getFileName(filePath) {
  59. // 使用 '/' 或 '\' 作为分隔符,分割路径
  60. const parts = filePath.split(/[/\\]/);
  61. // 返回最后一个部分,即文件名
  62. return parts.pop();
  63. }
  64. module.exports.uploadFileToOSS = uploadFileToOSS
  65. /**
  66. * 将给定的文件路径数组打包成指定名称的zip压缩包
  67. * @param {Array<string>} filePathList - 要打包的文件路径数组
  68. * @param {string} outputZipName - 输出的zip文件名称
  69. */
  70. export function createZip(filePathList, outputZipName) {
  71. let zipStream = new compressing.zip.Stream();
  72. return new Promise((resolve)=>{
  73. try {
  74. let outputPath = path.join(tempDir,outputZipName)
  75. // 遍历文件路径列表,将每个文件添加到zip流中
  76. for (const filePath of filePathList) {
  77. // 检查文件是否存在
  78. if (fs.existsSync(filePath)) {
  79. // 将文件添加到zip流中
  80. zipStream.addEntry(filePath);
  81. } else {
  82. console.error(`文件不存在: ${filePath}`);
  83. }
  84. }
  85. // 创建一个写入流
  86. const output = fs.createWriteStream(outputPath);
  87. // 使用 compressing 库的 zip 方法将文件打包
  88. // console.log(filePathList)
  89. // await compressing.zip.compressDir(filePathList, output);
  90. // 将zip流写入文件
  91. zipStream.pipe(output);
  92. output.on('finish', () => {
  93. // console.log(`成功创建压缩包: ${outputPath}`);
  94. resolve(outputPath)
  95. });
  96. output.on('error', (error) => {
  97. console.error('写入压缩包时出错:', error);
  98. resolve(null)
  99. });
  100. // console.log(`成功创建压缩包: ${outputPath}`);
  101. // return outputPath
  102. } catch (error) {
  103. console.error('创建压缩包时出错:', error);
  104. return null
  105. }
  106. })
  107. }
  108. module.exports.createZip = createZip
  109. const download = require('download')
  110. async function downloadUrl(url) {
  111. console.log(url)
  112. if(url?.startsWith("/")) {return url};
  113. let md5 = crypto.createHash('md5');
  114. let filename = md5.update(url).digest('hex') + path.extname(url)
  115. let filepath = path.join(tempDir,filename)
  116. // console.log(filename,filepath)
  117. try{
  118. // if(fs.existsSync(filepath)){fs.rmSync(filepath)} // 存在则删除
  119. if(fs.existsSync(filepath)){return filepath} // 存在则直接返回(md5相同)
  120. fs.writeFileSync(filepath, await download(url));
  121. return filepath
  122. }catch(err){
  123. console.error(err)
  124. return null
  125. }
  126. }
  127. /**
  128. * 将 DOCX 文件转换为 PDF
  129. *
  130. * @param {string} docxPath - 要转换的 DOCX 文件的路径
  131. * @param {string} outputPath - 输出 PDF 文件的路径
  132. * @returns {Promise<void>}
  133. */
  134. export async function docxToPdf(docxPath, outputPath,options) {
  135. let mergeFiles = options?.mergeFiles || []
  136. let merge = false;
  137. let mergeFileMap = {};
  138. if(mergeFiles?.length){
  139. let plist = []
  140. for (let index = 0; index < mergeFiles.length; index++) {
  141. let filePath
  142. plist.push((async ()=>{
  143. try{
  144. filePath = await downloadUrl(mergeFiles[index]);
  145. }catch(err){}
  146. if(filePath){
  147. mergeFileMap[index] = filePath // 按原有顺序整理
  148. // filePathList.push(filePath)
  149. }
  150. return
  151. })())
  152. }
  153. await Promise.all(plist);
  154. merge = true;
  155. }
  156. let filePathList = mergeFiles?.map((item,index)=>mergeFileMap[index]).filter(item=>item)
  157. // console.log("DOWNLOADED:",filePathList)
  158. filePathList = filePathList.map((filepath,index)=>{
  159. // 按顺序修改文件前缀数字为字母表顺序
  160. let fileDir = path.dirname(filepath);
  161. let abc = String.fromCharCode(96+(index+1)); // 字母顺序不会出现 把 1 10 11 12 放在一起的情况
  162. let num = index+110; // 数字顺序从百位开始,避免首数字排序错乱
  163. let fileName = num + "_" + path.basename(filepath)
  164. let orderPath = path.join(fileDir,fileName)
  165. fs.cpSync(filepath,orderPath);
  166. fs.readFileSync(filepath);
  167. return orderPath
  168. })
  169. try {
  170. let files = []
  171. if(docxPath){
  172. let docxBuffer = fs.readFileSync(docxPath);
  173. files.push({ data: docxBuffer, ext: "docx" })
  174. }
  175. files = [...files,...filePathList]
  176. console.log("files",files)
  177. let pdfBuffer = await LibreOffice.convert({
  178. files,
  179. properties: {
  180. // 设置页面属性,例如纸张大小和方向
  181. pageSize: 'A4',
  182. orientation: 'portrait',
  183. margin: {
  184. top: 0,
  185. right: 0,
  186. bottom: 0,
  187. left: 0
  188. }
  189. },
  190. pdfa: false, // 根据需要设置
  191. pdfUA: false, // 根据需要设置
  192. merge: merge, // 如果只转换一个文件,设置为false
  193. // metadata: {
  194. // // 你可以在这里添加元数据
  195. // },
  196. // losslessImageCompression: false,
  197. // reduceImageResolution: false,
  198. // quality: 90, // JPG 导出质量
  199. // maxImageResolution: 300 // 最大图像分辨率
  200. });
  201. // 将 Buffer 写入输出文件
  202. fs.writeFileSync(outputPath, pdfBuffer);
  203. console.log(`成功输出 ${outputPath}`);
  204. return outputPath
  205. } catch (error) {
  206. console.error('转换失败:', error);
  207. return null
  208. }
  209. }
  210. module.exports.docxToPdf = docxToPdf
  211. const ImageModule = require("@slosarek/docxtemplater-image-module-free");
  212. const sizeOf = require("image-size");
  213. export function renderDocx(inputDocxPath, outputDocxName, options){
  214. let imageOptions = {
  215. getImage(tagValue,tagName) {
  216. if(!fs.existsSync(tagValue)){
  217. throw new Error(`Image not found: ${tagValue}`);
  218. }
  219. return fs.readFileSync(tagValue);
  220. },
  221. getSize(img) {
  222. const sizeObj = sizeOf(img);
  223. console.log(sizeObj);
  224. return [sizeObj.width, sizeObj.height];
  225. },
  226. };
  227. let outputDocxPath = path.join(tempDir,outputDocxName)
  228. // Load the docx file as binary content
  229. let content = fs.readFileSync(
  230. inputDocxPath,
  231. "binary"
  232. );
  233. // Unzip the content of the file
  234. let zip = new PizZip(content);
  235. let doc = new Docxtemplater(zip, {
  236. paragraphLoop: true,
  237. linebreaks: true,
  238. modules: [new ImageModule(imageOptions)],
  239. });
  240. // Render the document (Replace {first_name} by John, {last_name} by Doe, ...)
  241. Object.keys(options).forEach(key=>{ // 除去空值
  242. if(options[key]==undefined){
  243. options[key] = ""
  244. }
  245. })
  246. doc.render(options);
  247. // Get the zip document and generate it as a nodebuffer
  248. let buf = doc.getZip().generate({
  249. type: "nodebuffer",
  250. // compression: DEFLATE adds a compression step.
  251. // For a 50MB output document, expect 500ms additional CPU time
  252. compression: "DEFLATE",
  253. });
  254. // buf is a nodejs Buffer, you can either write it to a
  255. // file or res.send it with express for example.
  256. fs.writeFileSync(outputDocxPath, buf);
  257. return outputDocxPath
  258. }
  259. /**
  260. * docx 替换模板字符串内容
  261. * @example
  262. // 要替换内容的模板
  263. let inputDocx = 'cs.docx'
  264. // 替换完成的docx文件
  265. let outputDocx = 'dd.docx'
  266. // {{xx}} 处要替换的内容
  267. let replaceData = {
  268. name: '替换name处的内容',
  269. age: '替换age处的内容',
  270. }
  271. replaceDocx(inputDocx, outputDocx, replaceData)
  272. */
  273. export function replaceDocx(inputDocxPath, outputDocxPath, options,eventMap) {
  274. return new Promise((resolve,reject)=>{
  275. // 解压出来的临时目录
  276. let md5 = crypto.createHash('md5');
  277. let outmd5 = md5.update(outputDocxPath).digest('hex')
  278. let tempDocxPath = path.join(tempDir , outmd5)
  279. // 要替换的xml文件位置
  280. let tempDocxXMLName = path.join(tempDocxPath,`word/document.xml`)
  281. // 压缩文件夹为文件
  282. let dir_to_docx = (inputFilePath, outputFilePath) => {
  283. outputFilePath = path.join(tempDir,outputFilePath)
  284. // 创建压缩流
  285. let zipStream = new compressing.zip.Stream()
  286. // 写出流
  287. let outStream = fs.createWriteStream(outputFilePath)
  288. fs.readdir(inputFilePath, null, (err, files) => {
  289. if (!err) {
  290. files.map(file => path.join(inputFilePath, file))
  291. .forEach(file => {
  292. zipStream.addEntry(file)
  293. })
  294. }
  295. })
  296. // 写入文件内容
  297. zipStream.pipe(outStream)
  298. .on('close', () => {
  299. // 打包完成后删除临时目录
  300. // console.log(tempDocxPath)
  301. eventMap["onDocxComplete"]&&eventMap["onDocxComplete"](outputFilePath)
  302. shell.rm("-r",tempDocxPath)
  303. // rimrif.rimrafSync(tempDocxPath)
  304. resolve(true)
  305. })
  306. }
  307. // 替换word/document.xml文件中{{xx}}处的内容
  308. let replaceXML = (data, text) => {
  309. Object.keys(data).forEach(key => {
  310. text = text.replaceAll(`{{${key}}}`, data[key])
  311. })
  312. return text
  313. }
  314. // 解压docx文件替换内容重新打包成docx文件
  315. compressing.zip.uncompress(inputDocxPath, tempDocxPath)
  316. .then(() => {
  317. // 读写要替换内容的xml文件
  318. fs.readFile(tempDocxXMLName, null, (err, data) => {
  319. if (!err) {
  320. let text = data.toString()
  321. text = replaceXML(options, text)
  322. fs.writeFile(tempDocxXMLName, text, (err) => {
  323. if (!err) {
  324. dir_to_docx(tempDocxPath, outputDocxPath)
  325. } else {
  326. reject(err)
  327. }
  328. })
  329. } else {
  330. reject(err)
  331. }
  332. })
  333. }).catch(err => {
  334. reject(err)
  335. })
  336. })
  337. }
  338. module.exports.replaceDocx = replaceDocx
  339. function generateObjectId(inputString) {
  340. inputString = inputString || ""
  341. inputString = String(inputString)
  342. const hash = crypto.createHash('sha256').update(inputString).digest('hex');
  343. const objectId = hash;
  344. return objectId;
  345. }