feat: feat: Add forwardRef for Badge

oceanbase · Nov 2, 2023 · 8ff84da · 8ff84da
1 parent 995b66e
commit 8ff84da
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/packages/design/src/button/index.tsx b/packages/design/src/button/index.tsx
@@ -9,13 +9,17 @@ export * from 'antd/es/button';
 
 export type ButtonProps = AntButtonProps;
 
-const Button = ({ prefixCls: customizePrefixCls, className, ...restProps }: ButtonProps) => {
-  const { getPrefixCls } = useContext(ConfigProvider.ConfigContext);
-  const prefixCls = getPrefixCls('btn', customizePrefixCls);
-  const { wrapSSR } = useStyle(prefixCls);
-  const buttonCls = classNames(className);
-  return wrapSSR(<AntButton prefixCls={customizePrefixCls} className={buttonCls} {...restProps} />);
-};
+const Button = React.forwardRef<HTMLButtonElement | HTMLAnchorElement, ButtonProps>(
+  ({ prefixCls: customizePrefixCls, className, ...restProps }, ref) => {
+    const { getPrefixCls } = useContext(ConfigProvider.ConfigContext);
+    const prefixCls = getPrefixCls('btn', customizePrefixCls);
+    const { wrapSSR } = useStyle(prefixCls);
+    const buttonCls = classNames(className);
+    return wrapSSR(
+      <AntButton ref={ref} prefixCls={customizePrefixCls} className={buttonCls} {...restProps} />
+    );
+  }
+);
 
 Button.Group = AntButton.Group;
 Button.__ANT_BUTTON = (