WeFlow/electron/services/accountDirResolver.ts

/**
 * 账号目录解析器（Worker 线程 / 主进程通用）
 *
 * 职责：在 dbPath 根目录下，根据传入的 wxid，找出微信"实际写入数据"
 *       的那个账号子目录，例如：
 *         dbPath = <微信数据根目录>
 *         wxid   = customwxid_abcd  或  customwxid
 *       期望返回 <微信数据根目录>/customwxid_abcd（带后缀、有 session.db 的那个）
 *
 * 与 ConfigService.getAccountDir 行为保持一致；二者实现独立是因为本文件
 * 也会在 Worker 线程中被加载，无法依赖 electron-store。
 */
import { existsSync, readdirSync, statSync } from 'fs'
import { join } from 'path'

// 解析结果缓存（进程内，避免重复 IO）。key = `${dbPath}|${cleanedWxid}`
const accountDirCache = new Map<string, string>()

/**
 * 把 wxid 字符串"标准化"为目录前缀。
 *  - wxid_xxx_yyyy           → wxid_xxx     （wxid_ 后只取第一段）
 *  - 自定义微信号_后缀(4 位)  → 自定义微信号  （例如 customwxid_abcd → customwxid）
 *  - 其他形式                  → 原样返回
 *
 * 注意：清洗只是为了得到"前缀"用于扫描匹配，并不代表清洗结果就是真实目录名。
 *       真实目录名仍需在 dbPath 下按"前缀 + 任意后缀"扫描得出。
 */
const cleanAccountDirName = (dirName: string): string => {
  const trimmed = dirName.trim()
  if (!trimmed) return trimmed

  if (trimmed.toLowerCase().startsWith('wxid_')) {
    const match = trimmed.match(/^(wxid_[^_]+)/i)
    if (match) return match[1]
    return trimmed
  }

  const suffixMatch = trimmed.match(/^(.+)_([a-zA-Z0-9]{4})$/)
  if (suffixMatch) return suffixMatch[1]

  return trimmed
}

const isDirectory = (path: string): boolean => {
  try {
    return statSync(path).isDirectory()
  } catch {
    return false
  }
}

/**
 * 解析账号目录的真实绝对路径。
 *
 * ## 修复 #996（错误码 -3001：未找到数据库目录）
 *
 * ### 旧实现存在的两处严重缺陷
 * 1. **对 wxid_ 开头的目录强制要求"带后缀"**：
 *    未自定义微信号的普通用户，目录就叫 `wxid_X`（无任何后缀），
 *    旧逻辑会因为段数不足而把它过滤掉，导致这类用户根本匹配不到。
 *
 * 2. **对非 wxid_ 开头（自定义微信号）走短路返回，且不校验目录有效性**：
 *    旧实现写法是
 *      ```
 *      if (!lowerWxid.startsWith('wxid_')) {
 *        const direct = join(root, cleanedWxid)
 *        if (existsSync(direct)) return direct  // ← 直接返回，没校验里面有没有 db_storage
 *      }
 *      ```
 *    叠加 `cleanAccountDirName` 会把 `<自定义号>_<4位后缀>` 清洗成
 *    `<自定义号>`，于是无论用户存的是哪个 wxid，都会命中旧的、无后缀的
 *    空目录（它真实存在但里面没有 db_storage），最终触发 -3001。
 *
 * ### 修复后的统一匹配流程
 * 1. 扫描 dbPath 下所有子目录；
 * 2. 同时接受**精确匹配**(`entry == cleanedWxid`) 与
 *    **后缀匹配**(`entry.startsWith(cleanedWxid + '_')`) 两种命中方式；
 * 3. 用 {@link accountDirLooksValid} 过滤掉"看起来根本不像账号目录"的项
 *    （没有 db_storage 也没有 FileStorage/Image[2]）；
 * 4. 在剩余候选中按以下优先级排序，取最优：
 *    - **有 session.db** > 没有：区分"真正写入数据"与"残留空目录"；
 *    - **后缀匹配** > 精确匹配：与微信 4.x 实际写入目录的命名习惯一致；
 *    - **修改时间更新** > 更旧：兜底。
 */
