From d024fa71bc15522eacd3588ba7a638a8b5858a0b Mon Sep 17 00:00:00 2001
From: Nate Moore <nate@astro.build>
Date: Mon, 25 Jul 2022 23:08:47 -0400
Subject: [PATCH 1/5] feat: add generic page-ssr plugin

---
 packages/astro/src/core/create-vite.ts        |  2 +
 .../astro/src/vite-plugin-scripts/page.ts     | 53 +++++++++++++++++++
 2 files changed, 55 insertions(+)
 create mode 100644 packages/astro/src/vite-plugin-scripts/page.ts

diff --git a/packages/astro/src/core/create-vite.ts b/packages/astro/src/core/create-vite.ts
index 7b3a8070f72b..42d736ff6b8e 100644
--- a/packages/astro/src/core/create-vite.ts
+++ b/packages/astro/src/core/create-vite.ts
@@ -14,6 +14,7 @@ import astroIntegrationsContainerPlugin from '../vite-plugin-integrations-contai
 import jsxVitePlugin from '../vite-plugin-jsx/index.js';
 import markdownVitePlugin from '../vite-plugin-markdown/index.js';
 import astroScriptsPlugin from '../vite-plugin-scripts/index.js';
+import astroScriptsPagePlugin from '../vite-plugin-scripts/page.js';
 import { createCustomViteLogger } from './errors.js';
 import { resolveDependency } from './util.js';
 
@@ -80,6 +81,7 @@ export async function createVite(
 			jsxVitePlugin({ config: astroConfig, logging }),
 			astroPostprocessVitePlugin({ config: astroConfig }),
 			astroIntegrationsContainerPlugin({ config: astroConfig }),
+			astroScriptsPagePlugin({ config: astroConfig }),
 		],
 		publicDir: fileURLToPath(astroConfig.publicDir),
 		root: fileURLToPath(astroConfig.root),
diff --git a/packages/astro/src/vite-plugin-scripts/page.ts b/packages/astro/src/vite-plugin-scripts/page.ts
new file mode 100644
index 000000000000..5cc09b276d70
--- /dev/null
+++ b/packages/astro/src/vite-plugin-scripts/page.ts
@@ -0,0 +1,53 @@
+import { Plugin as VitePlugin } from 'vite';
+import { AstroConfig } from '../@types/astro.js';
+import { PAGE_SSR_SCRIPT_ID } from './index.js';
+
+import { resolvePages } from '../core/util.js';
+import ancestor from 'common-ancestor-path';
+import MagicString from 'magic-string';
+
+export default function astroScriptsPostPlugin({ config }: { config: AstroConfig }): VitePlugin {
+	function normalizeFilename(filename: string) {
+		if (filename.startsWith('/@fs')) {
+			filename = filename.slice('/@fs'.length);
+		} else if (filename.startsWith('/') && !ancestor(filename, config.root.pathname)) {
+			filename = new URL('.' + filename, config.root).pathname;
+		}
+		return filename;
+	}
+
+	return {
+		name: 'astro:scripts:post',
+		enforce: 'post',
+		
+		transform(this, code, id, options) {
+			if (!options?.ssr) return;
+
+			const filename = normalizeFilename(id);
+			let fileUrl: URL;
+			try {
+				fileUrl = new URL(`file://${filename}`);
+			} catch (e) {
+				// If we can't construct a valid URL, exit early
+				return;
+			}
+
+			const isPage = fileUrl.pathname.startsWith(resolvePages(config).pathname);
+			if (!isPage) return;
+			const parts = fileUrl.pathname.slice(resolvePages(config).pathname.length).split('/');
+			for (const part of parts) {
+				if (part.startsWith('_')) return;
+			}
+			const hasInjectedScript = config._ctx.scripts.some((s) => s.stage === 'page-ssr');
+			if (!hasInjectedScript) return;
+
+			const s = new MagicString(code, { filename });
+			s.prepend(`import '${PAGE_SSR_SCRIPT_ID}';\n`);
+
+			return {
+				code: s.toString(),
+				map: s.generateMap(),
+			}
+		},
+	};
+}

