/*
 * Copyright 2020 Spotify AB
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import React, { PropsWithChildren } from 'react';
import { Box, useTheme } from '@material-ui/core';
import { LegendItem } from '../LegendItem';
import { currencyFormatter } from '../../utils/formatters';
import { CostInsightsTheme } from '../../types';
import { useBarChartLayoutStyles as useStyles } from '../../utils/styles';

export type BarChartLegendOptions = {
  previousName: string;
  previousFill: string;
  currentName: string;
  currentFill: string;
  hideMarker?: boolean;
};

export type BarChartLegendProps = {
  costStart: number;
  costEnd: number;
  options?: Partial<BarChartLegendOptions>;
};

export const BarChartLegend = ({
  costStart,
  costEnd,
  options = {},
  children,
}: PropsWithChildren<BarChartLegendProps>) => {
  const theme = useTheme<CostInsightsTheme>();
  const classes = useStyles();

  const data = Object.assign(
    {
      previousName: 'Previous',
      previousFill: theme.palette.lightBlue,
      currentName: 'Current',
      currentFill: theme.palette.darkBlue,
    },
    options,
  );

  return (
    <Box className={classes.legend} display="flex" flexDirection="row">
      <Box marginRight={2}>
        <LegendItem
          title={data.previousName}
          markerColor={options.hideMarker ? undefined : data.previousFill}
        >
          {currencyFormatter.format(costStart)}
        </LegendItem>
      </Box>
      <Box marginRight={2}>
        <LegendItem
          title={data.currentName}
          markerColor={options.hideMarker ? undefined : data.currentFill}
        >
          {currencyFormatter.format(costEnd)}
        </LegendItem>
      </Box>
      {children}
    </Box>
  );
};
