Paginate over template results

src/commands/templates/helpers.ts

@@ -4,6 +4,8 @@ import { readJsonSync, readFileSync, existsSync } from 'fs-extra'
 import traverse from 'traverse'
 import dirTree from 'directory-tree'
 import { TemplateManifest, MetaFileTraverse, MetaFile } from '../../types'
+import { ServerClient } from 'postmark'
+import { Templates } from 'postmark/dist/client/models'
 
 /**
  * Parses templates folder and files
@@ -102,3 +104,19 @@ export function sameContent(
 
   return str1 === str2
 }
+
+export async function fetchAllTemplates(client: ServerClient) {
+  const pageSize = 300
+  let offset = 0
+  let totalCount = Infinity
+  let templates: Templates['Templates'] = []
+
+  while (offset < totalCount) {
+    const response = await client.getTemplates({ count: pageSize, offset })
+    totalCount = response.TotalCount
+    offset += response.Templates.length
+    templates.push(...response.Templates)
+  }
+
+  return { Templates: templates, TotalCount: totalCount }
+}

src/commands/templates/pull.ts

@@ -15,6 +15,7 @@ import {
   logError,
   fatalError,
 } from '../../utils'
+import { fetchAllTemplates } from './helpers'
 
 export const command = 'pull <output directory> [options]'
 export const desc = 'Pull templates from a server to <output directory>'
@@ -106,7 +107,7 @@ async function fetchTemplateList(options: TemplateListOptions) {
   }
 
   try {
-    const templates = await client.getTemplates({ count: 300 })
+    const templates = await fetchAllTemplates(client)
 
     if (templates.TotalCount === 0) {
       spinner.stop()

src/commands/templates/push.ts

@@ -18,7 +18,12 @@ import {
   logError,
 } from '../../utils'
 
-import { createManifest, sameContent, templatesDiff } from './helpers'
+import {
+  createManifest,
+  fetchAllTemplates,
+  sameContent,
+  templatesDiff,
+} from './helpers'
 
 const debug = require('debug')('postmark-cli:templates:push')
 
@@ -119,7 +124,7 @@ async function push(
     }
 
     try {
-      const templateList = await client.getTemplates({ count: 300 })
+      const templateList = await fetchAllTemplates(client)
       const newList =
         templateList.TotalCount === 0
           ? []