From 1b966cfc7a33a5ea5825e4d66c6e1b2eee68dece Mon Sep 17 00:00:00 2001
From: Nate Moore <nate@astro.build>
Date: Mon, 1 Aug 2022 15:23:36 -0500
Subject: [PATCH 2/5] refactor: remove page-specific logic from
 astro/markdown/mdx plugins

---
 .changeset/plenty-hats-grow.md                |  6 ++
 packages/astro/src/core/create-vite.ts        |  4 +-
 packages/astro/src/core/render/dev/index.ts   |  7 ++
 packages/astro/src/vite-plugin-astro/index.ts | 14 ---
 .../astro/src/vite-plugin-markdown/index.ts   |  8 +-
 .../astro/src/vite-plugin-scripts/index.ts    | 88 ++++++++++---------
 .../{page.ts => page-ssr.ts}                  | 21 ++---
 packages/integrations/mdx/src/index.ts        |  7 --
 8 files changed, 71 insertions(+), 84 deletions(-)
 create mode 100644 .changeset/plenty-hats-grow.md
 rename packages/astro/src/vite-plugin-scripts/{page.ts => page-ssr.ts} (74%)

diff --git a/.changeset/plenty-hats-grow.md b/.changeset/plenty-hats-grow.md
new file mode 100644
index 000000000000..d7238f840cac
--- /dev/null
+++ b/.changeset/plenty-hats-grow.md
@@ -0,0 +1,6 @@
+---
+'astro': patch
+'@astrojs/mdx': patch
+---
+
+Improve `injectScript` handling for non-Astro pages
diff --git a/packages/astro/src/core/create-vite.ts b/packages/astro/src/core/create-vite.ts
index 42d736ff6b8e..631c1dc2b9c3 100644
--- a/packages/astro/src/core/create-vite.ts
+++ b/packages/astro/src/core/create-vite.ts
@@ -14,7 +14,7 @@ import astroIntegrationsContainerPlugin from '../vite-plugin-integrations-contai
 import jsxVitePlugin from '../vite-plugin-jsx/index.js';
 import markdownVitePlugin from '../vite-plugin-markdown/index.js';
 import astroScriptsPlugin from '../vite-plugin-scripts/index.js';
-import astroScriptsPagePlugin from '../vite-plugin-scripts/page.js';
+import astroScriptsPageSSRPlugin from '../vite-plugin-scripts/page-ssr.js';
 import { createCustomViteLogger } from './errors.js';
 import { resolveDependency } from './util.js';
 
@@ -81,7 +81,7 @@ export async function createVite(
 			jsxVitePlugin({ config: astroConfig, logging }),
 			astroPostprocessVitePlugin({ config: astroConfig }),
 			astroIntegrationsContainerPlugin({ config: astroConfig }),
-			astroScriptsPagePlugin({ config: astroConfig }),
+			astroScriptsPageSSRPlugin({ config: astroConfig }),
 		],
 		publicDir: fileURLToPath(astroConfig.publicDir),
 		root: fileURLToPath(astroConfig.root),
diff --git a/packages/astro/src/core/render/dev/index.ts b/packages/astro/src/core/render/dev/index.ts
index a6643fc359cb..de45a3ae182b 100644
--- a/packages/astro/src/core/render/dev/index.ts
+++ b/packages/astro/src/core/render/dev/index.ts
@@ -9,6 +9,7 @@ import type {
 	SSRElement,
 	SSRLoadedRenderer,
 } from '../../../@types/astro';
+import { PAGE_SCRIPT_ID } from '../../../vite-plugin-scripts/index.js';
 import { prependForwardSlash } from '../../../core/path.js';
 import { LogOptions } from '../../logger/core.js';
 import { isPage } from '../../util.js';
@@ -124,6 +125,7 @@ export async function render(
 			children: '',
 		});
 	}
