🔧 fix: handle .then

Author: SaltyAom

CHANGELOG.md

@@ -1,3 +1,7 @@
+# 1.1.23 - 22 Oct 2024
+Bug fix:
+- Handle object with `.then` even if it's not promise (looking at you, Drizzle)
+
 # 1.1.22 - 13 Oct 2024
 Bug fix:
 - Fix `set-cookie` to resent if value is accessed even without set

package.json

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

src/handler.ts

@@ -448,7 +448,13 @@ export const mapResponse = (
 					// @ts-expect-error
 					return handleStream(response as any, set, request)
 
-				if ('toResponse' in (response as any))
+				// @ts-expect-error
+				if (typeof response?.then === 'function')
+					// @ts-expect-error
+					return response.then((x) => mapResponse(x, set)) as any
+
+				// @ts-expect-error
+				if (typeof response?.toResponse === 'function')
 					return mapResponse((response as any).toResponse(), set)
 
 				if ('charCodeAt' in (response as any)) {
@@ -588,7 +594,13 @@ export const mapResponse = (
 					// @ts-expect-error
 					return handleStream(response as any, set, request)
 
-				if ('toResponse' in (response as any))
+				// @ts-expect-error
+				if (typeof response?.then === 'function')
+					// @ts-expect-error
+					return response.then((x) => mapResponse(x, set)) as any
+
+				// @ts-expect-error
+				if (typeof response?.toResponse === 'function')
 					return mapResponse((response as any).toResponse(), set)
 
 				if ('charCodeAt' in (response as any)) {
@@ -832,8 +844,14 @@ export const mapEarlyResponse = (
 					// @ts-expect-error
 					return handleStream(response as any, set, request)
 
-				if ('toResponse' in (response as any))
-					return mapEarlyResponse((response as any).toResponse(), set)
+				// @ts-expect-error
+				if (typeof response?.then === 'function')
+					// @ts-expect-error
+					return response.then((x) => mapResponse(x, set)) as any
+
+				// @ts-expect-error
+				if (typeof response?.toResponse === 'function')
+					return mapResponse((response as any).toResponse(), set)
 
 				if ('charCodeAt' in (response as any)) {
 					const code = (response as any).charCodeAt(0)
@@ -966,8 +984,14 @@ export const mapEarlyResponse = (
 					// @ts-expect-error
 					return handleStream(response as any, set, request)
 
-				if ('toResponse' in (response as any))
-					return mapEarlyResponse((response as any).toResponse(), set)
+				// @ts-expect-error
+				if (typeof response?.then === 'function')
+					// @ts-expect-error
+					return response.then((x) => mapResponse(x, set)) as any
+
+				// @ts-expect-error
+				if (typeof response?.toResponse === 'function')
+					return mapResponse((response as any).toResponse(), set)
 
 				if ('charCodeAt' in (response as any)) {
 					const code = (response as any).charCodeAt(0)
@@ -1097,8 +1121,14 @@ export const mapCompactResponse = (
 				// @ts-expect-error
 				return handleStream(response as any, undefined, request)
 
-			if ('toResponse' in (response as any))
-				return mapCompactResponse((response as any).toResponse())
+			// @ts-expect-error
+			if (typeof response?.then === 'function')
+				// @ts-expect-error
+				return response.then((x) => mapResponse(x, set)) as any
+
+			// @ts-expect-error
+			if (typeof response?.toResponse === 'function')
+				return mapResponse((response as any).toResponse(), set)
 
 			if ('charCodeAt' in (response as any)) {
 				const code = (response as any).charCodeAt(0)