pug-site-core 3.0.35 → 3.0.36

package/lib/dbConnection.js

@@ -13,6 +13,8 @@
 
 // Cloudflare D1 API 基础 URL
 const CLOUDFLARE_API_BASE = 'https://api.cloudflare.com/client/v4';
+const DEFAULT_D1_BINDING = 'db';
+const D1_BINDING_NAME_RE = /^[A-Za-z_][A-Za-z0-9_]*$/;
 
 /**
  * 创建 D1 远程连接
@@ -146,56 +148,122 @@ function createD1RemoteAdapter(accountId, databaseId, apiToken) {
   };
 }
 
-/**
- * 从配置中获取数据库连接信息
- * @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;
+function normalizeConfigImportPath(configPath) {
+  if (configPath.startsWith('file://') || configPath.startsWith('http://') || configPath.startsWith('https://')) {
+    return configPath;
+  }
+  if (configPath.match(/^[A-Za-z]:/)) {
+    return `file:///${configPath.replace(/\\/g, '/')}`;
+  }
+  return null;
+}
 
-  if (accountId && databaseId && apiToken) {
-    return { accountId, databaseId, apiToken };
+function getEnvD1Parts() {
+  return {
+    accountId: process.env.D1_ACCOUNT_ID || process.env.CF_ACCOUNT_ID,
+    databaseId: process.env.D1_DATABASE_ID,
+    apiToken: process.env.CLOUDFLARE_API_TOKEN || process.env.CF_API_TOKEN,
+  };
+}
+
+function assertValidBindingName(binding) {
+  if (!D1_BINDING_NAME_RE.test(binding)) {
+    throw new Error(`D1 binding 名称无效: ${binding}`);
   }
+}
 
-  // 如果环境变量未配置，尝试从项目的 SITE_CONFIGS_JSON 读取
+function getDatabaseIdFromValue(binding, value) {
+  if (typeof value === 'string') {
+    return value;
+  }
+  if (value && typeof value === 'object') {
+    return value.database_id || value.databaseId;
+  }
+  throw new Error(`D1 binding ${binding} 的配置必须是 database id 字符串或 { database_id } 对象`);
+}
+
+function normalizeD1Bindings(siteConfig, envDatabaseId) {
+  const bindings = {};
+  const d1Databases = siteConfig.d1_databases && typeof siteConfig.d1_databases === 'object'
+    ? siteConfig.d1_databases
+    : {};
+
+  const defaultDbConfig = Object.prototype.hasOwnProperty.call(d1Databases, DEFAULT_D1_BINDING)
+    ? d1Databases[DEFAULT_D1_BINDING]
+    : siteConfig.d1_database_id;
+
+  if (defaultDbConfig || envDatabaseId) {
+    bindings[DEFAULT_D1_BINDING] = envDatabaseId || getDatabaseIdFromValue(DEFAULT_D1_BINDING, defaultDbConfig);
+  }
+
+  for (const [binding, value] of Object.entries(d1Databases)) {
+    assertValidBindingName(binding);
+    if (binding === DEFAULT_D1_BINDING) continue;
+    const databaseId = getDatabaseIdFromValue(binding, value);
+    if (!databaseId) {
+      throw new Error(`D1 binding ${binding} 缺少 database_id`);
+    }
+    bindings[binding] = databaseId;
+  }
+
+  return bindings;
+}
+
+async function loadSiteConfig(configPath) {
   try {
-    // 确保路径是有效的 file:// URL 格式（Windows 需要）
-    let importPath = configPath;
-    if (!configPath.startsWith('file://') && !configPath.startsWith('http://') && !configPath.startsWith('https://')) {
-      // 如果是绝对路径，转换为 file:// URL
-      if (configPath.match(/^[A-Za-z]:/)) {
-        importPath = `file:///${configPath.replace(/\\/g, '/')}`;
-      } else {
-        // 相对路径，使用 pathToFileURL
-        const { pathToFileURL } = await import('url');
-        importPath = pathToFileURL(configPath).href;
-      }
+    let importPath = normalizeConfigImportPath(configPath);
+    if (!importPath) {
+      const { pathToFileURL } = await import('url');
+      importPath = pathToFileURL(configPath).href;
     }
+
     const { config, SITE_CONFIGS_JSON } = await import(importPath);
     const siteName = config.siteConfig?.siteName || 'default';
-    
-    // 检查是否有 SITE_CONFIGS_JSON 配置
     if (SITE_CONFIGS_JSON && SITE_CONFIGS_JSON[siteName]) {
-      const siteConfig = SITE_CONFIGS_JSON[siteName];
-      return {
-        accountId: siteConfig.cf_account_id,
-        databaseId: siteConfig.d1_database_id,
-        apiToken: siteConfig.cf_api_token,
-      };
-    } else {
-      console.warn(`SITE_CONFIGS_JSON 中未找到 ${siteName} 的配置`);
+      return SITE_CONFIGS_JSON[siteName];
     }
+    console.warn(`SITE_CONFIGS_JSON 中未找到 ${siteName} 的配置`);
   } catch (error) {
-    // 忽略配置读取错误
     console.warn('读取 SITE_CONFIGS_JSON 配置失败:', error.message);
   }
 
   return null;
 }
 
+/**
+ * 从配置中获取所有 D1 binding 连接信息
+ * @param {string} configPath - 项目 config.js 的路径
+ */
+async function getD1Configs(configPath) {
+  const envConfig = getEnvD1Parts();
+  const siteConfig = await loadSiteConfig(configPath);
+
+  if (!siteConfig) {
+    if (envConfig.accountId && envConfig.databaseId && envConfig.apiToken) {
+      return {
+        [DEFAULT_D1_BINDING]: {
+          accountId: envConfig.accountId,
+          databaseId: envConfig.databaseId,
+          apiToken: envConfig.apiToken,
+        },
+      };
+    }
+    return null;
+  }
+
+  const accountId = envConfig.accountId || siteConfig.cf_account_id;
+  const apiToken = envConfig.apiToken || siteConfig.cf_api_token;
+  const bindings = normalizeD1Bindings(siteConfig, envConfig.databaseId);
+
+  const configs = {};
+  for (const [binding, databaseId] of Object.entries(bindings)) {
+    assertValidBindingName(binding);
+    configs[binding] = { accountId, databaseId, apiToken };
+  }
+
+  return Object.keys(configs).length ? configs : null;
+}
+
 /**
  * 验证 API Token 是否有效
  * @param {string} apiToken - Cloudflare API Token
@@ -222,26 +290,48 @@ async function verifyApiToken(apiToken) {
   }
 }
 
+async function verifyD1Tokens(configs) {
+  const apiTokens = [...new Set(Object.values(configs).map((config) => config.apiToken).filter(Boolean))];
+  for (const apiToken of apiTokens) {
+    const isValid = await verifyApiToken(apiToken);
+    if (!isValid) {
+      console.warn('⚠️  API Token 验证失败，但将继续尝试连接数据库...');
+      console.warn('   如果后续数据库查询失败，请检查：');
+      console.warn('   1. Token 是否正确（检查 config.js 中的 cf_api_token）');
+      console.warn('   2. Token 是否在 Cloudflare Dashboard 中仍然有效');
+      console.warn('   3. Token 是否有 D1 数据库的访问权限');
+    }
+  }
+}
+
 /**
- * 创建数据库连接
+ * 创建所有 D1 binding 连接
  * @param {string} configPath - 项目 config.js 的路径
  */
