Fixed bugs in memberships management

packages/features/team-accounts/src/components/members/account-members-table.tsx

@@ -56,16 +56,23 @@ export function AccountMembersTable({
   const { t } = useTranslation('teams');
 
   const permissions = {
-    canUpdateRole: (targetRole: number) =>
-      canManageRoles && targetRole < userRoleHierarchy,
-    canRemoveFromAccount: (targetRole: number) =>
-      canManageRoles && targetRole < userRoleHierarchy,
+    canUpdateRole: (targetRole: number) => {
+      return (
+        isPrimaryOwner || (canManageRoles && userRoleHierarchy < targetRole)
+      );
+    },
+    canRemoveFromAccount: (targetRole: number) => {
+      return (
+        isPrimaryOwner || (canManageRoles && userRoleHierarchy < targetRole)
+      );
+    },
     canTransferOwnership: isPrimaryOwner,
   };
 
   const columns = useGetColumns(permissions, {
     currentUserId,
     currentAccountId,
+    currentRoleHierarchy: userRoleHierarchy,
   });
 
   const filteredMembers = members.filter((member) => {
@@ -96,6 +103,7 @@ function useGetColumns(
   params: {
     currentUserId: string;
     currentAccountId: string;
+    currentRoleHierarchy: number;
   },
 ): ColumnDef<Members[0]>[] {
   const { t } = useTranslation('teams');
@@ -173,6 +181,7 @@ function useGetColumns(
             member={row.original}
             currentUserId={params.currentUserId}
             accountId={params.currentAccountId}
+            currentRoleHierarchy={params.currentRoleHierarchy}
           />
         ),
       },
@@ -185,12 +194,13 @@ function ActionsDropdown({
   permissions,
   member,
   currentUserId,
-  accountId,
+  currentRoleHierarchy,
 }: {
   permissions: Permissions;
   member: Members[0];
   currentUserId: string;
   accountId: string;
+  currentRoleHierarchy: number;
 }) {
   const [isRemoving, setIsRemoving] = useState(false);
   const [isTransferring, setIsTransferring] = useState(false);
@@ -262,10 +272,10 @@ function ActionsDropdown({
         <UpdateMemberRoleDialog
           isOpen
           setIsOpen={setIsUpdatingRole}
-          accountId={member.id}
           userId={member.user_id}
           userRole={member.role}
-          userRoleHierarchy={memberRoleHierarchy}
+          teamAccountId={member.account_id}
+          userRoleHierarchy={currentRoleHierarchy}
         />
       </If>
 
@@ -274,7 +284,7 @@ function ActionsDropdown({
           isOpen
           setIsOpen={setIsTransferring}
           targetDisplayName={member.name ?? member.email}
-          accountId={accountId}
+          accountId={member.account_id}
           userId={member.user_id}
         />
       </If>

packages/features/team-accounts/src/components/members/update-member-role-dialog.tsx

@@ -36,14 +36,14 @@ export const UpdateMemberRoleDialog: React.FC<{
   isOpen: boolean;
   setIsOpen: (isOpen: boolean) => void;
   userId: string;
-  accountId: string;
+  teamAccountId: string;
   userRole: Role;
   userRoleHierarchy: number;
 }> = ({
   isOpen,
   setIsOpen,
   userId,
-  accountId,
+  teamAccountId,
   userRole,
   userRoleHierarchy,
 }) => {
@@ -61,14 +61,14 @@ export const UpdateMemberRoleDialog: React.FC<{
         </DialogHeader>
 
         <RolesDataProvider
-          accountId={accountId}
+          accountId={teamAccountId}
           maxRoleHierarchy={userRoleHierarchy}
         >
           {(data) => (
             <UpdateMemberForm
               setIsOpen={setIsOpen}
               userId={userId}
-              accountId={accountId}
+              teamAccountId={teamAccountId}
               userRole={userRole}
               roles={data}
             />
@@ -82,13 +82,13 @@ export const UpdateMemberRoleDialog: React.FC<{
 function UpdateMemberForm({
   userId,
   userRole,
-  accountId,
+  teamAccountId,
   setIsOpen,
   roles,
 }: React.PropsWithChildren<{
   userId: string;
   userRole: Role;
-  accountId: string;
+  teamAccountId: string;
   setIsOpen: (isOpen: boolean) => void;
   roles: Role[];
 }>) {
@@ -99,7 +99,11 @@ function UpdateMemberForm({
   const onSubmit = ({ role }: { role: Role }) => {
     startTransition(async () => {
       try {
-        await updateMemberRoleAction({ accountId, userId, role });
+        await updateMemberRoleAction({
+          accountId: teamAccountId,
+          userId,
+          role,
+        });
 
         setIsOpen(false);
       } catch (e) {