webnovel-writer/v7/test/commands/_helper.js

import path from 'node:path'
import os from 'node:os'
import { fileURLToPath } from 'node:url'
import { mkdtemp, mkdir, writeFile, rm } from 'node:fs/promises'
import { CacheManager } from '../../src/cache/index.js'

const __dirname = path.dirname(fileURLToPath(import.meta.url))

// 共享的示例书仓库（只读，命令测试默认喂它）
export const fixtureRoot = path.join(__dirname, '../fixtures/sample-book')

/**
 * 用共享 sample-book fixture 建命令运行上下文。
 * @returns {Promise<{ctx: {repoPath, cache}, cleanup: () => Promise<void>}>}
 */
export function fixtureCtx() {
  return repoCtx(fixtureRoot, null)
}

/**
 * 建命令运行上下文。
 * @param {string|null} repoPath 现成书仓库路径（与 files 二选一）
 * @param {object|null} files {相对路径: 内容} → 在临时目录现造一个书仓库
 */
export async function repoCtx(repoPath, files) {
  let tmpRepo = null
  if (files) {
    tmpRepo = await mkdtemp(path.join(os.tmpdir(), 'wnw-cmd-repo-'))
    for (const [rel, content] of Object.entries(files)) {
      const full = path.join(tmpRepo, rel)
      await mkdir(path.dirname(full), { recursive: true })
      await writeFile(full, content, 'utf8')
    }
    repoPath = tmpRepo
  }
  // db 放独立临时目录，绝不写进书仓库
  const dbDir = await mkdtemp(path.join(os.tmpdir(), 'wnw-cmd-db-'))
  const cache = new CacheManager(path.join(dbDir, 'index.db'))
  await cache.ensureReady(repoPath)
  return {
    ctx: { repoPath, cache },
    cleanup: async () => {
      await cache.close()
      await rm(dbDir, { recursive: true, force: true })
      if (tmpRepo) await rm(tmpRepo, { recursive: true, force: true })
    },
  }
}