From eeda959130dbb8febe33a6060f34a213759675db Mon Sep 17 00:00:00 2001
From: Naor Peled <me@naor.dev>
Date: Thu, 16 May 2024 01:28:54 +0300
Subject: [PATCH] fix(packages/http-router): refine return type to include
 event and response

---
 packages/http-router/index.d.ts      | 12 +++++++-----
 packages/http-router/index.test-d.ts | 22 ++++++++++++++++++++--
 2 files changed, 27 insertions(+), 7 deletions(-)

diff --git a/packages/http-router/index.d.ts b/packages/http-router/index.d.ts
index bfe685ab7..6ff043fad 100644
--- a/packages/http-router/index.d.ts
+++ b/packages/http-router/index.d.ts
@@ -11,9 +11,7 @@ import {
 
 export type Method = 'GET' | 'POST' | 'PUT' | 'PATCH' | 'DELETE' | 'OPTIONS' | 'HEAD' | 'ANY'
 
-type TResult = ALBResult | APIGatewayProxyResult | APIGatewayProxyResultV2
-
-export interface Route<TEvent> {
+export interface Route<TEvent, TResult> {
   method: Method
   path: string
   handler: LambdaHandler<TEvent, TResult> | MiddyfiedHandler<TEvent, TResult, any, any>
@@ -23,7 +21,11 @@ declare function httpRouterHandler<
   TEvent extends
   | ALBEvent
   | APIGatewayProxyEvent
-  | APIGatewayProxyEventV2 = APIGatewayProxyEvent
-> (routes: Array<Route<TEvent>>): middy.MiddyfiedHandler
+  | APIGatewayProxyEventV2 = APIGatewayProxyEvent,
+  TResult extends
+  | ALBResult
+  | APIGatewayProxyResult
+  | APIGatewayProxyResultV2 = APIGatewayProxyResult
+> (routes: Array<Route<TEvent, TResult>>): middy.MiddyfiedHandler<TEvent, TResult>
 
 export default httpRouterHandler
diff --git a/packages/http-router/index.test-d.ts b/packages/http-router/index.test-d.ts
index eea386b3e..c26d12a11 100644
--- a/packages/http-router/index.test-d.ts
+++ b/packages/http-router/index.test-d.ts
@@ -2,6 +2,7 @@ import { expectType } from 'tsd'
 import middy from '@middy/core'
 import httpRouterHandler from '.'
 import {
+  ALBEvent, ALBResult,
   APIGatewayProxyEvent,
   APIGatewayProxyEventV2,
   APIGatewayProxyResult,
@@ -23,7 +24,7 @@ const middleware = httpRouterHandler([
     handler: lambdaHandler
   }
 ])
-expectType<middy.MiddyfiedHandler>(middleware)
+expectType<middy.MiddyfiedHandler<APIGatewayProxyEvent, APIGatewayProxyResult>>(middleware)
 
 const lambdaHandlerV2: LambdaHandler<APIGatewayProxyEventV2, APIGatewayProxyResultV2> = async (event) => {
   return {
@@ -39,4 +40,21 @@ const middlewareV2 = httpRouterHandler([
     handler: lambdaHandlerV2
   }
 ])
-expectType<middy.MiddyfiedHandler>(middlewareV2)
+expectType<middy.MiddyfiedHandler<APIGatewayProxyEventV2, APIGatewayProxyResultV2>>(middlewareV2)
+
+const lambdaHandlerALB: LambdaHandler<ALBEvent, ALBResult> = async (event) => {
+  return {
+    statusCode: 200,
+    body: 'Hello world'
+  }
+}
+
+const middlewareALB = httpRouterHandler([
+  {
+    method: 'GET',
+    path: '/',
+    handler: lambdaHandlerALB
+  }
+])
+
+expectType<middy.MiddyfiedHandler<ALBEvent, ALBResult>>(middlewareALB)