1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
import { normalizePath } from "../core/utils/path.ts";import reloadClient from "./reload_client.js";
import type { Middleware } from "../core/server.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, headers } = response;
      return new Response(body, { status, statusText, headers });    }
    return response;  };}