vite-plugin-env.ts — astro Source File

Source Code

import { readFileSync } from 'node:fs';
import type { Plugin } from 'vite';
import { AstroError, AstroErrorData } from '../core/errors/index.js';
import { isAstroClientEnvironment } from '../environments.js';
import type { AstroSettings } from '../types/astro.js';
import {
	CLIENT_VIRTUAL_MODULE_ID,
	INTERNAL_VIRTUAL_MODULE_ID,
	MODULE_TEMPLATE_URL,
	RESOLVED_CLIENT_VIRTUAL_MODULE_ID,
	RESOLVED_INTERNAL_VIRTUAL_MODULE_ID,
	RESOLVED_SERVER_VIRTUAL_MODULE_ID,
	SERVER_VIRTUAL_MODULE_ID,
} from './constants.js';
import type { EnvLoader } from './env-loader.js';
import { type InvalidVariable, invalidVariablesToError } from './errors.js';
import type { EnvSchema } from './schema.js';
import { getEnvFieldType, validateEnvVariable } from './validators.js';

interface AstroEnvPluginParams {
	settings: AstroSettings;
	sync: boolean;
	envLoader: EnvLoader;
}

export function astroEnv({ settings, sync, envLoader }: AstroEnvPluginParams): Plugin {
	const { schema, validateSecrets } = settings.config.env;
	let isBuild: boolean;
	let populated = false;

	return {
		name: 'astro-env-plugin',
		enforce: 'pre',
		config(_, { command }) {
			isBuild = command === 'build';
		},
		buildStart() {
			if (!isBuild || populated) {
				return;
			}

			const loadedEnv = envLoader.get();

			// During the build, we populate process.env so that secrets can work
			for (const [key, value] of Object.entries(loadedEnv)) {
				if (value !== undefined) {
					process.env[key] = value;
				}
			}

			populated = true;
		},
		resolveId: {
			filter: {
				id: new RegExp(
					`^(${CLIENT_VIRTUAL_MODULE_ID}|${SERVER_VIRTUAL_MODULE_ID}|${INTERNAL_VIRTUAL_MODULE_ID})$`,
				),
			},
			handler(id) {
				if (id === CLIENT_VIRTUAL_MODULE_ID) {
// ... (154 more lines)