From bd6f2a7c2900620618ba543f9fc77cb2cc043da7 Mon Sep 17 00:00:00 2001
From: Diablohu <Diablohu@users.noreply.github.com>
Date: Tue, 26 Nov 2019 16:09:26 +0800
Subject: [PATCH] fix #191

---
 CHANGELOG.md                                  |  4 +-
 packages/koot/React/component-extender.d.ts   | 42 ++++++++++++++++++-
 .../src/components/ssr/controled-ssr.tsx      |  2 +-
 .../standard/src/components/ssr/index.tsx     |  2 +-
 4 files changed, 43 insertions(+), 7 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index dd117b8f..604a1d5c 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -28,6 +28,7 @@
         -   所有注入/插入的 `<script>` 标签现在均会新增 `[data-koot-entry]` 属性
         -   特殊 JS 入口 _CLIENT_RUN_FIRST_ 现默认引入 `regenerator-runtime/runtime`
     -   现在开发环境下会默认启用 _Service Worker_
+    -   更新 TS 定义 ([#191](https://github.com/cmux/koot/issues/191))
 
 **koot-boilerplate**
 
@@ -40,9 +41,6 @@
 
 -   错误修正
     -   新建项目时的“覆盖”模式会删除原有文件的问题 ([#180](https://github.com/cmux/koot/issues/180))
--   更新依赖包 **major** 版本号
-    -   `chalk` -> _^3.0.0_
-    -   `ora` -> _^4.0.3_
 
 ## [0.11.15] - 2019-10-15
 
diff --git a/packages/koot/React/component-extender.d.ts b/packages/koot/React/component-extender.d.ts
index 51d78384..e6d91f59 100644
--- a/packages/koot/React/component-extender.d.ts
+++ b/packages/koot/React/component-extender.d.ts
@@ -11,7 +11,13 @@ import {
     ComponentState
 } from 'react';
 import { Store, Dispatch } from 'redux';
-import { MapStateToPropsParam } from 'react-redux';
+import {
+    MapStateToPropsParam,
+    MapDispatchToPropsNonObject,
+    MapDispatchToPropsParam,
+    MergeProps,
+    Options as ReactReduxOptions
+} from 'react-redux';
 
 // ============================================================================
 
@@ -45,7 +51,39 @@ class HOC extends Component {}
 
 export function extend<P = {}, S = ComponentState>(options: {
     // connect?: Connect;
-    connect?: MapStateToPropsParam<any, P, any>;
+    connect?:
+        | MapStateToPropsParam<any, P, S>
+        | [null | undefined | MapStateToPropsParam<any, P, S>]
+        | [
+              null | undefined | MapStateToPropsParam<any, P, S>,
+              (
+                  | null
+                  | undefined
+                  | MapDispatchToPropsNonObject<any, P>
+                  | MapDispatchToPropsParam<any, P>
+              )
+          ]
+        | [
+              null | undefined | MapStateToPropsParam<any, P, S>,
+              (
+                  | null
+                  | undefined
+                  | MapDispatchToPropsNonObject<any, P>
+                  | MapDispatchToPropsParam<any, P>
+              ),
+              null | undefined | MergeProps<any, any, P, any>
+          ]
+        | [
+              null | undefined | MapStateToPropsParam<any, P, S>,
+              (
+                  | null
+                  | undefined
+                  | MapDispatchToPropsNonObject<any, P>
+                  | MapDispatchToPropsParam<any, P>
+              ),
+              null | undefined | MergeProps<any, any, P, any>,
+              null | undefined | ReactReduxOptions<S, any, P>
+          ];
     /** 提供页面的 title 和 meta 标签信息 */
     pageinfo?: Pageinfo | ExtendPageinfoFunction;
     data?: DataFetchFunction | ExtendData;
diff --git a/test/projects/standard/src/components/ssr/controled-ssr.tsx b/test/projects/standard/src/components/ssr/controled-ssr.tsx
index 54343805..2db14624 100644
--- a/test/projects/standard/src/components/ssr/controled-ssr.tsx
+++ b/test/projects/standard/src/components/ssr/controled-ssr.tsx
@@ -15,7 +15,7 @@ const SSR: React.ComponentClass = extend({
     ssr: <SSR />
 })
 class ControledSSR extends React.Component<ExtendedProps> {
-    render() {
+    render(): React.ReactNode {
         return <div id="koot-test-controled-ssr">Controled SSR</div>;
     }
 }
diff --git a/test/projects/standard/src/components/ssr/index.tsx b/test/projects/standard/src/components/ssr/index.tsx
index 03d55056..a9b3b503 100644
--- a/test/projects/standard/src/components/ssr/index.tsx
+++ b/test/projects/standard/src/components/ssr/index.tsx
@@ -8,7 +8,7 @@ import Controled from './controled-ssr';
     styles: require('./index.module.less')
 })
 class SSRSamples extends React.Component<ExtendedProps> {
-    render() {
+    render(): React.ReactNode {
         return (
             <div className={this.props.className} id="koot-test-ssr">
                 <NoSSR />