fix: Add extractMembership function

techmatters · Aug 28, 2023 · b233d92 · b233d92
1 parent c2500db
commit b233d92
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/src/collaboration/membershipsUtils.ts b/src/collaboration/membershipsUtils.ts
@@ -17,6 +17,7 @@
 import _ from 'lodash/fp';
 import type {
   AccountCollaborationMembershipFragment,
+  CollaborationMembershipFieldsFragment,
   CollaborationMembershipsFragment,
   CollaborationMembershipsInfoFragment,
   CollaborationMembershipsPendingFragment,
@@ -38,6 +39,15 @@ export const extractMembershipsInfo = (
   membershipsSample: extractMemberships(membershipList),
 });
 
+export const extractMembership = (
+  membership: Partial<CollaborationMembershipFieldsFragment>,
+) => ({
+  ...membership.user,
+  ..._.omit('user', membership),
+  membershipId: membership.id,
+  userId: membership.user?.id,
+});
+
 export const extractMemberships = (membershipList?: MembershipQuery | null) =>
   (
     (
@@ -47,11 +57,7 @@ export const extractMemberships = (membershipList?: MembershipQuery | null) =>
         | null
         | undefined
     )?.memberships?.edges || []
-  ).map(edge => ({
-    ...edge.node.user,
-    ..._.omit('user', edge.node),
-    membershipId: edge.node.id,
-  }));
+  ).map(edge => extractMembership(edge.node));
 
 export const extractAccountMembership = (
   membershipList?: AccountCollaborationMembershipFragment | null,