123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
import { normalizePath } from "../core/utils.ts";import reloadClient from "./reload_client.js";
import type { Middleware } from "../core.ts";import type { Watcher } from "../core/watcher.ts";
export interface Options {  watcher: Watcher;}
/** Middleware to hot reload changes */export default function reload(options: Options): Middleware {  const sockets = new Set<WebSocket>();  const { watcher } = options;
  watcher.addEventListener("change", (event) => {    if (!sockets.size) {      return;    }
    const files = event.files!;    const urls = Array.from(files).map((file) => normalizePath(file));    const message = JSON.stringify(urls);    sockets.forEach((socket) => socket.send(message));    console.log("Changes sent to the browser");  });
  watcher.start();
  return async (request, next) => {    // Is a websocket    if (request.headers.get("upgrade") === "websocket") {      const { socket, response } = Deno.upgradeWebSocket(request);
      socket.onopen = () => sockets.add(socket);      socket.onclose = () => sockets.delete(socket);      socket.onerror = (e) => console.log("Socket errored", e);
      return response;    }
    // It's a regular request    const response = await next(request);
    if (!response.body || response.status !== 200) {      return response;    }
    // Insert live-reload script in the body    if (response.headers.get("content-type")?.includes("html")) {      const reader = response.body.getReader();
      let body = "";      let result = await reader.read();      const decoder = new TextDecoder();
      while (!result.done) {        body += decoder.decode(result.value);        result = await reader.read();      }
      body +=        `<script type="module" id="lume-live-reload">${reloadClient}; liveReload();</script>`;
      const { status, statusText } = response;      const headers = new Headers();      headers.set("content-type", "text/html");
      return new Response(body, { status, statusText, headers });    }
    return response;  };}