From 96e5f38644ea06e2d796870446b348f16b45e66c Mon Sep 17 00:00:00 2001
From: Matthew Russell <mattrussell36@gmail.com>
Date: Mon, 31 Jul 2023 18:55:18 +0100
Subject: [PATCH] feat: allow tabs to wrap and apply bg color

---
 libs/ui-toolkit/src/components/tabs/tabs.tsx | 44 ++++++++++++++++++--
 1 file changed, 40 insertions(+), 4 deletions(-)

diff --git a/libs/ui-toolkit/src/components/tabs/tabs.tsx b/libs/ui-toolkit/src/components/tabs/tabs.tsx
index 11827b703..d20720556 100644
--- a/libs/ui-toolkit/src/components/tabs/tabs.tsx
+++ b/libs/ui-toolkit/src/components/tabs/tabs.tsx
@@ -2,10 +2,11 @@ import * as TabsPrimitive from '@radix-ui/react-tabs';
 import {
   useLocalStorageSnapshot,
   getValidItem,
+  useResizeObserver,
 } from '@vegaprotocol/react-helpers';
 import classNames from 'classnames';
 import type { ReactElement, ReactNode } from 'react';
-import { Children, isValidElement, useState } from 'react';
+import { Children, isValidElement, useRef, useState } from 'react';
 export interface TabsProps extends TabsPrimitive.TabsProps {
   children: (ReactElement<TabProps> | null)[];
 }
@@ -24,6 +25,19 @@ export const Tabs = ({
     return children.find((v) => v)?.props.id;
   });
 
+  // Bunch of refs in order to detect wrapping in side the tabs so that we
+  // can apply a bg color
+  const wrapperRef = useRef<HTMLDivElement | null>(null);
+  const tabsRef = useRef<HTMLDivElement | null>(null);
+  const menuRef = useRef<HTMLDivElement | null>(null);
+  const [wrapped, setWrapped] = useState(() =>
+    isWrapped(tabsRef.current, menuRef.current)
+  );
+
+  useResizeObserver(wrapperRef.current, () => {
+    setWrapped(isWrapped(tabsRef.current, menuRef.current));
+  });
+
   return (
     <TabsPrimitive.Root
       {...props}
@@ -31,10 +45,14 @@ export const Tabs = ({
       onValueChange={onValueChange || setActiveTab}
       className="h-full grid grid-rows-[min-content_1fr]"
     >
-      <div className="flex border-b border-default min-w-0">
+      <div
+        ref={wrapperRef}
+        className="flex flex-wrap justify-between border-b border-default min-w-0"
+      >
         <TabsPrimitive.List
           className="flex flex-nowrap overflow-visible"
           role="tablist"
+          ref={tabsRef}
         >
           {Children.map(children, (child) => {
             if (!isValidElement(child) || child.props.hidden) return null;
@@ -67,13 +85,21 @@ export const Tabs = ({
             );
           })}
         </TabsPrimitive.List>
-        <div className="ml-auto p-1">
+        <div
+          ref={menuRef}
+          className={classNames('flex-1 p-1', {
+            'bg-vega-clight-700 dark:bg-vega-cdark-700': wrapped,
+            '': wrapped,
+          })}
+        >
           {Children.map(children, (child) => {
             if (!isValidElement(child) || child.props.hidden) return null;
             return (
               <TabsPrimitive.Content
                 value={child.props.id}
-                className="flex flex-nowrap gap-1 justify-end"
+                className={classNames('flex flex-nowrap gap-1', {
+                  'justify-end': !wrapped,
+                })}
               >
                 {child.props.menu}
               </TabsPrimitive.Content>
@@ -136,3 +162,13 @@ export const LocalStoragePersistTabs = ({
     />
   );
 };
+
+const isWrapped = (
+  tabs: HTMLDivElement | null,
+  menu: HTMLDivElement | null
+) => {
+  if (!tabs || !menu) return;
+  const listRect = tabs.getBoundingClientRect();
+  const menuRect = menu.getBoundingClientRect();
+  return menuRect.y > listRect.y;
+};