123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154
import { join } from "https://deno.land/std@0.145.0/path/mod.ts";import { parseExportNames } from "https://deno.land/x/aleph_compiler@0.6.6/mod.ts";import type { RouteConfig } from "../framework/core/route.ts";import log, { blue } from "../lib/log.ts";import util from "../lib/util.ts";import { initRoutes, toRouteRegExp } from "./routing.ts";import { createFsEmitter, removeFsEmitter, watchFs } from "./watch_fs.ts";import type { AlephConfig } from "./types.ts";
export function handleHMRSocket(req: Request): Response {  const { socket, response } = Deno.upgradeWebSocket(req, {});  const emitter = createFsEmitter();  const send = (message: Record<string, unknown>) => {    try {      socket.send(JSON.stringify(message));    } catch (err) {      log.warn("socket.send:", err.message);    }  };  socket.addEventListener("open", () => {    emitter.on("create", ({ specifier }) => {      const config: AlephConfig | undefined = Reflect.get(globalThis, "__ALEPH_CONFIG");      if (config && config.routes) {        const reg = toRouteRegExp(config.routes);        const routePattern = reg.exec(specifier);        if (routePattern) {          send({ type: "create", specifier, routePattern });          return;        }      }      send({ type: "create", specifier });    });    emitter.on("remove", ({ specifier }) => {      emitter.off(`hotUpdate:${specifier}`);      send({ type: "remove", specifier });    });  });  socket.addEventListener("message", (e) => {    if (util.isFilledString(e.data)) {      try {        const { type, specifier } = JSON.parse(e.data);        if (type === "hotAccept" && util.isFilledString(specifier)) {          emitter.on(            `hotUpdate:${specifier}`,            () => send({ type: "modify", specifier }),          );        }      } catch (_e) {        log.error("invlid socket message:", e.data);      }    }  });  socket.addEventListener("close", () => {    removeFsEmitter(emitter);  });  return response;}
export function watchFS(appDir?: string) {  const config: AlephConfig | undefined = Reflect.get(globalThis, "__ALEPH_CONFIG");  const emitter = createFsEmitter();
  // update global route config when fs changess  const updateRoutes = async ({ specifier }: { specifier: string }) => {    if (config?.routes) {      const reg = toRouteRegExp(config.routes);      if (reg.test(specifier)) {        const routeConfig = await initRoutes(reg, appDir);        Reflect.set(globalThis, "__ALEPH_ROUTE_CONFIG", routeConfig);        if (!Deno.env.get("ALEPH_CLI")) {          generateRoutesModule(routeConfig).catch((error) => log.error(error));        }      }    } else {      Reflect.set(globalThis, "__ALEPH_ROUTE_CONFIG", null);    }  };  emitter.on("create", updateRoutes);  emitter.on("remove", updateRoutes);
  if (config?.routes) {    initRoutes(config.routes, appDir).then((routeConfig) => {      Reflect.set(globalThis, "__ALEPH_ROUTE_CONFIG", routeConfig);      if (!Deno.env.get("ALEPH_CLI")) {        generateRoutesModule(routeConfig).catch((error) => log.error(error));      }    });  }
  // apply user `watchFS` handler  if (typeof config?.devServer?.watchFS === "function") {    const { watchFS } = config.devServer;    emitter.on("*", (kind, { specifier }) => {      if (kind.startsWith("modify:")) {        watchFS("modify", specifier);      } else if (kind === "create" || kind === "remove") {        watchFS(kind, specifier);      }    });  }
  log.info(`Watching files for changes...`);  watchFs(appDir);}
/** generate the `_export.ts` module follows the routes config */export async function generateRoutesModule(routeConfig: RouteConfig, cwd = Deno.cwd()) {  const genFile = join(cwd, routeConfig.prefix, "_export.ts");
  const routeFiles: [filename: string, pattern: string, exportNames: string[]][] = await Promise.all(    routeConfig.routes.map(async ([_, { filename, pattern }]) => {      const code = await Deno.readTextFile(join(cwd, filename));      const exportNames = await parseExportNames(filename, code);      return [filename, pattern.pathname, exportNames];    }),  );
  const imports: string[] = [];  const revives: string[] = [];
  routeFiles.forEach(([filename, pattern, exportNames], idx) => {    if (exportNames.length > 0) {      const importUrl = JSON.stringify("." + util.trimPrefix(filename, routeConfig.prefix));      imports.push(`import * as $${idx} from ${importUrl};`);      revives.push(`  ${JSON.stringify(pattern)}: $${idx},`);    }  });
  if (revives.length > 0) {    await Deno.writeTextFile(      genFile,      [        "// Exports route modules for serverless env that doesn't support the dynamic imports.",        "// This module will be updated automaticlly in develoment mode, do NOT edit it manually.",        "",        ...imports,        "",        "export default {",        ...revives,        "}",      ].join("\n"),    );    log.debug(`${blue("routes.gen.ts")} generated`);  } else {    try {      await Deno.remove(genFile);    } catch (error) {      if (!(error instanceof Deno.errors.NotFound)) {        throw error;      }    }  }}