Use checkpoint pagination to get all organizations

src/tools/auth0/client.ts

@@ -38,12 +38,15 @@ function checkpointPaginator(
     const { checkpoint, ...newArgs } = _.cloneDeep(args[0]);
 
     // fetch the total to validate records match
-    const { total } = await client.pool
-      .addSingleTask({
-        data: newArgs,
-        generator: (requestArgs) => target[name](requestArgs),
-      })
-      .promise();
+    const total =
+      (
+        await client.pool
+          .addSingleTask({
+            data: newArgs,
+            generator: (requestArgs) => target[name](requestArgs),
+          })
+          .promise()
+      ).data?.total || 0;
 
     let done = false;
     // use checkpoint pagination to allow fetching 1000+ results
@@ -57,11 +60,11 @@ function checkpointPaginator(
         })
         .promise();
 
-      data.push(...getEntity(rsp));
-      if (!rsp.next) {
+      data.push(...getEntity(rsp.data));
+      if (!rsp.data.next) {
         done = true;
       } else {
-        newArgs.from = rsp.next;
+        newArgs.from = rsp.data.next;
       }
     }
 
@@ -175,7 +178,7 @@ export default function pagedClient(client: ManagementClient): Auth0APIClient {
 // eslint-disable-next-line no-unused-vars
 export async function paginate<T>(
   fetchFunc: (...paginateArgs: any) => any,
-  args: PagePaginationParams
+  args: PagePaginationParams | CheckpointPaginationParams
 ): Promise<T[]> {
   // override default <T>.getAll() behaviour using pagedClient
   const allItems = (await fetchFunc(args)) as unknown as T[];

src/tools/auth0/handlers/organizations.ts

@@ -321,7 +321,7 @@ export default class OrganizationsHandler extends DefaultHandler {
     try {
       const [organizations, clients] = await Promise.all([
         paginate<GetOrganizations200ResponseOneOfInner>(this.client.organizations.getAll, {
-          paginate: true,
+          checkpoint: true,
           include_totals: true,
         }),
         paginate<Client>(this.client.clients.getAll, {

src/types.ts

@@ -22,8 +22,8 @@ type SharedPaginationParams = {
 };
 
 export type CheckpointPaginationParams = SharedPaginationParams & {
-  from: string;
-  take: number;
+  from?: string;
+  take?: number;
 };
 
 export type PagePaginationParams = SharedPaginationParams & {