Support immutable cache headers for _astro assets

.changeset/twelve-fishes-fail.md

@@ -0,0 +1,5 @@
+---
+'@astrojs/node': minor
+---
+
+Automatically set immutable cache headers for assets served from the `/_astro` directory

packages/integrations/node/src/http-server.ts

@@ -10,6 +10,7 @@ interface CreateServerOptions {
 	port: number;
 	host: string | undefined;
 	removeBase: (pathname: string) => string;
+	assets: string;
 }
 
 function parsePathname(pathname: string, host: string | undefined, port: number) {
@@ -22,9 +23,16 @@ function parsePathname(pathname: string, host: string | undefined, port: number)
 }
 
 export function createServer(
-	{ client, port, host, removeBase }: CreateServerOptions,
+	{ client, port, host, removeBase, assets }: CreateServerOptions,
 	handler: http.RequestListener
 ) {
+	// The `base` is removed before passed to this function, so we don't
+	// need to check for it here.
+	const assetsPrefix = `/${assets}/`;
+	function isImmutableAsset(pathname: string) {
+		return pathname.startsWith(assetsPrefix);
+	}
+
 	const listener: http.RequestListener = (req, res) => {
 		if (req.url) {
 			let pathname: string | undefined = removeBase(req.url);
@@ -54,6 +62,12 @@ export function createServer(
 				// File not found, forward to the SSR handler
 				handler(req, res);
 			});
+			stream.on('headers', (_res: http.ServerResponse<http.IncomingMessage>) => {
+				if(isImmutableAsset(encodedURI)) {
+					// Taken from https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Cache-Control#immutable
+					_res.setHeader('Cache-Control', 'public, max-age=604800, immutable')
+				}
+			});
 			stream.on('directory', () => {
 				// On directory find, redirect to the trailing slash
 				let location: string;

packages/integrations/node/src/index.ts

@@ -6,7 +6,7 @@ export function getAdapter(options: Options): AstroAdapter {
 		name: '@astrojs/node',
 		serverEntrypoint: '@astrojs/node/server.js',
 		previewEntrypoint: '@astrojs/node/preview.js',
-		exports: ['handler', 'startServer'],
+		exports: ['handler', 'startServer', 'options'],
 		args: options,
 		supportedAstroFeatures: {
 			hybridOutput: 'stable',
@@ -49,6 +49,7 @@ export default function createIntegration(userOptions: UserOptions): AstroIntegr
 					server: config.build.server?.toString(),
 					host: config.server.host,
 					port: config.server.port,
+					assets: config.build.assets,
 				};
 				setAdapter(getAdapter(_options));
 

packages/integrations/node/src/preview.ts

@@ -17,11 +17,13 @@ const preview: CreatePreviewServer = async function ({
 	type ServerModule = ReturnType<typeof createExports>;
 	type MaybeServerModule = Partial<ServerModule>;
 	let ssrHandler: ServerModule['handler'];
+	let options: ServerModule['options'];
 	try {
 		process.env.ASTRO_NODE_AUTOSTART = 'disabled';
 		const ssrModule: MaybeServerModule = await import(serverEntrypoint.toString());
 		if (typeof ssrModule.handler === 'function') {
 			ssrHandler = ssrModule.handler;
+			options = ssrModule.options!;
 		} else {
 			throw new AstroError(
 				`The server entrypoint doesn't have a handler. Are you sure this is the right file?`
@@ -59,6 +61,7 @@ const preview: CreatePreviewServer = async function ({
 			port,
 			host,
 			removeBase,
+			assets: options.assets,
 		},
 		handler
 	);

packages/integrations/node/src/server.ts

@@ -8,6 +8,7 @@ applyPolyfills();
 export function createExports(manifest: SSRManifest, options: Options) {
 	const app = new NodeApp(manifest);
 	return {
+		options: options,
 		handler: middleware(app, options.mode),
 		startServer: () => startServer(app, options),
 	};

packages/integrations/node/src/standalone.ts

@@ -52,6 +52,7 @@ export default function startServer(app: NodeApp, options: Options) {
 			port,
 			host,
 			removeBase: app.removeBase.bind(app),
+			assets: options.assets,
 		},
 		handler
 	);

packages/integrations/node/src/types.ts

@@ -15,6 +15,7 @@ export interface Options extends UserOptions {
 	port: number;
 	server: string;
 	client: string;
+	assets: string;
 }
 
 export type RequestHandlerParams = [

packages/integrations/node/test/assets.test.js

@@ -0,0 +1,43 @@
+import { expect } from 'chai';
+import nodejs from '../dist/index.js';
+import { loadFixture } from './test-utils.js';
+import * as cheerio from 'cheerio';
+
+describe('Assets', () => {
+	/** @type {import('./test-utils').Fixture} */
+	let fixture;
+	let devPreview;
+
+	before(async () => {
+		fixture = await loadFixture({
+			root: './fixtures/image/',
+			output: 'server',
+			adapter: nodejs({ mode: 'standalone' }),
+			vite: {
+				build: {
+					assetsInlineLimit: 0,
+				}
+			}
+		});
+		await fixture.build();
+		devPreview = await fixture.preview();
+	});
+
+	after(async () => {
+		await devPreview.stop();
+	});
+
+	it('Assets within the _astro folder should be given immutable headers', async () => {
+		let response = await fixture.fetch('/text-file');
+		let cacheControl = response.headers.get('cache-control');
+		expect(cacheControl).to.equal(null);
+		const html = await response.text();
+		const $ = cheerio.load(html);
+
+		// Fetch the asset
+		const fileURL = $('a').attr('href');
+		response = await fixture.fetch(fileURL);
+		cacheControl = response.headers.get('cache-control');
+		expect(cacheControl).to.equal('public, max-age=604800, immutable');
+	});
+});

packages/integrations/node/test/fixtures/image/src/assets/file.txt

@@ -0,0 +1 @@
+this is a text file

packages/integrations/node/test/fixtures/image/src/pages/text-file.astro

@@ -0,0 +1,14 @@
+---
+import txt from '../assets/file.txt?url';
+---
+<html>
+	<head>
+		<title>Testing</title>
+	</head>
+	<body>
+		<h1>Testing</h1>
+		<main>
+			<a href={txt} download>Download text file</a>
+		</main>
+	</body>
+</html>