+
 	// TODO: We should allow adding generic HTML elements to the head, not just scripts
 	for (const script of astroConfig._ctx.scripts) {
 		if (script.stage === 'head-inline') {
@@ -131,6 +133,11 @@ export async function render(
 				props: {},
 				children: script.content,
 			});
+		} else if (script.stage === 'page' && isPage(filePath, astroConfig)) {
+			scripts.add({
+				props: { type: 'module', src: PAGE_SCRIPT_ID },
+				children: '',
+			});
 		}
 	}
 
diff --git a/packages/astro/src/vite-plugin-astro/index.ts b/packages/astro/src/vite-plugin-astro/index.ts
index b4925d0fd66e..f38d5e5ca11b 100644
--- a/packages/astro/src/vite-plugin-astro/index.ts
+++ b/packages/astro/src/vite-plugin-astro/index.ts
@@ -9,8 +9,6 @@ import esbuild from 'esbuild';
 import slash from 'slash';
 import { fileURLToPath } from 'url';
 import { isRelativePath, startsWithForwardSlash } from '../core/path.js';
-import { resolvePages } from '../core/util.js';
-import { PAGE_SCRIPT_ID, PAGE_SSR_SCRIPT_ID } from '../vite-plugin-scripts/index.js';
 import { getFileInfo } from '../vite-plugin-utils/index.js';
 import { cachedCompilation, CompileProps, getCachedSource } from './compile.js';
 import { handleHotUpdate, trackCSSDependencies } from './hmr.js';
@@ -215,14 +213,6 @@ export default function astro({ config, logging }: AstroPluginOptions): vite.Plu
 			}
 
 			const filename = normalizeFilename(parsedId.filename);
-			let isPage = false;
-			try {
-				const fileUrl = new URL(`file://${filename}`);
-				isPage = fileUrl.pathname.startsWith(resolvePages(config).pathname);
-			} catch {}
-			if (isPage && config._ctx.scripts.some((s) => s.stage === 'page')) {
-				source += `\n<script src="${PAGE_SCRIPT_ID}" />`;
-			}
 			const compileProps: CompileProps = {
 				config,
 				filename,
@@ -269,10 +259,6 @@ export default function astro({ config, logging }: AstroPluginOptions): vite.Plu
 						i++;
 					}
 				}
