util.ts

import { sep } from 'path'

import type { NetlifyAPI } from '@netlify/api'
import pWaitFor from 'p-wait-for'

import { DEPLOY_POLL } from './constants.js'
import type { FileObject } from './upload-files.js'

type Deploy = Awaited<ReturnType<NetlifyAPI['getSiteDeploy']>>

// normalize windows paths to unix paths
export const normalizePath = (relname: string): string => {
  if (relname.includes('#') || relname.includes('?')) {
    throw new Error(`Invalid filename ${relname}. Deployed filenames cannot contain # or ? characters`)
  }
  return relname.split(sep).join('/')
}

// poll an async deployId until its done diffing
export const waitForDiff = async (
  api: NetlifyAPI,
  deployId: string,
  siteId: string,
  timeout: number,
): Promise<Deploy> => {
  // capture ready deploy during poll
  let deploy: Deploy

  const loadDeploy = async () => {
    const siteDeploy = await api.getSiteDeploy({ siteId, deployId })

    switch (siteDeploy.state) {
      // https://github.com/netlify/bitballoon/blob/master/app/models/deploy.rb#L21-L33
      case 'error': {
        const deployError = new Error(siteDeploy.error_message || `Deploy ${deployId} had an error`)
        Object.assign(deployError, { deploy: siteDeploy })
        throw deployError
      }
      case 'prepared':
      case 'uploading':
      case 'uploaded':
      case 'ready': {
        deploy = siteDeploy
        return true
      }
      case 'preparing':
      default: {
        return false
      }
    }
  }

  await pWaitFor(loadDeploy, {
    interval: DEPLOY_POLL,
    timeout: {
      milliseconds: timeout,
      message: 'Timeout while waiting for deploy',
    },
  })

  // @ts-expect-error TS(2454) FIXME: Variable 'deploy' is used before being assigned.
  return deploy
}

// Poll a deployId until its ready
export const waitForDeploy = async (
  api: NetlifyAPI,
  deployId: string,
  siteId: string,
  timeout: number,
): Promise<Deploy> => {
  // capture ready deploy during poll
  let deploy: Deploy

  const loadDeploy = async () => {
    const siteDeploy = await api.getSiteDeploy({ siteId, deployId })
    switch (siteDeploy.state) {
      // https://github.com/netlify/bitballoon/blob/master/app/models/deploy.rb#L21-L33
      case 'error': {
        const deployError = new Error(siteDeploy.error_message || `Deploy ${deployId} had an error`)
        Object.assign(deployError, { deploy: siteDeploy })
        throw deployError
      }
      case 'ready': {
        deploy = siteDeploy
        return true
      }
      case 'preparing':
      case 'prepared':
      case 'uploaded':
      case 'uploading':
      default: {
        return false
      }
    }
  }

  await pWaitFor(loadDeploy, {
    interval: DEPLOY_POLL,
    timeout: {
      milliseconds: timeout,
      message: 'Timeout while waiting for deploy',
    },
  })

  // @ts-expect-error TS(2454) FIXME: Variable 'deploy' is used before being assigned.
  return deploy
}

// Transform the fileShaMap and fnShaMap into a generic shaMap that file-uploader.js can use
export const getUploadList = (required?: string[] | null, shaMap?: Record<string, FileObject[]> | null) => {
  if (!required || !shaMap) return []
  return required.flatMap((sha) => shaMap[sha])
}