fix(striker-ui): make IconButton change based on state

striker-ui/components/IconButton/IconButton.tsx

@@ -1,9 +1,16 @@
-import { FC } from 'react';
+import {
+  Done as MUIDoneIcon,
+  Edit as MUIEditIcon,
+  Visibility as MUIVisibilityIcon,
+  VisibilityOff as MUIVisibilityOffIcon,
+} from '@mui/icons-material';
 import {
   IconButton as MUIIconButton,
   IconButtonProps as MUIIconButtonProps,
   inputClasses as muiInputClasses,
+  styled,
 } from '@mui/material';
+import { createElement, FC, ReactNode, useMemo } from 'react';
 
 import {
   BLACK,
@@ -13,17 +20,9 @@ import {
   TEXT,
 } from '../../lib/consts/DEFAULT_THEME';
 
-export type IconButtonProps = MUIIconButtonProps;
+type IconButtonProps = IconButtonOptionalProps & MUIIconButtonProps;
 
-const IconButton: FC<IconButtonProps> = ({
-  children,
-  sx,
-  ...iconButtonRestProps
-}) => (
-  <MUIIconButton
-    {...{
-      ...iconButtonRestProps,
-      sx: {
+const ContainedIconButton = styled(MUIIconButton)({
   borderRadius: BORDER_RADIUS,
   backgroundColor: GREY,
   color: BLACK,
@@ -35,13 +34,79 @@ const IconButton: FC<IconButtonProps> = ({
   [`&.${muiInputClasses.disabled}`]: {
     backgroundColor: DISABLED,
   },
+});
 
-        ...sx,
-      },
-    }}
-  >
-    {children}
-  </MUIIconButton>
+const NormalIconButton = styled(MUIIconButton)({
+  color: GREY,
+});
+
+const MAP_TO_VISIBILITY_ICON: IconButtonMapToStateIcon = {
+  false: MUIVisibilityIcon,
+  true: MUIVisibilityOffIcon,
+};
+
+const MAP_TO_EDIT_ICON: IconButtonMapToStateIcon = {
+  false: MUIEditIcon,
+  true: MUIDoneIcon,
+};
+
+const MAP_TO_MAP_PRESET: Record<
+  IconButtonPresetMapToStateIcon,
+  IconButtonMapToStateIcon
+> = {
+  edit: MAP_TO_EDIT_ICON,
+  visibility: MAP_TO_VISIBILITY_ICON,
+};
+
+const MAP_TO_VARIANT: Record<IconButtonVariant, CreatableComponent> = {
+  contained: ContainedIconButton,
+  normal: NormalIconButton,
+};
+
+const IconButton: FC<IconButtonProps> = ({
+  children,
+  defaultIcon,
+  iconProps,
+  mapPreset,
+  mapToIcon: externalMapToIcon,
+  state,
+  variant = 'contained',
+  ...restIconButtonProps
+}) => {
+  const mapToIcon = useMemo<IconButtonMapToStateIcon | undefined>(
+    () => externalMapToIcon ?? (mapPreset && MAP_TO_MAP_PRESET[mapPreset]),
+    [externalMapToIcon, mapPreset],
   );
 
+  const iconButtonContent = useMemo(() => {
+    let result: ReactNode;
+
+    if (mapToIcon) {
+      const iconElementType: CreatableComponent | undefined = state
+        ? mapToIcon[state] ?? defaultIcon
+        : defaultIcon;
+
+      if (iconElementType) {
+        result = createElement(iconElementType, iconProps);
+      }
+    } else {
+      result = children;
+    }
+
+    return result;
+  }, [children, mapToIcon, state, defaultIcon, iconProps]);
+  const iconButtonElementType = useMemo(
+    () => MAP_TO_VARIANT[variant],
+    [variant],
+  );
+
+  return createElement(
+    iconButtonElementType,
+    restIconButtonProps,
+    iconButtonContent,
+  );
+};
+
+export type { IconButtonProps };
+
 export default IconButton;

striker-ui/types/IconButton.d.ts

@@ -0,0 +1,16 @@
+type CreatableComponent = Parameters<typeof import('react').createElement>[0];
+
+type IconButtonPresetMapToStateIcon = 'edit' | 'visibility';
+
+type IconButtonMapToStateIcon = Record<string, CreatableComponent>;
+
+type IconButtonVariant = 'contained' | 'normal';
+
+type IconButtonOptionalProps = {
+  defaultIcon?: CreatableComponent;
+  iconProps?: import('@mui/material').SvgIconProps;
+  mapPreset?: IconButtonPresetMapToStateIcon;
+  mapToIcon?: IconButtonMapToStateIcon;
+  state?: string;
+  variant?: IconButtonVariant;
+};