/*
Copyright (c) Uber Technologies, Inc.

This source code is licensed under the MIT license found in the
LICENSE file in the root directory of this source tree.
*/
import * as React from 'react';
import { withStyle, withWrapper } from '../styles';
import {
  StyledTable as FlexStyledTable,
  StyledHeadCell as FlexStyledHeadCell,
  StyledCell as FlexStyledBodyCell,
} from '../table';

const StyledTableElement = withStyle<
  typeof FlexStyledTable,
  {
    $gridTemplateColumns: string;
  }
>(FlexStyledTable, (props) => {
  return {
    display: 'grid',
    gridTemplateColumns: props.$gridTemplateColumns,
    flexDirection: 'unset',
    // Creates a stacking context so we can use z-index on the StyledHeadCell
    // without affecting anything outside of this component.
    transform: 'scale(1)',
  };
});

StyledTableElement.displayName = 'StyledTableElement';

export const StyledTable = withWrapper(
  StyledTableElement,
  (StyledComponent) =>
    function StyledTable(props) {
      return <StyledComponent data-baseweb="table-grid" {...props} />;
    }
);

export const StyledHeadCell = withStyle<
  typeof FlexStyledHeadCell,
  {
    $sticky?: boolean;
    $isFocusVisible?: boolean;
  }
  // @ts-ignore
>(FlexStyledHeadCell, ({ $sticky = true, $isFocusVisible, $theme }) => {
  return {
    backgroundColor: $theme.colors.tableHeadBackgroundColor,
    boxShadow: $theme.lighting.shadow400,
    position: $sticky ? 'sticky' : null,
    top: $sticky ? 0 : null,
    width: 'unset',
    ':focus': {
      outline: $isFocusVisible ? `3px solid ${$theme.colors.borderAccent}` : 'none',
      outlineOffset: '-3px',
    },
    zIndex: $sticky ? 1 : 'auto',
  };
});

StyledHeadCell.displayName = 'StyledHeadCell';

export const StyledBodyCell = withStyle<
  typeof FlexStyledBodyCell,
  {
    $gridColumn?: string;
    $gridRow?: string;
    $isFocusVisible?: boolean;
  }
  // @ts-ignore
>(FlexStyledBodyCell, (props) => {
  return {
    display: 'block',
    flex: 'unset',
    gridColumn: props.$gridColumn || null,
    gridRow: props.$gridRow || null,
    ':focus': {
      outline: props.$isFocusVisible ? `3px solid ${props.$theme.colors.borderAccent}` : 'none',
      outlineOffset: '-3px',
    },
  };
});
StyledBodyCell.displayName = 'StyledBodyCell';