Include partytown scripts in SSR manifest (#3686)

* Include partytown scripts in SSR manifst * Adds a changeset
withastro · Jun 23, 2022 · b36ecb7 · b36ecb7
1 parent aa883a5
commit b36ecb7
Show file tree

Hide file tree

Showing 7 changed files with 79 additions and 10 deletions.
diff --git a/.changeset/cuddly-llamas-arrive.md b/.changeset/cuddly-llamas-arrive.md
@@ -0,0 +1,5 @@
+---
+'@astrojs/partytown': patch
+---
+
+Include partytown scripts in SSR manifest
diff --git a/packages/astro/test/fixtures/ssr-partytown/astro.config.mjs b/packages/astro/test/fixtures/ssr-partytown/astro.config.mjs
@@ -0,0 +1,7 @@
+import { defineConfig } from 'astro/config';
+import partytown from '@astrojs/partytown';
+
+// https://astro.build/config
+export default defineConfig({
+	integrations: [partytown()],
+});
diff --git a/packages/astro/test/fixtures/ssr-partytown/package.json b/packages/astro/test/fixtures/ssr-partytown/package.json
@@ -0,0 +1,9 @@
+{
+  "name": "@test/ssr-partytown",
+  "version": "0.0.0",
+  "private": true,
+  "dependencies": {
+    "astro": "workspace:*",
+    "@astrojs/partytown": "workspace:*"
+  }
+}
diff --git a/packages/astro/test/fixtures/ssr-partytown/src/pages/index.astro b/packages/astro/test/fixtures/ssr-partytown/src/pages/index.astro
@@ -0,0 +1,6 @@
+<html>
+	<head><title>testing</title></head>
+	<body>
+		<h1>testing</h1>
+	</body>
+</html>
diff --git a/packages/astro/test/ssr-partytown.test.js b/packages/astro/test/ssr-partytown.test.js
@@ -0,0 +1,34 @@
+import { expect } from 'chai';
+import { load as cheerioLoad } from 'cheerio';
+import { loadFixture } from './test-utils.js';
+import testAdapter from './test-adapter.js';
+
+describe('Using the Partytown integration in SSR', () => {
+	/** @type {import('./test-utils').Fixture} */
+	let fixture;
+
+	before(async () => {
+		fixture = await loadFixture({
+			root: './fixtures/ssr-partytown/',
+			adapter: testAdapter(),
+			experimental: {
+				ssr: true,
+			},
+		});
+		await fixture.build();
+	});
+
+	it('Has the scripts in the page', async () => {
+		const app = await fixture.loadTestAdapterApp();
+		const request = new Request('http://example.com/');
+		const response = await app.render(request);
+		const html = await response.text();
+		const $ = cheerioLoad(html);
+		expect($('script')).to.have.a.lengthOf(1);
+	});
+
+	it('The partytown scripts are in the manifest', async () => {
+		const app = await fixture.loadTestAdapterApp();
+		expect(app.manifest.assets).to.contain('/~partytown/partytown-sw.js');
+	});
+});
diff --git a/packages/integrations/partytown/src/index.ts b/packages/integrations/partytown/src/index.ts
@@ -1,8 +1,9 @@
 import { partytownSnippet } from '@builder.io/partytown/integration';
-import { copyLibFiles } from '@builder.io/partytown/utils';
+import { copyLibFiles, libDirPath } from '@builder.io/partytown/utils';
 import type { AstroConfig, AstroIntegration } from 'astro';
 import { createRequire } from 'module';
 import path from 'path';
+import * as fs from 'fs';
 import { fileURLToPath } from 'url';
 import sirv from './sirv.js';
 const resolve = createRequire(import.meta.url).resolve;
@@ -50,6 +51,14 @@ export default function createPlugin(options: PartytownOptions): AstroIntegratio
 					debugDir: false,
 				});
 			},
+			'astro:build:ssr': async ({ manifest }) => {
+				const dirpath = libDirPath({ debugDir: false });
+				const files = await fs.promises.readdir(dirpath);
+				for(const file of files) {
+					if(file === 'debug') continue;
+					manifest.assets.push(`/~partytown/${file}`)
+				}
+			}
 		},
 	};
 }
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml