Function createPostgresConnection

apps/sim/app/api/tools/postgresql/utils.ts:5–35

(config: PostgresConnectionConfig)

import type { PostgresConnectionConfig } from '@/tools/postgresql/types'

export async function createPostgresConnection(config: PostgresConnectionConfig) {
  const hostValidation = await validateDatabaseHost(config.host, 'host')
  if (!hostValidation.isValid) {
    throw new Error(hostValidation.error)
  }

  const resolvedHost = hostValidation.resolvedIP ?? config.host

  const sslConfig: boolean | 'prefer' | { rejectUnauthorized: boolean; servername?: string } =
    config.ssl === 'disabled'
      ? false
      : config.ssl === 'preferred'
        ? 'prefer'
        : { rejectUnauthorized: false, servername: config.host }

  const sql = postgres({
    // Pin the validated IP (never the hostname) to prevent DNS rebinding; SNI stays the hostname above.
    host: resolvedHost,
    port: config.port,
    database: config.database,
    username: config.username,
    password: config.password,
    ssl: sslConfig,
    connect_timeout: 10, // 10 seconds
    idle_timeout: 20, // 20 seconds
    max_lifetime: 60 * 30, // 30 minutes
    max: 1, // Single connection for tool usage
  })

  return sql
}

export async function executeQuery(
  sql: any,