🔧 fix: #1596 handle context pass to function with sub context

Author: SaltyAom

CHANGELOG.md

@@ -7,6 +7,7 @@ Improvement
 Bug fix:
 - [#1591](https://github.com/elysiajs/elysia/pull/1591), [#1590](https://github.com/elysiajs/elysia/pull/1591) merge set.headers without duplicating Response
 - add set immediate fallback
+- [#1596](https://github.com/elysiajs/elysia/issues/1596) handle context pass to function with sub context
 - [#1597](https://github.com/elysiajs/elysia/issues/1597) safeParse, parse to static type
 
 # 1.4.18 - 4 Dec 2025

example/a.ts

@@ -1,13 +1,13 @@
-import { Elysia, t } from '../src'
+import { t, getSchemaValidator } from '../src'
+import { sucrose } from '../src/sucrose'
 
-new Elysia({ aot: false })
-	.guard(
-		{
-			afterResponse: () => {
-				console.log('afterResponse')
-			}
-		},
-		(app) => app.get('/test', () => 'afterResponse')
-	)
-	.get('/', () => 'hi')
-	.listen(3000)
+const v = sucrose({
+	handler: (context) => {
+		console.log('path >>> ', context.path)
+		console.log(context)
+
+		return { id: 'test' }
+	}
+})
+
+console.log(v)

src/sucrose.ts

@@ -551,8 +551,36 @@ export const isContextPassToFunction = (
 ) => {
 	// ! Function is passed to another function, assume as all is accessed
 	try {
-		const captureFunction = new RegExp(`\\w\\((.*?)?${context}`, 'gs')
-		captureFunction.test(body)
+		const captureFunction = new RegExp(
+			`\\w\\((?:.*?)?${context}(?:.*?)?\\)`,
+			'gs'
+		)
+		const exactParameter = new RegExp(`${context}(,|\\))`, 'gs')
+
+		const length = body.length
+		let fn
+
+		fn = captureFunction.exec(body) + ''
+		while (
+			captureFunction.lastIndex !== 0 &&
+			captureFunction.lastIndex < length + (fn ? fn.length : 0)
+		) {
+			if (fn && exactParameter.test(fn)) {
+				inference.query = true
+				inference.headers = true
+				inference.body = true
+				inference.cookie = true
+				inference.set = true
+				inference.server = true
+				inference.url = true
+				inference.route = true
+				inference.path = true
+
+				return true
+			}
+
+			fn = captureFunction.exec(body) + ''
+		}
 
 		/*
 		Since JavaScript engine already format the code (removing whitespace, newline, etc.),
@@ -700,7 +728,11 @@ export const sucrose = (
 				code.charCodeAt(0) === 123 &&
 				code.charCodeAt(body.length - 1) === 125
 			)
-				code = code.slice(1, -1)
+				code = code.slice(1, -1).trim()
+
+			console.log(
+				isContextPassToFunction(mainParameter, code, fnInference)
+			)
 
 			if (!isContextPassToFunction(mainParameter, code, fnInference))
 				inferBodyReference(code, aliases, fnInference)

test/sucrose/sucrose.test.ts

@@ -307,4 +307,25 @@ describe('sucrose', () => {
 			route: true
 		})
 	})
+
+	it('handle context pass to function with sub context', () => {
+		expect(
+			sucrose({
+				handler: (context) => {
+					console.log('path >>> ', context.path)
+					console.log(context)
+				}
+			})
+		).toEqual({
+			query: true,
+			headers: true,
+			body: true,
+			cookie: true,
+			set: true,
+			server: true,
+			path: true,
+			url: true,
+			route: true
+		})
+	})
 })