123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169
import { join, relative } from "../deps/path.ts";import { normalizePath } from "./utils.ts";import Events from "./events.ts";
import type { Event, EventListener, EventOptions, Site } from "../core.ts";
/** The options to configure the local server */export interface Options {  /** The folder root to watch */  root: string;
  /** Paths ignored by the watcher */  ignore?: (string | ((path: string) => boolean))[];
  /** The debounce waiting time */  debounce?: number;}
/** Custom events for server */export interface WatchEvent extends Event {  /** The event type */  type: WatchEventType;
  /** The list of changed files (only for "change" events) */  files?: Set<string>;
  /** The error object (only for "error" events) */  error?: Error;}
/** The available event types */export type WatchEventType =  | "start"  | "change"  | "error";
export interface Watcher {  /** Add a listener to an event */  addEventListener(    type: WatchEventType,    listener: EventListener<WatchEvent>,    options?: EventOptions,  ): this;
  /** Dispatch an event */  dispatchEvent(event: WatchEvent): Promise<boolean>;
  /** Start the watcher */  start(): Promise<void>;}
export default class FSWatcher implements Watcher {  events: Events<WatchEvent> = new Events<WatchEvent>();  options: Options;
  constructor(options: Options) {    this.options = options;  }
  /** Add a listener to an event */  addEventListener(    type: WatchEventType,    listener: EventListener<WatchEvent>,    options?: EventOptions,  ) {    this.events.addEventListener(type, listener, options);    return this;  }
  /** Dispatch an event */  dispatchEvent(event: WatchEvent) {    return this.events.dispatchEvent(event);  }
  /** Start the file watcher */  async start() {    const { root, ignore, debounce } = this.options;    const watcher = Deno.watchFs(root);    const changes = new Set<string>();    let timer = 0;    let runningCallback = false;
    await this.dispatchEvent({ type: "start" });
    const callback = async () => {      if (!changes.size || runningCallback) {        return;      }
      const files = new Set(changes);      changes.clear();
      runningCallback = true;
      try {        const result = await this.dispatchEvent({ type: "change", files });        if (false === result) {          return watcher.close();        }      } catch (error) {        await this.dispatchEvent({ type: "error", error });      }      runningCallback = false;    };
    for await (const event of watcher) {      let paths = event.paths.map(normalizePath);
      // Filter ignored paths      paths = paths.filter((path) =>        ignore          ? !ignore.some((ignore) =>            typeof ignore === "string"              ? (path.startsWith(normalizePath(join(root, ignore, "/"))) ||                path === normalizePath(join(root, ignore)))              : ignore(path)          )          : true      );
      if (!paths.length) {        continue;      }
      paths.forEach((path) => changes.add(normalizePath(relative(root, path))));
      // Debounce      clearTimeout(timer);      timer = setTimeout(callback, debounce ?? 100);    }  }}
export class SiteWatcher implements Watcher {  site: Site;  events: Events<WatchEvent> = new Events<WatchEvent>();
  constructor(site: Site) {    this.site = site;  }
  /** Add a listener to an event */  addEventListener(    type: WatchEventType,    listener: EventListener<WatchEvent>,    options?: EventOptions,  ) {    this.events.addEventListener(type, listener, options);    return this;  }
  /** Dispatch an event */  dispatchEvent(event: WatchEvent) {    return this.events.dispatchEvent(event);  }
  /** Start the watcher */  async start() {    await this.dispatchEvent({ type: "start" });    this.site.addEventListener("afterUpdate", (event) => {      const files = new Set([        ...event.pages!.map((page) => page.outputPath!),        ...event.staticFiles!.map((file) => file.outputPath!),      ]);      this.dispatchEvent({ type: "change", files });    });  }}