From 55af46db34cd14898fe163d383b751b29f9d0ab0 Mon Sep 17 00:00:00 2001
From: huangxuan <huangxuan20@huawei.com>
Date: Thu, 30 Nov 2023 15:38:05 +0800
Subject: [PATCH] =?UTF-8?q?[inulax]=20connect=20API=20=E6=94=AF=E6=8C=81fo?=
 =?UTF-8?q?rwardRef=E5=8F=82=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../inula/src/inulax/adapters/reduxReact.ts    | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/packages/inula/src/inulax/adapters/reduxReact.ts b/packages/inula/src/inulax/adapters/reduxReact.ts
index 51650f98..160dec37 100644
--- a/packages/inula/src/inulax/adapters/reduxReact.ts
+++ b/packages/inula/src/inulax/adapters/reduxReact.ts
@@ -17,6 +17,7 @@ import { useState, useContext, useEffect, useRef } from '../../renderer/hooks/Ho
 import { createContext } from '../../renderer/components/context/CreateContext';
 import { createElement } from '../../external/JSXElement';
 import type { ReduxStoreHandler, ReduxAction, BoundActionCreator } from './redux';
+import { forwardRef } from '../../renderer/components/ForwardRef';
 
 const DefaultContext = createContext(null);
 type Context = typeof DefaultContext;
@@ -90,6 +91,11 @@ type MergePropsP<StateProps, DispatchProps, OwnProps, MergedProps> = (
 type WrappedComponent<OwnProps> = (props: OwnProps) => ReturnType<typeof createElement>;
 type OriginalComponent<MergedProps> = (props: MergedProps) => ReturnType<typeof createElement>;
 type Connector<OwnProps, MergedProps> = (Component: OriginalComponent<MergedProps>) => WrappedComponent<OwnProps>;
+type ConnectOption<State = any> = {
+  areStatesEqual?: (oldState: State, newState: State) => boolean;
+  context?: Context;
+  forwardRef?: boolean
+}
 
 export function connect<StateProps, DispatchProps, OwnProps, MergedProps>(
   mapStateToProps: MapStateToPropsP<StateProps, OwnProps> = () => ({} as StateProps),
@@ -99,10 +105,7 @@ export function connect<StateProps, DispatchProps, OwnProps, MergedProps>(
     dispatchProps,
     ownProps
   ): MergedProps => ({ ...stateProps, ...dispatchProps, ...ownProps } as unknown as MergedProps),
-  options?: {
-    areStatesEqual?: (oldState: any, newState: any) => boolean;
-    context?: Context;
-  }
+  options: ConnectOption,
 ): Connector<OwnProps, MergedProps> {
   if (!options) {
     options = {};
@@ -172,6 +175,13 @@ export function connect<StateProps, DispatchProps, OwnProps, MergedProps>(
       return node;
     };
 
+    if (options.forwardRef) {
+      const forwarded = forwardRef((props, ref) => {
+        return Wrapper({ ...props, ref: ref });
+      });
+      return forwarded as WrappedComponent<OwnProps>;
+    }
+
     return Wrapper;
   };
 }