Add new Check in the useIsResourceRestricted logic

linode · Oct 14, 2024 · 5748b02 · 5748b02
1 parent 032a6c0
commit 5748b02
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 11 deletions.
diff --git a/packages/manager/src/features/VPCs/VPCLanding/VPCRow.tsx b/packages/manager/src/features/VPCs/VPCLanding/VPCRow.tsx
@@ -9,7 +9,7 @@ import { TableCell } from 'src/components/TableCell';
 import { TableRow } from 'src/components/TableRow';
 import { useRegionsQuery } from 'src/queries/regions/regions';
 import { getRestrictedResourceText } from 'src/features/Account/utils';
-import { useGrants } from 'src/queries/profile/profile';
+import { useIsResourceRestricted } from 'src/hooks/useIsResourceRestricted';
 
 interface Props {
   handleDeleteVPC: () => void;
@@ -27,14 +27,11 @@ export const VPCRow = ({ handleDeleteVPC, handleEditVPC, vpc }: Props) => {
     0
   );
 
-  const { data: grants } = useGrants();
-
-  const isVPCReadOnly =
-    grants &&
-    (grants['vpc'].some(
-      (grant) => grant.id === vpc.id && grant.permissions === 'read_only'
-    ) ||
-      !grants['vpc'].some((grant) => grant.id === vpc.id));
+  const isVPCReadOnly = useIsResourceRestricted({
+    grantLevel: 'read_only',
+    grantType: 'vpc',
+    id: vpc.id,
+  });
 
   const actions: Action[] = [
     {

diff --git a/packages/manager/src/hooks/useIsResourceRestricted.ts b/packages/manager/src/hooks/useIsResourceRestricted.ts
@@ -19,7 +19,9 @@ export const useIsResourceRestricted = ({
   if (!grants) {
     return false;
   }
-  return grants[grantType].some(
-    (grant) => grant.id === id && grant.permissions === grantLevel
+  return (
+    grants[grantType].some(
+      (grant) => grant.id === id && grant.permissions === grantLevel
+    ) || !grants[grantType].some((grant) => grant.id === id)
   );
 };