pug-site-core 3.0.28 → 3.0.30

package/lib/dbConnection.js

@@ -0,0 +1,179 @@
+/**
+ * 数据库连接辅助文件
+ * 用于在本地 dev 模式下通过 Cloudflare API 连接远程 D1 数据库
+ * 
+ * 配置方式：
+ * 1. 通过环境变量配置（推荐）
+ *    - D1_ACCOUNT_ID 或 CF_ACCOUNT_ID: Cloudflare 账户 ID
+ *    - D1_DATABASE_ID: D1 数据库 ID
+ *    - CLOUDFLARE_API_TOKEN 或 CF_API_TOKEN: Cloudflare API Token
+ * 
+ * 2. 通过 SITE_CONFIGS_JSON 配置（在项目的 config.js 中）
+ */
+
+// Cloudflare D1 API 基础 URL
+const CLOUDFLARE_API_BASE = 'https://api.cloudflare.com/client/v4';
+
+/**
+ * 创建 D1 远程连接
+ * @param {string} accountId - Cloudflare 账户 ID
+ * @param {string} databaseId - D1 数据库 ID
+ * @param {string} apiToken - Cloudflare API Token
+ */
+async function createD1RemoteConnection(accountId, databaseId, apiToken) {
+  if (!accountId || !databaseId || !apiToken) {
+    throw new Error('缺少必要的 D1 连接配置：accountId、databaseId 和 apiToken');
+  }
+
+  console.log(`正在连接到 Cloudflare D1 数据库: ${databaseId}`);
+
+  // 创建 D1 远程适配器
+  return createD1RemoteAdapter(accountId, databaseId, apiToken);
+}
+
+/**
+ * 创建 D1 远程适配器
+ * 通过 Cloudflare REST API 调用 D1 数据库
+ */
+function createD1RemoteAdapter(accountId, databaseId, apiToken) {
+  const apiUrl = `${CLOUDFLARE_API_BASE}/accounts/${accountId}/d1/database/${databaseId}/query`;
+
+  // 执行查询的通用函数
+  const executeQuery = async (sql, params = []) => {
+    try {
+      const response = await fetch(apiUrl, {
+        method: 'POST',
+        headers: {
+          'Authorization': `Bearer ${apiToken}`,
+          'Content-Type': 'application/json',
+        },
+        body: JSON.stringify({
+          sql: sql,
+          params: params.length > 0 ? params : undefined,
+        }),
+      });
+
+      if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(`D1 API 错误 (${response.status}): ${errorText}`);
+      }
+
+      const data = await response.json();
+      
+      if (!data.success) {
+        throw new Error(`D1 查询失败: ${JSON.stringify(data.errors || data)}`);
+      }
+
+      return data;
+    } catch (error) {
+      console.error('D1 查询错误:', error);
+      throw error;
+    }
+  };
+
+  return {
+    prepare: (sql) => {
+      // 创建一个可以链式调用的对象
+      // 支持两种调用方式：
+      // 1. prepare(sql).bind(...args).all() / first() / run()
+      // 2. prepare(sql).all() / first() / run() (当没有参数时)
+      const prepared = {
+        bind: (...args) => {
+          return {
+            all: async () => {
+              const data = await executeQuery(sql, args);
+              const results = data.result && data.result[0] ? data.result[0].results : [];
+              return { results };
+            },
+            first: async () => {
+              const data = await executeQuery(sql, args);
+              const results = data.result && data.result[0] ? data.result[0].results : [];
+              return results.length > 0 ? results[0] : null;
+            },
+            run: async () => {
+              const data = await executeQuery(sql, args);
+              const meta = data.result && data.result[0] ? data.result[0].meta : {};
+              return { success: true, meta };
+            }
+          };
+        },
+        // 支持直接调用 all()、first()、run()（当没有参数时）
+        all: async () => {
+          const data = await executeQuery(sql, []);
+          const results = data.result && data.result[0] ? data.result[0].results : [];
+          return { results };
+        },
+        first: async () => {
+          const data = await executeQuery(sql, []);
+          const results = data.result && data.result[0] ? data.result[0].results : [];
+          return results.length > 0 ? results[0] : null;
+        },
+        run: async () => {
+          const data = await executeQuery(sql, []);
+          const meta = data.result && data.result[0] ? data.result[0].meta : {};
+          return { success: true, meta };
+        }
+      };
+      
+      return prepared;
+    }
+  };
+}
+
+/**
+ * 从配置中获取数据库连接信息
+ * @param {string} configPath - 项目 config.js 的路径
+ */
+async function getD1Config(configPath) {
+  // 优先从环境变量读取
+  const accountId = process.env.D1_ACCOUNT_ID || process.env.CF_ACCOUNT_ID;
+  const databaseId = process.env.D1_DATABASE_ID;
+  const apiToken = process.env.CLOUDFLARE_API_TOKEN || process.env.CF_API_TOKEN;
+
+  if (accountId && databaseId && apiToken) {
+    console.log('使用环境变量配置 D1 数据库连接');
+    return { accountId, databaseId, apiToken };
+  }
+
+  // 如果环境变量未配置，尝试从项目的 SITE_CONFIGS_JSON 读取
+  try {
+    const { config, SITE_CONFIGS_JSON } = await import(configPath);
+    const siteName = config.siteConfig?.siteName || 'default';
+    
+    console.log(`尝试从 SITE_CONFIGS_JSON 读取配置，站点名称: ${siteName}`);
+    
+    // 检查是否有 SITE_CONFIGS_JSON 配置
+    if (SITE_CONFIGS_JSON && SITE_CONFIGS_JSON[siteName]) {
+      const siteConfig = SITE_CONFIGS_JSON[siteName];
+      console.log(`找到 ${siteName} 的配置`);
+      return {
+        accountId: siteConfig.cf_account_id,
+        databaseId: siteConfig.d1_database_id,
+        apiToken: siteConfig.cf_api_token,
+      };
+    } else {
+      console.warn(`SITE_CONFIGS_JSON 中未找到 ${siteName} 的配置`);
+    }
+  } catch (error) {
+    // 忽略配置读取错误
+    console.warn('读取 SITE_CONFIGS_JSON 配置失败:', error.message);
+  }
+
+  return null;
+}
+
+/**
+ * 创建数据库连接
+ * @param {string} configPath - 项目 config.js 的路径
+ */
+export async function getDatabaseConnection(configPath) {
+  const config = await getD1Config(configPath);
+  
+  if (!config) {
+    throw new Error('未配置 D1 数据库连接信息。请设置环境变量或配置 SITE_CONFIGS_JSON');
+  }
+
+  console.log(`使用账户 ID: ${config.accountId}, 数据库 ID: ${config.databaseId}`);
+
+  return await createD1RemoteConnection(config.accountId, config.databaseId, config.apiToken);
+}

