Function createHmrFileEditor

e2e/e2e-utils/src/hmrFileEditor.ts:14–127

(
  options: HmrFileEditorOptions<TFileKey>,
)

}

export function createHmrFileEditor<TFileKey extends string>(
  options: HmrFileEditorOptions<TFileKey>,
) {
  const files = Object.fromEntries(
    Object.entries(options.files).map(([key, filePath]) => [
      key,
      options.rootDir && !path.isAbsolute(filePath as string)
        ? path.join(options.rootDir, filePath as string)
        : (filePath as string),
    ]),
  ) as Record<TFileKey, string>
  const originalContents: Partial<Record<TFileKey, string>> = {}
  const pendingRestoreKeys = new Set<TFileKey>()
  const normalizeSource =
    options.normalizeSource ?? ((_fileKey: TFileKey, source: string) => source)

  async function captureOriginals() {
    for (const [key, filePath] of Object.entries(files) as Array<
      [TFileKey, string]
    >) {
      const current = await readFile(filePath, 'utf8')
      const normalized = normalizeSource(key, current)

      if (normalized !== current) {
        await writeFile(filePath, normalized)
        pendingRestoreKeys.add(key)
      }

      originalContents[key] = normalized
    }
  }

  const capturePromise = captureOriginals()

  async function restoreFiles(forceFileKeys: Iterable<TFileKey> = []) {
    const forceRestoreKeys = new Set(forceFileKeys)
    const restoredFileKeys: Array<TFileKey> = []

    for (const [key, filePath] of Object.entries(files) as Array<
      [TFileKey, string]
    >) {
      const content = originalContents[key]
      if (content === undefined) continue

      const current = await readFile(filePath, 'utf8')

      if (current !== content || forceRestoreKeys.has(key)) {
        await writeFile(filePath, content)
        restoredFileKeys.push(key)
      }
    }

    return restoredFileKeys
  }

  async function replaceText(fileKey: TFileKey, from: string, to: string) {
    const filePath = files[fileKey]
    const source = await readFile(filePath, 'utf8')