123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183
import { join } from "https://deno.land/std@0.144.0/path/mod.ts";import { parseDeps, parseExportNames } from "https://deno.land/x/aleph_compiler@0.6.4/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, appDir).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, appDir).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 `routes.gen.ts` module follows the routes config */export async function generateRoutesModule(routeConfig: RouteConfig, appDir?: string, cwd = Deno.cwd()) {  const genFile = join(cwd, appDir ? join(appDir, "routes.gen.ts") : "routes.gen.ts");
  try {    const sourceCode = await Deno.readTextFile(genFile);    const deps = await parseDeps("./routes.gen.ts", sourceCode);    if (      routeConfig.routes.every((route) =>        deps.findIndex(({ specifier }) => {          const filename = appDir ? "." + util.trimPrefix(route[1].filename, appDir) : route[1].filename;          return specifier === filename;        }) >= 0      )    ) {      return;    }  } catch (error) {    if (!(error instanceof Deno.errors.NotFound)) {      throw error;    }  }
  const routeFiles: [filename: string, exportNames: string[]][] = await Promise.all(    routeConfig.routes.map(async ([_, { filename }]) => {      const code = await Deno.readTextFile(join(cwd, filename));      const exportNames = await parseExportNames(filename, code);      return [filename, exportNames];    }),  );
  const imports: string[] = [];  const revives: string[] = [];
  routeFiles.forEach(([filename, exportNames], idx) => {    const hasDefaultExport = exportNames.includes("default");    const hasNameExports = exportNames.filter((name) => name !== "default").length > 0;    const importUrl = JSON.stringify(appDir ? "." + util.trimPrefix(filename, appDir) : filename);    if (hasDefaultExport) {      imports.push(`import d${idx} from ${importUrl};`);    }    if (hasNameExports) {      imports.push(`import * as m${idx} from ${importUrl};`);    }    if (hasDefaultExport && hasNameExports) {      revives.push(`  ${JSON.stringify(filename)}: { ...m${idx}, default: d${idx} },`);    } else if (hasDefaultExport) {      revives.push(`  ${JSON.stringify(filename)}: { default: d${idx} },`);    } else if (hasNameExports) {      revives.push(`  ${JSON.stringify(filename)}: m${idx},`);    }  });
  if (revives.length > 0) {    await Deno.writeTextFile(      genFile,      [        "/*! Generated by Aleph.js, do NOT change and NO git-ignore. */",        "",        ...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;      }    }  }}