refactor: extract shared hasPermission helper (MODSetter/SurfSense#1366)

- Add canPerform() helper function to members-query.atoms.ts
- Add usePermissionGate() hook for convenience
- Update team-content.tsx to use canPerform()
- Update roles-manager.tsx to use canPerform()
- Eliminates duplicated permission check logic
- Centralizes permission policy in one location

Fixes #1366
This commit is contained in:
guangyang1206 2026-05-22 12:08:00 +08:00
parent 334729754f
commit a66d65a835
3 changed files with 41 additions and 8 deletions

View file

@ -31,7 +31,7 @@ import {
deleteMemberMutationAtom, deleteMemberMutationAtom,
updateMemberMutationAtom, updateMemberMutationAtom,
} from "@/atoms/members/members-mutation.atoms"; } from "@/atoms/members/members-mutation.atoms";
import { membersAtom, myAccessAtom } from "@/atoms/members/members-query.atoms"; import { membersAtom, myAccessAtom, canPerform } from "@/atoms/members/members-query.atoms";
import { import {
AlertDialog, AlertDialog,
AlertDialogAction, AlertDialogAction,
@ -126,14 +126,9 @@ export function TeamContent({ searchSpaceId }: TeamContentProps) {
const { data: access = null, isLoading: accessLoading } = useAtomValue(myAccessAtom); const { data: access = null, isLoading: accessLoading } = useAtomValue(myAccessAtom);
const hasPermission = useCallback( const hasPermission = useCallback(
(permission: string) => { (permission: string) => canPerform(access, permission),
if (!access) return false;
if (access.is_owner) return true;
return access.permissions?.includes(permission) ?? false;
},
[access] [access]
); );
const { data: members = [], isLoading: membersLoading } = useAtomValue(membersAtom); const { data: members = [], isLoading: membersLoading } = useAtomValue(membersAtom);
const { mutateAsync: updateMember } = useAtomValue(updateMemberMutationAtom); const { mutateAsync: updateMember } = useAtomValue(updateMemberMutationAtom);

View file

@ -39,3 +39,38 @@ export const myAccessAtom = atomWithQuery((get) => {
}, },
}; };
}); });
/**
* Helper function to check if the current user has a specific permission.
*
* @param access - The access object from useAtomValue(myAccessAtom)
* @param permission - The permission string to check
* @returns boolean indicating if the user has the permission
*
* @example
* const access = useAtomValue(myAccessAtom);
* if (canPerform(access, 'manage_members')) { ... }
*/
export function canPerform(
access: { is_owner: boolean; permissions?: string[] } | null | undefined,
permission: string
): boolean {
if (!access) return false;
if (access.is_owner) return true;
return access.permissions?.includes(permission) ?? false;
}
/**
* Hook wrapper for canPerform that reads from myAccessAtom internally.
* Use this if you want to avoid calling useAtomValue(myAccessAtom) separately.
*
* @param permission - The permission string to check
* @returns boolean indicating if the user has the permission
*
* @example
* const canManageMembers = usePermissionGate('manage_members');
*/
export function usePermissionGate(permission: string): boolean {
const access = useAtomValue(myAccessAtom);
return canPerform(access, permission);
}

View file

@ -26,7 +26,7 @@ import {
} from "lucide-react"; } from "lucide-react";
import { useCallback, useEffect, useMemo, useState } from "react"; import { useCallback, useEffect, useMemo, useState } from "react";
import { toast } from "sonner"; import { toast } from "sonner";
import { myAccessAtom } from "@/atoms/members/members-query.atoms"; import { myAccessAtom, canPerform } from "@/atoms/members/members-query.atoms";
import { permissionsAtom } from "@/atoms/permissions/permissions-query.atoms"; import { permissionsAtom } from "@/atoms/permissions/permissions-query.atoms";
import { import {
createRoleMutationAtom, createRoleMutationAtom,
@ -257,6 +257,9 @@ export function RolesManager({ searchSpaceId }: { searchSpaceId: number }) {
const { data: access = null } = useAtomValue(myAccessAtom); const { data: access = null } = useAtomValue(myAccessAtom);
const hasPermission = useCallback( const hasPermission = useCallback(
(permission: string) => canPerform(access, permission),
[access]
);
(permission: string) => { (permission: string) => {
if (!access) return false; if (!access) return false;
if (access.is_owner) return true; if (access.is_owner) return true;