feat: add plugin handle404 middleware (#420)

Author: stevenle

.changeset/dry-pans-buy.md

@@ -0,0 +1,5 @@
+---
+'@blinkk/root': patch
+---
+
+feat: add plugin handle404 middleware

packages/root/src/cli/dev.ts

@@ -119,6 +119,15 @@ export async function createDevServer(options?: {
 
       // Add the root.js dev server middlewares.
       server.use(rootDevServerMiddleware());
+
+      // Add any custom plugin 404 handlers.
+      plugins.forEach((plugin) => {
+        if (plugin.handle404) {
+          server.use(plugin.handle404);
+        }
+      });
+
+      // Add error handlers.
       server.use(rootDevServer404Middleware());
       server.use(rootDevServer500Middleware());
     },

packages/root/src/cli/preview.ts

@@ -106,6 +106,13 @@ export async function createPreviewServer(options: {
       // Add the root.js preview server middlewares.
       server.use(rootPreviewServerMiddleware());
 
+      // Add any custom plugin 404 handlers.
+      plugins.forEach((plugin) => {
+        if (plugin.handle404) {
+          server.use(plugin.handle404);
+        }
+      });
+
       // Add error handlers.
       server.use(rootPreviewServer404Middleware());
       server.use(rootPreviewServer500Middleware());

packages/root/src/cli/start.ts

@@ -101,6 +101,13 @@ export async function createProdServer(options: {
       // Add the root.js preview server middlewares.
       server.use(rootProdServerMiddleware());
 
+      // Add any custom plugin 404 handlers.
+      plugins.forEach((plugin) => {
+        if (plugin.handle404) {
+          server.use(plugin.handle404);
+        }
+      });
+
       // Add error handlers.
       server.use(rootProdServer404Middleware());
       server.use(rootProdServer500Middleware());

packages/root/src/core/plugin.ts

@@ -1,6 +1,6 @@
 import {ViteDevServer, PluginOption as VitePlugin} from 'vite';
 import {RootConfig} from './config.js';
-import {Server} from './types.js';
+import {NextFunction, Request, Response, Server} from './types.js';
 
 type MaybePromise<T> = T | Promise<T>;
 
@@ -55,6 +55,12 @@ export interface Plugin {
   vitePlugins?: VitePlugin[];
   /** Plugin lifecycle callback hooks. */
   hooks?: PluginHooks;
+  /** Custom 404 handler. */
+  handle404?: (
+    req: Request,
+    res: Response,
+    next: NextFunction
+  ) => void | Promise<void>;
 }
 
 /**