@8btc/office-assistant-mcp 0.0.7 → 0.0.9

package/dist/index.cjs

@@ -76998,6 +76998,22 @@ const localImageToBase64 = async (imagePath) => {
   }
   return null;
 };
+const extractAllImageSrc = (html2) => {
+  const regex = /<img[^>]*?src=["']([^"']+)["'][^>]*>/gi;
+  const srcs = [];
+  let match;
+  while ((match = regex.exec(html2)) !== null) {
+    if (!match[1].startsWith("data:image/")) {
+      logger.info("extractAllImageSrc match[0]:" + match[0], match[1]);
+      logger.info("extractAllImageSrc match[1]:" + match[1]);
+    }
+    if (isLocalPath(match[1])) {
+      srcs.push(match[1]);
+    }
+  }
+  return srcs;
+};
+const reg = /<img[^>]+>/gi;
 const createPage = async (page, item) => {
   const {
     tempFileName,
@@ -77014,12 +77030,10 @@ const createPage = async (page, item) => {
       </div>
     </header>
   `;
-  await page.setContent(html2, { waitUntil: "networkidle" });
-  const imgElements = await page.evaluate(async () => {
-    return Array.from(document.querySelectorAll("img")).map((it) => it.src);
-  });
-  const fileInfoArray = await Promise.all(
-    imgElements.filter((src) => isLocalPath(src)).map(
+  const localSrcs = extractAllImageSrc(html2);
+  logger.info("localSrcs:::" + localSrcs.join(","));
+  const localFileInfoList = await Promise.all(
+    localSrcs.map(
       (src) => new Promise((resolve) => {
         localImageToBase64(src).then((fileInfo) => {
           resolve(fileInfo);
@@ -77029,20 +77043,19 @@ const createPage = async (page, item) => {
       })
     )
   );
-  await page.evaluate(
-    async (fileInfoArray2) => Array.from(document.querySelectorAll("img")).forEach((img) => {
-      const imgInfo = fileInfoArray2.find(
-        (it) => it && img.src && it.src === img.src
+  const _html = html2.replaceAll(reg, (m) => {
+    const item2 = localFileInfoList.find(
+      (info) => m.indexOf(info.src) !== -1
+    );
+    if (item2) {
+      return m.replace(
+        /(<img[^>]*?)src=["'][^"']*["']([^>]*>)/gi,
+        `$1src="${item2.base64}"$2`
       );
-      if (!imgInfo) {
-        return;
-      }
-      console.log("img.src:", img.src);
-      console.log("base64.length:", imgInfo.base64.length);
-      img.src = imgInfo.base64;
-    }),
-    fileInfoArray
-  );
+    }
+    return m;
+  });
+  await page.setContent(_html, { waitUntil: "networkidle" });
   const pageRes = await page.pdf({
     path: tempFileName,
     format: "A4",
@@ -77223,6 +77236,10 @@ const tool = {
 const toolHandler = async (args) => {
   try {
     logger.info(`start: ${(/* @__PURE__ */ new Date()).getTime()}`);
+    try {
+      logger.info(JSON.stringify(args));
+    } catch (e) {
+    }
     const generatedTemplates = await generateBasePdf(args);
     const pdfUrl = await generatePDF({
       list: generatedTemplates,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@8btc/office-assistant-mcp",
-  "version": "0.0.7",
+  "version": "0.0.9",
   "description": "办公agent辅助mcp工具",
   "main": "dist/index.cjs",
   "type": "module",