🔧 fix: handle .then

elysiajs · Oct 21, 2024 · 9007186 · 9007186
1 parent 243f8b5
commit 9007186
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/handler.ts b/src/handler.ts
@@ -847,11 +847,11 @@ export const mapEarlyResponse = (
 				// @ts-expect-error
 				if (typeof response?.then === 'function')
 					// @ts-expect-error
-					return response.then((x) => mapResponse(x, set)) as any
+					return response.then((x) => mapEarlyResponse(x, set)) as any
 
 				// @ts-expect-error
 				if (typeof response?.toResponse === 'function')
-					return mapResponse((response as any).toResponse(), set)
+					return mapEarlyResponse((response as any).toResponse(), set)
 
 				if ('charCodeAt' in (response as any)) {
 					const code = (response as any).charCodeAt(0)
@@ -987,11 +987,11 @@ export const mapEarlyResponse = (
 				// @ts-expect-error
 				if (typeof response?.then === 'function')
 					// @ts-expect-error
-					return response.then((x) => mapResponse(x, set)) as any
+					return response.then((x) => mapEarlyResponse(x, set)) as any
 
 				// @ts-expect-error
 				if (typeof response?.toResponse === 'function')
-					return mapResponse((response as any).toResponse(), set)
+					return mapEarlyResponse((response as any).toResponse(), set)
 
 				if ('charCodeAt' in (response as any)) {
 					const code = (response as any).charCodeAt(0)
@@ -1128,7 +1128,7 @@ export const mapCompactResponse = (
 
 			// @ts-expect-error
 			if (typeof response?.toResponse === 'function')
-				return mapResponse((response as any).toResponse(), set)
+				return mapCompactResponse((response as any).toResponse())
 
 			if ('charCodeAt' in (response as any)) {
 				const code = (response as any).charCodeAt(0)