fix: hide groups in account page if not enabled (coder#10898)

Parkreiner · web-flow · commit 1f6e39c0b0a8 · 2023-11-27T14:06:00.000-05:00
diff --git a/site/src/pages/UserSettingsPage/AccountPage/AccountPage.tsx b/site/src/pages/UserSettingsPage/AccountPage/AccountPage.tsx
@@ -5,19 +5,25 @@ import { useQuery } from "react-query";
 import { groupsForUser } from "api/queries/groups";
 import { useOrganizationId } from "hooks";
 import { useAuth } from "components/AuthProvider/AuthProvider";
+import { useDashboard } from "components/Dashboard/DashboardProvider";
 
 import { Stack } from "@mui/system";
 import { AccountUserGroups } from "./AccountUserGroups";
 import { AccountForm } from "./AccountForm";
 import { Section } from "components/SettingsLayout/Section";
 
 export const AccountPage: FC = () => {
-  const { updateProfile, updateProfileError, isUpdatingProfile } = useAuth();
-  const permissions = usePermissions();
-
   const me = useMe();
+  const permissions = usePermissions();
   const organizationId = useOrganizationId();
-  const groupsQuery = useQuery(groupsForUser(organizationId, me.id));
+  const { updateProfile, updateProfileError, isUpdatingProfile } = useAuth();
+  const { entitlements } = useDashboard();
+
+  const hasGroupsFeature = entitlements.features.user_role_management.enabled;
+  const groupsQuery = useQuery({
+    ...groupsForUser(organizationId, me.id),
+    enabled: hasGroupsFeature,
+  });
 
   return (
     <Stack spacing={6}>
@@ -33,11 +39,13 @@ export const AccountPage: FC = () => {
       </Section>
 
       {/* Has <Section> embedded inside because its description is dynamic */}
-      <AccountUserGroups
-        groups={groupsQuery.data}
-        loading={groupsQuery.isLoading}
-        error={groupsQuery.error}
-      />
+      {hasGroupsFeature && (
+        <AccountUserGroups
+          groups={groupsQuery.data}
+          loading={groupsQuery.isLoading}
+          error={groupsQuery.error}
+        />
+      )}
     </Stack>
   );
 };
