import { RankedMode } from "@prisma/client";
import { Box, Grid, useMediaQuery } from "@chakra-ui/react";
import ModeImage from "../common/ModeImage";
import SubText from "../common/SubText";
import { t } from "@lingui/macro";
import { FaCheck } from "react-icons/fa";
import { CSSVariables } from "../../utils/CSSVariables";
import NewTable, { TableRow } from "../common/NewTable";

function getModeCells(
  enabledModes: RankedMode[]
): { [mode in RankedMode]: JSX.Element | null } {
  return Object.values(RankedMode).reduce((map, mode) => {
    map[mode] = enabledModes.includes(mode) ? (
      <FaCheck color={CSSVariables.themeColor} style={{ margin: "auto" }} />
    ) : null;
    return map;
  }, {} as { [mode in RankedMode]: JSX.Element | null });
}

function getTableData(
  stagesSelected: Record<string, RankedMode[]>
): (TableRow | null)[] {
  return Object.entries(stagesSelected).map(([stage, modes], index) =>
    modes.length > 0
      ? {
          id: index,
          stage: t({ id: stage }),
          ...getModeCells(modes),
        }
      : null
  );
}

function getGridModeCells(
  stagesSelected: Record<string, RankedMode[]>
): JSX.Element[] {
  return Object.values(RankedMode).map((mode) => {
    return (
      <div key={mode}>
        <Box textAlign="center">
          <ModeImage mode={mode} />
        </Box>
        <Box textAlign="center">
          {Object.entries(stagesSelected).map(([stage, modes]) =>
            modes.includes(mode) ? (
              <SubText key={stage} mb={1}>
                {t({ id: stage })}
              </SubText>
            ) : null
          )}
        </Box>
      </div>
    );
  });
}

export function MapPoolDisplay({
  stagesSelected,
}: {
  stagesSelected: Record<string, RankedMode[]>;
}) {
  const [isMobile] = useMediaQuery("(max-width: 48em)");

  return isMobile ? (
    <Grid
      mb={8}
      templateColumns={{ base: "repeat(2, 1fr)", sm: "repeat(4, 1fr)" }}
      rowGap={4}
      columnGap={4}
      display="grid"
    >
      {getGridModeCells(stagesSelected)}
    </Grid>
  ) : (
    <NewTable
      size="sm"
      headers={[
        { name: t`Stage name`, dataKey: "stage" },
        { name: t`Splat Zones`, dataKey: RankedMode.SZ },
        { name: t`Tower Control`, dataKey: RankedMode.TC },
        { name: t`Rainmaker`, dataKey: RankedMode.RM },
        { name: t`Clam Blitz`, dataKey: RankedMode.CB },
      ]}
      data={getTableData(stagesSelected)}
    />
  );
}