Update hd compact story

jpmorganchase · May 22, 2024 · f260109 · f260109
1 parent 95c3e68
commit f260109
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/packages/ag-grid-theme/stories/examples/HDCompact.tsx b/packages/ag-grid-theme/stories/examples/HDCompact.tsx
@@ -1,4 +1,8 @@
-import { UNSTABLE_SaltProviderNext } from "@salt-ds/core";
+import {
+  SaltProvider,
+  UNSTABLE_SaltProviderNext,
+  useTheme,
+} from "@salt-ds/core";
 import { AgGridReact, AgGridReactProps } from "ag-grid-react";
 import { useAgGridThemeSwitcher } from "../dependencies/ThemeSwitcher";
 import dataGridExampleColumns from "../dependencies/dataGridExampleColumns";
@@ -18,15 +22,18 @@ const statusBar = {
 };
 
 const HDCompact = (props: AgGridReactProps) => {
+  const { themeNext } = useTheme();
   const { themeName } = useAgGridThemeSwitcher();
   const { agGridProps, containerProps } = useAgGridHelpers({
     agThemeName: `ag-theme-${themeName}`,
     compact: true,
     density: "high",
   });
 
+  const Provider = themeNext ? UNSTABLE_SaltProviderNext : SaltProvider;
+
   return (
-    <UNSTABLE_SaltProviderNext density="high">
+    <Provider density="high">
       <div {...containerProps}>
         <AgGridReact
           columnDefs={dataGridExampleColumns}
@@ -47,7 +54,7 @@ const HDCompact = (props: AgGridReactProps) => {
           }}
         />
       </div>
-    </UNSTABLE_SaltProviderNext>
+    </Provider>
   );
 };