fix(enhance): enhanceParams register

keverjs · Sep 5, 2023 · df0a4d8 · df0a4d8
1 parent e8d0a88
commit df0a4d8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/packages/enhance/src/context/handers.ts b/packages/enhance/src/context/handers.ts
@@ -39,15 +39,15 @@ export const enhanceQuery = (ctx: KoaContext): Context => {
   return ctx as Context
 }
 
-export const enhanceParams = (ctx: Context): Context => {
+export const enhanceParams = (ctx: KoaContext): Context => {
   ctx.getParams = <T>(key?: string) => {
     const params = Object.assign({}, ctx.getBody(), ctx.getQuery())
     if (key) {
       return params['key' as keyof typeof params]
     }
     return params as T
   }
-  return ctx
+  return ctx as Context
 }
 
 /**

diff --git a/packages/enhance/src/context/index.ts b/packages/enhance/src/context/index.ts
@@ -4,6 +4,7 @@ import {
   enhanceLogger,
   enhanceQuery,
   enhanceBody,
+  enhanceParams,
   enhanceHTML,
   enhanceJSON
 } from './handers'
@@ -18,6 +19,7 @@ export const enhanceContext = (ctx: KoaContext): Context => {
     enhanceLogger,
     enhanceQuery,
     enhanceBody,
+    enhanceParams,
     enhanceHTML,
     enhanceJSON
   ])(ctx)