123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218
import { listenAndServe } from "./deps/server.js";import { acceptWebSocket } from "./deps/ws.js";import { dirname, extname, join, posix, relative } from "./deps/path.js";import { brightGreen, red } from "./deps/colors.js";import { exists } from "./deps/fs.js";import localIp from "./deps/local_ip.js";import { mimes, normalizePath } from "./utils.js";import { readAll } from "./deps/util.js";
export async function server(site, options) {  const root = site.dest();  const port = parseInt(options.port) || site.options.server.port || 3000;  const ipAddr = await localIp();  const page404 = site.options.server.page404 || "/404.html";
  console.log();  console.log("  Server started at:");  console.log(brightGreen(`  http://localhost:${port}/`), "(local)");  console.log(brightGreen(`  http://${ipAddr}:${port}/`), "(network)");  console.log();
  if (options.open || site.options.server.open) {    const commands = {      darwin: "open",      linux: "xdg-open",      windows: "explorer",    };
    Deno.run({ cmd: [commands[Deno.build.os], `http://localhost:${port}/`] });  }
  // Live reload server  const watcher = Deno.watchFs(root);
  // Static files server  listenAndServe({ port }, (req) => {    // Is websocket    if (req.headers.get("upgrade") === "websocket") {      handleSocket(req);    } else {      handleFile(req);    }  });
  async function handleFile(req) {    let path = join(root, decodeURIComponent(req.url.split("?", 2).shift()));
    try {      const info = await Deno.stat(path);
      if (info.isDirectory) {        path = join(path, "index.html");        await Deno.stat(path);      }
      const mimeType = mimes.get(extname(path).toLowerCase()) ||        "application/octet-stream";
      try {        await req.respond({          status: 200,          headers: new Headers({            "content-type": mimeType,            "cache-control": "no-cache no-store must-revalidate",          }),          body: await (mimeType === "text/html; charset=utf-8"            ? getHtmlBody(path)            : getBody(path)),        });      } catch {        return;      }
      console.log(`${brightGreen("200")} ${req.url}`);    } catch {      console.log(`${red("404")} ${req.url}`);
      try {        await req.respond({          status: 404,          headers: new Headers({            "content-type": mimes.get(".html"),          }),          body: await getNotFoundBody(root, page404, path),        });      } catch {        return;      }    }  }
  let timer = 0;  let socket;  const changes = new Set();
  async function handleSocket(req) {    const { conn, r: bufReader, w: bufWriter, headers } = req;    socket = await acceptWebSocket({      conn,      bufReader,      bufWriter,      headers,    });
    async function sendChanges() {      if (!changes.size) {        return;      }
      const files = Array.from(changes).map((path) =>        posix.join("/", normalizePath(relative(root, path)))      );
      changes.clear();
      try {        await socket.send(JSON.stringify(files));        console.log("Changes sent to browser");      } catch (err) {        console.log(          `Changes couldn't be sent to browser due "${err.message.trim()}"`,        );      }    }
    console.log("Connected to browser");
    for await (const event of watcher) {      if (event.kind !== "modify") {        continue;      }
      event.paths.forEach((path) => changes.add(path));
      // Debounce      clearTimeout(timer);      timer = setTimeout(sendChanges, 100);    }  }}
let wsFile = new URL("./ws.js", import.meta.url);
if (wsFile.protocol === "file:") {  wsFile = await Deno.readTextFile(wsFile);}
async function getHtmlBody(path) {  const content = await Deno.readTextFile(path);
  if (typeof wsFile === "string") {    return `${content}<script id="lume-live-reload">${wsFile}</script>`;  }
  return `${content}<script type="module" src="${wsFile}" id="lume-live-reload"></script>`;}
async function getNotFoundBody(root, page404, file) {  const filepath = join(root, page404);
  if (await exists(filepath)) {    return getHtmlBody(filepath);  }
  const content = await listDirectory(dirname(file));
  return `<!DOCTYPE html>  <html lang="en">  <head>    <meta charset="UTF-8">    <meta name="viewport" content="width=device-width, initial-scale=1.0">    <title>404 - Not found</title>    <style> body { font-family: sans-serif; max-width: 40em; margin: auto; padding: 2em; line-height: 1.5; }</style>  </head>  <body>    <h1>404 - Not found</h1>    <p>The URL <code>${relative(root, file)}</code> does not exist</p>    <ul>      ${    content.map((item) =>      `<li>            <a href="${relative(root, item[1])}">              ${item[0]}            </a>          </li>`    ).join("\n")  }    </ul>  </body></html>`;}
async function getBody(path) {  const file = await Deno.open(path);  const content = await readAll(file);  Deno.close(file.rid);
  return content;}
async function listDirectory(directory) {  const files = [];
  if (!await exists(directory)) {    return files;  }
  for await (const info of Deno.readDir(directory)) {    const name = info.name;    const href = normalizePath(join(directory, name));
    files.push([name, href]);  }
  return files;}