Class VitestWatcher

packages/vitest/src/node/watcher.ts:8–240

import pm from 'picomatch'

export class VitestWatcher {
  /**
   * Modules that will be invalidated on the next run.
   */
  public readonly invalidates: Set<string> = new Set()
  /**
   * Test files that have changed and need to be rerun.
   */
  public readonly changedTests: Set<string> = new Set()

  private readonly _onRerun: ((file: string) => void)[] = []

  constructor(private vitest: Vitest) {}

  /**
   * Register a handler that will be called when test files need to be rerun.
   * The callback can receive several files in case the changed file is imported by several test files.
   * Several invocations of this method will add multiple handlers.
   * @internal
   */
  onWatcherRerun(cb: (file: string) => void): this {
    this._onRerun.push(cb)
    return this
  }

  public close(): void {
    this.vitest.vite.watcher.close()
  }

  public unregisterWatcher: () => void = noop
  public registerWatcher(): this {
    const watcher = this.vitest.vite.watcher

    if (this.vitest.config.forceRerunTriggers.length) {
      watcher.add(this.vitest.config.forceRerunTriggers)
    }

    watcher.on('change', this.onFileChange)
    watcher.on('unlink', this.onFileDelete)
    watcher.on('add', this.onFileCreate)

    this.unregisterWatcher = () => {
      watcher.off('change', this.onFileChange)
      watcher.off('unlink', this.onFileDelete)
      watcher.off('add', this.onFileCreate)
      this.unregisterWatcher = noop
    }
    return this
  }

  private scheduleRerun(file: string): void {
    this._onRerun.forEach(cb => cb(file))
  }

  private getTestFilesFromWatcherTrigger(id: string): boolean {
    if (!this.vitest.config.watchTriggerPatterns) {
      return false
    }