vite.ts — astro Source File

Source Code

import npath from 'node:path';
import { type EnvironmentModuleNode, isCSSRequest, type RunnableDevEnvironment } from 'vite';
import { SUPPORTED_MARKDOWN_FILE_EXTENSIONS } from '../core/constants.js';
import { unwrapId } from '../core/util.js';
import { hasSpecialQueries } from '../vite-plugin-utils/index.js';
import { PROPAGATED_ASSET_QUERY_PARAM } from '../content/consts.js';

/**
 * List of file extensions signalling we can (and should) SSR ahead-of-time
 * See usage below
 */
const fileExtensionsToSSR = new Set(['.astro', '.mdoc', ...SUPPORTED_MARKDOWN_FILE_EXTENSIONS]);

const STRIP_QUERY_PARAMS_REGEX = /\?.*$/;

/** recursively crawl the module graph to get all style files imported by parent id */
export async function* crawlGraph(
	environment: RunnableDevEnvironment,
	_id: string,
	isRootFile: boolean,
	scanned = new Set<string>(),
): AsyncGenerator<EnvironmentModuleNode, void, unknown> {
	const id = unwrapId(_id);
	const importedModules = new Set<EnvironmentModuleNode>();

	const moduleEntriesForId = isRootFile
		? // "getModulesByFile" pulls from a delayed module cache (fun implementation detail),
			// So we can get up-to-date info on initial server load.
			// Needed for slower CSS preprocessing like Tailwind
			(environment.moduleGraph.getModulesByFile(id) ?? new Set())
		: // For non-root files, we're safe to pull from "getModuleById" based on testing.
			// TODO: Find better invalidation strategy to use "getModuleById" in all cases!
			new Set([environment.moduleGraph.getModuleById(id)]);

	// Collect all imported modules for the module(s).
	for (const entry of moduleEntriesForId) {
		// Handle this in case an module entries weren't found for ID
		// This seems possible with some virtual IDs (ex: `astro:markdown/*.md`)
		if (!entry) {
			continue;
		}
		if (id === entry.id) {
			scanned.add(id);

			// NOTE: It may be worth revisiting if we can crawl direct imports of the module since
			// `.importedModules` would also include modules that are dynamically watched, not imported.
			// That way we no longer need the below `continue` skips.

			// CSS requests `importedModules` are usually from `@import`, but we don't really need
			// to crawl into those as the `@import` code are already inlined into this `id`.
			// If CSS requests `importedModules` contain non-CSS files, e.g. Tailwind might add HMR
			// dependencies as `importedModules`, we should also skip them as they aren't really
			// imported. Without this, every hoisted script in the project is added to every page!
			if (isCSSRequest(id)) {
				continue;
			}
			// Some special Vite queries like `?url` or `?raw` are known to be a simple default export
			// and doesn't have any imports to crawl. However, since they would `this.addWatchFile` the
			// underlying module, our logic would crawl into them anyways which is incorrect as they
			// don't take part in the final rendering, so we skip it here.
// ... (68 more lines)