button

Author: liuliu-dev

packages/styled-react/src/index.tsx

@@ -41,10 +41,15 @@ type StyledProps = SxProp &
 
 type SegmentedControlProps = PrimerSegmentedControlProps & Omit<StyledProps, keyof PrimerSegmentedControlProps>
 
-const SegmentedControl = forwardRef<HTMLUListElement, SegmentedControlProps>(function SegmentedControl(props, ref) {
+const SegmentedControlImpl = forwardRef<HTMLUListElement, SegmentedControlProps>(function SegmentedControl(props, ref) {
   return <Box as={PrimerSegmentedControl} ref={ref} {...props} />
 })
 
+const SegmentedControl = Object.assign(SegmentedControlImpl, {
+  Button: PrimerSegmentedControl.Button,
+  IconButton: PrimerSegmentedControl.IconButton,
+})
+
 type StateLabelProps = PrimerStateLabelProps & SxProp
 
 const StateLabel = forwardRef<HTMLSpanElement, StateLabelProps>(function StateLabel(props, ref) {