From 89de6cf4d48e72d4f5e0eb49ee3054694fa11386 Mon Sep 17 00:00:00 2001
From: HendrikThePendric <hendrik@dhis2.org>
Date: Wed, 27 Sep 2023 11:07:40 +0200
Subject: [PATCH] fix(pivot-table): correctly show tooltip after resizing the
 container

---
 src/__demo__/PivotTable.stories.js            | 17 ++++++++++-
 .../PivotTable/PivotTableTitleRow.js          | 28 ++++++++++++++++---
 2 files changed, 40 insertions(+), 5 deletions(-)

diff --git a/src/__demo__/PivotTable.stories.js b/src/__demo__/PivotTable.stories.js
index 2c631d079..e06782146 100644
--- a/src/__demo__/PivotTable.stories.js
+++ b/src/__demo__/PivotTable.stories.js
@@ -1123,6 +1123,13 @@ storiesOf('PivotTable', module).add('DEGS', (_, { pivotTableOptions }) => {
 storiesOf('PivotTable', module).add(
     'Truncated header cell',
     (_, { pivotTableOptions }) => {
+        const widths = [250, 200, 500]
+        const [width, setWidth] = useState(250)
+        const toggleWidth = () =>
+            setWidth(
+                (currentWidth) =>
+                    widths[widths.indexOf(currentWidth) + 1] ?? widths[0]
+            )
         const visualization = {
             ...narrativeVisualization,
             ...visualizationReset,
@@ -1139,7 +1146,15 @@ storiesOf('PivotTable', module).add(
         }
 
         return (
-            <div style={{ width: 250, height: 600, marginTop: 50 }}>
+            <div
+                style={{
+                    width,
+                    height: 600,
+                    marginTop: 50,
+                    transition: 'width 1s',
+                }}
+            >
+                <button onClick={toggleWidth}>Toggle width</button>
                 <PivotTable data={data} visualization={visualization} />
             </div>
         )
diff --git a/src/components/PivotTable/PivotTableTitleRow.js b/src/components/PivotTable/PivotTableTitleRow.js
index 7ce48226a..902a1e3b8 100644
--- a/src/components/PivotTable/PivotTableTitleRow.js
+++ b/src/components/PivotTable/PivotTableTitleRow.js
@@ -1,6 +1,6 @@
 import { Tooltip } from '@dhis2/ui'
 import PropTypes from 'prop-types'
-import React, { useRef, useMemo } from 'react'
+import React, { useRef, useMemo, useState, useEffect } from 'react'
 import { PivotTableCell } from './PivotTableCell.js'
 import { usePivotTableEngine } from './PivotTableEngineContext.js'
 import { cell as cellStyle } from './styles/PivotTable.style.js'
@@ -12,6 +12,8 @@ export const PivotTableTitleRow = ({
     totalWidth,
 }) => {
     const containerRef = useRef(null)
+    const [scrollWidth, setScrollWidth] = useState(0)
+    const [isTitleTruncated, setIsTitleTruncated] = useState(false)
     const engine = usePivotTableEngine()
     const columnCount = engine.width + engine.rowDepth
     const maxWidth = useMemo(
@@ -26,8 +28,26 @@ export const PivotTableTitleRow = ({
             ),
         [containerWidth, scrollPosition.x, totalWidth]
     )
-    const titleIsTruncated =
-        containerRef.current?.scrollWidth > containerRef.current?.clientWidth
+
+    useEffect(() => {
+        if (containerRef.current) {
+            /* Only set `scrollWidth` once, because during a CSS transition
+             * the reported value can sometimes be equal to `clientWidth`
+             * even though the text doesn't fit. Due to `white-space: nowrap`
+             * and `overflow: hidden` the `scrollWidth` should remain constant,
+             * so we can assume this initial value is correct. */
+            if (!scrollWidth) {
+                setScrollWidth(containerRef.current.scrollWidth)
+            }
+            const currentScrollWidth =
+                scrollWidth ?? containerRef.current.scrollWidth
+            const newIsTitleTruncated =
+                currentScrollWidth > containerRef.current.clientWidth
+            if (newIsTitleTruncated !== isTitleTruncated) {
+                setIsTitleTruncated(newIsTitleTruncated)
+            }
+        }
+    }, [containerWidth, scrollWidth, isTitleTruncated])
 
     return (
         <tr>
@@ -43,7 +63,7 @@ export const PivotTableTitleRow = ({
                     data-test="visualization-title"
                     className="title-cell-content"
                 >
-                    {titleIsTruncated ? (
+                    {isTitleTruncated ? (
                         <Tooltip content={title}>
                             {({ ref: tooltipRef, onMouseOver, onMouseOut }) => (
                                 <div