🎉 feat: mount trace

Author: SaltyAom

example/a.ts

@@ -2,14 +2,16 @@ import { Elysia, t } from '../src'
 
 const a = (request: Request) => new Response(request.url)
 
-const app = new Elysia({ systemRouter: false })
-	.trace((a) => {
-		a.onHandle(() => {
-			// @ts-expect-error private property
-			a.context.url
-		})
+let url = ''
+let hasWrap = false
+
+const app = new Elysia()
+	.wrap((fn) => {
+		console.log('A')
+
+		return fn
 	})
-	.get('/', () => 'ok')
+	.mount('/', () => new Response('OK'))
 	.listen(3000)
 
 fetch('http://localhost:3000/a')

package.json

@@ -1,7 +1,7 @@
 {
 	"name": "elysia",
 	"description": "Ergonomic Framework for Human",
-	"version": "1.3.0-exp.68",
+	"version": "1.3.0-exp.69",
 	"author": {
 		"name": "saltyAom",
 		"url": "https://github.com/SaltyAom",

src/adapter/bun/index.ts

@@ -127,8 +127,14 @@ export const BunAdapter: ElysiaAdapter = {
 
 					if (method === 'ALL') {
 						if (!(`WS_${route.path}` in tree))
-							routes[route.path] =
-								route.hooks?.config?.mount ?? route.handler
+							routes[route.path] = route.hooks?.config?.mount
+								? route.hooks.trace ||
+									app.event.trace ||
+									// @ts-expect-error private property
+									app.extender.higherOrderFunctions
+									? createBunRouteHandler(app, route)
+									: route.hooks.mount || route.handler
+								: route.handler
 
 						continue
 					}

test/bun/router.test.ts

@@ -160,4 +160,83 @@ describe('Bun router', () => {
 		expect(url).toBe(`http://localhost:${app.server!.port}/`)
 		expect(hasRequestId).toBe(true)
 	})
+
+	it('handle wrap in mount', async () => {
+		let url = ''
+		let hasWrap = false
+
+		const app = new Elysia()
+			.wrap((fn) => {
+				hasWrap = true
+
+				return fn
+			})
+			.mount('/', (request) => new Response((url = request.url)))
+			.listen(0)
+
+		await fetch(`http://localhost:${app.server!.port}/`)
+
+		expect(url).toBe(`http://localhost:${app.server!.port}/`)
+		expect(hasWrap).toBe(true)
+	})
+
+	it('handle trace url with wrap', async () => {
+		let url = ''
+		let hasRequestId = false
+		let hasWrap = false
+
+		const app = new Elysia()
+			.wrap((fn) => {
+				hasWrap = true
+
+				return fn
+			})
+			.trace((a) => {
+				a.onHandle(() => {
+					// @ts-expect-error private property
+					url = a.context.url
+
+					// @ts-expect-error private property
+					hasRequestId = !!a.context[ELYSIA_REQUEST_ID]
+				})
+			})
+			.get('/', () => 'ok')
+			.listen(0)
+
+		await fetch(`http://localhost:${app.server!.port}/`)
+
+		expect(url).toBe(`http://localhost:${app.server!.port}/`)
+		expect(hasWrap).toBe(true)
+		expect(hasRequestId).toBe(true)
+	})
+
+	it('handle mount', async () => {
+		let url = ''
+		let hasRequestId = false
+		let hasWrap = false
+
+		const app = new Elysia()
+			.wrap((fn) => {
+				hasWrap = true
+
+				return fn
+			})
+			.trace((a) => {
+				a.onHandle(() => {
+					// @ts-expect-error private property
+					url = a.context.url
+
+					// @ts-expect-error private property
+					hasRequestId = !!a.context[ELYSIA_REQUEST_ID]
+				})
+			})
+			.get('/', () => 'ok')
+			.listen(0)
+
+		await fetch(`http://localhost:${app.server!.port}/`)
+
+		expect(url).toBe(`http://localhost:${app.server!.port}/`)
+		expect(hasWrap).toBe(true)
+		expect(hasRequestId).toBe(true)
+	})
 })