vitePluginDb() — astro Function Reference

Source Code

packages/db/src/core/integration/vite-plugin-db.ts lines 63–131

export function vitePluginDb(params: VitePluginDBParams): VitePlugin {
	let command: 'build' | 'serve' = 'build';
	return {
		name: 'astro:db',
		enforce: 'pre',
		configResolved(resolvedConfig) {
			command = resolvedConfig.command;
		},
		resolveId: {
			filter: {
				id: new RegExp(`^${VIRTUAL_MODULE_ID}$`),
			},
			handler() {
				if (params.seedHandler.inProgress) {
					return resolved.importedFromSeedFile;
				}
				return resolved.module;
			},
		},
		load: {
			filter: {
				id: new RegExp(`^(${resolved.module}|${resolved.importedFromSeedFile})$`),
			},
			async handler(id) {
				if (params.connectToRemote) {
					return getRemoteVirtualModContents({
						appToken: params.appToken,
						tables: params.tables.get(),
						isBuild: command === 'build',
						output: params.output,
						localExecution: false,
					});
				}

				// When seeding, we resolved to a different virtual module.
				// this prevents an infinite loop attempting to rerun seed files.
				// Short circuit with the module contents in this case.
				if (id === resolved.importedFromSeedFile) {
					return getLocalVirtualModContents({
						root: params.root,
						tables: params.tables.get(),
						localExecution: false,
					});
				}

				await recreateTables(params);
				const seedFiles = getResolvedSeedFiles(params);
				for await (const seedFile of seedFiles) {
					// Use `addWatchFile()` to invalidate the `astro:db` module
					// when a seed file changes.
					this.addWatchFile(fileURLToPath(seedFile));
					if (existsSync(seedFile)) {
						params.seedHandler.inProgress = true;
						await params.seedHandler.execute(seedFile);
					}
				}
				if (params.seedHandler.inProgress) {
					(params.logger ?? console).info('Seeded database.');
					params.seedHandler.inProgress = false;
				}
				return getLocalVirtualModContents({
					root: params.root,
					tables: params.tables.get(),
					localExecution: false,
				});
			},
		},
	};
}