From 4cf740aa68ea842ec7cc1433d99c4fa6e2a480c7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Matuzal=C3=A9m=20Teles?= <matuzalemteles@gmail.com>
Date: Tue, 22 Sep 2020 15:13:59 -0300
Subject: [PATCH] fix(@clayui/shared): removes the onTransitionEnd listener
 from elements by setTimeout

Trusting `onTransitionEnd` has been showing a little uncertainty as to whether the transition happened, so instead of adding a fallback to this we will add the implementation only in setTimeout as the trigger of the final transition, calculating the duration time of the style properties.
---
 packages/clay-nav/src/Vertical.tsx            | 13 ++-
 packages/clay-navigation-bar/src/index.tsx    | 13 ++-
 packages/clay-panel/src/index.tsx             | 13 ++-
 .../clay-shared/src/useTransitionHeight.ts    | 87 +++++++------------
 4 files changed, 50 insertions(+), 76 deletions(-)

diff --git a/packages/clay-nav/src/Vertical.tsx b/packages/clay-nav/src/Vertical.tsx
index 4d8503144b..47d35436cb 100644
--- a/packages/clay-nav/src/Vertical.tsx
+++ b/packages/clay-nav/src/Vertical.tsx
@@ -110,11 +110,11 @@ function Item({
 	const menuRef = React.useRef(null);
 	const [expanded, setExpaned] = React.useState(!initialExpanded);
 
-	const [
-		transitioning,
-		handleTransitionEnd,
-		handleClickToggler,
-	] = useTransitionHeight(expanded, setExpaned, menuRef);
+	const {animate, transitioning} = useTransitionHeight(
+		expanded,
+		setExpaned,
+		menuRef
+	);
 
 	const showIconCollapsed = !(
 		(!expanded && transitioning) ||
@@ -128,7 +128,7 @@ function Item({
 				collapsed={showIconCollapsed}
 				href={href}
 				onClick={(e) => {
-					handleClickToggler(e);
+					animate(e);
 
 					if (onClick) {
 						onClick();
@@ -148,7 +148,6 @@ function Item({
 						collapsing: transitioning,
 						show: expanded,
 					})}
-					onTransitionEnd={handleTransitionEnd}
 					ref={menuRef}
 				>
 					<Nav stacked>
diff --git a/packages/clay-navigation-bar/src/index.tsx b/packages/clay-navigation-bar/src/index.tsx
index 4403983280..b971f2635b 100644
--- a/packages/clay-navigation-bar/src/index.tsx
+++ b/packages/clay-navigation-bar/src/index.tsx
@@ -47,11 +47,11 @@ const ClayNavigationBar: React.FunctionComponent<IProps> & {
 }: IProps) => {
 	const [visible, setVisible] = React.useState(false);
 	const contentRef = React.useRef<HTMLDivElement>(null);
-	const [
-		transitioning,
-		handleTransitionEnd,
-		handleClickToggler,
-	] = useTransitionHeight(visible, setVisible, contentRef);
+	const {animate, transitioning} = useTransitionHeight(
+		visible,
+		setVisible,
+		contentRef
+	);
 
 	const activeElementsCount = children.filter((child) => child.props.active)
 		.length;
@@ -89,7 +89,7 @@ const ClayNavigationBar: React.FunctionComponent<IProps> & {
 					)}
 					data-testid="navbarToggler"
 					displayType="unstyled"
-					onClick={handleClickToggler}
+					onClick={animate}
 				>
 					<span className="navbar-text-truncate">{triggerLabel}</span>
 
@@ -103,7 +103,6 @@ const ClayNavigationBar: React.FunctionComponent<IProps> & {
 						show: visible,
 					})}
 					data-testid="NavigationBarDropdown"
-					onTransitionEnd={handleTransitionEnd}
 					ref={contentRef}
 				>
 					<ClayLayout.ContainerFluid>
diff --git a/packages/clay-panel/src/index.tsx b/packages/clay-panel/src/index.tsx
index 06133ff009..2c2efc2d6b 100644
--- a/packages/clay-panel/src/index.tsx
+++ b/packages/clay-panel/src/index.tsx
@@ -72,11 +72,11 @@ const ClayPanel: React.FunctionComponent<IProps> & {
 	const panelRef = React.useRef<HTMLDivElement>(null);
 	const [expanded, setExpaned] = React.useState<boolean>(defaultExpanded);
 
-	const [
-		transitioning,
-		handleTransitionEnd,
-		handleClickToggler,
-	] = useTransitionHeight(expanded, setExpaned, panelRef);
+	const {animate, transitioning} = useTransitionHeight(
+		expanded,
+		setExpaned,
+		panelRef
+	);
 
 	const showIconCollapsed = !(
 		(!expanded && transitioning) ||
@@ -120,7 +120,7 @@ const ClayPanel: React.FunctionComponent<IProps> & {
 								collapsed: showIconCollapsed,
 							}
 						)}
-						onClick={handleClickToggler}
+						onClick={animate}
 						role="tab"
 					>
 						{displayTitle &&
@@ -160,7 +160,6 @@ const ClayPanel: React.FunctionComponent<IProps> & {
 								show: expanded,
 							}
 						)}
-						onTransitionEnd={handleTransitionEnd}
 						ref={panelRef}
 						role="tabpanel"
 					>
diff --git a/packages/clay-shared/src/useTransitionHeight.ts b/packages/clay-shared/src/useTransitionHeight.ts
index 4c765968ce..0a54b8de68 100644
--- a/packages/clay-shared/src/useTransitionHeight.ts
+++ b/packages/clay-shared/src/useTransitionHeight.ts
@@ -3,7 +3,7 @@
  * SPDX-License-Identifier: BSD-3-Clause
  */
 
-import React from 'react';
+import React, {useCallback, useEffect, useState} from 'react';
 
 /**
  * Removes the height style that triggers a transition when collapse
@@ -35,74 +35,51 @@ function setCollapseHeight(collapseElementRef: React.RefObject<any>) {
 
 export function useTransitionHeight(
 	visible: boolean,
-	setVisible: any,
+	setVisible: React.Dispatch<React.SetStateAction<boolean>>,
 	contentRef: React.RefObject<any>
 ) {
-	const [transitioning, setTransitioning] = React.useState<boolean>(false);
+	const [transitioning, setTransitioning] = useState<boolean>(false);
 
-	const transitionBuffer = 0.01;
-	const transitionTimerRef = React.useRef<NodeJS.Timeout | null>(null);
-
-	const fakeTransitionEnd = () => {
-		const {transitionDelay, transitionDuration} = window.getComputedStyle(
-			contentRef.current
-		);
-		const totalDuration =
-			(parseFloat(transitionDelay) +
-				parseFloat(transitionDuration) +
-				transitionBuffer) *
-			1000;
-
-		transitionTimerRef.current = setTimeout(() => {
-			setTransitioning(false);
-			setVisible(!visible);
-
-			if (!visible) {
-				removeCollapseHeight(contentRef);
-			}
-		}, totalDuration);
-	};
-
-	React.useEffect(() => {
+	useEffect(() => {
 		if (transitioning) {
 			setCollapseHeight(contentRef);
+
 			if (visible) {
 				removeCollapseHeight(contentRef);
 			}
 
-			fakeTransitionEnd();
-		}
-	}, [transitioning]);
+			const {
+				transitionDelay,
+				transitionDuration,
+			} = window.getComputedStyle(contentRef.current);
 
-	const handleTransitionEnd = (event: React.TransitionEvent) => {
-		if (transitionTimerRef.current) {
-			clearTimeout(transitionTimerRef.current);
-		}
+			const totalDuration =
+				(parseFloat(transitionDelay) + parseFloat(transitionDuration)) *
+				1000;
 
-		if (event.target === contentRef.current && transitioning && !visible) {
-			setVisible(true);
-			setTransitioning(false);
-			removeCollapseHeight(contentRef);
-		} else if (event.target === contentRef.current) {
-			setVisible(false);
-			setTransitioning(false);
-		}
-	};
+			setTimeout(() => {
+				setVisible(!visible);
+				setTransitioning(false);
 
-	const willTriggerTransition = (event: React.MouseEvent | MouseEvent) => {
-		event.preventDefault();
-		if (visible && !transitioning) {
-			setCollapseHeight(contentRef);
+				if (!visible) {
+					removeCollapseHeight(contentRef);
+				}
+			}, totalDuration);
 		}
+	}, [transitioning]);
+
+	const animate = useCallback(
+		(event: React.SyntheticEvent) => {
+			event.preventDefault();
+
+			if (visible && !transitioning) {
+				setCollapseHeight(contentRef);
+			}
 
-		if (!transitioning) {
 			setTransitioning(true);
-		}
-	};
+		},
+		[transitioning, visible]
+	);
 
-	return [transitioning, handleTransitionEnd, willTriggerTransition] as [
-		boolean,
-		(event: React.TransitionEvent) => void,
-		(event: React.MouseEvent | MouseEvent) => void
-	];
+	return {animate, transitioning};
 }