fix: fix handling not found in compose.ts (#204)

honojs · May 6, 2022 · d399609 · d399609
1 parent c4cdad8
commit d399609
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/src/compose.test.ts b/src/compose.test.ts
@@ -157,8 +157,7 @@ describe('compose with Context - 401 not authorized', () => {
     return c.text('onError', 500)
   }
   const handler = (c: Context, next: Function) => {
-    c.text('Hello')
-    next()
+    return c.text('Hello')
   }
   const mHandler = async (c: Context, next: Function) => {
     await next()

diff --git a/src/compose.ts b/src/compose.ts
@@ -17,10 +17,12 @@ export const compose = <C>(
       let handler = middleware[i]
       index = i
       if (i === middleware.length) handler = next
-      if (!handler) return Promise.resolve(context)
 
-      if (onNotFound && context instanceof Context && !context.res) {
-        context.res = onNotFound(context)
+      if (!handler) {
+        if (context instanceof Context && !context.res) {
+          context.res = onNotFound(context)
+        }
+        return Promise.resolve(context)
       }
 
       return Promise.resolve(handler(context, dispatch.bind(null, i + 1)))