feat(blade): Changes in components for X migration

packages/blade/src/components/Button/IconButton/IconButton.tsx

@@ -9,6 +9,7 @@ import type { BladeCommonEvents } from '~components/types';
 import type { Platform } from '~utils';
 import type { SubtleOrIntense } from '~tokens/theme/theme';
 import { makeAnalyticsAttribute } from '~utils/makeAnalyticsAttribute';
+import type { StyledPropsBlade } from '~components/Box/styledProps';
 
 type IconButtonProps = {
   /**
@@ -46,6 +47,7 @@ type IconButtonProps = {
   _tabIndex?: number;
 } & DataAnalyticsAttribute &
   BladeCommonEvents &
+  StyledPropsBlade &
   Platform.Select<{
     web: {
       onClick: (event: React.MouseEvent<HTMLButtonElement>) => void;
@@ -108,6 +110,7 @@ const _IconButton: React.ForwardRefRenderFunction<BladeElementRef, IconButtonPro
       onTouchEnd={onTouchEnd}
       onTouchStart={onTouchStart}
       {...makeAnalyticsAttribute(rest)}
+      {...rest}
     />
   );
 };

packages/blade/src/components/Button/IconButton/StyledIconButton.web.tsx

@@ -13,6 +13,7 @@ import { getFocusRingStyles } from '~utils/getFocusRingStyles';
 import { throwBladeError } from '~utils/logger';
 import getIn from '~utils/lodashButBetter/get';
 import { makeAnalyticsAttribute } from '~utils/makeAnalyticsAttribute';
+import { useStyledProps } from '~components/Box/styledProps';
 
 type StyledButtonProps = {
   emphasis: SubtleOrIntense;
@@ -23,7 +24,7 @@ type StyledButtonProps = {
 const StyledButton = styled.button<StyledButtonProps>((props) => {
   const { theme, emphasis } = props;
   const motionToken = theme.motion;
-
+  const styledPropsCSSObject = useStyledProps(props);
   const emphasisColor = emphasis === 'intense' ? 'gray' : 'staticWhite';
 
   if (__DEV__) {
@@ -74,6 +75,7 @@ const StyledButton = styled.button<StyledButtonProps>((props) => {
     '&:active': {
       color: theme.colors.interactive.icon[emphasisColor].subtle,
     },
+    ...styledPropsCSSObject,
   };
 });
 
@@ -121,6 +123,7 @@ const StyledIconButton = React.forwardRef<HTMLButtonElement, StyledIconButtonPro
       {...makeAccessible({ label: accessibilityLabel })}
       {...metaAttribute({ name: MetaConstants.IconButton, testID })}
       {...makeAnalyticsAttribute(rest)}
+      {...rest}
     >
       <Icon size={size} color={isDisabled ? 'interactive.icon.gray.disabled' : 'currentColor'} />
     </StyledButton>

packages/blade/src/components/Button/IconButton/types.ts

@@ -3,6 +3,7 @@ import type { IconComponent } from '~components/Icons';
 import type { DataAnalyticsAttribute, RemoveUndefinedFromUnion, TestID } from '~utils/types';
 import type { BladeCommonEvents } from '~components/types';
 import type { SubtleOrIntense } from '~tokens/theme/theme';
+import type { StyledPropsBlade } from '~components/Box/styledProps';
 
 export type StyledIconButtonProps = {
   icon: IconComponent;
@@ -15,4 +16,5 @@ export type StyledIconButtonProps = {
   onClick?: IconButtonProps['onClick'];
 } & TestID &
   BladeCommonEvents &
-  DataAnalyticsAttribute;
+  DataAnalyticsAttribute &
+  StyledPropsBlade;

packages/blade/src/components/ButtonGroup/ButtonGroup.web.tsx

@@ -1,4 +1,3 @@
-import type { ReactElement } from 'react';
 import React from 'react';
 import styled from 'styled-components';
 import type { ButtonGroupProps } from './types';
@@ -13,10 +12,9 @@ import type { DotNotationToken } from '~utils/lodashButBetter/get';
 import getIn from '~utils/lodashButBetter/get';
 import { getBackgroundColorToken } from '~components/Button/BaseButton/BaseButton';
 import type { Theme } from '~components/BladeProvider';
-import { throwBladeError } from '~utils/logger';
-import { isValidAllowedChildren } from '~utils/isValidAllowedChildren';
 import type { BladeElementRef } from '~utils/types';
 import { makeAnalyticsAttribute } from '~utils/makeAnalyticsAttribute';
+import { useVerifyAllowedChildren } from '~utils/useVerifyAllowedChildren';
 
 const getDividerColorToken = ({
   color,
@@ -74,6 +72,12 @@ const _ButtonGroup = (
     isFullWidth,
   };
 
+  useVerifyAllowedChildren({
+    allowedComponents: ['Button', 'Dropdown', 'Tooltip', 'Popover'],
+    componentName: 'ButtonGroup',
+    children,
+  });
+
   return (
     <ButtonGroupProvider value={contextValue}>
       <StyledButtonGroup
@@ -88,26 +92,6 @@ const _ButtonGroup = (
         role="group"
       >
         {React.Children.map(children, (child, index) => {
-          if (__DEV__) {
-            // throw error if child is not a button or dropdown with button trigger
-            /* eslint-disable no-restricted-properties */
-            if (
-              !isValidAllowedChildren(child, 'Button') &&
-              !(
-                isValidAllowedChildren(child, 'Dropdown') &&
-                (child as ReactElement).props.children.some((c: ReactElement) =>
-                  isValidAllowedChildren(c, 'DropdownButton'),
-                )
-              )
-            ) {
-              throwBladeError({
-                moduleName: 'ButtonGroup',
-                message: `Only "Button" or "Dropdown" component with Button trigger are allowed as children.`,
-              });
-            }
-            /* eslint-enable no-restricted-properties */
-          }
-
           return (
             <>
               {child}

packages/blade/src/components/ButtonGroup/__tests__/ButtonGroup.web.test.tsx

@@ -7,7 +7,6 @@ import { Button } from '~components/Button/Button';
 import { ChevronDownIcon, PlusIcon } from '~components/Icons';
 import { Dropdown, DropdownButton, DropdownOverlay } from '~components/Dropdown';
 import { ActionList, ActionListItem } from '~components/ActionList';
-import { AutoComplete } from '~components/Input/DropdownInputTriggers';
 
 beforeAll(() => jest.spyOn(console, 'error').mockImplementation());
 afterAll(() => jest.restoreAllMocks());
@@ -168,40 +167,7 @@ describe('<ButtonGroup />', () => {
         </ButtonGroup>,
       ),
     ).toThrowError(
-      '[Blade: ButtonGroup]: Only "Button" or "Dropdown" component with Button trigger are allowed as children.',
-    );
-  });
-
-  it('should throw error with invalid dropdown children', () => {
-    expect(() =>
-      renderWithTheme(
-        <ButtonGroup>
-          <Button icon={PlusIcon}>Payout</Button>
-          <Dropdown selectionType="single">
-            <AutoComplete
-              label="City"
-              placeholder="Select your City"
-              name="action"
-              onChange={({ name, values }) => {
-                console.log({ name, values });
-              }}
-              onInputValueChange={({ name, value }) => {
-                console.log({ name, value });
-              }}
-            />
-            <DropdownOverlay>
-              <ActionList>
-                <ActionListItem title="Mumbai" value="mumbai" />
-                <ActionListItem title="Pune" value="pune" />
-                <ActionListItem title="Bangalore" value="bangalore" />
-                <ActionListItem title="Mysore" value="mysore" />
-              </ActionList>
-            </DropdownOverlay>
-          </Dropdown>
-        </ButtonGroup>,
-      ),
-    ).toThrowError(
-      '[Blade: ButtonGroup]: Only "Button" or "Dropdown" component with Button trigger are allowed as children.',
+      '[Blade: ButtonGroup]: Only `Button, Dropdown, Tooltip, Popover` components are accepted in `ButtonGroup` children',
     );
   });
 });

