From df0a4d8849bc395302de5b77127a25b16e52a8b2 Mon Sep 17 00:00:00 2001
From: kimwangchong <kimwangchong@didiglobal.com>
Date: Tue, 5 Sep 2023 20:41:46 +0800
Subject: [PATCH] fix(enhance): enhanceParams register

---
 packages/enhance/src/context/handers.ts | 4 ++--
 packages/enhance/src/context/index.ts   | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/packages/enhance/src/context/handers.ts b/packages/enhance/src/context/handers.ts
index 2b51089..a4f3e3b 100644
--- a/packages/enhance/src/context/handers.ts
+++ b/packages/enhance/src/context/handers.ts
@@ -39,7 +39,7 @@ 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) {
@@ -47,7 +47,7 @@ export const enhanceParams = (ctx: Context): Context => {
     }
     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
index 13e4065..3c41d7c 100644
--- 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)