pug-site-core 3.0.30 → 3.0.32

package/lib/dbConnection.js

@@ -41,30 +41,57 @@ function createD1RemoteAdapter(accountId, databaseId, apiToken) {
   // 执行查询的通用函数
   const executeQuery = async (sql, params = []) => {
     try {
+      // 参考 Python 代码：params 始终为数组，即使为空也传空数组
+      const requestBody = {
+        sql: sql,
+        params: params || [],
+      };
+      
       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,
-        }),
+        body: JSON.stringify(requestBody),
       });
 
       if (!response.ok) {
         const errorText = await response.text();
+        // 如果是 401 错误，提供更详细的提示
+        if (response.status === 401) {
+          console.error('❌ D1 API 认证失败 (401)');
+          console.error('可能的原因：');
+          console.error('1. API Token 无效或已过期');
+          console.error('2. API Token 没有 D1 数据库的访问权限');
+          console.error('3. API Token 格式不正确');
+          console.error('请检查 Cloudflare Dashboard 中的 API Token 配置');
+        }
         throw new Error(`D1 API 错误 (${response.status}): ${errorText}`);
       }
 
       const data = await response.json();
       
+      // 检查顶层 success
       if (!data.success) {
-        throw new Error(`D1 查询失败: ${JSON.stringify(data.errors || data)}`);
+        const errors = data.errors || [];
+        throw new Error(`D1 请求被拒绝: ${JSON.stringify(errors)}`);
+      }
+
+      // 检查 result 数组
+      const resultList = data.result || [];
+      if (resultList.length === 0) {
+        return { results: [], meta: {} };
+      }
+
+      const result = resultList[0];
+      
+      // 检查 result 的 success（参考 Python 代码）
+      if (result.success === false) {
+        throw new Error(`D1 SQL 执行失败: ${result.error || JSON.stringify(result)}`);
       }
 
-      return data;
+      return result;
     } catch (error) {
       console.error('D1 查询错误:', error);
       throw error;
@@ -81,36 +108,37 @@ function createD1RemoteAdapter(accountId, databaseId, apiToken) {
         bind: (...args) => {
           return {
             all: async () => {
-              const data = await executeQuery(sql, args);
-              const results = data.result && data.result[0] ? data.result[0].results : [];
-              return { results };
+              const result = await executeQuery(sql, args);
+              // executeQuery 现在直接返回 result 对象，包含 results 和 meta
+              const rows = result.results || [];
+              return { results: Array.isArray(rows) ? rows : [] };
             },
             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;
+              const result = await executeQuery(sql, args);
+              const rows = result.results || [];
+              return Array.isArray(rows) && rows.length > 0 ? rows[0] : null;
             },
             run: async () => {
-              const data = await executeQuery(sql, args);
-              const meta = data.result && data.result[0] ? data.result[0].meta : {};
+              const result = await executeQuery(sql, args);
+              const meta = result.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 };
+          const result = await executeQuery(sql, []);
+          const rows = result.results || [];
+          return { results: Array.isArray(rows) ? rows : [] };
         },
         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;
+          const result = await executeQuery(sql, []);
+          const rows = result.results || [];
+          return Array.isArray(rows) && rows.length > 0 ? rows[0] : null;
         },
         run: async () => {
-          const data = await executeQuery(sql, []);
-          const meta = data.result && data.result[0] ? data.result[0].meta : {};
+          const result = await executeQuery(sql, []);
+          const meta = result.meta || {};
           return { success: true, meta };
         }
       };
@@ -137,7 +165,19 @@ async function getD1Config(configPath) {
 
   // 如果环境变量未配置，尝试从项目的 SITE_CONFIGS_JSON 读取
   try {
-    const { config, SITE_CONFIGS_JSON } = await import(configPath);
+    // 确保路径是有效的 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;
+      }
+    }
+    const { config, SITE_CONFIGS_JSON } = await import(importPath);
     const siteName = config.siteConfig?.siteName || 'default';
     
     console.log(`尝试从 SITE_CONFIGS_JSON 读取配置，站点名称: ${siteName}`);
@@ -162,6 +202,32 @@ async function getD1Config(configPath) {
   return null;
 }
 
+/**
+ * 验证 API Token 是否有效
+ * @param {string} apiToken - Cloudflare API Token
+ * @returns {Promise<boolean>} - Token 是否有效
+ */
+async function verifyApiToken(apiToken) {
+  try {
+    const response = await fetch('https://api.cloudflare.com/client/v4/user/tokens/verify', {
+      method: 'GET',
+      headers: {
+        'Authorization': `Bearer ${apiToken}`,
+        'Content-Type': 'application/json',
+      },
+    });
+    
+    if (response.ok) {
+      const data = await response.json();
+      return data.success === true;
+    }
+    return false;
+  } catch (error) {
+    console.warn('验证 API Token 时出错:', error.message);
+    return false;
+  }
+}
+
 /**
  * 创建数据库连接
  * @param {string} configPath - 项目 config.js 的路径
@@ -174,6 +240,22 @@ export async function getDatabaseConnection(configPath) {
   }
 
   console.log(`使用账户 ID: ${config.accountId}, 数据库 ID: ${config.databaseId}`);
+  // 只显示 Token 的前 10 个字符用于调试（不显示完整 Token）
+  const tokenPreview = config.apiToken ? `${config.apiToken.substring(0, 10)}...` : '未设置';
+  console.log(`API Token: ${tokenPreview}`);
+
+  // 验证 API Token 是否有效（可选，仅用于诊断）
+  console.log('正在验证 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 数据库的访问权限');
+  } else {
+    console.log('✅ API Token 验证成功');
+  }
 
   return await createD1RemoteConnection(config.accountId, config.databaseId, config.apiToken);
 }

package/package.json

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