packages/blade/src/components/ButtonGroup/docs/ButtonGroup.stories.tsx

@@ -10,6 +10,7 @@ import { getStyledPropsArgTypes } from '~components/Box/BaseBox/storybookArgType
 import { RefreshIcon, ShareIcon, DownloadIcon, ChevronDownIcon, PlusIcon } from '~components/Icons';
 import { Dropdown, DropdownButton, DropdownOverlay } from '~components/Dropdown';
 import { ActionList, ActionListItem } from '~components/ActionList';
+import { Tooltip } from '~components/Tooltip';
 
 const Page = (): React.ReactElement => {
   return (
@@ -76,7 +77,9 @@ const ButtonGroupDropdownTemplate: StoryFn<typeof ButtonGroupComponent> = (args)
   return (
     <Box display="flex" alignItems="center" justifyContent="center">
       <ButtonGroupComponent {...args}>
-        <Button icon={PlusIcon}>Payout</Button>
+        <Tooltip content="Create a new payout">
+          <Button icon={PlusIcon}>Payout</Button>
+        </Tooltip>
         <Dropdown>
           <DropdownButton icon={ChevronDownIcon} />
           <DropdownOverlay defaultPlacement="bottom-end">

packages/blade/src/components/ButtonGroup/types.ts

@@ -1,3 +1,4 @@
+import type { StyledPropsBlade } from '~components/Box/styledProps';
 import type { ButtonProps } from '~components/Button/Button';
 import type { DataAnalyticsAttribute } from '~utils/types';
 
@@ -38,7 +39,8 @@ type ButtonGroupProps = {
    * Test ID for automation
    */
   testID?: string;
-} & DataAnalyticsAttribute;
+} & DataAnalyticsAttribute &
+  StyledPropsBlade;
 
 type StyledButtonGroupProps = Pick<
   ButtonGroupProps,

packages/blade/src/components/Drawer/Drawer.web.tsx

@@ -21,6 +21,8 @@ import { metaAttribute, MetaConstants } from '~utils/metaAttribute';
 import { useId } from '~utils/useId';
 import { useVerifyAllowedChildren } from '~utils/useVerifyAllowedChildren';
 import { makeAnalyticsAttribute } from '~utils/makeAnalyticsAttribute';
+import type { BladeElementRef } from '~utils/types';
+import { mergeRefs } from '~utils/useMergeRefs';
 
 const SHOW_DRAWER = 'show-drawer';
 
@@ -65,18 +67,21 @@ const DrawerOverlay = styled(FloatingOverlay)(({ theme }) => {
   };
 });
 
-const _Drawer = ({
-  isOpen,
-  onDismiss,
-  zIndex = componentZIndices.drawer,
-  children,
-  accessibilityLabel,
-  showOverlay = true,
-  initialFocusRef,
-  isLazy = true,
-  testID,
-  ...rest
-}: DrawerProps): React.ReactElement => {
+const _Drawer: React.ForwardRefRenderFunction<BladeElementRef, DrawerProps> = (
+  {
+    isOpen,
+    onDismiss,
+    zIndex = componentZIndices.drawer,
+    children,
+    accessibilityLabel,
+    showOverlay = true,
+    initialFocusRef,
+    isLazy = true,
+    testID,
+    ...rest
+  },
+  ref,
+) => {
   const closeButtonRef = React.useRef<HTMLDivElement>(null);
   const [zIndexState, setZIndexState] = React.useState<number>(zIndex);
 
@@ -155,7 +160,7 @@ const _Drawer = ({
               {...makeAnalyticsAttribute(rest)}
               zIndex={zIndexState}
             >
-              {showOverlay || stackingLevel === 2 ? (
+              {showOverlay ? (
                 <DrawerOverlay
                   onClick={() => {
                     onDismiss();
@@ -184,7 +189,7 @@ const _Drawer = ({
                 height="100%"
                 display="flex"
                 flexDirection="column"
-                ref={refs.setFloating}
+                ref={mergeRefs(ref, refs.setFloating)}
                 onKeyDown={(event) => {
                   if (event?.key === 'Escape' || event?.code === 'Escape') {
                     onDismiss();
@@ -238,7 +243,8 @@ const _Drawer = ({
  * 
  * 
  */
-const Drawer = assignWithoutSideEffects(_Drawer, {
+const Drawer = assignWithoutSideEffects(React.forwardRef(_Drawer), {
+  displayName: 'Drawer',
   componentId: drawerComponentIds.Drawer,
 });
 

packages/blade/src/components/Radio/Radio.tsx

@@ -88,9 +88,9 @@ const _Radio: React.ForwardRefRenderFunction<BladeElementRef, RadioProps> = (
   const isReactNative = getPlatformType() === 'react-native';
   const _size = groupProps.size ?? size;
 
-  const handleChange: OnChange = ({ isChecked, value }) => {
+  const handleChange: OnChange = ({ isChecked, value, event }) => {
     if (isChecked) {
-      groupProps?.state?.setValue(value!);
+      groupProps?.state?.setValue(value!, event);
     } else {
       groupProps?.state?.removeValue();
     }

packages/blade/src/components/Radio/RadioGroup/RadioGroup.tsx

@@ -76,7 +76,15 @@ type RadioGroupProps = {
   /**
    * The callback invoked when any of the radio's state changes
    */
-  onChange?: ({ name, value }: { name: string | undefined; value: string }) => void;
+  onChange?: ({
+    name,
+    value,
+    event,
+  }: {
+    name: string | undefined;
+    value: string;
+    event: React.ChangeEvent<HTMLInputElement>;
+  }) => void;
   /**
    * The name of the input field in a radio
    * (Useful for form submission).

packages/blade/src/components/Radio/RadioGroup/useRadioGroup.ts

@@ -22,7 +22,7 @@ type UseRadioGroupProps = Pick<
 
 export type State = {
   value: string;
-  setValue(value: string): void;
+  setValue(value: string, event: React.ChangeEvent<Element>): void;
   removeValue(): void;
   isChecked(value: string): boolean;
 };
@@ -52,18 +52,20 @@ const useRadioGroup = ({
   const [checkedValue, setValue] = useControllableState({
     value,
     defaultValue,
-    onChange: (v: string) => onChange?.({ value: v, name: fallbackName }),
+    onChange: (v, event) => {
+      onChange?.({ value: v, name: fallbackName, event });
+    },
   });
 
   const state = React.useMemo<State>(() => {
     return {
       value: checkedValue,
-      setValue(v: string): void {
+      setValue(v, event): void {
         if (isDisabled) {
           return;
         }
 
-        setValue(() => v);
+        setValue(() => v, false, event);
       },
       removeValue(): void {
         if (isDisabled) {

packages/blade/src/components/Radio/useRadio.ts

@@ -14,7 +14,7 @@ export type OnChange = ({
   value,
 }: {
   isChecked: boolean;
-  event?: React.ChangeEvent;
+  event: React.ChangeEvent;
   value?: string;
 }) => void;
 

packages/blade/src/components/SideNav/SideNavItems/SideNavLink.web.tsx

@@ -183,6 +183,7 @@ const SideNavLink = ({
   tooltip,
   as,
   target,
+  onClick,
   ...rest
 }: SideNavLinkProps): React.ReactElement => {
   const {
@@ -249,7 +250,7 @@ const SideNavLink = ({
                 href={as ? undefined : href}
                 target={target}
                 ref={refs.setReference}
-                onClick={() => {
+                onClick={(e: React.MouseEvent<HTMLAnchorElement, MouseEvent>) => {
                   // Close the mobile nav when item is clicked and its not trigger for next menu
                   if (!isL2Trigger) {
                     closeMobileNav?.();
@@ -264,6 +265,8 @@ const SideNavLink = ({
                       isFirstRender: false,
                     });
                   }
+
+                  onClick?.(e);
                 }}
                 onFocus={(e: { target: HTMLDivElement }) => {
                   // FloatinFocusManager by default focusses on last clicked element when you move to different tab and come back to the original tab

packages/blade/src/components/SideNav/docs/SideNav.stories.tsx

@@ -149,7 +149,10 @@ const DashboardSkeleton = ({ children }: { children: React.ReactElement }): Reac
   );
 };
 
-type ItemsType = Pick<SideNavLinkProps, 'icon' | 'title' | 'href' | 'trailing' | 'tooltip'>;
+type ItemsType = Pick<
+  SideNavLinkProps,
+  'icon' | 'title' | 'href' | 'trailing' | 'tooltip' | 'onClick'
+>;
 type NavItemsJSONType = {
   type: 'section';
   title?: SideNavSectionProps['title'];