Fixed issue in disabling the side menu items

Zipstack · Feb 4, 2025 · dbd24b2 · dbd24b2
1 parent cc56b29
commit dbd24b2
Showing 1 changed file with 24 additions and 7 deletions.
diff --git a/frontend/src/components/navigations/side-nav-bar/SideNavBar.jsx b/frontend/src/components/navigations/side-nav-bar/SideNavBar.jsx
@@ -46,6 +46,14 @@ try {
   // Plugin unavailable.
 }
 
+let selectedProductStore;
+let selectedProduct;
+try {
+  selectedProductStore = require("../../../plugins/llm-whisperer/store/select-product-store.js");
+} catch {
+  // Ignore if hook not available
+}
+
 const SideNavBar = ({ collapsed }) => {
   const navigate = useNavigate();
   const { sessionDetails } = useSessionStore();
@@ -62,6 +70,12 @@ const SideNavBar = ({ collapsed }) => {
     // Do nothing
   }
 
+  if (selectedProductStore?.useSelectedProductStore) {
+    selectedProduct = selectedProductStore.useSelectedProductStore(
+      (state) => state?.selectedProduct
+    );
+  }
+
   let menu;
   if (sideMenu) {
     menu = sideMenu.useSideMenu();
@@ -193,7 +207,12 @@ const SideNavBar = ({ collapsed }) => {
   }
 
   const shouldDisableAll = useMemo(() => {
-    if (!unstractSubscriptionPlan || !UNSTRACT_SUBSCRIPTION_PLANS) {
+    const isUnstract = !(selectedProduct && selectedProduct !== "unstract");
+    if (
+      !unstractSubscriptionPlan ||
+      !UNSTRACT_SUBSCRIPTION_PLANS ||
+      !isUnstract
+    ) {
       return false;
     }
 
@@ -203,13 +222,11 @@ const SideNavBar = ({ collapsed }) => {
     );
   }, [unstractSubscriptionPlan]);
 
-  if (shouldDisableAll) {
-    data.forEach((mainMenuItem) => {
-      mainMenuItem.subMenu.forEach((subMenuItem) => {
-        subMenuItem.disable = true;
-      });
+  data.forEach((mainMenuItem) => {
+    mainMenuItem.subMenu.forEach((subMenuItem) => {
+      subMenuItem.disable = shouldDisableAll;
     });
-  }
+  });
 
   return (
     <Sider