feat(type): value support generics

src/index.tsx

@@ -10,25 +10,31 @@ export type SegmentedValue = string | number;
 
 export type SegmentedRawOption = SegmentedValue;
 
-export interface SegmentedLabeledOption {
+export interface SegmentedLabeledOption<ValueType = SegmentedRawOption> {
   className?: string;
   disabled?: boolean;
   label: React.ReactNode;
-  value: SegmentedRawOption;
+  value: ValueType;
   /**
    * html `title` property for label
    */
   title?: string;
 }
 
-type SegmentedOptions = (SegmentedRawOption | SegmentedLabeledOption)[];
-
-export interface SegmentedProps
-  extends Omit<React.HTMLProps<HTMLDivElement>, 'onChange'> {
-  options: SegmentedOptions;
-  defaultValue?: SegmentedValue;
-  value?: SegmentedValue;
-  onChange?: (value: SegmentedValue) => void;
+type SegmentedOptions<T = SegmentedRawOption> = (
+  | T
+  | SegmentedLabeledOption<T>
+)[];
+
+export interface SegmentedProps<ValueType = SegmentedValue>
+  extends Omit<
+    React.HTMLProps<HTMLDivElement>,
+    'defaultValue' | 'value' | 'onChange'
+  > {
+  options: SegmentedOptions<ValueType>;
+  defaultValue?: ValueType;
+  value?: ValueType;
+  onChange?: (value: ValueType) => void;
   disabled?: boolean;
   prefixCls?: string;
   direction?: 'ltr' | 'rtl';
@@ -46,7 +52,9 @@ function getValidTitle(option: SegmentedLabeledOption) {
   }
 }
 
-function normalizeOptions(options: SegmentedOptions): SegmentedLabeledOption[] {
+function normalizeOptions(
+  options: SegmentedProps['options'],
+): SegmentedLabeledOption[] {
   return options.map((option) => {
     if (typeof option === 'object' && option !== null) {
       const validTitle = getValidTitle(option);
@@ -219,6 +227,14 @@ const Segmented = React.forwardRef<HTMLDivElement, SegmentedProps>(
   },
 );
 
-Segmented.displayName = 'Segmented';
+if (process.env.NODE_ENV !== 'production') {
+  Segmented.displayName = 'Segmented';
+}
+
+const TypedSegmented = Segmented as <ValueType>(
+  props: SegmentedProps<ValueType> & {
+    ref?: React.ForwardedRef<HTMLDivElement>;
+  },
+) => ReturnType<typeof Segmented>;
 
-export default Segmented;
+export default TypedSegmented;