feat: creates paginated organizations query

CHANGELOG.md

@@ -7,6 +7,9 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 
 ## [Unreleased]
 
+### Added
+- Introduced the getOrganizationsPaginatedByEmail function to retrieve organizations with pagination, preventing timeouts.
+
 ## [0.62.0] - 2024-12-03
 ### Added
 

graphql/schema.graphql

@@ -92,7 +92,13 @@ type Query {
   getOrganizationsByEmail(email: String): [B2BOrganization]
     @checkUserAccess
     @cacheControl(scope: PRIVATE)
-
+  getOrganizationsPaginatedByEmail(
+    email: String
+    page: Int = 1
+    pageSize: Int = 25
+  ): B2BOrganizationPaginated
+    @checkUserAccess
+    @cacheControl(scope: PRIVATE)
   getActiveOrganizationsByEmail(email: String): [B2BOrganization]
     @cacheControl(scope: PRIVATE)
     @validateStoreUserAccess
@@ -490,6 +496,11 @@ type B2BOrganization {
   costCenterName: String
 }
 
+type B2BOrganizationPaginated {
+  data: [B2BOrganization]
+  pagination: Pagination
+}
+
 type SimpleRole {
   id: ID
   name: String!

node/clients/storefrontPermissions.ts

@@ -8,6 +8,7 @@ import saveUser from '../mutations/saveUser'
 import updateUser from '../mutations/updateUser'
 import getB2BUser from '../queries/getB2BUser'
 import getOrganizationsByEmail from '../queries/getOrganizationsByEmail'
+import getOrganizationsPaginatedByEmail from '../queries/getOrganizationsPaginatedByEmail'
 import getPermission from '../queries/getPermission'
 import getRole from '../queries/getRole'
 import getUser from '../queries/getUser'
@@ -40,6 +41,22 @@ export default class StorefrontPermissions extends AppGraphQLClient {
     })
   }
 
+  public getOrganizationsPaginatedByEmail = async (
+    email: string,
+    page: number,
+    pageSize: number
+  ): Promise<any> => {
+    return this.query({
+      extensions: this.getPersistedQuery(),
+      query: getOrganizationsPaginatedByEmail,
+      variables: {
+        email,
+        page,
+        pageSize,
+      },
+    })
+  }
+
   public listRoles = async (): Promise<any> => {
     return this.query({
       extensions: this.getPersistedQuery(),

node/queries/getOrganizationsPaginatedByEmail.ts

@@ -0,0 +1,25 @@
+import { print } from 'graphql'
+import gql from 'graphql-tag'
+
+export default print(gql`
+  query OrganizationsPaginated($email: String!, $page: Int, $pageSize: Int) {
+    getOrganizationsPaginatedByEmail(
+      email: $email
+      page: $page
+      pageSize: $pageSize
+    ) {
+      data {
+        costId
+        orgId
+        roleId
+        id
+        clId
+      }
+      pagination {
+        page
+        pageSize
+        total
+      }
+    }
+  }
+`)

node/resolvers/Queries/Organizations.ts

@@ -276,6 +276,58 @@ const Organizations = {
     }
   },
 
+  getOrganizationsPaginatedByEmail: async (
+    _: void,
+    {
+      email,
+      page = 1,
+      pageSize = 25,
+    }: {
+      email?: string
+      page: number
+      pageSize: number
+    },
+    {
+      clients: { storefrontPermissions },
+      vtex: { logger },
+    }: any
+  ) => {
+    try {
+      const organizationFilters: string[] = []
+      const fromSession = false
+
+      const response =
+        await storefrontPermissions.getOrganizationsPaginatedByEmail(
+          email,
+          page,
+          pageSize
+        )
+
+      const {
+        data: rawOrganizations = [],
+        pagination,
+      } = response?.data?.getOrganizationsPaginatedByEmail || {}
+
+      const organizations =
+        organizationFilters.length > 0
+          ? rawOrganizations.filter(({ orgId }: { orgId: string }) =>
+              organizationFilters.includes(orgId) || fromSession
+            )
+          : rawOrganizations
+
+      return {
+        data: organizations,
+        pagination,
+      }
+    } catch (error) {
+      logger.error({
+        error,
+        message: 'getOrganizationsPaginatedByEmail-error',
+      })
+      throw new GraphQLError(getErrorMessage(error))
+    }
+  },
+
   getActiveOrganizationsByEmail: async (
     _: void,
     { email }: { email: string },