utils.ts

import pgcs from 'pg-connection-string'
import { FastifyRequest } from 'fastify'
import { compile } from 'json-schema-to-typescript'

export const extractRequestForLogging = (request: FastifyRequest) => {
  let pg: string = 'unknown'
  try {
    if (request.headers.pg) {
      pg = pgcs.parse(request.headers.pg as string).host || pg
    }
  } catch (e: any) {
    console.warn('failed to parse PG connstring for ' + request.url)
  }

  const additional = request.headers['x-supabase-info']?.toString() || ''

  return {
    method: request.method,
    url: request.url,
    pg,
    opt: additional,
  }
}

export function translateErrorToResponseCode(
  error: { message: string },
  defaultResponseCode = 400
): number {
  if (error.message === 'Connection terminated due to connection timeout') {
    return 504
  } else if (error.message === 'sorry, too many clients already') {
    return 503
  }
  return defaultResponseCode
}

export async function generateTypeFromCheckConstraint(
  checkConstraints: string | null
): Promise<string> {
  if (!checkConstraints) {
    throw new Error('check constraint is empty')
  }

  if (typeof checkConstraints !== 'string') {
    throw new Error('invalid input type')
  }

  const match = /jsonb?_matches_schema\(\'([\{|\[].*[\}|\]])/gms.exec(checkConstraints)
  const extractedJsonStr = match ? match[1] : null
  const jsonSchema = JSON.parse(extractedJsonStr ?? '{}')
  const tsType = await compile(jsonSchema, 'Type', {
    bannerComment: '',
    style: {
      singleQuote: true,
      semi: false,
    },
    format: false
  })

  return tsType.replaceAll('export interface Type ', '')
}