package/lib/devServer.js

@@ -19,6 +19,7 @@ import { Worker } from "worker_threads";
 import { paths } from "./paths.js";
 import { injectDebugScript } from "./debug/pugDebug.js";
 import { setupDebugRoutes } from "./debug/debugRouter.js";
+import { getDatabaseConnection } from "./dbConnection.js";
 
 const { config } = await import(paths.config);
 const pagsTemplatePath = config.devServer.isDebug ? paths.template.debugPages : paths.template.pages;
@@ -27,6 +28,32 @@ const port = await getIdleProt(config.devServer.port);
 process.env._port = port;
 process.env._localIp = localIp;
 
+// 初始化数据库连接（如果配置了）- 在模块加载时初始化一次
+let dbConnection = null;
+if (config.devServer?.db) {
+  try {
+    if (typeof config.devServer.db === 'function') {
+      dbConnection = await config.devServer.db();
+      console.log('数据库连接已初始化（dev 模式）');
+    } else if (config.devServer.db === true) {
+      // 如果设置为 true，使用内置的数据库连接功能
+      try {
+        dbConnection = await getDatabaseConnection(paths.config);
+        console.log('✅ D1 数据库连接成功（使用内置连接）');
+      } catch (error) {
+        console.error('❌ 数据库连接失败:', error.message);
+        console.warn('将使用 null 作为数据库连接');
+      }
+    } else {
+      dbConnection = config.devServer.db;
+      console.log('使用已配置的数据库连接（dev 模式）');
+    }
+  } catch (error) {
+    console.error('初始化数据库连接失败:', error);
+    console.warn('将使用 null 作为数据库连接');
+  }
+}
+
 function createServer() {
   const app = express();
   const server = http.createServer(app);
@@ -251,7 +278,18 @@ async function matchRouter(url, language, device) {
     let data = await getJsonData(jsonPath);
     return data;
   };
-  let params = { url, language, device, getR2Data };
+  
+  // 创建 params 对象，包含 env 属性以模拟线上环境（Cloudflare Workers）
+  let params = { 
+    url, 
+    language, 
+    device, 
+    getR2Data,
+    env: {
+      db: dbConnection  // 使用预初始化的数据库连接，模拟线上环境的 this.env.db
+    }
+  };
+  
   if(config.abtest && config.abtest.enabled && abtestRouter && abtestRouter[config.abtest.curVariant]){
     router.unshift(...abtestRouter[config.abtest.curVariant]);
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pug-site-core",
-  "version": "3.0.28",
+  "version": "3.0.30",
   "main": "index.js",
   "type": "module",
   "scripts": {
@@ -52,7 +52,7 @@
     "ws": "^8.18.0"
   },
   "license": "ISC",
-  "description": "feat: 添加匈牙利语和菲律宾语支持",
+  "description": "feat: 添加 dev 模式下的 D1 数据库连接支持",
   "files": [
     "lib/",
     "index.js"