whistle.mockbubu 1.0.0-dev.4 → 2.0.0-beta.0

package/lib/storage-adapter.js

@@ -0,0 +1,518 @@
+/**
+ * Storage Adapter - 适配器模式
+ *
+ * 将新的 FileSystemStorage 包装成 Whistle Storage 兼容的 API
+ * 使得现有代码无需大改即可使用新存储
+ */
+
+const FileSystemStorage = require('./storage-v3')
+const path = require('path')
+
+class StorageAdapter {
+  constructor(options) {
+    const baseDir = options.baseDir || path.join(
+      process.env.HOME,
+      '.WhistleAppData/.whistle/.plugins/whistle.mockbubu',
+    )
+
+    this.v3Storage = new FileSystemStorage(baseDir)
+    this.currentGroupId = 'default'
+    this.initialized = false
+  }
+
+  /**
+   * 初始化（必须在使用前调用）
+   */
+  async init() {
+    if (this.initialized) return
+    await this.v3Storage.init()
+    // 从文件加载当前组ID
+    this.currentGroupId = await this.v3Storage.getCurrentGroupId()
+    this.initialized = true
+  }
+
+  /**
+   * 确保已初始化
+   */
+  async _ensureInit() {
+    if (!this.initialized) {
+      await this.init()
+    }
+  }
+
+  /**
+   * 设置当前组ID
+   */
+  async setCurrentGroupId(groupId) {
+    this.currentGroupId = groupId
+    // V3 架构：持久化到文件
+    if (this.getV3Storage) {
+      const v3Storage = this.getV3Storage()
+      await v3Storage.setCurrentGroupId(groupId)
+    }
+  }
+
+  /**
+   * 获取当前组ID
+   */
+  async getCurrentGroupId() {
+    // V3 架构：从文件读取
+    if (this.getV3Storage) {
+      const v3Storage = this.getV3Storage()
+      this.currentGroupId = await v3Storage.getCurrentGroupId()
+    }
+    return this.currentGroupId
+  }
+
+  // ========================================
+  // 兼容旧 API：文件操作
+  // ========================================
+
+  /**
+   * 读取文件
+   * @param {string} name - 文件名（格式：{groupId}/{filename}）
+   */
+  async readFile(name) {
+    await this._ensureInit()
+
+    try {
+      const { groupId, filename } = this._parsePath(name)
+
+      // ⚠️ 强制刷新索引缓存，确保读取最新数据
+      this.v3Storage.invalidateIndex(groupId)
+
+      // 从索引查找对应的 fileId（因为 fileId 是随机生成的，无法从 URL 推导）
+      const index = await this.v3Storage.getIndex(groupId)
+      const entry = index.files.find(f => f.url === filename)
+
+      if (!entry) {
+        console.log(`[storage-adapter] ⚠️  文件不存在: ${filename}`)
+        return null
+      }
+
+      // 使用索引中的 fileId 读取文件
+      const file = await this.v3Storage.getFile(groupId, entry.id)
+
+      if (!file || !file.session) {
+        console.log(`[storage-adapter] ⚠️  未找到 session 数据: ${groupId}/${entry.id}`)
+        return null
+      }
+
+      // 返回 session 数据的 JSON 字符串（兼容旧格式）
+      return JSON.stringify(file.session)
+    } catch (err) {
+      console.error('[storage-adapter] ❌ readFile 失败:', err.message)
+      return null
+    }
+  }
+
+  /**
+   * 写入文件
+   * @param {string} name - 文件名（格式：{groupId}/{filename}）
+   * @param {string} data - JSON 字符串
+   */
+  async writeFile(name, data) {
+    await this._ensureInit()
+
+    const { groupId, filename } = this._parsePath(name)
+
+    try {
+      // 解析 session 数据
+      let session
+      let hasParseError = false
+
+      try {
+        session = JSON.parse(data)
+        console.log(`[storage-adapter] ✅ JSON 解析成功: method=${session.req?.method}, status=${session.res?.statusCode}`)
+      } catch (parseErr) {
+        // JSON 解析失败 - 可能是非标准格式,但仍然保存
+        console.warn(`[storage-adapter] ⚠️ JSON 解析失败,将保存原始数据: ${filename}`)
+        console.warn(`[storage-adapter] 解析错误: ${parseErr.message}`)
+        hasParseError = true
+
+        // 构造最小可用的 session 对象,保存原始响应数据
+        session = {
+          req: {
+            method: 'UNKNOWN',
+            headers: {},
+          },
+          res: {
+            statusCode: 0,
+            headers: {},
+            body: data, // 保存原始的未解析数据
+          },
+          _parseError: parseErr.message, // 标记解析错误
+        }
+      }
+
+      // ⚠️ 强制刷新索引缓存
+      this.v3Storage.invalidateIndex(groupId)
+
+      // 从索引查找是否已有相同URL的文件
+      const index = await this.v3Storage.getIndex(groupId)
+      const existingEntry = index.files.find(f => f.url === filename)
+
+      if (existingEntry) {
+        // 更新现有文件（使用已有的 fileId）
+        console.log(`[storage-adapter] 🔄 更新现有文件: ${existingEntry.id}`)
+        await this.v3Storage.updateFile(groupId, existingEntry.id, {
+          session,
+          updateTime: Date.now(),
+          hasParseError, // 标记是否有解析错误
+        })
+        console.log(`[storage-adapter] ✅ 文件更新成功${hasParseError ? ' (包含解析错误)' : ''}`)
+      } else {
+        // 创建新文件（生成新的随机 fileId）
+        const newFileId = this.v3Storage.generateFileId()
+        console.log(`[storage-adapter] ➕ 创建新文件: ${newFileId}`)
+        await this.v3Storage.createFile(groupId, newFileId, {
+          url: filename,
+          method: session.req?.method || 'UNKNOWN',
+          status: session.res?.statusCode || 0,
+          headers: session.res?.headers || {},
+          session,
+          config: {
+            mock: false,
+            locked: false,
+            mockVersion: 'source',
+          },
+          hasParseError, // 标记是否有解析错误
+        })
+        console.log(`[storage-adapter] ✅ 文件创建成功${hasParseError ? ' (包含解析错误)' : ''}`)
+      }
+    } catch (err) {
+      // 存储层错误 - 记录但不抛出,避免影响正常流量
+      console.error(`[storage-adapter] ❌ 写入文件失败: ${filename}`)
+      console.error(`[storage-adapter] 错误详情: ${err.message}`)
+      console.error('[storage-adapter] 错误堆栈:', err.stack)
+      // 不再抛出错误,避免中断请求流程
+    }
+  }
+
+  /**
+   * 删除文件
+   * @param {string} name - 文件名（格式：{groupId}/{filename}）
+   */
+  async removeFile(name) {
+    await this._ensureInit()
+
+    const { groupId, filename } = this._parsePath(name)
+
+    try {
+      // ⚠️ 强制刷新索引缓存
+      this.v3Storage.invalidateIndex(groupId)
+
+      // 从索引查找对应的 fileId
+      const index = await this.v3Storage.getIndex(groupId)
+      const entry = index.files.find(f => f.url === filename)
+
+      if (!entry) {
+        console.log(`[storage-adapter] ⚠️  文件不存在，无需删除: ${filename}`)
+        return
+      }
+
+      console.log(`[storage-adapter] 🗑️  删除文件: ${entry.id} (${filename})`)
+      await this.v3Storage.deleteFile(groupId, entry.id)
+      console.log('[storage-adapter] ✅ 删除成功')
+    } catch (err) {
+      console.error('[storage-adapter] ❌ 删除失败:', err.message, err.stack)
+      // 文件不存在也视为成功
+    }
+  }
+
+  /**
+   * 获取文件列表
+   * @returns {Array} - 文件列表（兼容旧格式）
+   */
+  async getFileList() {
+    await this._ensureInit()
+
+    try {
+      const groups = await this.v3Storage.listGroups()
+      const allFiles = []
+
+      for (const group of groups) {
+        const files = await this.v3Storage.listFiles(group.id)
+
+        files.forEach(file => {
+          // 构造兼容旧格式的文件名
+          const name = `${group.id}/${file.url}`
+
+          allFiles.push({
+            name, // 不带索引前缀（新架构不需要）
+            size: file.size,
+          })
+        })
+      }
+
+      return allFiles
+    } catch (err) {
+      console.error('[storage-adapter] getFileList failed:', err.message)
+      return []
+    }
+  }
+
+  // ========================================
+  // 兼容旧 API：属性操作
+  // ========================================
+
+  /**
+   * 获取属性
+   * @param {string} key - 属性键
+   */
+  async getProperty(key) {
+    await this._ensureInit()
+
+    try {
+      // 特殊处理：组列表
+      if (key === '__groups__') {
+        return await this._getGroupsProperty()
+      }
+
+      // 特殊处理：组文件配置
+      if (key.startsWith('group.') && key.includes('.file.')) {
+        return await this._getFileConfigProperty(key)
+      }
+
+      // 通用属性：从 V3 storage 的 properties.json 读取
+      return await this.v3Storage.getGenericProperty(key)
+    } catch (err) {
+      return null
+    }
+  }
+
+  /**
+   * 设置属性
+   * @param {string} key - 属性键
+   * @param {*} value - 属性值
+   */
+  async setProperty(key, value) {
+    await this._ensureInit()
+
+    try {
+      // 特殊处理：组列表
+      if (key === '__groups__') {
+        // 新架构不需要手动管理组列表
+        return
+      }
+
+      // 特殊处理：组文件配置
+      if (key.startsWith('group.') && key.includes('.file.')) {
+        await this._setFileConfigProperty(key, value)
+        return
+      }
+
+      // 通用属性：写入 V3 storage 的 properties.json
+      await this.v3Storage.setGenericProperty(key, value)
+    } catch (err) {
+      console.error('[storage-adapter] setProperty failed:', key, err.message)
+    }
+  }
+
+  /**
+   * 删除属性
+   * @param {string} key - 属性键
+   */
+  async removeProperty(key) {
+    await this._ensureInit()
+
+    try {
+      // 从 Whistle storage 中删除
+      await this.storage.removeProperty(key)
+    } catch (err) {
+      console.error('[storage-adapter] removeProperty failed:', key, err.message)
+      throw err
+    }
+  }
+
+  /**
+   * 检查属性是否存在
+   * @param {string} key - 属性键
+   */
+  async hasProperty(key) {
+    await this._ensureInit()
+
+    const value = await this.getProperty(key)
+    return value !== null && value !== undefined
+  }
+
+  /**
+   * 检查文件是否在索引中（用于 Issue #4 检测）
+   * 与 hasProperty 的区别：
+   * - hasProperty：检查物理文件是否存在
+   * - hasFileInIndex：检查索引中是否有记录（即使物理文件被删除）
+   *
+   * @param {string} key - 配置键，格式：group.{groupId}.file.{filename}
+   * @returns {boolean} 索引中是否存在该文件
+   */
+  async hasFileInIndex(key) {
+    await this._ensureInit()
+
+    try {
+      // 解析配置键
+      if (!key.startsWith('group.') || !key.includes('.file.')) {
+        return false
+      }
+
+      const { groupId, filename } = this._parseConfigKey(key)
+
+      // ⚠️ 强制刷新索引缓存，确保读取最新数据（避免删除后立即捕获时读到旧缓存）
+      this.v3Storage.invalidateIndex(groupId)
+
+      // 检查索引中是否存在（通过 URL 而不是 fileId，因为 fileId 是创建时生成的随机ID）
+      const index = await this.v3Storage.getIndex(groupId)
+      const exists = index.files.some(f => f.url === filename)
+
+      return exists
+    } catch (err) {
+      return false
+    }
+  }
+
+  /**
+   * 获取所有属性
+   */
+  async getProperties() {
+    await this._ensureInit()
+
+    // 新架构不需要这个方法，返回空对象
+    return {}
+  }
+
+  // ========================================
+  // 内部辅助方法
+  // ========================================
+
+  /**
+   * 解析文件路径
+   * @param {string} path - 格式：{groupId}/{filename}
+   */
+  _parsePath(pathStr) {
+    const parts = pathStr.split('/')
+    if (parts.length < 2) {
+      return {
+        groupId: this.currentGroupId,
+        filename: pathStr,
+      }
+    }
+
+    return {
+      groupId: parts[0],
+      filename: parts.slice(1).join('/'),
+    }
+  }
+
+  /**
+   * 解析配置键
+   * @param {string} key - 格式：group.{groupId}.file.{filename}
+   */
+  _parseConfigKey(key) {
+    const parts = key.split('.file.')
+    if (parts.length !== 2) {
+      throw new Error(`Invalid config key: ${key}`)
+    }
+
+    const groupId = parts[0].replace('group.', '')
+    const filename = parts[1]
+
+    return { groupId, filename }
+  }
+
+  /**
+   * 获取组列表属性（兼容旧格式）
+   */
+  async _getGroupsProperty() {
+    const groups = await this.v3Storage.listGroups()
+    const currentGroupId = await this.v3Storage.getCurrentGroupId()
+
+    return {
+      groups: groups.map(g => ({
+        id: g.id,
+        name: g.name,
+        description: g.description,
+        createTime: g.createTime,
+        updateTime: g.updateTime,
+        isDefault: g.isDefault,
+      })),
+      currentGroupId,
+    }
+  }
+
+  /**
+   * 获取文件配置属性（兼容旧格式）
+   */
+  async _getFileConfigProperty(key) {
+    const { groupId, filename } = this._parseConfigKey(key)
+
+    try {
+      // ⚠️ 强制刷新索引缓存
+      this.v3Storage.invalidateIndex(groupId)
+
+      // 从索引查找对应的 fileId
+      const index = await this.v3Storage.getIndex(groupId)
+      const entry = index.files.find(f => f.url === filename)
+
+      if (!entry) {
+        return null
+      }
+
+      const file = await this.v3Storage.getFile(groupId, entry.id)
+
+      return {
+        ...file.config,
+        url: file.url,
+        method: file.method,
+        status: file.status,
+        date: file.createTime,
+      }
+    } catch (err) {
+      return null
+    }
+  }
+
+  /**
+   * 设置文件配置属性
+   */
+  async _setFileConfigProperty(key, value) {
+    const { groupId, filename } = this._parseConfigKey(key)
+
+    try {
+      // ⚠️ 强制刷新索引缓存
+      this.v3Storage.invalidateIndex(groupId)
+
+      // 从索引查找对应的 fileId
+      const index = await this.v3Storage.getIndex(groupId)
+      const entry = index.files.find(f => f.url === filename)
+
+      if (!entry) {
+        console.error('[storage-adapter] _setFileConfigProperty: 文件不存在:', filename)
+        return
+      }
+
+      const file = await this.v3Storage.getFile(groupId, entry.id)
+
+      await this.v3Storage.updateFile(groupId, entry.id, {
+        config: {
+          ...file.config,
+          ...value,
+        },
+      })
+    } catch (err) {
+      console.error('[storage-adapter] _setFileConfigProperty failed:', err.message)
+    }
+  }
+
+  // ========================================
+  // 新增：直接访问 V3 Storage
+  // ========================================
+
+  /**
+   * 获取底层的 V3 Storage 实例
+   * 用于需要使用新 API 的场景
+   */
+  getV3Storage() {
+    return this.v3Storage
+  }
+}
+
+module.exports = StorageAdapter