Function createHtmlAssetRegistry

packages/server/html-assets.ts:17–100

()

export { inlineHtmlLocalAssets };

export function createHtmlAssetRegistry() {
  const rootsByToken = new Map<string, string>();
  const tokensByRoot = new Map<string, string>();

  function register(baseDir: string): string {
    const root = resolvePath(baseDir);
    const existing = tokensByRoot.get(root);
    if (existing) return existing;
    const token = crypto.randomUUID().replace(/-/g, "").slice(0, 16);
    tokensByRoot.set(root, token);
    rootsByToken.set(token, root);
    return token;
  }

  function rewriteHtml(html: string, htmlFilePath: string): string {
    if (/^https?:\/\//i.test(htmlFilePath)) return html;
    try {
      const token = register(dirname(resolvePath(htmlFilePath)));
      return rewriteHtmlAssetReferences(
        html,
        (assetPath) => `${HTML_ASSET_ROUTE_PREFIX}/${token}/${encodeHtmlAssetPath(assetPath)}`,
      );
    } catch {
      return html;
    }
  }

  function inlineHtml(html: string, htmlFilePath: string): string {
    return inlineHtmlLocalAssets(html, htmlFilePath);
  }

  async function handle(_req: Request, url: URL): Promise<Response | null> {
    const prefix = `${HTML_ASSET_ROUTE_PREFIX}/`;
    if (!url.pathname.startsWith(prefix)) return null;

    const rest = url.pathname.slice(prefix.length);
    const slash = rest.indexOf("/");
    if (slash <= 0) {
      return Response.json({ error: "Missing asset token or path" }, { status: 404 });
    }

    const token = rest.slice(0, slash);
    const root = rootsByToken.get(token);
    if (!root) {
      return Response.json({ error: "Unknown asset root" }, { status: 404 });
    }

    const assetPath = normalizeHtmlAssetRoutePath(rest.slice(slash + 1));
    if (!assetPath) {
      return Response.json({ error: "Invalid asset path" }, { status: 400 });
    }

    const contentType = htmlAssetContentType(assetPath);
    if (!contentType) {
      return Response.json({ error: "Unsupported asset type" }, { status: 415 });
    }

    const resolved = resolvePath(root, assetPath);