-export async function getDatabaseConnection(configPath) {
-  const config = await getD1Config(configPath);
-  
-  if (!config) {
+export async function getDatabaseConnections(configPath) {
+  const configs = await getD1Configs(configPath);
+
+  if (!configs) {
     throw new Error('未配置 D1 数据库连接信息。请设置环境变量或配置 SITE_CONFIGS_JSON');
   }
 
-  // 验证 API Token 是否有效（可选，仅用于诊断）
-  const isValid = await verifyApiToken(config.apiToken);
-  if (!isValid) {
-    console.warn('⚠️  API Token 验证失败，但将继续尝试连接数据库...');
-    console.warn('   如果后续数据库查询失败，请检查：');
-    console.warn('   1. Token 是否正确（检查 config.js 中的 cf_api_token）');
-    console.warn('   2. Token 是否在 Cloudflare Dashboard 中仍然有效');
-    console.warn('   3. Token 是否有 D1 数据库的访问权限');
+  await verifyD1Tokens(configs);
+
+  const connections = {};
+  for (const [binding, config] of Object.entries(configs)) {
+    connections[binding] = await createD1RemoteConnection(config.accountId, config.databaseId, config.apiToken);
   }
+  return connections;
+}
 
-  return await createD1RemoteConnection(config.accountId, config.databaseId, config.apiToken);
+/**
+ * 创建默认数据库连接
+ * @param {string} configPath - 项目 config.js 的路径
+ */
+export async function getDatabaseConnection(configPath) {
+  const connections = await getDatabaseConnections(configPath);
+  if (!connections[DEFAULT_D1_BINDING]) {
+    throw new Error(`未配置默认 D1 binding: ${DEFAULT_D1_BINDING}`);
+  }
+  return connections[DEFAULT_D1_BINDING];
 }

package/lib/devServer.js

@@ -19,7 +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";
+import { getDatabaseConnections } from "./dbConnection.js";
 
 const { config } = await import(paths.config);
 const pagsTemplatePath = config.devServer.isDebug ? paths.template.debugPages : paths.template.pages;
@@ -28,22 +28,29 @@ const port = await getIdleProt(config.devServer.port);
 process.env._port = port;
 process.env._localIp = localIp;
 
+function normalizeDbBindings(dbConfig) {
+  if (dbConfig && typeof dbConfig === 'object' && typeof dbConfig.prepare !== 'function') {
+    return dbConfig;
+  }
+  return { db: dbConfig };
+}
+
 // 初始化数据库连接（如果配置了）- 在模块加载时初始化一次
-let dbConnection = null;
+let dbBindings = { db: null };
 if (config.devServer?.db) {
   try {
     if (typeof config.devServer.db === 'function') {
-      dbConnection = await config.devServer.db();
+      dbBindings = normalizeDbBindings(await config.devServer.db());
     } else if (config.devServer.db === true) {
       // 如果设置为 true，使用内置的数据库连接功能
       try {
-        dbConnection = await getDatabaseConnection(paths.config);
+        dbBindings = await getDatabaseConnections(paths.config);
       } catch (error) {
         console.error('❌ 数据库连接失败:', error.message);
         console.warn('将使用 null 作为数据库连接');
       }
     } else {
-      dbConnection = config.devServer.db;
+      dbBindings = normalizeDbBindings(config.devServer.db);
     }
   } catch (error) {
     console.error('初始化数据库连接失败:', error);
@@ -282,9 +289,7 @@ async function matchRouter(url, language, device) {
     language, 
     device, 
     getR2Data,
-    env: {
-      db: dbConnection  // 使用预初始化的数据库连接，模拟线上环境的 this.env.db
-    }
+    env: dbBindings
   };
   
   if(config.abtest && config.abtest.enabled && abtestRouter && abtestRouter[config.abtest.curVariant]){

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pug-site-core",
-  "version": "3.0.35",
+  "version": "3.0.36",
   "main": "index.js",
   "type": "module",
   "scripts": {