pug-site-core 3.0.11 → 3.0.13

package/lib/devServer.js

@@ -11,6 +11,7 @@ import {
   pathSymbol,
   getJsonData,
   addTemplateScopeIsolation,
+  getFilePathByFunctionName,
 } from "./utils.js";
 import http from "http";
 import WebSocket, { WebSocketServer } from "ws";
@@ -258,8 +259,7 @@ async function matchRouter(url, language, device) {
     const obj = router[index];
     if (obj.matchFn.call(params)) {
       let data = await obj.getData.call(params);
-      let pugPath = obj.getPagesFnName.call(params);
-      pugPath = pugPath.split("_").join(pathSymbol);
+      let pugPath = await getFilePathByFunctionName(obj.getPagesFnName.call(params));
       return {
         pugPath,
         data,

package/lib/generate.js

@@ -62,7 +62,7 @@ export async function compilePagesPugToFn(pugPath) {
           .split(pathSymbol)
           .join("_")
           .slice(0, -4)
-          .replace(/[-]/g, "_");
+          .replaceAll(/[-]/g, "_");
 
         // 读取并编译pug文件
         const pugValue = await fse.readFile(filePath, "utf8");
@@ -153,7 +153,7 @@ export async function generateGetDataFn() {
     await async.each(pagesPugFilePathArr, async (fileName) => {
       const funName =
         "get_" +
-        fileName.split(pathSymbol).join("_").slice(0, -4).replace(/[-]/g, "_") +
+        fileName.split(pathSymbol).join("_").slice(0, -4).replaceAll(/[-]/g, "_") +
         "_data";
 
       // 使用正则表达式检查特定的数据获取函数是否存在
@@ -346,7 +346,7 @@ export async function fetchDataToJsonFile(args) {
       for (const fileName of pugFilePathList) {
         let funName =
           "get_" +
-          fileName.split(pathSymbol).join("_").slice(0, -4).replace(/[-]/g, "_") +
+          fileName.split(pathSymbol).join("_").slice(0, -4).replaceAll(/[-]/g, "_") +
           "_data";
 
         let jsonFilePath = fileName.slice(0, -4).split(pathSymbol);
@@ -767,14 +767,6 @@ export async function buildStatic() {
           return;
         }
 
-        const html = generateHtml(
-          pagesPugToFn,
-          funName,
-          data,
-          pugTemplate,
-          commonData
-        );
-
         // 构建输出路径
         let finalPugTemplate = pugTemplate;
         if (data.page_name) {
@@ -790,6 +782,15 @@ export async function buildStatic() {
           finalPugTemplate.replace(/\.[^.]*$/, ".html")
         );
         console.log(htmlPath);
+
+        const html = generateHtml(
+          pagesPugToFn,
+          funName,
+          data,
+          pugTemplate,
+          commonData
+        );
+
         const compressedHtml = await compressHtml(html, htmlPath);
         await fse.outputFile(htmlPath, compressedHtml);
       });

package/lib/utils.js

@@ -642,3 +642,42 @@ export function addTemplateScopeIsolation(htmlString, scopePrefix = "xy", option
     return htmlString; // 出错时返回原始内容
   }
 }
+
+/**
+ * 根据函数名查找对应的pug文件路径
+ * @param {string} functionName - 函数名（可以是模板函数名或数据获取函数名）
+ * @returns {Promise<string|null>} 返回对应的pug文件路径，如果未找到则返回null
+ */
+export async function getFilePathByFunctionName(functionName) {
+  try {
+    // 获取所有pug文件路径
+    const pagesPugFilePathArr = await getPagesPugFilePathArr();
+    
+    // 清理函数名，移除get_前缀和_data后缀（如果存在）
+    let cleanFunctionName = functionName;
+    if (functionName.startsWith("get_") && functionName.endsWith("_data")) {
+      // 数据获取函数名格式：get_xxx_data
+      cleanFunctionName = functionName.slice(4, -5); // 移除 "get_" 和 "_data"
+    }
+    
+    // 遍历所有pug文件，找到匹配的文件
+    for (const fileName of pagesPugFilePathArr) {
+      // 根据fileName生成函数名（与generate.js中的逻辑保持一致）
+      const generatedFunctionName = fileName
+        .split(pathSymbol)
+        .join("_")
+        .slice(0, -4) // 移除.pug扩展名
+        .replaceAll(/[-]/g, "_");
+      
+      // 检查是否匹配
+      if (generatedFunctionName === cleanFunctionName || generatedFunctionName === functionName) {
+        return fileName;
+      }
+    }
+    
+    return null;
+  } catch (error) {
+    console.error("根据函数名查找文件路径失败:", error);
+    return null;
+  }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pug-site-core",
-  "version": "3.0.11",
+  "version": "3.0.13",
   "main": "index.js",
   "type": "module",
   "scripts": {
@@ -52,7 +52,7 @@
     "ws": "^8.18.0"
   },
   "license": "ISC",
-  "description": "修复语言数据打包失败的问题以及commonData的合并问题",
+  "description": "优化自定义路由的pug文件路径获取",
   "files": [
     "lib/",
     "index.js"