Function createCustomConsole

packages/vitest/src/runtime/console.ts:35–220

(defaultState?: WorkerGlobalState)

}

export function createCustomConsole(defaultState?: WorkerGlobalState): Console {
  const stdoutBuffer = new Map<string, any[]>()
  const stderrBuffer = new Map<string, any[]>()
  const timers = new Map<
    string,
    { stdoutTime: number; stderrTime: number; cancel?: () => void }
  >()

  const { queueMicrotask } = getSafeTimers()

  function queueCancelableMicrotask(callback: () => void) {
    let canceled = false
    queueMicrotask(() => {
      if (!canceled) {
        callback()
      }
    })
    return () => {
      canceled = true
    }
  }

  const state = () => defaultState || getWorkerState()

  // group sync console.log calls with micro task
  function schedule(taskId: string) {
    const timer = timers.get(taskId)!
    const { stdoutTime, stderrTime } = timer
    timer.cancel?.()
    timer.cancel = queueCancelableMicrotask(() => {
      if (stderrTime < stdoutTime) {
        sendStderr(taskId)
        sendStdout(taskId)
      }
      else {
        sendStdout(taskId)
        sendStderr(taskId)
      }
    })
  }
  function sendStdout(taskId: string) {
    sendBuffer('stdout', taskId)
  }

  function sendStderr(taskId: string) {
    sendBuffer('stderr', taskId)
  }

  function sendBuffer(type: 'stdout' | 'stderr', taskId: string) {
    const buffers = type === 'stdout' ? stdoutBuffer : stderrBuffer
    const buffer = buffers.get(taskId)
    if (!buffer) {
      return
    }
    if (state().config.printConsoleTrace) {
      buffer.forEach(([buffer, origin]) => {
        sendLog(type, taskId, String(buffer), buffer.length, origin)
      })