fs.ts

import type { BrowserCommands } from 'vitest/browser'
import type { BrowserCommand, TestProject } from 'vitest/node'
import fs, { promises as fsp } from 'node:fs'
import { basename, dirname, resolve } from 'node:path'
import mime from 'mime/lite'
import { isFileServingAllowed } from 'vitest/node'

function assertFileAccess(path: string, project: TestProject) {
  if (
    !isFileServingAllowed(path, project.vite)
    && !isFileServingAllowed(path, project.vitest.vite)
  ) {
    throw new Error(
      `Access denied to "${path}". See Vite config documentation for "server.fs": https://vitejs.dev/config/server-options.html#server-fs-strict.`,
    )
  }
}

export const readFile: BrowserCommand<
  Parameters<BrowserCommands['readFile']>
> = async ({ project }, path, options = {}) => {
  const filepath = resolve(project.config.root, path)
  assertFileAccess(filepath, project)
  // never return a Buffer
  if (typeof options === 'object' && !options.encoding) {
    options.encoding = 'utf-8'
  }
  return fsp.readFile(filepath, options)
}

export const writeFile: BrowserCommand<
  Parameters<BrowserCommands['writeFile']>
> = async ({ project }, path, data, options) => {
  const filepath = resolve(project.config.root, path)
  assertFileAccess(filepath, project)
  const dir = dirname(filepath)
  if (!fs.existsSync(dir)) {
    await fsp.mkdir(dir, { recursive: true })
  }
  await fsp.writeFile(filepath, data, options)
}

export const removeFile: BrowserCommand<
  Parameters<BrowserCommands['removeFile']>
> = async ({ project }, path) => {
  const filepath = resolve(project.config.root, path)
  assertFileAccess(filepath, project)
  await fsp.rm(filepath)
}

export const _fileInfo: BrowserCommand<[path: string, encoding: BufferEncoding]> = async ({ project }, path, encoding) => {
  const filepath = resolve(project.config.root, path)
  assertFileAccess(filepath, project)
  const content = await fsp.readFile(filepath, encoding || 'base64')
  return {
    content,
    basename: basename(filepath),
    mime: mime.getType(filepath),
  }
}