payloadcms/test/helpers/reInitializeDB.ts

import * as qs from 'qs-esm'

export const path = '/re-initialize'

export const reInitializeDB = async ({
  serverURL,
  snapshotKey,
  uploadsDir,
  deleteOnly,
}: {
  deleteOnly?: boolean
  serverURL: string
  snapshotKey: string
  uploadsDir?: string | string[]
}) => {
  const maxAttempts = 50
  let attempt = 1
  const startTime = Date.now()

  while (attempt <= maxAttempts) {
    try {
      console.log(`Attempting to reinitialize DB (attempt ${attempt}/${maxAttempts})...`)

      const queryParams = qs.stringify(
        {
          snapshotKey,
          uploadsDir,
          deleteOnly,
        },
        {
          addQueryPrefix: true,
        },
      )

      const response = await fetch(`${serverURL}/api${path}${queryParams}`, {
        method: 'get',
        headers: {
          'Content-Type': 'application/json',
        },
      })

      if (!response.ok) {
        throw new Error(`HTTP error! status: ${response.status}`)
      }

      const timeTaken = Date.now() - startTime
      console.log(`Successfully reinitialized DB (took ${timeTaken}ms)`)
      return
    } catch (error) {
      console.error(`Failed to reinitialize DB`, error)

      if (attempt === maxAttempts) {
        console.error('Max retry attempts reached. Giving up.')
        throw error
      }

      console.log('Retrying in 3 seconds...')
      await new Promise((resolve) => setTimeout(resolve, 3000))
      attempt++
    }
  }
}