fix: antd issue 35170

Author: vagusX

docs/examples/controlled.tsx

@@ -13,15 +13,27 @@ export default class Demo extends React.Component<
 
   render() {
     return (
-      <Segmented
-        options={['iOS', 'Android', 'Web3']}
-        value={this.state.value}
-        onChange={(e) =>
-          this.setState({
-            value: e.target.value,
-          })
-        }
-      />
+      <>
+        <Segmented
+          options={['iOS', 'Android', 'Web3']}
+          value={this.state.value}
+          onChange={(e) =>
+            this.setState({
+              value: e.target.value,
+            })
+          }
+        />
+        &nbsp;&nbsp;
+        <Segmented
+          options={['iOS', 'Android', 'Web3']}
+          value={this.state.value}
+          onChange={(e) =>
+            this.setState({
+              value: e.target.value,
+            })
+          }
+        />
+      </>
     );
   }
 }

src/index.tsx

@@ -104,7 +104,7 @@ const InternalSegmentedOption: React.FC<{
 
   return (
     <label
-      className={classNames(`${prefixCls}-item`, className, {
+      className={classNames(className, {
         [`${prefixCls}-item-disabled`]: disabled,
       })}
     >
@@ -155,34 +155,66 @@ const Segmented = React.forwardRef<HTMLDivElement, SegmentedProps>(
     }, [options]);
 
     const [selected, setSelected] = useMergedState(segmentedOptions[0]?.value, {
-      value,
+      value: props.value,
       defaultValue,
     });
 
     const [visualSelected, setVisualSelected] = React.useState<
       SegmentedRawOption | undefined
     >(selected);
 
-    const [thumbShow, setThumbShow] = React.useState(false);
+    const latestVisualSelected = React.useRef(visualSelected);
 
-    const calcThumbMoveStatus = (
-      event: React.ChangeEvent<HTMLInputElement>,
-    ) => {
-      const toElement = event.target.closest(`.${prefixCls}-item`);
+    React.useEffect(() => {
+      latestVisualSelected.current = visualSelected;
+    });
+
+    const [thumbShow, setThumbShow] = React.useState(false);
 
-      const fromElement = containerRef.current?.querySelector(
-        `.${prefixCls}-item-selected`,
-      );
+    const doThumbAnimation = React.useCallback(
+      (selectedValue: SegmentedRawOption) => {
+        const segmentedItemIndex = segmentedOptions.findIndex(
+          (n) => n.value === selectedValue,
+        );
 
-      if (fromElement && toElement && thumbMoveStatus.current) {
-        thumbMoveStatus.current.from = calcThumbStyle(
-          fromElement as HTMLElement,
+        // find target element
+        const toElement = containerRef.current?.querySelector(
+          `.${prefixCls}-item:nth-child(${segmentedItemIndex + 1})`,
         );
-        thumbMoveStatus.current.to = calcThumbStyle(toElement as HTMLElement);
 
-        setThumbShow(true);
+        if (toElement) {
+          // find source element
+          const fromElement = containerRef.current?.querySelector(
+            `.${prefixCls}-item-selected`,
+          );
+
+          if (fromElement && toElement && thumbMoveStatus.current) {
+            // calculate for thumb moving animation
+            thumbMoveStatus.current.from = calcThumbStyle(
+              fromElement as HTMLElement,
+            );
+            thumbMoveStatus.current.to = calcThumbStyle(
+              toElement as HTMLElement,
+            );
+
+            // trigger css-motion starts
+            setThumbShow(true);
+          }
+        }
+      },
+      [prefixCls, segmentedOptions],
+    );
+
+    React.useEffect(() => {
+      // Syncing `visualSelected` when `selected` changed
+      // and do thumb animation
+      if (
+        (typeof selected === 'string' || typeof selected === 'number') &&
+        selected !== latestVisualSelected.current
+      ) {
+        doThumbAnimation(selected);
       }
-    };
+    }, [selected]);
 
     const handleChange = (
       event: React.ChangeEvent<HTMLInputElement>,
@@ -192,8 +224,6 @@ const Segmented = React.forwardRef<HTMLDivElement, SegmentedProps>(
         return;
       }
 
-      calcThumbMoveStatus(event);
-
       setSelected(val);
 
       if (onChange) {
@@ -275,10 +305,14 @@ const Segmented = React.forwardRef<HTMLDivElement, SegmentedProps>(
           <InternalSegmentedOption
             key={segmentedOption.value}
             prefixCls={prefixCls}
-            className={classNames(segmentedOption.className, {
-              [`${prefixCls}-item-selected`]:
-                segmentedOption.value === visualSelected,
-            })}
+            className={classNames(
+              segmentedOption.className,
+              `${prefixCls}-item`,
+              {
+                [`${prefixCls}-item-selected`]:
+                  segmentedOption.value === visualSelected,
+              },
+            )}
             checked={segmentedOption.value === selected}
             onChange={handleChange}
             {...segmentedOption}

tests/index.spec.tsx

@@ -163,7 +163,7 @@ describe('rc-segmented', () => {
     expect(wrapper.render()).toMatchSnapshot();
     expect(
       wrapper
-        .find('.rc-segmented-item')
+        .find('label.rc-segmented-item')
         .at(1)
         .hasClass('rc-segmented-item-disabled'),
     ).toBeTruthy();