export const resolveAccountDir = (dbPath?: string, wxid?: string): string | null => {
  if (!dbPath || !wxid) return null

  const cleanedWxid = cleanAccountDirName(wxid)
  const normalized = dbPath.replace(/[\\/]+$/, '')
  const cacheKey = `${normalized}|${cleanedWxid.toLowerCase()}`

  // 命中缓存且目标仍存在则直接返回；目标已被删除的过期缓存项会被剔除
  const cached = accountDirCache.get(cacheKey)
  if (cached && existsSync(cached)) return cached
  if (cached && !existsSync(cached)) {
    accountDirCache.delete(cacheKey)
  }

  const lowerWxid = cleanedWxid.toLowerCase()

  try {
    const entries = readdirSync(normalized)
    type Candidate = { entryPath: string; isExact: boolean; hasSession: boolean; mtime: number }
    const candidates: Candidate[] = []

    for (const entry of entries) {
      const entryPath = join(normalized, entry)
      if (!isDirectory(entryPath)) continue

      const lowerEntry = entry.toLowerCase()
      const isExactMatch = lowerEntry === lowerWxid
      const isSuffixMatch = lowerEntry.startsWith(`${lowerWxid}_`)
      // 既不是精确命中、也不是前缀命中 → 与本 wxid 无关，跳过
      if (!isExactMatch && !isSuffixMatch) continue

      // 看起来不像账号目录（连 db_storage 与 FileStorage/Image 都没有）→ 跳过
      // 这一步是修复 #996 的关键：自定义微信号场景下旧的、无后缀空目录
      // 会在这里被过滤掉，避免后续 wcdbCore.open 误判为真实账号目录。
      if (!accountDirLooksValid(entryPath)) continue

      let mtime = 0
      try { mtime = statSync(entryPath).mtimeMs } catch { /* 忽略 stat 异常 */ }
      candidates.push({
        entryPath,
        isExact: isExactMatch,
        hasSession: accountDirHasSessionDb(entryPath),
        mtime,
      })
    }

    if (candidates.length > 0) {
      candidates.sort((a, b) => {
        // 1) 优先选有 session.db 的（真实写入数据的目录）
        if (a.hasSession !== b.hasSession) return a.hasSession ? -1 : 1
        // 2) 其次优先选"带后缀"的（更接近微信 4.x 实际写入目录）
        if (a.isExact !== b.isExact) return a.isExact ? 1 : -1
        // 3) 最后按修改时间倒序（最新的优先）
        return b.mtime - a.mtime
      })
      const best = candidates[0].entryPath
      accountDirCache.set(cacheKey, best)
      return best
    }
  } catch { /* 扫描目录失败时直接 fallthrough 返回 null */ }

  return null
}

/**
 * 浅层判定一个目录"看起来像不像账号目录"：
 *   存在 db_storage 子目录，或存在 FileStorage/Image[2] 子目录之一即认为是。
 *
 * 用于在候选阶段剔除"同名但实际无数据"的残留空目录
 *（例如自定义微信号后遗留下来的旧 wxid 主目录）。
 */
const accountDirLooksValid = (entryPath: string): boolean => {
  return (
    existsSync(join(entryPath, 'db_storage')) ||
    existsSync(join(entryPath, 'FileStorage', 'Image')) ||
    existsSync(join(entryPath, 'FileStorage', 'Image2'))
  )
}

/**
 * 检测账号目录下是否存在 session.db。
 *
 * 是排序优先级里"区分真实写入数据 vs 仅有空 db_storage 骨架"的关键判据，
 * 同时兼容微信 4.x 两种已知布局：
 *   - db_storage/session/session.db （新版本嵌套布局）
 *   - db_storage/session.db          （部分版本扁平布局）
 */
const accountDirHasSessionDb = (entryPath: string): boolean => {
  const candidates = [
    join(entryPath, 'db_storage', 'session', 'session.db'),
    join(entryPath, 'db_storage', 'session.db'),
  ]
  for (const candidate of candidates) {
    if (existsSync(candidate)) return true
  }
  return false
}