From 966f7bf3a10f2b074fe2c2abe4a7298c909c432a Mon Sep 17 00:00:00 2001
From: Bero <berislav.grgicak@gmail.com>
Date: Thu, 4 Jul 2024 12:49:45 +0200
Subject: [PATCH] Generate a list assets to cache for offline support

---
 packages/playground/website/vite.config.ts    |  6 ++
 .../vite-website-cache-paths.ts               | 77 +++++++++++++++++++
 2 files changed, 83 insertions(+)
 create mode 100644 packages/vite-extensions/vite-website-cache-paths.ts

diff --git a/packages/playground/website/vite.config.ts b/packages/playground/website/vite.config.ts
index 564dc9f0ad..361c62a9c9 100644
--- a/packages/playground/website/vite.config.ts
+++ b/packages/playground/website/vite.config.ts
@@ -19,6 +19,7 @@ import { fileURLToPath } from 'node:url';
 import { copyFileSync, existsSync, cpSync } from 'node:fs';
 import { join } from 'node:path';
 import { buildVersionPlugin } from '../../vite-extensions/vite-build-version';
+import { websiteCachePathsPlugin } from '../../vite-extensions/vite-website-cache-paths';
 
 const proxy = {
 	'^/plugin-proxy': {
@@ -151,6 +152,10 @@ export default defineConfig(({ command, mode }) => {
 					}
 				},
 			} as Plugin,
+			/**
+			 * Merge cache manifest files into a single `cache-files.json` file.
+			 */
+			websiteCachePathsPlugin() as Plugin,
 		],
 
 		// Configuration for building your library.
@@ -195,6 +200,7 @@ export default defineConfig(({ command, mode }) => {
 				// },
 				external: [],
 			},
+			manifest: 'website-cache-files.json',
 		},
 
 		test: {
diff --git a/packages/vite-extensions/vite-website-cache-paths.ts b/packages/vite-extensions/vite-website-cache-paths.ts
new file mode 100644
index 0000000000..f8b5b88a23
--- /dev/null
+++ b/packages/vite-extensions/vite-website-cache-paths.ts
@@ -0,0 +1,77 @@
+import { readdirSync, statSync, writeFileSync } from 'node:fs';
+import { join } from 'node:path';
+
+const patternsToCache = [
+	'/',
+	'/index.html',
+	'/sw.js',
+	'/favicon.ico',
+	'/remote.html',
+	/^\/logo-[\w]+\.png/,
+	/^\/wp-[\w]\/wordpress-static.zip/,
+	/^\/worker-thread-[\w]+\.js/,
+	/^\/assets\/index-[\w]+\.js/,
+	/^\/assets\/modulepreload-polyfill-[\w]+\.js/,
+	/^\/assets\/index-[\w]+\.css/,
+	/^\/assets\/preload-helper-[\w]+\.js/,
+	/^\/assets\/main-[\w]+\.css/,
+	/^\/assets\/client-[\w]+\.js/,
+	/^\/assets\/config-[\w]+\.js/,
+	/^\/assets\/main-[\w]+\.js/,
+	/^\/assets\/wordpress-[\w]+\.js/,
+	/^\/assets\/remote-[\w]+\.css/,
+	/^\/assets\/sqlite-database-integration-[\w]+\.zip/,
+	/^\/assets\/wp-\d+(\.\d+)+-[\w]+.zip/,
+	/^\/assets\/php_[\w-]+\.js/,
+	/^\/assets\/php_[\w-]+\.wasm/,
+	/^\/wp-\w+((\.\d+)?)+\/wordpress-static.zip/,
+];
+
+export const websiteCachePathsPlugin = () => {
+	function listFiles(dirPath: string, fileList: string[] = []) {
+		const files = readdirSync(dirPath);
+
+		files.forEach((file) => {
+			const filePath = join(dirPath, file);
+			const fileStat = statSync(filePath);
+
+			if (fileStat.isDirectory()) {
+				listFiles(filePath, fileList);
+			} else {
+				fileList.push(filePath);
+			}
+		});
+
+		return fileList;
+	}
+	return {
+		name: 'website-cache-paths-plugin',
+		apply: 'build',
+		writeBundle({ dir: outputDir }: { dir: string }) {
+			const outputManifestPath = join(outputDir, 'cache-files.json');
+			const directoriesToList = ['/', '../remote', '../client'];
+
+			const files = directoriesToList.flatMap((dir) => {
+				const fullDirPath = join(outputDir, dir);
+				return listFiles(fullDirPath)
+					.map((file) => {
+						file = file.replace(fullDirPath, '');
+						if (file.startsWith('/')) {
+							return file;
+						}
+						return `/${file}`;
+					})
+					.filter((item) => {
+						return patternsToCache.some((pattern) => {
+							if (pattern instanceof RegExp) {
+								return pattern.test(item);
+							}
+							return pattern === item;
+						});
+					});
+			});
+
+			writeFileSync(outputManifestPath, JSON.stringify(files, null, 2));
+		},
+	};
+};