@analogjs/vite-plugin-angular 2.0.0-alpha.1 → 2.0.0-alpha.10

package/src/lib/angular-vite-plugin.js

@@ -1,14 +1,15 @@
-import { dirname, resolve } from 'node:path';
+import { basename, dirname, isAbsolute, relative, resolve } from 'node:path';
+import { mkdirSync, writeFileSync } from 'node:fs';
 import * as compilerCli from '@angular/compiler-cli';
-import * as ts from 'typescript';
 import { createRequire } from 'node:module';
 import { normalizePath, preprocessCSS, } from 'vite';
+import * as ngCompiler from '@angular/compiler';
 import { createCompilerPlugin } from './compiler-plugin.js';
 import { StyleUrlsResolver, TemplateUrlsResolver, } from './component-resolvers.js';
 import { augmentHostWithCaching, augmentHostWithResources, augmentProgramWithVersioning, mergeTransformers, } from './host.js';
 import { jitPlugin } from './angular-jit-plugin.js';
 import { buildOptimizerPlugin } from './angular-build-optimizer-plugin.js';
-import { createJitResourceTransformer, SourceFileCache, } from './utils/devkit.js';
+import { createJitResourceTransformer, SourceFileCache, angularMajor, } from './utils/devkit.js';
 import { angularVitestPlugins } from './angular-vitest-plugin.js';
 import { angularStorybookPlugin } from './angular-storybook-plugin.js';
 const require = createRequire(import.meta.url);
@@ -16,22 +17,22 @@ import { getFrontmatterMetadata } from './authoring/frontmatter.js';
 import { defaultMarkdownTemplateTransforms, } from './authoring/markdown-transform.js';
 import { routerPlugin } from './router-plugin.js';
 import { pendingTasksPlugin } from './angular-pending-tasks.plugin.js';
+import { analyzeFileUpdates } from './utils/hmr-candidates.js';
 /**
  * TypeScript file extension regex
  * Match .(c or m)ts, .ts extensions with an optional ? for query params
  * Ignore .tsx extensions
  */
 const TS_EXT_REGEX = /\.[cm]?(ts|analog|ag)[^x]?\??/;
+const ANGULAR_COMPONENT_PREFIX = '/@ng/component';
+const classNames = new Map();
 export function angular(options) {
     /**
      * Normalize plugin options so defaults
      * are used for values not provided.
      */
     const pluginOptions = {
-        tsconfig: options?.tsconfig ??
-            (process.env['NODE_ENV'] === 'test'
-                ? './tsconfig.spec.json'
-                : './tsconfig.app.json'),
+        tsconfig: options?.tsconfig || '',
         workspaceRoot: options?.workspaceRoot ?? process.cwd(),
         inlineStylesExtension: options?.inlineStylesExtension ?? 'css',
         advanced: {
@@ -50,20 +51,19 @@ export function angular(options) {
             : defaultMarkdownTemplateTransforms,
         include: options?.include ?? [],
         additionalContentDirs: options?.additionalContentDirs ?? [],
+        liveReload: options?.liveReload ?? false,
+        disableTypeChecking: options?.disableTypeChecking ?? true,
     };
-    // The file emitter created during `onStart` that will be used during the build in `onLoad` callbacks for TS files
-    let fileEmitter;
-    let compilerOptions = {};
-    const ts = require('typescript');
     let resolvedConfig;
-    let rootNames;
-    let host;
     let nextProgram;
     let builderProgram;
     let watchMode = false;
-    let testWatchMode = false;
+    let testWatchMode = isTestWatchMode();
+    let inlineComponentStyles;
+    let externalComponentStyles;
     const sourceFileCache = new SourceFileCache();
     const isTest = process.env['NODE_ENV'] === 'test' || !!process.env['VITEST'];
+    const isVitestVscode = !!process.env['VITEST_VSCODE'];
     const isStackBlitz = !!process.versions['webcontainer'];
     const isAstroIntegration = process.env['ANALOG_ASTRO'] === 'true';
     const isStorybook = process.env['npm_lifecycle_script']?.includes('storybook') ||
@@ -72,28 +72,27 @@ export function angular(options) {
         process.env['ANALOG_STORYBOOK'] === 'true';
     const jit = typeof pluginOptions?.jit !== 'undefined' ? pluginOptions.jit : isTest;
     let viteServer;
-    let styleTransform;
     const styleUrlsResolver = new StyleUrlsResolver();
     const templateUrlsResolver = new TemplateUrlsResolver();
+    const outputFiles = new Map();
+    const fileEmitter = (file) => {
+        return outputFiles.get(normalizePath(file));
+    };
+    let initialCompilation = false;
+    const declarationFiles = [];
     function angularPlugin() {
         let isProd = false;
+        if (angularMajor < 19 || isTest) {
+            pluginOptions.liveReload = false;
+        }
         return {
             name: '@analogjs/vite-plugin-angular',
-            async watchChange() {
-                if (isTest) {
-                    await buildAndAnalyze();
-                }
-            },
             async config(config, { command }) {
                 watchMode = command === 'serve';
                 isProd =
                     config.mode === 'production' ||
                         process.env['NODE_ENV'] === 'production';
-                pluginOptions.tsconfig =
-                    options?.tsconfig ??
-                        resolve(config.root || '.', process.env['NODE_ENV'] === 'test'
-                            ? './tsconfig.spec.json'
-                            : './tsconfig.app.json');
+                pluginOptions.tsconfig = getTsConfigPath(config.root || '.', pluginOptions, isProd, isTest, !!config?.build?.lib);
                 return {
                     esbuild: config.esbuild ?? false,
                     optimizeDeps: {
@@ -123,44 +122,94 @@ export function angular(options) {
             },
             configResolved(config) {
                 resolvedConfig = config;
-                // set test watch mode
-                // - vite override from vitest-angular
-                // - @nx/vite executor set server.watch explicitly to undefined (watch)/null (watch=false)
-                // - vite config for test.watch variable
-                testWatchMode =
-                    !(config.server.watch === null) ||
-                        config.test?.watch === true;
+                if (isTest) {
+                    // set test watch mode
+                    // - vite override from vitest-angular
+                    // - @nx/vite executor set server.watch explicitly to undefined (watch)/null (watch=false)
+                    // - vite config for test.watch variable
+                    // - vitest watch mode detected from the command line
+                    testWatchMode =
+                        !(config.server.watch === null) ||
+                            config.test?.watch === true ||
+                            testWatchMode;
+                }
             },
             configureServer(server) {
                 viteServer = server;
                 server.watcher.on('add', async () => {
-                    setupCompilation(resolvedConfig);
-                    await buildAndAnalyze();
+                    await performCompilation(resolvedConfig);
                 });
                 server.watcher.on('unlink', async () => {
-                    setupCompilation(resolvedConfig);
-                    await buildAndAnalyze();
+                    await performCompilation(resolvedConfig);
                 });
+                if (pluginOptions.liveReload) {
+                    const angularComponentMiddleware = async (req, res, next) => {
+                        if (req.url === undefined || res.writableEnded) {
+                            return;
+                        }
+                        if (!req.url.includes(ANGULAR_COMPONENT_PREFIX)) {
+                            next();
+                            return;
+                        }
+                        const requestUrl = new URL(req.url, 'http://localhost');
+                        const componentId = requestUrl.searchParams.get('c');
+                        if (!componentId) {
+                            res.statusCode = 400;
+                            res.end();
+                            return;
+                        }
+                        const [fileId] = decodeURIComponent(componentId).split('@');
+                        const resolvedId = resolve(process.cwd(), fileId);
+                        const invalidated = !!server.moduleGraph.getModuleById(resolvedId)
+                            ?.lastInvalidationTimestamp && classNames.get(resolvedId);
+                        // don't send an HMR update until the file has been invalidated
+                        if (!invalidated) {
+                            res.setHeader('Content-Type', 'text/javascript');
+                            res.setHeader('Cache-Control', 'no-cache');
+                            res.end('');
+                            return;
+                        }
+                        const result = fileEmitter(resolvedId);
+                        res.setHeader('Content-Type', 'text/javascript');
+                        res.setHeader('Cache-Control', 'no-cache');
+                        res.end(`${result?.hmrUpdateCode || ''}`);
+                    };
+                    viteServer.middlewares.use(angularComponentMiddleware);
+                }
             },
             async buildStart() {
-                setupCompilation(resolvedConfig);
-                // Only store cache if in watch mode
-                if (watchMode) {
-                    augmentHostWithCaching(host, sourceFileCache);
+                // Defer the first compilation in test mode
+                if (!isVitestVscode) {
+                    const { host } = await performCompilation(resolvedConfig);
+                    initialCompilation = true;
+                    // Only store cache if in watch mode
+                    if (watchMode) {
+                        augmentHostWithCaching(host, sourceFileCache);
+                    }
                 }
-                await buildAndAnalyze();
             },
             async handleHotUpdate(ctx) {
-                // The `handleHotUpdate` hook may be called before the `buildStart`,
-                // which sets the compilation. As a result, the `host` may not be available
-                // yet for use, leading to build errors such as "cannot read properties of undefined"
-                // (because `host` is undefined).
-                if (!host) {
-                    return;
-                }
                 if (TS_EXT_REGEX.test(ctx.file)) {
-                    sourceFileCache.invalidate([ctx.file.replace(/\?(.*)/, '')]);
-                    await buildAndAnalyze();
+                    let [fileId] = ctx.file.split('?');
+                    if (pluginOptions.supportAnalogFormat &&
+                        ['ag', 'analog', 'agx'].some((ext) => fileId.endsWith(ext))) {
+                        fileId += '.ts';
+                    }
+                    sourceFileCache.invalidate([fileId]);
+                    await performCompilation(resolvedConfig, [fileId]);
+                    const result = fileEmitter(fileId);
+                    if (pluginOptions.liveReload &&
+                        !!result?.hmrEligible &&
+                        classNames.get(fileId)) {
+                        const relativeFileId = `${relative(process.cwd(), fileId)}@${classNames.get(fileId)}`;
+                        sendHMRComponentUpdate(ctx.server, relativeFileId);
+                        return ctx.modules.map((mod) => {
+                            if (mod.id === ctx.file) {
+                                return markModuleSelfAccepting(mod);
+                            }
+                            return mod;
+                        });
+                    }
                 }
                 if (/\.(html|htm|css|less|sass|scss)$/.test(ctx.file)) {
                     /**
@@ -169,28 +218,116 @@ export function angular(options) {
                      */
                     const isDirect = ctx.modules.find((mod) => ctx.file === mod.file && mod.id?.includes('?direct'));
                     if (isDirect) {
+                        if (pluginOptions.liveReload && isDirect?.id && isDirect.file) {
+                            const isComponentStyle = isDirect.type === 'css' && isComponentStyleSheet(isDirect.id);
+                            if (isComponentStyle) {
+                                const { encapsulation } = getComponentStyleSheetMeta(isDirect.id);
+                                // Track if the component uses ShadowDOM encapsulation
+                                // Shadow DOM components currently require a full reload.
+                                // Vite's CSS hot replacement does not support shadow root searching.
+                                if (encapsulation !== 'shadow') {
+                                    ctx.server.ws.send({
+                                        type: 'update',
+                                        updates: [
+                                            {
+                                                type: 'css-update',
+                                                timestamp: Date.now(),
+                                                path: isDirect.url,
+                                                acceptedPath: isDirect.file,
+                                            },
+                                        ],
+                                    });
+                                    return ctx.modules
+                                        .filter((mod) => {
+                                        // Component stylesheets will have 2 modules (*.component.scss and *.component.scss?direct&ngcomp=xyz&e=x)
+                                        // We remove the module with the query params to prevent vite double logging the stylesheet name "hmr update *.component.scss, *.component.scss?direct&ngcomp=xyz&e=x"
+                                        return mod.file !== ctx.file || mod.id !== isDirect.id;
+                                    })
+                                        .map((mod) => {
+                                        if (mod.file === ctx.file) {
+                                            return markModuleSelfAccepting(mod);
+                                        }
+                                        return mod;
+                                    });
+                                }
+                            }
+                        }
                         return ctx.modules;
                     }
                     const mods = [];
+                    const updates = [];
                     ctx.modules.forEach((mod) => {
                         mod.importers.forEach((imp) => {
                             sourceFileCache.invalidate([imp.id]);
                             ctx.server.moduleGraph.invalidateModule(imp);
-                            mods.push(imp);
+                            if (pluginOptions.liveReload && classNames.get(imp.id)) {
+                                updates.push(imp.id);
+                            }
+                            else {
+                                mods.push(imp);
+                            }
                         });
                     });
-                    await buildAndAnalyze();
+                    await performCompilation(resolvedConfig, updates);
+                    if (updates.length > 0) {
+                        updates.forEach((updateId) => {
+                            const impRelativeFileId = `${relative(process.cwd(), updateId)}@${classNames.get(updateId)}`;
+                            sendHMRComponentUpdate(ctx.server, impRelativeFileId);
+                        });
+                        return ctx.modules.map((mod) => {
+                            if (mod.id === ctx.file) {
+                                return markModuleSelfAccepting(mod);
+                            }
+                            return mod;
+                        });
+                    }
                     return mods;
                 }
+                // clear HMR updates with a full reload
+                classNames.clear();
                 return ctx.modules;
             },
-            resolveId(id, importer) {
+            resolveId(id, importer, options) {
                 if (id.startsWith('angular:jit:')) {
                     const path = id.split(';')[1];
                     return `${normalizePath(resolve(dirname(importer), path))}?raw`;
                 }
+                // Map angular external styleUrls to the source file
+                if (isComponentStyleSheet(id)) {
+                    const componentStyles = externalComponentStyles?.get(getFilenameFromPath(id));
+                    if (componentStyles) {
+                        return componentStyles + new URL(id, 'http://localhost').search;
+                    }
+                }
+                if (options?.ssr && id.includes(ANGULAR_COMPONENT_PREFIX)) {
+                    const requestUrl = new URL(id.slice(1), 'http://localhost');
+                    const componentId = requestUrl.searchParams.get('c');
+                    const res = resolve(process.cwd(), decodeURIComponent(componentId).split('@')[0]);
+                    return res;
+                }
                 return undefined;
             },
+            async load(id, options) {
+                // Map angular inline styles to the source text
+                if (isComponentStyleSheet(id)) {
+                    const componentStyles = inlineComponentStyles?.get(getFilenameFromPath(id));
+                    if (componentStyles) {
+                        return componentStyles;
+                    }
+                }
+                if (pluginOptions.liveReload &&
+                    options?.ssr &&
+                    id.includes(ANGULAR_COMPONENT_PREFIX)) {
+                    const requestUrl = new URL(id.slice(1), 'http://localhost');
+                    const componentId = requestUrl.searchParams.get('c');
+                    if (!componentId) {
+                        return;
+                    }
+                    const result = fileEmitter(resolve(process.cwd(), decodeURIComponent(componentId).split('@')[0]));
+                    return result?.hmrUpdateCode || '';
+                }
+                return;
+            },
             async transform(code, id) {
                 // Skip transforming node_modules
                 if (id.includes('node_modules')) {
@@ -220,6 +357,19 @@ export function angular(options) {
                 if (id.includes('analog-content-')) {
                     return;
                 }
+                /**
+                 * Encapsulate component stylesheets that use emulated encapsulation
+                 */
+                if (pluginOptions.liveReload && isComponentStyleSheet(id)) {
+                    const { encapsulation, componentId } = getComponentStyleSheetMeta(id);
+                    if (encapsulation === 'emulated' && componentId) {
+                        const encapsulated = ngCompiler.encapsulateStyle(code, componentId);
+                        return {
+                            code: encapsulated,
+                            map: null,
+                        };
+                    }
+                }
                 if (TS_EXT_REGEX.test(id)) {
                     if (id.includes('.ts?')) {
                         // Strip the query string off the ID
@@ -231,11 +381,17 @@ export function angular(options) {
                      * for test(Vitest)
                      */
                     if (isTest) {
+                        if (isVitestVscode && !initialCompilation) {
+                            // Do full initial compilation
+                            await performCompilation(resolvedConfig);
+                            initialCompilation = true;
+                        }
                         const tsMod = viteServer?.moduleGraph.getModuleById(id);
                         if (tsMod) {
-                            sourceFileCache.invalidate([id]);
-                            if (testWatchMode) {
-                                await buildAndAnalyze();
+                            const invalidated = tsMod.lastInvalidationTimestamp;
+                            if (testWatchMode && invalidated) {
+                                sourceFileCache.invalidate([id]);
+                                await performCompilation(resolvedConfig, [id]);
                             }
                         }
                     }
@@ -250,7 +406,7 @@ export function angular(options) {
                             this.addWatchFile(absoluteFileUrl);
                         }
                     }
-                    const typescriptResult = await fileEmitter?.(id);
+                    const typescriptResult = fileEmitter(id);
                     if (typescriptResult?.warnings &&
                         typescriptResult?.warnings.length > 0) {
                         this.warn(`${typescriptResult.warnings.join('\n')}`);
@@ -283,7 +439,7 @@ export function angular(options) {
                         pluginOptions.supportAnalogFormat &&
                         fileEmitter) {
                         sourceFileCache.invalidate([`${id}.ts`]);
-                        const ngFileResult = await fileEmitter(`${id}.ts`);
+                        const ngFileResult = fileEmitter(`${id}.ts`);
                         data = ngFileResult?.content || '';
                         if (id.includes('.agx')) {
                             const metadata = await getFrontmatterMetadata(code, id, pluginOptions.markdownTemplateTransforms || []);
@@ -297,6 +453,12 @@ export function angular(options) {
                 }
                 return undefined;
             },
+            closeBundle() {
+                declarationFiles.forEach(({ declarationFileDir, declarationPath, data }) => {
+                    mkdirSync(declarationFileDir, { recursive: true });
+                    writeFileSync(declarationPath, data, 'utf-8');
+                });
+            },
         };
     }
     return [
@@ -331,7 +493,7 @@ export function angular(options) {
         const globs = [
             `${appRoot}/**/*.{analog,agx,ag}`,
             ...extraGlobs.map((glob) => `${workspaceRoot}${glob}.{analog,agx,ag}`),
-            ...(pluginOptions.additionalContentDirs || [])?.map((glob) => `${workspaceRoot}${glob}/**/*.agx`),
+            ...(pluginOptions.additionalContentDirs || []).map((glob) => `${workspaceRoot}${glob}/**/*.agx`),
             ...pluginOptions.include.map((glob) => `${workspaceRoot}${glob}`.replace(/\.ts$/, '.analog')),
         ];
         return fg
@@ -350,11 +512,29 @@ export function angular(options) {
             dot: true,
         });
     }
-    function setupCompilation(config, context) {
+    function getTsConfigPath(root, options, isProd, isTest, isLib) {
+        if (options.tsconfig && isAbsolute(options.tsconfig)) {
+            return options.tsconfig;
+        }
+        let tsconfigFilePath = './tsconfig.app.json';
+        if (isLib) {
+            tsconfigFilePath = isProd
+                ? './tsconfig.lib.prod.json'
+                : './tsconfig.lib.json';
+        }
+        if (isTest) {
+            tsconfigFilePath = './tsconfig.spec.json';
+        }
+        if (options.tsconfig) {
+            tsconfigFilePath = options.tsconfig;
+        }
+        return resolve(root, tsconfigFilePath);
+    }
+    async function performCompilation(config, ids) {
         const isProd = config.mode === 'production';
         const analogFiles = findAnalogFiles(config);
         const includeFiles = findIncludes();
-        const { options: tsCompilerOptions, rootNames: rn } = compilerCli.readConfiguration(pluginOptions.tsconfig, {
+        let { options: tsCompilerOptions, rootNames } = compilerCli.readConfiguration(pluginOptions.tsconfig, {
             suppressOutputPathCheck: true,
             outDir: undefined,
             sourceMap: false,
@@ -377,89 +557,262 @@ export function angular(options) {
             // AOT and virtually compiled .analog files.
             tsCompilerOptions.compilationMode = 'experimental-local';
         }
-        rootNames = rn.concat(analogFiles, includeFiles);
-        compilerOptions = tsCompilerOptions;
-        host = ts.createIncrementalCompilerHost(compilerOptions);
-        styleTransform = (code, filename) => preprocessCSS(code, filename, config);
+        if (pluginOptions.liveReload && watchMode) {
+            tsCompilerOptions['_enableHmr'] = true;
+            tsCompilerOptions['externalRuntimeStyles'] = true;
+            // Workaround for https://github.com/angular/angular/issues/59310
+            // Force extra instructions to be generated for HMR w/defer
+            tsCompilerOptions['supportTestBed'] = true;
+        }
+        if (tsCompilerOptions.compilationMode === 'partial') {
+            // These options can't be false in partial mode
+            tsCompilerOptions['supportTestBed'] = true;
+            tsCompilerOptions['supportJitMode'] = true;
+        }
+        if (!isTest && config.build?.lib) {
+            tsCompilerOptions['declaration'] = true;
+            tsCompilerOptions['declarationMap'] = watchMode;
+            tsCompilerOptions['inlineSources'] = true;
+        }
+        rootNames = rootNames.concat(analogFiles, includeFiles);
+        const ts = require('typescript');
+        const host = ts.createIncrementalCompilerHost(tsCompilerOptions);
         if (!jit) {
+            const styleTransform = (code, filename) => preprocessCSS(code, filename, config);
+            inlineComponentStyles = tsCompilerOptions['externalRuntimeStyles']
+                ? new Map()
+                : undefined;
+            externalComponentStyles = tsCompilerOptions['externalRuntimeStyles']
+                ? new Map()
+                : undefined;
             augmentHostWithResources(host, styleTransform, {
                 inlineStylesExtension: pluginOptions.inlineStylesExtension,
                 supportAnalogFormat: pluginOptions.supportAnalogFormat,
                 isProd,
                 markdownTemplateTransforms: pluginOptions.markdownTemplateTransforms,
+                inlineComponentStyles,
+                externalComponentStyles,
             });
         }
-    }
-    /**
-     * Creates a new NgtscProgram to analyze/re-analyze
-     * the source files and create a file emitter.
-     * This is shared between an initial build and a hot update.
-     */
-    async function buildAndAnalyze() {
+        /**
+         * Creates a new NgtscProgram to analyze/re-analyze
+         * the source files and create a file emitter.
+         * This is shared between an initial build and a hot update.
+         */
         let builder;
         let typeScriptProgram;
         let angularCompiler;
         if (!jit) {
             // Create the Angular specific program that contains the Angular compiler
-            const angularProgram = new compilerCli.NgtscProgram(rootNames, compilerOptions, host, nextProgram);
+            const angularProgram = new compilerCli.NgtscProgram(ids && ids.length > 0 ? ids : rootNames, tsCompilerOptions, host, nextProgram);
             angularCompiler = angularProgram.compiler;
             typeScriptProgram = angularProgram.getTsProgram();
             augmentProgramWithVersioning(typeScriptProgram);
-            builder = builderProgram =
-                ts.createEmitAndSemanticDiagnosticsBuilderProgram(typeScriptProgram, host, builderProgram);
+            builder = ts.createEmitAndSemanticDiagnosticsBuilderProgram(typeScriptProgram, host, builderProgram);
             await angularCompiler.analyzeAsync();
             nextProgram = angularProgram;
+            builderProgram =
+                builder;
         }
         else {
-            builder = builderProgram =
-                ts.createEmitAndSemanticDiagnosticsBuilderProgram(rootNames, compilerOptions, host, nextProgram);
+            builder = ts.createEmitAndSemanticDiagnosticsBuilderProgram(rootNames, tsCompilerOptions, host, nextProgram);
             typeScriptProgram = builder.getProgram();
-            nextProgram = builderProgram;
         }
         if (!watchMode) {
             // When not in watch mode, the startup cost of the incremental analysis can be avoided by
             // using an abstract builder that only wraps a TypeScript program.
             builder = ts.createAbstractBuilder(typeScriptProgram, host);
         }
-        const getTypeChecker = () => builder.getProgram().getTypeChecker();
-        fileEmitter = createFileEmitter(builder, mergeTransformers({
-            before: [
-                ...(jit
-                    ? [
-                        compilerCli.constructorParametersDownlevelTransform(builder.getProgram()),
-                        createJitResourceTransformer(getTypeChecker),
-                    ]
-                    : []),
-                ...pluginOptions.advanced.tsTransformers.before,
-            ],
-            after: pluginOptions.advanced.tsTransformers.after,
-            afterDeclarations: pluginOptions.advanced.tsTransformers.afterDeclarations,
-        }, jit ? {} : angularCompiler.prepareEmit().transformers), () => [], angularCompiler);
+        const beforeTransformers = jit
+            ? [
+                compilerCli.constructorParametersDownlevelTransform(builder.getProgram()),
+                createJitResourceTransformer(() => builder.getProgram().getTypeChecker()),
+            ]
+            : [];
+        const transformers = mergeTransformers({ before: beforeTransformers }, jit ? {} : angularCompiler.prepareEmit().transformers);
+        const fileMetadata = getFileMetadata(builder, angularCompiler, pluginOptions.liveReload, pluginOptions.disableTypeChecking);
+        const writeFileCallback = (_filename, content, _a, _b, sourceFiles) => {
+            if (!sourceFiles?.length) {
+                return;
+            }
+            const filename = normalizePath(sourceFiles[0].fileName);
+            if (filename.includes('ngtypecheck.ts') || filename.includes('.d.')) {
+                return;
+            }
+            const metadata = watchMode
+                ? fileMetadata(filename, sourceFileCache.get(filename))
+                : {};
+            outputFiles.set(filename, {
+                content,
+                dependencies: [],
+                errors: metadata.errors,
+                warnings: metadata.warnings,
+                hmrUpdateCode: metadata.hmrUpdateCode,
+                hmrEligible: metadata.hmrEligible,
+            });
+        };
+        const writeOutputFile = (id) => {
+            const sourceFile = builder.getSourceFile(id);
+            if (!sourceFile) {
+                return;
+            }
+            let content = '';
+            builder.emit(sourceFile, (filename, data) => {
+                if (/\.[cm]?js$/.test(filename)) {
+                    content = data;
+                }
+                if (!watchMode &&
+                    !isTest &&
+                    /\.d\.ts/.test(filename) &&
+                    !filename.includes('.ngtypecheck.')) {
+                    // output to library root instead /src
+                    const declarationPath = resolve(config.root, config.build.outDir, relative(config.root, filename)).replace('/src/', '/');
+                    const declarationFileDir = declarationPath
+                        .replace(basename(filename), '')
+                        .replace('/src/', '/');
+                    declarationFiles.push({
+                        declarationFileDir,
+                        declarationPath,
+                        data,
+                    });
+                }
+            }, undefined /* cancellationToken */, undefined /* emitOnlyDtsFiles */, transformers);
+            writeFileCallback(id, content, false, undefined, [sourceFile]);
+        };
+        if (!watchMode) {
+            for (const sf of builder.getSourceFiles()) {
+                const id = sf.fileName;
+                writeOutputFile(id);
+            }
+        }
+        else {
+            if (ids && ids.length > 0) {
+                ids.forEach((id) => writeOutputFile(id));
+            }
+            else {
+                // TypeScript will loop until there are no more affected files in the program
+                while (builder.emitNextAffectedFile(writeFileCallback, undefined, undefined, transformers)) {
+                    /* empty */
+                }
+            }
+        }
+        return { host };
     }
 }
-export function createFileEmitter(program, transformers = {}, onAfterEmit, angularCompiler) {
-    return async (file) => {
+function sendHMRComponentUpdate(server, id) {
+    server.ws.send('angular:component-update', {
+        id: encodeURIComponent(id),
+        timestamp: Date.now(),
+    });
+    classNames.delete(id);
+}
+export function getFileMetadata(program, angularCompiler, liveReload, disableTypeChecking) {
+    const ts = require('typescript');
+    return (file, stale) => {
         const sourceFile = program.getSourceFile(file);
         if (!sourceFile) {
-            return undefined;
+            return {};
         }
-        const diagnostics = angularCompiler
-            ? angularCompiler.getDiagnosticsForFile(sourceFile, 1)
-            : [];
+        const hmrEligible = liveReload && stale
+            ? !!analyzeFileUpdates(stale, sourceFile, angularCompiler)
+            : false;
+        const diagnostics = getDiagnosticsForSourceFile(sourceFile, !!disableTypeChecking, program, angularCompiler);
         const errors = diagnostics
             .filter((d) => d.category === ts.DiagnosticCategory?.Error)
-            .map((d) => d.messageText);
+            .map((d) => typeof d.messageText === 'object'
+            ? d.messageText.messageText
+            : d.messageText);
         const warnings = diagnostics
             .filter((d) => d.category === ts.DiagnosticCategory?.Warning)
             .map((d) => d.messageText);
-        let content;
-        program.emit(sourceFile, (filename, data) => {
-            if (/\.[cm]?js$/.test(filename)) {
-                content = data;
+        let hmrUpdateCode = undefined;
+        if (liveReload) {
+            for (const node of sourceFile.statements) {
+                if (ts.isClassDeclaration(node) && node.name != null) {
+                    hmrUpdateCode = angularCompiler?.emitHmrUpdateModule(node);
+                    !!hmrUpdateCode && classNames.set(file, node.name.getText());
+                }
             }
-        }, undefined /* cancellationToken */, undefined /* emitOnlyDtsFiles */, transformers);
-        onAfterEmit?.(sourceFile);
-        return { content, dependencies: [], errors, warnings };
+        }
+        return { errors, warnings, hmrUpdateCode, hmrEligible };
     };
 }
+function getDiagnosticsForSourceFile(sourceFile, disableTypeChecking, program, angularCompiler) {
+    const syntacticDiagnostics = program.getSyntacticDiagnostics(sourceFile);
+    if (disableTypeChecking) {
+        // Syntax errors are cheap to compute and the app will not run if there are any
+        // So always show these types of errors regardless if type checking is disabled
+        return syntacticDiagnostics;
+    }
+    const semanticDiagnostics = program.getSemanticDiagnostics(sourceFile);
+    const angularDiagnostics = angularCompiler
+        ? angularCompiler.getDiagnosticsForFile(sourceFile, 1)
+        : [];
+    return [
+        ...syntacticDiagnostics,
+        ...semanticDiagnostics,
+        ...angularDiagnostics,
+    ];
+}
+function markModuleSelfAccepting(mod) {
+    // support Vite 6
+    if ('_clientModule' in mod) {
+        mod['_clientModule'].isSelfAccepting = true;
+    }
+    return {
+        ...mod,
+        isSelfAccepting: true,
+    };
+}
+function isComponentStyleSheet(id) {
+    return id.includes('ngcomp=');
+}
+function getComponentStyleSheetMeta(id) {
+    const params = new URL(id, 'http://localhost').searchParams;
+    const encapsulationMapping = {
+        '0': 'emulated',
+        '2': 'none',
+        '3': 'shadow',
+    };
+    return {
+        componentId: params.get('ngcomp'),
+        encapsulation: encapsulationMapping[params.get('e')],
+    };
+}
+/**
+ * Removes leading / and query string from a url path
+ * e.g. /foo.scss?direct&ngcomp=ng-c3153525609&e=0 returns foo.scss
+ * @param id
+ */
+function getFilenameFromPath(id) {
+    return new URL(id, 'http://localhost').pathname.replace(/^\//, '');
+}
+/**
+ * Checks for vitest run from the command line
+ * @returns boolean
+ */
+export function isTestWatchMode(args = process.argv) {
+    // vitest --run
+    const hasRun = args.find((arg) => arg.includes('--run'));
+    if (hasRun) {
+        return false;
+    }
+    // vitest --no-run
+    const hasNoRun = args.find((arg) => arg.includes('--no-run'));
+    if (hasNoRun) {
+        return true;
+    }
+    // check for --watch=false or --no-watch
+    const hasWatch = args.find((arg) => arg.includes('watch'));
+    if (hasWatch && ['false', 'no'].some((neg) => hasWatch.includes(neg))) {
+        return false;
+    }
+    // check for --watch false
+    const watchIndex = args.findIndex((arg) => arg.includes('watch'));
+    const watchArg = args[watchIndex + 1];
+    if (watchArg && watchArg === 'false') {
+        return false;
+    }
+    return true;
+}
 //# sourceMappingURL=angular-vite-plugin.js.map