From 82b0f94fdacc5f4665d00eeb1c93fcfc104b0cc8 Mon Sep 17 00:00:00 2001
From: Alan Agius <alanagius@google.com>
Date: Wed, 16 Aug 2023 16:07:20 +0000
Subject: [PATCH] fix(@angular-devkit/build-angular): handle HMR updates of
 global CSS when using Vite

This commit updates the Vite based dev-server to send updates for global styles using the HMR update instead of a full page reload when the the dev-server builder `hmr` option to be enabled.

(cherry picked from commit b09ce572ed72f19c353e71b7d9d751484c5db63a)
---
 .../src/builders/dev-server/vite-server.ts    | 75 ++++++++++++++-----
 1 file changed, 57 insertions(+), 18 deletions(-)

diff --git a/packages/angular_devkit/build_angular/src/builders/dev-server/vite-server.ts b/packages/angular_devkit/build_angular/src/builders/dev-server/vite-server.ts
index 441b13712cb5..9adff6640e5b 100644
--- a/packages/angular_devkit/build_angular/src/builders/dev-server/vite-server.ts
+++ b/packages/angular_devkit/build_angular/src/builders/dev-server/vite-server.ts
@@ -7,7 +7,7 @@
  */
 
 import type { BuilderContext } from '@angular-devkit/architect';
-import type { json } from '@angular-devkit/core';
+import type { json, logging } from '@angular-devkit/core';
 import type { OutputFile } from 'esbuild';
 import { lookup as lookupMimeType } from 'mrmime';
 import assert from 'node:assert';
@@ -86,23 +86,7 @@ export async function* serveWithVite(
     }
 
     if (server) {
-      // Invalidate any updated files
-      for (const [file, record] of generatedFiles) {
-        if (record.updated) {
-          const updatedModules = server.moduleGraph.getModulesByFile(file);
-          updatedModules?.forEach((m) => server?.moduleGraph.invalidateModule(m));
-        }
-      }
-
-      // Send reload command to clients
-      if (serverOptions.liveReload) {
-        context.logger.info('Reloading client(s)...');
-
-        server.ws.send({
-          type: 'full-reload',
-          path: '*',
-        });
-      }
+      handleUpdate(generatedFiles, server, serverOptions, context.logger);
     } else {
       // Setup server and start listening
       const serverConfiguration = await setupServer(
@@ -137,6 +121,61 @@ export async function* serveWithVite(
   }
 }
 
+function handleUpdate(
+  generatedFiles: Map<string, OutputFileRecord>,
+  server: ViteDevServer,
+  serverOptions: NormalizedDevServerOptions,
+  logger: logging.LoggerApi,
+): void {
+  const updatedFiles: string[] = [];
+
+  // Invalidate any updated files
+  for (const [file, record] of generatedFiles) {
+    if (record.updated) {
+      updatedFiles.push(file);
+      const updatedModules = server.moduleGraph.getModulesByFile(file);
+      updatedModules?.forEach((m) => server?.moduleGraph.invalidateModule(m));
+    }
+  }
+
+  if (!updatedFiles.length) {
+    return;
+  }
+
+  if (serverOptions.hmr) {
+    if (updatedFiles.every((f) => f.endsWith('.css'))) {
+      const timestamp = Date.now();
+      server.ws.send({
+        type: 'update',
+        updates: updatedFiles.map((f) => {
+          const filePath = f.slice(1); // Remove leading slash.
+
+          return {
+            type: 'css-update',
+            timestamp,
+            path: filePath,
+            acceptedPath: filePath,
+          };
+        }),
+      });
+
+      logger.info('HMR update sent to client(s)...');
+
+      return;
+    }
+  }
+
+  // Send reload command to clients
+  if (serverOptions.liveReload) {
+    logger.info('Reloading client(s)...');
+
+    server.ws.send({
+      type: 'full-reload',
+      path: '*',
+    });
+  }
+}
+
 function analyzeResultFiles(
   resultFiles: OutputFile[],
   generatedFiles: Map<string, OutputFileRecord>,