shlinkio / shlink-web-client

import { Card, CardHeader, CardBody, CardFooter } from 'reactstrap';

import { Doughnut, HorizontalBar } from 'react-chartjs-2';

import PropTypes from 'prop-types';

import React from 'react';

import { keys, values } from 'ramda';

import { fillTheGaps } from '../utils/helpers/visits';

import './GraphCard.scss';

const propTypes = {

  title: PropTypes.oneOfType([ PropTypes.string, PropTypes.func ]),

  footer: PropTypes.oneOfType([ PropTypes.string, PropTypes.node ]),

  isBarChart: PropTypes.bool,

  stats: PropTypes.object,

  max: PropTypes.number,

  highlightedStats: PropTypes.object,

  onClick: PropTypes.func,

};

const generateGraphData = (title, isBarChart, labels, data, highlightedData) => ({

  labels,

  datasets: [

    {

      title,

      label: highlightedData ? 'Non-selected' : 'Visits',

      data,

      backgroundColor: isBarChart ? 'rgba(70, 150, 229, 0.4)' : [

        '#97BBCD',

        '#F7464A',

        '#46BFBD',

        '#FDB45C',

        '#949FB1',

        '#57A773',

        '#414066',

        '#08B2E3',

        '#B6C454',

        '#DCDCDC',

        '#463730',

      ],

      borderColor: isBarChart ? 'rgba(70, 150, 229, 1)' : 'white',

      borderWidth: 2,

    },

    highlightedData && {

      title,

      label: 'Selected',

      data: highlightedData,

      backgroundColor: 'rgba(247, 127, 40, 0.4)',

      borderColor: '#F77F28',

      borderWidth: 2,

    },

  ].filter(Boolean),

});

const dropLabelIfHidden = (label) => label.startsWith('hidden') ? '' : label;

const determineHeight = (isBarChart, labels) => {

  if (!isBarChart && labels.length > 8) {

    return 200;

  }

  return isBarChart && labels.length > 20 ? labels.length * 8 : null;

};

const renderGraph = (title, isBarChart, stats, max, highlightedStats, onClick) => {

  const hasHighlightedStats = highlightedStats && Object.keys(highlightedStats).length > 0;

  const Component = isBarChart ? HorizontalBar : Doughnut;

  const labels = keys(stats).map(dropLabelIfHidden);

  const data = values(!hasHighlightedStats ? stats : keys(highlightedStats).reduce((acc, highlightedKey) => {

    if (acc[highlightedKey]) {

      acc[highlightedKey] -= highlightedStats[highlightedKey];

    }

    return acc;

  }, { ...stats }));

  const highlightedData = hasHighlightedStats && fillTheGaps(highlightedStats, labels);

  const options = {

    legend: isBarChart ? { display: false } : { position: 'right' },

    scales: isBarChart && {

      xAxes: [

        {

          ticks: { beginAtZero: true, precision: 0, max },

          stacked: true,

        },

      ],

      yAxes: [{ stacked: true }],

    },

    tooltips: {

      intersect: !isBarChart,

      // Do not show tooltip on items with empty label when in a bar chart

      filter: ({ yLabel }) => !isBarChart || yLabel !== '',

    },

    onHover: isBarChart && (({ target }, chartElement) => {

      target.style.cursor = chartElement[0] ? 'pointer' : 'default';

    }),

  };

  const graphData = generateGraphData(title, isBarChart, labels, data, highlightedData);

  const height = determineHeight(isBarChart, labels);

  // Provide a key based on the height, so that every time the dataset changes, a new graph is rendered

  return (

    <Component

      key={height}

      data={graphData}

      options={options}

      height={height}

      getElementAtEvent={([ chart ]) => {

        if (!onClick || !chart) {

          return;

        }

        const { _index, _chart: { data } } = chart;

        const { labels } = data;

        onClick(labels[_index]);

      }}

    />

  );

};

const GraphCard = ({ title, footer, isBarChart, stats, max, highlightedStats, onClick }) => (

  <Card>

    <CardHeader className="graph-card__header">{typeof title === 'function' ? title() : title}</CardHeader>

    <CardBody>{renderGraph(title, isBarChart, stats, max, highlightedStats, onClick)}</CardBody>

    {footer && <CardFooter className="graph-card__footer--sticky">{footer}</CardFooter>}

  </Card>

);

GraphCard.propTypes = propTypes;

export default GraphCard;