From 2e245c4d7215fe1a30c0c133ec2cfa65b7711ca3 Mon Sep 17 00:00:00 2001
From: MadCcc <1075746765@qq.com>
Date: Fri, 15 Dec 2023 11:22:49 +0800
Subject: [PATCH] refactor: better classNames logic (#612)

---
 src/InputNumber.tsx | 17 ++++-------------
 1 file changed, 4 insertions(+), 13 deletions(-)

diff --git a/src/InputNumber.tsx b/src/InputNumber.tsx
index 0886c130..b8c0b2ac 100644
--- a/src/InputNumber.tsx
+++ b/src/InputNumber.tsx
@@ -17,6 +17,7 @@ import { getDecupleSteps } from './utils/numberUtil';
 
 import { InputFocusOptions, triggerFocus } from 'rc-input/lib/utils/commonUtils';
 import useFrame from './hooks/useFrame';
+import { BaseInputProps } from 'rc-input/lib/interface';
 
 export type { ValueType };
 
@@ -67,15 +68,7 @@ export interface InputNumberProps<T extends ValueType = ValueType>
   suffix?: React.ReactNode;
   addonBefore?: React.ReactNode;
   addonAfter?: React.ReactNode;
-  classes?: {
-    affixWrapper?: string;
-    group?: string;
-    wrapper?: string;
-  };
-  classNames?: {
-    affixWrapper?: string;
-    group?: string;
-    wrapper?: string;
+  classNames?: BaseInputProps['classNames'] & {
     input?: string;
   };
 
@@ -583,7 +576,7 @@ const InternalInputNumber = React.forwardRef(
     // ============================ Render ============================
     return (
       <div
-        className={clsx(prefixCls, classNames?.input, className, {
+        className={clsx(prefixCls, className, {
           [`${prefixCls}-focused`]: focus,
           [`${prefixCls}-disabled`]: disabled,
           [`${prefixCls}-readonly`]: readOnly,
@@ -644,7 +637,6 @@ const InputNumber = React.forwardRef(
       suffix,
       addonBefore,
       addonAfter,
-      classes,
       className,
       classNames,
       ...rest
@@ -670,7 +662,6 @@ const InputNumber = React.forwardRef(
         suffix={suffix}
         addonAfter={addonAfter}
         addonBefore={addonBefore}
-        classes={classes}
         classNames={classNames}
         components={{
           affixWrapper: 'div',
@@ -682,8 +673,8 @@ const InputNumber = React.forwardRef(
         <InternalInputNumber
           prefixCls={prefixCls}
           disabled={disabled}
-          classNames={classNames}
           ref={composeRef(inputFocusRef, ref)}
+          className={classNames?.input}
           {...rest}
         />
       </BaseInput>