-				// Add handling to inject scripts into each page JS bundle, if needed.
-				if (isPage) {
-					SUFFIX += `\nimport "${PAGE_SSR_SCRIPT_ID}";`;
-				}
 
 				// Prefer live reload to HMR in `.astro` files
 				if (!resolvedConfig.isProduction) {
diff --git a/packages/astro/src/vite-plugin-markdown/index.ts b/packages/astro/src/vite-plugin-markdown/index.ts
index ab6f6a7ef13f..6794b499ab09 100644
--- a/packages/astro/src/vite-plugin-markdown/index.ts
+++ b/packages/astro/src/vite-plugin-markdown/index.ts
@@ -10,11 +10,9 @@ import { pagesVirtualModuleId } from '../core/app/index.js';
 import { collectErrorMetadata } from '../core/errors.js';
 import type { LogOptions } from '../core/logger/core.js';
 import { warn } from '../core/logger/core.js';
-import { resolvePages } from '../core/util.js';
 import { cachedCompilation, CompileProps } from '../vite-plugin-astro/compile.js';
 import { getViteTransform, TransformHook } from '../vite-plugin-astro/styles.js';
 import type { PluginMetadata as AstroPluginMetadata } from '../vite-plugin-astro/types';
-import { PAGE_SSR_SCRIPT_ID } from '../vite-plugin-scripts/index.js';
 import { getFileInfo } from '../vite-plugin-utils/index.js';
 
 interface AstroPluginOptions {
@@ -147,8 +145,6 @@ export default function markdown({ config, logging }: AstroPluginOptions): Plugi
 				const isAstroFlavoredMd = config.legacy.astroFlavoredMarkdown;
 
 				const fileUrl = new URL(`file://${filename}`);
-				const isPage = fileUrl.pathname.startsWith(resolvePages(config).pathname);
-				const hasInjectedScript = isPage && config._ctx.scripts.some((s) => s.stage === 'page-ssr');
 
 				// Extract special frontmatter keys
 				let { data: frontmatter, content: markdownContent } = safeMatter(source, filename);
@@ -187,7 +183,6 @@ export default function markdown({ config, logging }: AstroPluginOptions): Plugi
 import Slugger from 'github-slugger';
 ${layout ? `import Layout from '${layout}';` : ''}
 ${isAstroFlavoredMd && components ? `import * from '${components}';` : ''}
-${hasInjectedScript ? `import '${PAGE_SSR_SCRIPT_ID}';` : ''}
 ${isAstroFlavoredMd ? setup : ''}
 
 const slugger = new Slugger();
@@ -224,7 +219,8 @@ ${isAstroFlavoredMd ? setup : ''}`.trim();
 				if (/\bLayout\b/.test(imports)) {
 					astroResult = `${prelude}\n<Layout content={$$content}>\n\n${astroResult}\n\n</Layout>`;
 				} else {
-					astroResult = `${prelude}\n${astroResult}`;
+					// Note: without a Layout, we need to inject `head` manually so `maybeRenderHead` runs
+					astroResult = `${prelude}\n<head></head>${astroResult}`;
 				}
 
 				// Transform from `.astro` to valid `.ts`
diff --git a/packages/astro/src/vite-plugin-scripts/index.ts b/packages/astro/src/vite-plugin-scripts/index.ts
index d055cf59d918..898b69a3dbe8 100644
--- a/packages/astro/src/vite-plugin-scripts/index.ts
+++ b/packages/astro/src/vite-plugin-scripts/index.ts
@@ -11,52 +11,54 @@ export const BEFORE_HYDRATION_SCRIPT_ID = `${SCRIPT_ID_PREFIX}${
 export const PAGE_SCRIPT_ID = `${SCRIPT_ID_PREFIX}${'page' as InjectedScriptStage}.js`;
 export const PAGE_SSR_SCRIPT_ID = `${SCRIPT_ID_PREFIX}${'page-ssr' as InjectedScriptStage}.js`;
 
-export default function astroScriptsPlugin({ config }: { config: AstroConfig }): VitePlugin {
+export default function astroScriptsPlugin({ config }: { config: AstroConfig }): VitePlugin[] {
 	let env: ConfigEnv | undefined = undefined;
-	return {
-		name: 'astro:scripts',
+	return [
+		{
+			name: 'astro:scripts',
 
-		config(_config, _env) {
-			env = _env;
-		},
+			config(_config, _env) {
+				env = _env;
+			},
 
-		async resolveId(id) {
-			if (id.startsWith(SCRIPT_ID_PREFIX)) {
-				return id;
-			}
-			return undefined;
-		},
+			async resolveId(id) {
+				if (id.startsWith(SCRIPT_ID_PREFIX)) {
+					return id;
+				}
+				return undefined;
+			},
 
-		async load(id) {
-			if (id === BEFORE_HYDRATION_SCRIPT_ID) {
-				return config._ctx.scripts
-					.filter((s) => s.stage === 'before-hydration')
-					.map((s) => s.content)
-					.join('\n');
-			}
-			if (id === PAGE_SCRIPT_ID) {
-				return config._ctx.scripts
-					.filter((s) => s.stage === 'page')
-					.map((s) => s.content)
-					.join('\n');
-			}
-			if (id === PAGE_SSR_SCRIPT_ID) {
-				return config._ctx.scripts
-					.filter((s) => s.stage === 'page-ssr')
-					.map((s) => s.content)
-					.join('\n');
-			}
-			return null;
-		},
-		buildStart(options) {
-			const hasHydrationScripts = config._ctx.scripts.some((s) => s.stage === 'before-hydration');
-			if (hasHydrationScripts && env?.command === 'build' && !env?.ssrBuild) {
-				this.emitFile({
-					type: 'chunk',
-					id: BEFORE_HYDRATION_SCRIPT_ID,
-					name: BEFORE_HYDRATION_SCRIPT_ID,
-				});
-			}
+			async load(id) {
+				if (id === BEFORE_HYDRATION_SCRIPT_ID) {
+					return config._ctx.scripts
+						.filter((s) => s.stage === 'before-hydration')
+						.map((s) => s.content)
+						.join('\n');
+				}
+				if (id === PAGE_SCRIPT_ID) {
+					return config._ctx.scripts
+						.filter((s) => s.stage === 'page')
+						.map((s) => s.content)
+						.join('\n');
+				}
+				if (id === PAGE_SSR_SCRIPT_ID) {
+					return config._ctx.scripts
+						.filter((s) => s.stage === 'page-ssr')
+						.map((s) => s.content)
+						.join('\n');
+				}
+				return null;
+			},
+			buildStart(options) {
+				const hasHydrationScripts = config._ctx.scripts.some((s) => s.stage === 'before-hydration');
+				if (hasHydrationScripts && env?.command === 'build' && !env?.ssrBuild) {
+					this.emitFile({
+						type: 'chunk',
+						id: BEFORE_HYDRATION_SCRIPT_ID,
+						name: BEFORE_HYDRATION_SCRIPT_ID,
+					});
+				}
+			},
 		},
-	};
+	];
 }
diff --git a/packages/astro/src/vite-plugin-scripts/page.ts b/packages/astro/src/vite-plugin-scripts/page-ssr.ts
similarity index 74%
rename from packages/astro/src/vite-plugin-scripts/page.ts
rename to packages/astro/src/vite-plugin-scripts/page-ssr.ts
index 5cc09b276d70..1f726912e513 100644
--- a/packages/astro/src/vite-plugin-scripts/page.ts
+++ b/packages/astro/src/vite-plugin-scripts/page-ssr.ts
@@ -2,7 +2,7 @@ import { Plugin as VitePlugin } from 'vite';
 import { AstroConfig } from '../@types/astro.js';
 import { PAGE_SSR_SCRIPT_ID } from './index.js';
 
-import { resolvePages } from '../core/util.js';
+import { isPage } from '../core/util.js';
 import ancestor from 'common-ancestor-path';
 import MagicString from 'magic-string';
 
@@ -17,29 +17,26 @@ export default function astroScriptsPostPlugin({ config }: { config: AstroConfig
 	}
 
 	return {
-		name: 'astro:scripts:post',
+		name: 'astro:scripts:page-ssr',
 		enforce: 'post',
 		
 		transform(this, code, id, options) {
 			if (!options?.ssr) return;
 
+			const hasInjectedScript = config._ctx.scripts.some((s) => s.stage === 'page-ssr');
+			if (!hasInjectedScript) return;
+
 			const filename = normalizeFilename(id);
-			let fileUrl: URL;
+			let fileURL: URL;
 			try {
-				fileUrl = new URL(`file://${filename}`);
+				fileURL = new URL(`file://${filename}`);
 			} catch (e) {
 				// If we can't construct a valid URL, exit early
 				return;
 			}
 
-			const isPage = fileUrl.pathname.startsWith(resolvePages(config).pathname);
-			if (!isPage) return;
-			const parts = fileUrl.pathname.slice(resolvePages(config).pathname.length).split('/');
-			for (const part of parts) {
-				if (part.startsWith('_')) return;
-			}
-			const hasInjectedScript = config._ctx.scripts.some((s) => s.stage === 'page-ssr');
-			if (!hasInjectedScript) return;
+			const fileIsPage = isPage(fileURL, config);
+			if (!fileIsPage) return;
 
 			const s = new MagicString(code, { filename });
 			s.prepend(`import '${PAGE_SSR_SCRIPT_ID}';\n`);
diff --git a/packages/integrations/mdx/src/index.ts b/packages/integrations/mdx/src/index.ts
index 8b7831f2752b..12215bbf9e74 100644
--- a/packages/integrations/mdx/src/index.ts
+++ b/packages/integrations/mdx/src/index.ts
@@ -113,13 +113,6 @@ export default function mdx(mdxOptions: MdxOptions = {}): AstroIntegration {
 									if (!id.endsWith('.mdx')) return;
 									const [, moduleExports] = parseESM(code);
 
-									// This adds support for injected "page-ssr" scripts in MDX files.
-									// TODO: This should only be happening on page entrypoints, not all imported MDX.
-									// TODO: This code is copy-pasted across all Astro/Vite plugins that deal with page
-									// entrypoints (.astro, .md, .mdx). This should be handled in some centralized place,
-									// or otherwise refactored to not require copy-paste handling logic.
-									code += `\nimport "${'astro:scripts/page-ssr.js'}";`;
-
 									const { fileUrl, fileId } = getFileInfo(id, config);
 									if (!moduleExports.includes('url')) {
 										code += `\nexport const url = ${JSON.stringify(fileUrl)};`;

From fffdebbc39a9e99fbc4033b669f8163472d1de4b Mon Sep 17 00:00:00 2001
From: Nate Moore <nate@astro.build>
Date: Mon, 1 Aug 2022 15:29:03 -0500
Subject: [PATCH 3/5] refactor: revert changes to vite-plugin-scripts

---
 .../astro/src/vite-plugin-scripts/index.ts    | 88 +++++++++----------
 1 file changed, 43 insertions(+), 45 deletions(-)

diff --git a/packages/astro/src/vite-plugin-scripts/index.ts b/packages/astro/src/vite-plugin-scripts/index.ts
index 898b69a3dbe8..d055cf59d918 100644
--- a/packages/astro/src/vite-plugin-scripts/index.ts
+++ b/packages/astro/src/vite-plugin-scripts/index.ts
@@ -11,54 +11,52 @@ export const BEFORE_HYDRATION_SCRIPT_ID = `${SCRIPT_ID_PREFIX}${
 export const PAGE_SCRIPT_ID = `${SCRIPT_ID_PREFIX}${'page' as InjectedScriptStage}.js`;
 export const PAGE_SSR_SCRIPT_ID = `${SCRIPT_ID_PREFIX}${'page-ssr' as InjectedScriptStage}.js`;
 
-export default function astroScriptsPlugin({ config }: { config: AstroConfig }): VitePlugin[] {
+export default function astroScriptsPlugin({ config }: { config: AstroConfig }): VitePlugin {
 	let env: ConfigEnv | undefined = undefined;
-	return [
-		{
-			name: 'astro:scripts',
+	return {
+		name: 'astro:scripts',
 
-			config(_config, _env) {
-				env = _env;
-			},
+		config(_config, _env) {
+			env = _env;
+		},
 
-			async resolveId(id) {
-				if (id.startsWith(SCRIPT_ID_PREFIX)) {
-					return id;
-				}
-				return undefined;
-			},
+		async resolveId(id) {
+			if (id.startsWith(SCRIPT_ID_PREFIX)) {
+				return id;
+			}
+			return undefined;
+		},
 
-			async load(id) {
-				if (id === BEFORE_HYDRATION_SCRIPT_ID) {
-					return config._ctx.scripts
-						.filter((s) => s.stage === 'before-hydration')
-						.map((s) => s.content)
-						.join('\n');
-				}
-				if (id === PAGE_SCRIPT_ID) {
-					return config._ctx.scripts
-						.filter((s) => s.stage === 'page')
-						.map((s) => s.content)
-						.join('\n');
-				}
-				if (id === PAGE_SSR_SCRIPT_ID) {
-					return config._ctx.scripts
-						.filter((s) => s.stage === 'page-ssr')
-						.map((s) => s.content)
-						.join('\n');
-				}
-				return null;
-			},
-			buildStart(options) {
-				const hasHydrationScripts = config._ctx.scripts.some((s) => s.stage === 'before-hydration');
-				if (hasHydrationScripts && env?.command === 'build' && !env?.ssrBuild) {
-					this.emitFile({
-						type: 'chunk',
-						id: BEFORE_HYDRATION_SCRIPT_ID,
-						name: BEFORE_HYDRATION_SCRIPT_ID,
-					});
-				}
-			},
+		async load(id) {
+			if (id === BEFORE_HYDRATION_SCRIPT_ID) {
+				return config._ctx.scripts
+					.filter((s) => s.stage === 'before-hydration')
+					.map((s) => s.content)
+					.join('\n');
+			}
+			if (id === PAGE_SCRIPT_ID) {
+				return config._ctx.scripts
+					.filter((s) => s.stage === 'page')
+					.map((s) => s.content)
+					.join('\n');
+			}
+			if (id === PAGE_SSR_SCRIPT_ID) {
+				return config._ctx.scripts
+					.filter((s) => s.stage === 'page-ssr')
+					.map((s) => s.content)
+					.join('\n');
+			}
+			return null;
+		},
+		buildStart(options) {
+			const hasHydrationScripts = config._ctx.scripts.some((s) => s.stage === 'before-hydration');
+			if (hasHydrationScripts && env?.command === 'build' && !env?.ssrBuild) {
+				this.emitFile({
+					type: 'chunk',
+					id: BEFORE_HYDRATION_SCRIPT_ID,
+					name: BEFORE_HYDRATION_SCRIPT_ID,
+				});
+			}
 		},
-	];
+	};
 }

From 99f20d1d837df9b1cf2b9aa7aa541d6609b0a6cf Mon Sep 17 00:00:00 2001
From: Nate Moore <nate@astro.build>
Date: Mon, 1 Aug 2022 17:24:49 -0500
Subject: [PATCH 4/5] fix: handle injected `page` scripts in build

---
 packages/astro/src/core/build/generate.ts        | 14 +++++++++++++-
 packages/astro/src/core/build/static-build.ts    |  5 +++++
 packages/astro/src/core/build/vite-plugin-ssr.ts | 10 ++++++++--
 3 files changed, 26 insertions(+), 3 deletions(-)

diff --git a/packages/astro/src/core/build/generate.ts b/packages/astro/src/core/build/generate.ts
index fb125dd28b18..cbd8e1c2e93c 100644
--- a/packages/astro/src/core/build/generate.ts
+++ b/packages/astro/src/core/build/generate.ts
@@ -18,7 +18,7 @@ import {
 	removeTrailingForwardSlash,
 } from '../../core/path.js';
 import type { RenderOptions } from '../../core/render/core';
-import { BEFORE_HYDRATION_SCRIPT_ID } from '../../vite-plugin-scripts/index.js';
+import { BEFORE_HYDRATION_SCRIPT_ID, PAGE_SCRIPT_ID } from '../../vite-plugin-scripts/index.js';
 import { call as callEndpoint } from '../endpoint/index.js';
 import { debug, info } from '../logger/core.js';
 import { render } from '../render/core.js';
@@ -272,6 +272,18 @@ async function generatePath(
 	const links = createLinkStylesheetElementSet(linkIds.reverse(), site);
 	const scripts = createModuleScriptsSet(hoistedScripts ? [hoistedScripts] : [], site);
 
+	if (astroConfig._ctx.scripts.some((script) => script.stage === 'page')) {
+		const hashedFilePath = internals.entrySpecifierToBundleMap.get(PAGE_SCRIPT_ID);
+		if (typeof hashedFilePath !== 'string') {
+			throw new Error(`Cannot find the built path for ${PAGE_SCRIPT_ID}`);
+		}
+		const src = prependForwardSlash(npath.posix.join(astroConfig.base, hashedFilePath));
+		scripts.add({
+			props: { type: 'module', src },
+			children: '',
+		})
+	}
+
 	// Add all injected scripts to the page.
 	for (const script of astroConfig._ctx.scripts) {
 		if (script.stage === 'head-inline') {
diff --git a/packages/astro/src/core/build/static-build.ts b/packages/astro/src/core/build/static-build.ts
index 99b58318d097..19778ae4b87e 100644
--- a/packages/astro/src/core/build/static-build.ts
+++ b/packages/astro/src/core/build/static-build.ts
@@ -8,6 +8,7 @@ import { prependForwardSlash } from '../../core/path.js';
 import { emptyDir, isModeServerWithNoAdapter, removeDir } from '../../core/util.js';
 import { runHookBuildSetup } from '../../integrations/index.js';
 import { rollupPluginAstroBuildCSS } from '../../vite-plugin-build-css/index.js';
+import { PAGE_SCRIPT_ID } from '../../vite-plugin-scripts/index.js';
 import type { ViteConfigWithSSR } from '../create-vite';
 import { info } from '../logger/core.js';
 import { generatePages } from './generate.js';
@@ -85,6 +86,10 @@ Example:
 		...internals.discoveredScripts,
 	]);
 
+	if (astroConfig._ctx.scripts.some((script) => script.stage === 'page')) {
+		clientInput.add(PAGE_SCRIPT_ID);
+	}
+
 	// Run client build first, so the assets can be fed into the SSR rendered version.
 	timer.clientBuild = performance.now();
 	await clientBuild(opts, internals, clientInput);
diff --git a/packages/astro/src/core/build/vite-plugin-ssr.ts b/packages/astro/src/core/build/vite-plugin-ssr.ts
index 5ce5640a2787..e7fbfd995064 100644
--- a/packages/astro/src/core/build/vite-plugin-ssr.ts
+++ b/packages/astro/src/core/build/vite-plugin-ssr.ts
@@ -8,7 +8,7 @@ import glob from 'fast-glob';
 import * as fs from 'fs';
 import { fileURLToPath } from 'url';
 import { runHookBuildSsr } from '../../integrations/index.js';
-import { BEFORE_HYDRATION_SCRIPT_ID } from '../../vite-plugin-scripts/index.js';
+import { BEFORE_HYDRATION_SCRIPT_ID, PAGE_SCRIPT_ID } from '../../vite-plugin-scripts/index.js';
 import { pagesVirtualModuleId } from '../app/index.js';
 import { serializeRouteData } from '../routing/index.js';
 import { addRollupInput } from './add-rollup-input.js';
@@ -123,12 +123,19 @@ function buildManifest(
 	const { astroConfig } = opts;
 
 	const routes: SerializedRouteInfo[] = [];
+	const entryModules = Object.fromEntries(internals.entrySpecifierToBundleMap.entries());
+	if (astroConfig._ctx.scripts.some((script) => script.stage === 'page')) {
+		staticFiles.push(entryModules[PAGE_SCRIPT_ID]);
+	}
 
 	for (const pageData of eachPageData(internals)) {
 		const scripts: SerializedRouteInfo['scripts'] = [];
 		if (pageData.hoistedScript) {
 			scripts.unshift(pageData.hoistedScript);
 		}
+		if (astroConfig._ctx.scripts.some((script) => script.stage === 'page')) {
+			scripts.push({ type: 'external', value: entryModules[PAGE_SCRIPT_ID] });
+		}
 
 		routes.push({
 			file: '',
@@ -144,7 +151,6 @@ function buildManifest(
 	}
 
 	// HACK! Patch this special one.
-	const entryModules = Object.fromEntries(internals.entrySpecifierToBundleMap.entries());
 	if (!(BEFORE_HYDRATION_SCRIPT_ID in entryModules)) {
 		entryModules[BEFORE_HYDRATION_SCRIPT_ID] =
 			'data:text/javascript;charset=utf-8,//[no before-hydration script]';

From ecc1736a818f910126d514a411376ee3d26803ad Mon Sep 17 00:00:00 2001
From: Nate Moore <nate@astro.build>
Date: Mon, 1 Aug 2022 17:25:11 -0500
Subject: [PATCH 5/5] fix: prepend injected `page` scripts with `/@id/` in dev

---
 packages/astro/src/core/render/dev/index.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/astro/src/core/render/dev/index.ts b/packages/astro/src/core/render/dev/index.ts
index de45a3ae182b..6cf75a6ccb1e 100644
--- a/packages/astro/src/core/render/dev/index.ts
+++ b/packages/astro/src/core/render/dev/index.ts
@@ -135,7 +135,7 @@ export async function render(
 			});
 		} else if (script.stage === 'page' && isPage(filePath, astroConfig)) {
 			scripts.add({
-				props: { type: 'module', src: PAGE_SCRIPT_ID },
+				props: { type: 'module', src: `/@id/${PAGE_SCRIPT_ID}` },
 				children: '',
 			});
 		}