refactor: Drop netlify-lambda, use Vite middleware for Netlify functions (#1108)

* refactor: Drop netlify-lambda, use Vite middleware for Netlify functions

Co-authored-by: Marvin Hagemeister <hello@marvinh.dev>

* feat: Add middleware plugin for serving prerendered HTML

---------

Co-authored-by: Marvin Hagemeister <hello@marvinh.dev>

Author: rschristian

package-lock.json

@@ -7,13 +7,9 @@
     "node": ">=18"
   },
   "scripts": {
-    "dev": "concurrently \"npm run dev:client\" \"npm run dev:lambda\"",
-    "dev:client": "vite --port 8080",
-    "dev:lambda": "netlify-lambda serve src/lambda",
-    "start": "npm run -s server",
-    "prestart": "npm run -s build",
-    "server": "superstatic build -p ${PORT:-8080} --host 0.0.0.0 --compression -c '{\"headers\":[{\"source\":\"**\",\"headers\":[{\"key\":\"Cache-Control\",\"value\":\"max-age=31536000\"}]}]}'",
-    "build": "netlify-lambda build src/lambda && vite build",
+    "dev": "vite --port 8080",
+    "server": "vite preview --port 8080",
+    "build": "vite build",
     "lint": "eslint src test",
     "format": "prettier --write \"{src,test}/**/*.{css,js,json}\"",
     "prepare": "husky install",
@@ -63,11 +59,9 @@
     "feed": "^4.2.2",
     "husky": "^8.0.3",
     "lint-staged": "^15.2.0",
-    "netlify-lambda": "^2.0.16",
     "patch-package": "^8.0.0",
     "prettier": "^1.19.1",
     "rimraf": "^5.0.5",
-    "superstatic": "^9.0.3",
     "vite": "^5.0.11",
     "vite-plugin-static-copy": "1.0.2"
   },

package.json

@@ -10,8 +10,8 @@ export const handler = async event => {
 			url
 		}),
 		headers: {
-			'Cache-Control': 'public, max-age=3600, stale-if-error=86400',
-			'Access-Control-Allow-Origin': 'http://localhost:8080'
+			'Content-Type': 'application/json',
+			'Cache-Control': 'public, max-age=3600, stale-if-error=86400'
 		}
 	};
 };

patches/netlify-lambda+2.0.16.patch

@@ -6,8 +6,8 @@ export const handler = async event => {
 		statusCode: 200,
 		body: JSON.stringify(result),
 		headers: {
-			'Cache-Control': 'public, s-maxage=1800',
-			'Access-Control-Allow-Origin': 'http://localhost:8080'
+			'Content-Type': 'application/json',
+			'Cache-Control': 'public, max-age=3600, stale-if-error=86400'
 		}
 	};
 };

src/lambda/release.js

@@ -15,10 +15,7 @@ export const fallbackData = {
 	preactStargazers: 35783
 };
 
-const baseUrl =
-	process.env.NODE_ENV === 'production'
-		? '/.netlify/functions/'
-		: 'http://localhost:9000/.netlify/functions/';
+const baseUrl = '/.netlify/functions/';
 
 export const repoInfo = repo =>
 	fetch(`${baseUrl}repo?repo=${repo}`, { credentials: 'omit' })

src/lambda/repo.js

@@ -3,6 +3,8 @@ import preact from '@preact/preset-vite';
 import { viteStaticCopy } from 'vite-plugin-static-copy';
 import replace from '@rollup/plugin-replace';
 import yaml from 'yaml';
+import path from 'path';
+import { promises as fs } from 'fs';
 import { Feed } from 'feed';
 import config from './src/config.json';
 
@@ -79,10 +81,96 @@ export default defineConfig({
 				reloadPageOnChange: true
 			}
 		}),
+		netlifyPlugin(),
+		servePrerenderedHTML(),
 		rssFeedPlugin()
 	]
 });
 
+/**
+ * @returns {import('vite').Plugin}
+ */
+function servePrerenderedHTML() {
+	let outDir;
+
+	return {
+		name: 'serve-prerendered-html',
+		config(config) {
+			outDir = path.join(__dirname, config.build.outDir);
+		},
+		configurePreviewServer(server) {
+			server.middlewares.use(async (req, _res, next) => {
+				if (!req.url) return next();
+
+				const url = new URL(req.url, `http://${req.headers.host}`);
+				// If URL has a file extension, bail
+				if (url.pathname != url.pathname.split('.').pop()) return next();
+
+				const file = path.join(
+					outDir,
+					url.pathname
+						.split(path.posix.sep)
+						.join(path.sep),
+					'index.html'
+				);
+
+				try {
+					await fs.access(file);
+					req.url += '/index.html';
+				} catch {
+					req.url = '/404/index.html';
+				}
+
+				return next();
+			});
+		}
+	};
+}
+
+/**
+ * @returns {import('vite').Plugin}
+ */
+function netlifyPlugin() {
+	const lambdaDir = path.join(__dirname, 'src', 'lambda');
+
+	async function netlifyFunctionMiddleware(req, res, next) {
+		if (!req.url) return next();
+
+		const url = new URL(req.url, `http://${req.headers.host}`);
+		if (!url.pathname.startsWith('/.netlify/functions/')) return next();
+
+		const file = path.join(
+			lambdaDir,
+			url.pathname
+				.slice('/.netlify/functions/'.length)
+				.split(path.posix.sep)
+				.join(path.sep)
+		);
+
+		const m = await import(`${file}.js`);
+		const result = await m.handler({
+			queryStringParameters: Object.fromEntries(url.searchParams)
+		});
+
+		for (const [k, v] of Object.entries(result.headers)) {
+			res.setHeader(k, v);
+		}
+
+		res.statusCode = result.statusCode;
+		res.end(result.body);
+	}
+
+	return {
+		name: 'netlify-plugin',
+		configureServer(server) {
+			server.middlewares.use(netlifyFunctionMiddleware);
+		},
+		configurePreviewServer(server) {
+			server.middlewares.use(netlifyFunctionMiddleware);
+		}
+	};
+}
+
 /**
  * @returns {import('vite').Plugin}
  */