From 8a1cfd42ba7253c97f06cfacfc7663a31530ba82 Mon Sep 17 00:00:00 2001
From: Albert Yu <albert@albertyu.co>
Date: Tue, 22 Oct 2024 14:47:37 +0800
Subject: [PATCH] Move useContext call from hooks to components

---
 packages/mui-base/src/Tabs/Tab/Tab.tsx        | 24 ++++++++++++-------
 .../mui-base/src/Tabs/Tab/useTab.test.tsx     | 12 ++++++++--
 packages/mui-base/src/Tabs/Tab/useTab.ts      | 22 +++++++++--------
 3 files changed, 38 insertions(+), 20 deletions(-)

diff --git a/packages/mui-base/src/Tabs/Tab/Tab.tsx b/packages/mui-base/src/Tabs/Tab/Tab.tsx
index 217fef7cbb..805a95aa88 100644
--- a/packages/mui-base/src/Tabs/Tab/Tab.tsx
+++ b/packages/mui-base/src/Tabs/Tab/Tab.tsx
@@ -4,7 +4,8 @@ import PropTypes from 'prop-types';
 import { useTab } from './useTab';
 import { useComponentRenderer } from '../../utils/useComponentRenderer';
 import type { BaseUIComponentProps } from '../../utils/types';
-import { TabsOrientation } from '../Root/TabsRoot';
+import type { TabsOrientation } from '../Root/TabsRoot';
+import { useTabsRootContext } from '../Root/TabsRootContext';
 
 /**
  *
@@ -20,18 +21,25 @@ const Tab = React.forwardRef(function Tab(
   props: Tab.Props,
   forwardedRef: React.ForwardedRef<Element>,
 ) {
-  const { className, disabled = false, render, value, ...other } = props;
+  const { className, disabled = false, render, value: valueProp, ...other } = props;
 
-  const { selected, getRootProps, orientation } = useTab({
+  const { value: selectedValue, getTabPanelId, orientation } = useTabsRootContext();
+
+  const { selected, getRootProps } = useTab({
     ...props,
+    getTabPanelId,
+    isSelected: valueProp === selectedValue,
     rootRef: forwardedRef,
   });
 
-  const ownerState: Tab.OwnerState = {
-    disabled,
-    selected,
-    orientation,
-  };
+  const ownerState: Tab.OwnerState = React.useMemo(
+    () => ({
+      disabled,
+      selected,
+      orientation,
+    }),
+    [disabled, selected, orientation],
+  );
 
   const { renderElement } = useComponentRenderer({
     propGetter: getRootProps,
diff --git a/packages/mui-base/src/Tabs/Tab/useTab.test.tsx b/packages/mui-base/src/Tabs/Tab/useTab.test.tsx
index 7d7e5a3041..451f1e6c49 100644
--- a/packages/mui-base/src/Tabs/Tab/useTab.test.tsx
+++ b/packages/mui-base/src/Tabs/Tab/useTab.test.tsx
@@ -12,7 +12,11 @@ describe('useTab', () => {
     it('returns props for root slot', () => {
       function TestTab() {
         const rootRef = React.createRef<HTMLDivElement>();
-        const { getRootProps } = useTab({ rootRef });
+        const { getRootProps } = useTab({
+          rootRef,
+          isSelected: true,
+          getTabPanelId: () => undefined,
+        });
         return <button {...getRootProps()} />;
       }
 
@@ -37,7 +41,11 @@ describe('useTab', () => {
 
       function TestTab() {
         const rootRef = React.createRef<HTMLDivElement>();
-        const { getRootProps } = useTab({ rootRef });
+        const { getRootProps } = useTab({
+          rootRef,
+          isSelected: true,
+          getTabPanelId: () => undefined,
+        });
         return (
           <button
             {...getRootProps({
diff --git a/packages/mui-base/src/Tabs/Tab/useTab.ts b/packages/mui-base/src/Tabs/Tab/useTab.ts
index 204ba177c2..3fec1f4570 100644
--- a/packages/mui-base/src/Tabs/Tab/useTab.ts
+++ b/packages/mui-base/src/Tabs/Tab/useTab.ts
@@ -1,6 +1,6 @@
 'use client';
 import * as React from 'react';
-import { useTabsRootContext } from '../Root/TabsRootContext';
+import type { TabsRootContext } from '../Root/TabsRootContext';
 import { TabMetadata } from '../Root/useTabsRoot';
 import { useCompoundItem } from '../../useCompound';
 import { useListItem } from '../../useList';
@@ -8,20 +8,24 @@ import { useButton } from '../../useButton';
 import { useId } from '../../utils/useId';
 import { useForkRef } from '../../utils/useForkRef';
 import { mergeReactProps } from '../../utils/mergeReactProps';
-import { TabsOrientation } from '../Root/TabsRoot';
 
 function tabValueGenerator(otherTabValues: Set<any>) {
   return otherTabValues.size;
 }
 
 function useTab(parameters: useTab.Parameters): useTab.ReturnValue {
-  const { value: valueParam, rootRef: externalRef, disabled = false, id: idParam } = parameters;
+  const {
+    value: valueParam,
+    rootRef: externalRef,
+    disabled = false,
+    getTabPanelId,
+    id: idParam,
+    isSelected,
+  } = parameters;
 
   const tabRef = React.useRef<HTMLElement>(null);
   const id = useId(idParam);
 
-  const { value: selectedValue, getTabPanelId, orientation } = useTabsRootContext();
-
   const tabMetadata = React.useMemo(() => ({ disabled, ref: tabRef, id }), [disabled, tabRef, id]);
 
   const {
@@ -46,7 +50,6 @@ function useTab(parameters: useTab.Parameters): useTab.ReturnValue {
 
   const getRootProps = React.useCallback(
     (externalProps = {}) => {
-      //
       return mergeReactProps<'button'>(
         externalProps,
         mergeReactProps<'button'>(
@@ -70,14 +73,13 @@ function useTab(parameters: useTab.Parameters): useTab.ReturnValue {
     rootRef: handleRef,
     // the `selected` state isn't set on the server (it relies on effects to be calculated),
     // so we fall back to checking the `value` prop with the selectedValue from the TabsContext
-    selected: selected || value === selectedValue,
+    selected: selected || isSelected,
     totalTabsCount,
-    orientation,
   };
 }
 
 namespace useTab {
-  export interface Parameters {
+  export interface Parameters extends Pick<TabsRootContext, 'getTabPanelId'> {
     /**
      * The value of the tab.
      * It's used to associate the tab with a tab panel(s) with the same value.
@@ -97,6 +99,7 @@ namespace useTab {
      * If not provided, it will be automatically generated.
      */
     id?: string;
+    isSelected: boolean;
     /**
      * Ref to the root slot's DOM element.
      */
@@ -116,7 +119,6 @@ namespace useTab {
      * 0-based index of the tab in the list of tabs.
      */
     index: number;
-    orientation: TabsOrientation;
     /**
      * Ref to the root slot's DOM element.
      */