fetch-security-policy.ts

import constants from '../../constants'
import { handleApiCall, handleUnsuccessfulApiResponse } from '../../utils/api'
import { AuthError } from '../../utils/errors'
import { getDefaultToken, setupSdk } from '../../utils/sdk'

import type { SocketSdkReturnType } from '@socketsecurity/sdk'

export async function fetchSecurityPolicy(
  orgSlug: string
): Promise<SocketSdkReturnType<'getOrgSecurityPolicy'>['data'] | undefined> {
  const apiToken = getDefaultToken()
  if (!apiToken) {
    throw new AuthError(
      'User must be authenticated to run this command. To log in, run the command `socket login` and enter your API key.'
    )
  }

  return await fetchSecurityPolicyWithToken(apiToken, orgSlug)
}

async function fetchSecurityPolicyWithToken(
  apiToken: string,
  orgSlug: string
): Promise<SocketSdkReturnType<'getOrgSecurityPolicy'>['data'] | undefined> {
  // Lazily access constants.spinner.
  const { spinner } = constants

  const socketSdk = await setupSdk(apiToken)

  spinner.start('Fetching organization quota...')

  const result = await handleApiCall(
    socketSdk.getOrgSecurityPolicy(orgSlug),
    'looking up organization quota'
  )

  spinner?.successAndStop('Received organization quota response.')

  if (!result.success) {
    handleUnsuccessfulApiResponse('getOrgSecurityPolicy', result)
    return
  }

  return result.data
}