From 93cf65fc1134b0bb3da286c2253e656b7a09f1c1 Mon Sep 17 00:00:00 2001
From: Ezerous <ezerous@gmail.com>
Date: Thu, 8 Apr 2021 11:17:55 +0300
Subject: [PATCH] fix: minor poll view tweaks

---
 .../PollView/PollGraph/PollChartBar/index.jsx |   4 +-
 .../components/PollView/PollGraph/index.jsx   | 110 ++++++++++--------
 2 files changed, 62 insertions(+), 52 deletions(-)

diff --git a/packages/concordia-app/src/components/PollView/PollGraph/PollChartBar/index.jsx b/packages/concordia-app/src/components/PollView/PollGraph/PollChartBar/index.jsx
index c5ca9d7..63b1d4c 100644
--- a/packages/concordia-app/src/components/PollView/PollGraph/PollChartBar/index.jsx
+++ b/packages/concordia-app/src/components/PollView/PollGraph/PollChartBar/index.jsx
@@ -35,8 +35,8 @@ const PollChartBar = (props) => {
         show: false,
       },
       y: {
-        formatter(value, { seriesIndex }) {
-          return `<div>${voterNames[seriesIndex].join('</div><div>')}</div>`;
+        formatter(value, { dataPointIndex }) {
+          return `<div>${voterNames[dataPointIndex].join('</div><div>')}</div>`;
         },
         title: {
           formatter: () => null,
diff --git a/packages/concordia-app/src/components/PollView/PollGraph/index.jsx b/packages/concordia-app/src/components/PollView/PollGraph/index.jsx
index 8c0bb7d..3cea672 100644
--- a/packages/concordia-app/src/components/PollView/PollGraph/index.jsx
+++ b/packages/concordia-app/src/components/PollView/PollGraph/index.jsx
@@ -1,18 +1,24 @@
-import React, { useEffect, useMemo, useState } from 'react';
+import React, {
+  useCallback, useEffect, useMemo, useState,
+} from 'react';
 import {
-  Grid, Header, Statistic, Tab,
+  Grid,
+  Header, Statistic, Tab,
 } from 'semantic-ui-react';
 import PropTypes from 'prop-types';
 import { useTranslation } from 'react-i18next';
 import PollChartBar from './PollChartBar';
 import PollChartDonut from './PollChartDonut';
 import './styles.css';
+import { CHART_TYPE_BAR, CHART_TYPE_DONUT } from '../../../constants/polls/PollGraph';
 
 const PollGraph = (props) => {
   const {
     pollOptions, userVoteIndex, voteCounts, voterNames,
   } = props;
-  const [totalVotes, setTotalVotes] = useState(voteCounts.reduce((accumulator, voteCount) => accumulator + voteCount, 0));
+  const [totalVotes, setTotalVotes] = useState(
+    voteCounts.reduce((accumulator, voteCount) => accumulator + voteCount, 0),
+  );
   const { t } = useTranslation();
 
   useEffect(() => {
@@ -20,7 +26,7 @@ const PollGraph = (props) => {
   }, [voteCounts]);
 
   const footer = useMemo(() => (
-      <>
+      <div>
           <Statistic size="mini">
               <Statistic.Value>
                   { totalVotes }
@@ -36,84 +42,88 @@ const PollGraph = (props) => {
                   <span className="poll-voted-option">{pollOptions[userVoteIndex]}</span>
               </div>
           )}
-      </>
+      </div>
   ), [pollOptions, t, totalVotes, userVoteIndex]);
 
-  const panes = useMemo(() => {
-    // TODO: tidy up duplicated code below using CHART_TYPE_BAR and CHART_TYPE_DONUT
-    const chartBarPane = (
+  function calculateChartBarWidth(nOptions) {
+    if (nOptions < 6) return 8;
+    if (nOptions < 10) return 10;
+    if (nOptions < 14) return 12;
+    if (nOptions < 18) return 14;
+    return 16;
+  }
+
+  function calculateChartDonutWidth(nOptions) {
+    if (nOptions < 10) return 8;
+    if (nOptions < 16) return 10;
+    return 12;
+  }
+
+  const generatePane = useCallback(
+    (type) => (
         <Tab.Pane attached={false}>
             <Grid columns="equal">
                 <Grid.Row>
                     <Grid.Column />
-                    <Grid.Column width={8} textAlign="center">
-                        {totalVotes > 0
+                    <Grid.Column
+                      width={type === CHART_TYPE_BAR
+                        ? calculateChartBarWidth(pollOptions.length)
+                        : calculateChartDonutWidth(pollOptions.length)}
+                      textAlign="center"
+                    >
+                        {type === CHART_TYPE_BAR
                           ? (
                               <PollChartBar
                                 pollOptions={pollOptions}
                                 voteCounts={voteCounts}
                                 voterNames={voterNames}
                               />
-                          )
-                          : (
-                              <Header as="h4">
-                                  {t('topic.poll.tab.results.no.votes')}
-                              </Header>
-                          )}
-                    </Grid.Column>
-                    <Grid.Column />
-                </Grid.Row>
-                <Grid.Row>
-                    <Grid.Column textAlign="center">
-                        {totalVotes > 0 && footer}
-                    </Grid.Column>
-                </Grid.Row>
-            </Grid>
-        </Tab.Pane>
-    );
-    const chartDonutPane = (
-        <Tab.Pane attached={false}>
-            <Grid columns="equal">
-                <Grid.Row>
-                    <Grid.Column />
-                    <Grid.Column width={8} textAlign="center">
-                        {totalVotes > 0
-                          ? (
+                          ) : (
                               <PollChartDonut
                                 pollOptions={pollOptions}
                                 voteCounts={voteCounts}
                                 voterNames={voterNames}
                               />
-                          )
-                          : (
-                              <Header as="h4">
-                                  {t('topic.poll.tab.results.no.votes')}
-                              </Header>
                           )}
                     </Grid.Column>
                     <Grid.Column />
                 </Grid.Row>
                 <Grid.Row>
                     <Grid.Column textAlign="center">
-                        {totalVotes > 0 && footer}
+                        {footer}
                     </Grid.Column>
                 </Grid.Row>
             </Grid>
         </Tab.Pane>
-    );
+    ),
+    [footer, pollOptions, voteCounts, voterNames],
+  );
+
+  const panes = useMemo(() => {
+    const chartBarPane = generatePane(CHART_TYPE_BAR);
+    const chartDonutPane = generatePane(CHART_TYPE_DONUT);
 
     return ([
       { menuItem: { key: 'chart-bar', icon: 'chart bar' }, render: () => chartBarPane },
       { menuItem: { key: 'chart-donut', icon: 'chart pie' }, render: () => chartDonutPane },
     ]);
-  }, [footer, pollOptions, t, totalVotes, voteCounts, voterNames]);
-
-  return (
-      <Tab
-        menu={{ secondary: true }}
-        panes={panes}
-      />
-  );
+  }, [generatePane]);
+  return useMemo(() => (
+      <>
+          {totalVotes > 0
+            ? (
+                <Tab
+                  menu={{ secondary: true }}
+                  panes={panes}
+                />
+            )
+            : (
+                <Header as="h4" textAlign="center">
+                    {t('topic.poll.tab.results.no.votes')}
+                </Header>
+            )}
+      </>
+  ), [panes, t, totalVotes]);
 };
 
 PollGraph.defaultProps = {