feat(plugin): add configureServer hook

.changeset/early-cycles-drive.md

@@ -0,0 +1,5 @@
+---
+'@umijs/tnf': patch
+---
+
+feat(plugin): add configureServer hook

docs/plugin.md

@@ -34,9 +34,7 @@ Called when config resolved.
 
 ### configureServer
 
-> Not implemented
-
-- Type: `(server: DevServer) => void | Promise<void>`
+- Type: `(server: { middlewares: express.Application }) => void | Promise<void>`
 
 Can register pre or post middleware, or save server instance for other hooks.
 

src/dev.ts

@@ -2,12 +2,21 @@ import { build } from './build';
 import { createBundler } from './bundler/bundler';
 import { BundlerType } from './bundler/bundler';
 import { createServer } from './fishkit/server';
+import { PluginHookType } from './plugin/plugin_manager';
 import type { Context } from './types';
 
 export async function dev({ context }: { context: Context }) {
   const devServer = context.config?.devServer || {};
   const { port, host, server, app } = await createServer({
     devServer,
+    configureServer: () => {
+      context.pluginManager.apply({
+        hook: 'configureServer',
+        args: [],
+        type: PluginHookType.Series,
+        pluginContext: context.pluginContext,
+      });
+    },
   });
   const bundler = createBundler({ bundler: BundlerType.MAKO });
   await bundler.configDevServer({

src/fishkit/server.ts

@@ -10,6 +10,9 @@ import { createHttpsServer } from './https';
 
 export interface ServerOpts {
   devServer: Config['devServer'];
+  configureServer?: (opts: {
+    middlewares: express.Application;
+  }) => void | (() => void);
 }
 
 export async function createServer(opts: ServerOpts): Promise<{
@@ -28,6 +31,14 @@ export async function createServer(opts: ServerOpts): Promise<{
   const _port = await getPort(port);
   const app = express();
 
+  const context = {
+    middlewares: app,
+  };
+  let configureServerFn;
+  if (opts.configureServer) {
+    configureServerFn = opts.configureServer(context);
+  }
+
   // cors
   app.use(
     cors({
@@ -47,6 +58,8 @@ export async function createServer(opts: ServerOpts): Promise<{
     }),
   );
 
+  configureServerFn?.();
+
   // create server
   let server;
   if (https) {

src/plugin/types.ts

@@ -22,6 +22,12 @@ export const PluginSchema = z.object({
       z.union([z.any(), z.promise(z.any()), z.null()]),
     )
     .optional(),
+  configureServer: z
+    .function(
+      z.tuple([z.object({ middlewares: z.any() })]),
+      z.union([z.any(), z.promise(z.any()), z.null()]),
+    )
+    .optional(),
 });
 
 export type Plugin = z.infer<typeof PluginSchema>;

src/preview.ts

@@ -1,11 +1,22 @@
 import { resolve } from 'pathe';
 import sirv from 'sirv';
 import { createServer } from './fishkit/server';
+import { PluginHookType } from './plugin/plugin_manager';
 import type { Context } from './types';
 
 export async function preview({ context }: { context: Context }) {
   const devServer = context.config?.devServer || {};
-  const { app } = await createServer({ devServer });
+  const { app } = await createServer({
+    devServer,
+    configureServer: () => {
+      context.pluginManager.apply({
+        hook: 'configureServer',
+        args: [],
+        type: PluginHookType.Series,
+        pluginContext: context.pluginContext,
+      });
+    },
+  });
   const distDir = resolve(context.cwd, 'dist');
 
   app.use(