SearchBar.tsx

/* eslint-disable react-hooks/exhaustive-deps */
import { outlinedInputClasses } from '@mui/material/OutlinedInput';
import { Theme, ThemeProvider, createTheme } from '@mui/material/styles';
import debounce from 'lodash/debounce';
import React, { useCallback } from 'react';
import { ClickAwayListener } from '../base/ClickAwayListener';
import { TextField } from '../base/TextField';
import { CloseIcon, SearchIcon } from '../icons';
import { useTheme } from '../theme';
import { TooltipIcon } from './TooltipIconButton';

const customTheme = (theme: Theme) =>
  createTheme({
    typography: {
      fontFamily: theme.typography.fontFamily
    },
    components: {
      MuiTextField: {
        styleOverrides: {
          root: {
            '--TextField-brandBorderColor': theme.palette.border.strong,
            '--TextField-brandBorderHoverColor': theme.palette.background.graphics?.default,
            '--TextField-brandBorderFocusedColor': theme.palette.background.graphics?.default,
            '& label.Mui-focused': {
              color: 'var(--TextField-brandBorderFocusedColor)'
            }
          }
        }
      },
      MuiOutlinedInput: {
        styleOverrides: {
          notchedOutline: {
            borderColor: 'var(--TextField-brandBorderColor)'
          },
          root: {
            [`&:hover .${outlinedInputClasses.notchedOutline}`]: {
              borderColor: 'var(--TextField-brandBorderHoverColor)'
            },
            [`&.Mui-focused .${outlinedInputClasses.notchedOutline}`]: {
              borderColor: 'var(--TextField-brandBorderFocusedColor)'
            }
          }
        }
      },
      MuiInput: {
        styleOverrides: {
          root: {
            color: theme.palette.text.default,
            '&::before': {
              borderBottom: '2px solid var(--TextField-brandBorderColor)'
            },
            '&:hover:not(.Mui-disabled, .Mui-error):before': {
              borderBottom: '2px solid var(--TextField-brandBorderHoverColor)'
            },
            '&.Mui-focused:after': {
              borderBottom: '2px solid var(--TextField-brandBorderFocusedColor)'
            }
          }
        }
      },
      MuiButton: {
        styleOverrides: {
          root: {
            lineHeight: 'auto'
          }
        }
      }
    }
  });

export interface SearchBarProps {
  onSearch: (searchText: string) => void;
  style?: React.CSSProperties;
  placeholder?: string;
  onClear?: () => void;
  expanded: boolean;
  setExpanded: (expanded: boolean) => void;
  'data-testid'?: string;
  onKeyDown?: (event: React.KeyboardEvent<HTMLInputElement>) => void;
}

function SearchBar({
  onSearch,
  placeholder,
  onClear,
  expanded,
  setExpanded,
  'data-testid': testId = 'search-bar-wrapper',
  onKeyDown
}: SearchBarProps): JSX.Element {
  const [searchText, setSearchText] = React.useState('');
  const searchRef = React.useRef<HTMLInputElement | null>(null);
  const theme = useTheme();

  // Debounce the onSearch function for normal typing
  const debouncedOnSearch = useCallback(
    debounce((value) => {
      onSearch(value);
    }, 300),
    [onSearch]
  );

  const handleSearchChange = (event: React.ChangeEvent<HTMLInputElement>): void => {
    const value = event.target.value;
    setSearchText(value);
    debouncedOnSearch(value);
  };

  const handleClearIconClick = (e: React.MouseEvent): void => {
    e.stopPropagation();
    debouncedOnSearch('');
    setSearchText('');
    setExpanded(false);
    if (onClear) {
      onClear();
    }
  };

  const handleSearchIconClick = (e: React.MouseEvent): void => {
    e.stopPropagation();
    if (expanded) {
      debouncedOnSearch('');
      setSearchText('');
      setExpanded(false);
    } else {
      setExpanded(true);
      setTimeout(() => {
        if (searchRef.current) {
          searchRef.current.focus();
        }
      }, 300);
    }
  };

  // ✅ New unified keyDown handler
  const handleKeyDown = (event: React.KeyboardEvent<HTMLInputElement>) => {
    // Call external onKeyDown if provided
    if (onKeyDown) {
      onKeyDown(event);
    }

    // Trigger onSearch immediately when Enter is pressed
    if (event.key === 'Enter') {
      onSearch(searchText);
    }
  };

  return (
    <ClickAwayListener
      onClickAway={(event) => {
        event.stopPropagation();
        const isTable = (event.target as HTMLElement)?.closest('#ref');

        if (searchText !== '') return;

        if (isTable) {
          handleClearIconClick(event as unknown as React.MouseEvent);
        }
      }}
    >
      <div style={{ display: 'flex' }} data-testid={testId}>
        <ThemeProvider theme={customTheme(theme)}>
          <TextField
            variant="standard"
            value={searchText}
            onChange={handleSearchChange}
            inputRef={searchRef}
            placeholder={placeholder}
            data-testid="searchbar-input"
            onKeyDown={handleKeyDown} // <-- updated handler
            style={{
              width: expanded ? '150px' : '0',
              opacity: expanded ? 1 : 0,
              transition: 'width 0.3s ease, opacity 0.3s ease'
            }}
          />
        </ThemeProvider>
        {expanded ? (
          <TooltipIcon
            title="Close"
            onClick={handleClearIconClick}
            icon={
              <CloseIcon fill={theme.palette.icon.default} data-testid="searchbar-clear-button" />
            }
            arrow
          />
        ) : (
          <TooltipIcon
            title="Search"
            onClick={handleSearchIconClick}
            icon={
              <SearchIcon fill={theme.palette.icon.default} data-testid="searchbar-search-button" />
            }
            arrow
          />
        )}
      </div>
    </ClickAwayListener>
  );
}

export default SearchBar;