BIPMaterialManager/src/main/services/database/postgresql.ts

import { Pool } from 'pg'
import type {
  IDatabaseService,
  DatabaseType,
  QueryResult,
  PostgreSqlConfig
} from '../../types/database.types'
import { createLogger, trackDuration } from '../logger'

const log = createLogger('PostgreSqlService')

export type { PostgreSqlConfig } from '../../types/database.types'

/**
 * SQL keywords that should NOT be double-quoted during identifier preprocessing.
 * PostgreSQL lowercases unquoted identifiers, but SSMA-migrated databases
 * have uppercase column names that require double-quoting to preserve case.
 *
 * This list covers PostgreSQL reserved words across multiple categories:
 * - DML (Data Manipulation Language)
 * - DDL (Data Definition Language)
 * - Window functions
 * - CTEs (Common Table Expressions)
 * - Advanced GROUP BY clauses
 * - JSON operations
 * - Type system
 * - Table sampling
 * - Transaction control
 */
const SQL_KEYWORDS = new Set([
  // ==================== DML (Data Manipulation Language) ====================
  'SELECT',
  'FROM',
  'WHERE',
  'AND',
  'OR',
  'NOT',
  'IN',
  'IS',
  'NULL',
  'INSERT',
  'INTO',
  'VALUES',
  'UPDATE',
  'SET',
  'DELETE',

  // ==================== Ordering & Limiting ====================
  'ORDER',
  'BY',
  'ASC',
  'DESC',
  'LIMIT',
  'OFFSET',
  'FETCH',
  'NEXT',
  'ROWS',
  'ONLY',

  // ==================== Joins ====================
  'JOIN',
  'LEFT',
  'RIGHT',
  'INNER',
  'OUTER',
  'CROSS',
  'FULL',
  'ON',
  'NATURAL',
  'LATERAL',

  // ==================== Set Operations ====================
  'UNION',
  'ALL',
  'INTERSECT',
  'EXCEPT',

  // ==================== Grouping & Aggregation ====================
  'GROUP',
  'HAVING',
  'DISTINCT',
  'GROUPING',
  'SETS',
  'ROLLUP',
  'CUBE',
  'FILTER',
  'WITHIN',

  // ==================== Window Functions ====================
  'OVER',
  'PARTITION',
  'WINDOW',
  'RANGE',
  'UNBOUNDED',
  'PRECEDING',
  'FOLLOWING',
  'CURRENT',
  'ROW',
  'GROUPS',
  'EXCLUDE',
  'TIES',
  'RANK',
  'DENSE_RANK',
  'ROW_NUMBER',
  'NTILE',
  'LAG',
  'LEAD',
  'FIRST_VALUE',
  'LAST_VALUE',
  'NTH_VALUE',

  // ==================== CTE (Common Table Expressions) ====================
  'WITH',
  'RECURSIVE',
  'MATERIALIZED',
  'SEARCH',
  'CYCLE',
  'PATH',
  'ROOT',
  'SIBLINGS',

  // ==================== CASE Expressions ====================
  'CASE',
  'WHEN',
  'THEN',
  'ELSE',
  'END',

  // ==================== DDL (Data Definition Language) ====================
  'CREATE',
  'ALTER',
  'DROP',
  'TABLE',
  'INDEX',
  'COLUMN',
  'ADD',
  'MODIFY',
  'RENAME',
  'TO',
  'GENERATED',
  'ALWAYS',
  'IDENTITY',
  'INCLUDE',
  'TEMP',
  'TEMPORARY',
  'UNLOGGED',

  // ==================== PostgreSQL Specific - UPSERT/MERGE ====================
  'CONFLICT',
  'DO',
  'NOTHING',
  'EXCLUDED',
  'RETURNING',
  'MERGE',
  'USING',
  'MATCHED',
  'TARGET',
  'SOURCE',

  // ==================== Aggregate Functions ====================
  'COUNT',
  'SUM',
  'AVG',
  'MIN',
  'MAX',
  'EXISTS',
  'COALESCE',
  'NULLIF',
  'CAST',
  'AS',

  // ==================== JSON Operations ====================
  'JSON',
  'JSONB',
  'JSON_ARRAY',
  'JSON_OBJECT',
  'JSON_AGG',
  'JSONB_AGG',
  'JSONB_OBJECT_AGG',

  // ==================== Types & Casting ====================
  'DECIMAL',
  'NUMERIC',
  'BOOLEAN',
  'CHARACTER',
  'VARYING',
  'PRECISION',
  'REAL',
  'DOUBLE',
  'FLOAT',
  'TEXT',
  'INTEGER',
  'SERIAL',
  'BIGINT',
  'SMALLINT',
  'DATE',
  'TIME',
  'TIMESTAMP',
  'TIMESTAMPTZ',
  'TIMEZONE',
  'INTERVAL',
  'BIGSERIAL',
  'SMALLSERIAL',

  // ==================== Table Sampling ====================
  'TABLESAMPLE',
  'BERNOULLI',
  'SYSTEM',
  'REPEATABLE',
  'SEED',

  // ==================== Transaction Control ====================
  'BEGIN',
  'COMMIT',
  'ROLLBACK',
  'SAVEPOINT',
  'WORK',
  'ISOLATION',
  'LEVEL',
  'READ',
  'WRITE',
  'COMMITTED',
  'REPEATABLE',
  'SERIALIZABLE',

  // ==================== Types & Values ====================
  'TRUE',
  'FALSE',
  'DEFAULT',
  'PRIMARY',
  'KEY',
  'REFERENCES',
  'FOREIGN',
  'CONSTRAINT',
  'UNIQUE',
  'CHECK',
  'NULLS',
  'FIRST',
  'LAST',

  // ==================== Scalar & String Functions ====================
  'UPPER',
  'LOWER',
  'TRIM',
  'LTRIM',
  'RTRIM',
  'BTRIM',
  'SUBSTRING',
  'CONCAT',
  'LENGTH',
  'CHAR_LENGTH',
  'CHARACTER_LENGTH',
  'REPLACE',
  'POSITION',
  'OVERLAY',
  'LPAD',
  'RPAD',
  'REPEAT',
  'REVERSE',
  'SPLIT_PART',
  'INITCAP',
  'NORMALIZE',
  'CHR',
  'ASCII',
  'FORMAT',

  // ==================== Numeric Functions ====================
  'ABS',
  'CEIL',
  'CEILING',
  'FLOOR',
  'ROUND',
  'POWER',
  'SQRT',
  'MOD',
  'SIGN',
  'TRUNC',

  // ==================== Date/Time Functions ====================
  'EXTRACT',
  'DATE_TRUNC',
  'TO_CHAR',
  'TO_DATE',
  'TO_TIMESTAMP',
  'TO_NUMBER',
  'AGE',

  // ==================== Pattern Matching ====================
  'BETWEEN',
  'LIKE',
  'ILIKE',
  'SIMILAR',
  'ESCAPE',
  'ANY',
  'SOME',

  // ==================== Functions & Procedures ====================
  'AFTER',
  'BEFORE',
  'EACH',
  'STATEMENT',
  'TRIGGER',
  'FUNCTION',
  'PROCEDURE',
  'LANGUAGE',
  'SQL',
  'PLPGSQL',
  'RETURNS',
  'CALLED',
  'STRICT',
  'SECURITY',
  'INVOKER',
  'DEFINER',
  'VOLATILE',
  'STABLE',
  'IMMUTABLE',
  'PARALLEL',
  'SAFE',
  'RESTRICTED',
  'UNSAFE',

  // ==================== Utility Commands ====================
  'CONCURRENTLY',
  'REINDEX',
  'VACUUM',
  'ANALYZE',
  'EXPLAIN',
  'LOCAL',
  'GLOBAL',
  'ORDINALITY',
  'FREEZE',
  'VERBOSE',
  'BUFFERS',
  'FORMAT',
  'XML',
  'YAML',

  // ==================== Additional Reserved Words ====================
  'IF',
  'CURRENT_TIMESTAMP',
  'NOW',
  'GETDATE'
])

/**
 * Prepare SQL for PostgreSQL execution by quoting unquoted identifiers.
 *
 * PostgreSQL lowercases unquoted identifiers, but SSMA-migrated tables
 * have uppercase column names (e.g., "UserName", "ID") that require
 * double-quoting to preserve case.
 *
 * This function:
 * - Preserves string literals ('...')
 * - Preserves already-quoted identifiers ("...")
 * - Preserves parameter placeholders ($1, $2, ...)
 * - Preserves SQL keywords
 * - Double-quotes remaining identifiers
 */
export function prepareSql(sql: string): string {
  const result: string[] = []
  let i = 0
  const len = sql.length

  while (i < len) {
    const ch = sql[i]

    // Skip whitespace
    if (/\s/.test(ch)) {
      result.push(ch)
      i++
      continue
    }

    // Skip single-line comments (--)
    if (ch === '-' && i + 1 < len && sql[i + 1] === '-') {
      while (i < len && sql[i] !== '\n') {
        result.push(sql[i++])
      }
      continue
    }

    // Preserve string literals ('...')
    if (ch === "'") {
      result.push(ch)
      i++
      while (i < len) {
        if (sql[i] === "'") {
          result.push(sql[i++])
          // Handle escaped quotes ('')
          if (i < len && sql[i] === "'") {
            result.push(sql[i++])
          } else {
            break
          }
        } else {
          result.push(sql[i++])
        }
      }
      continue
    }

    // Preserve already-quoted identifiers ("...")
    if (ch === '"') {
      result.push(ch)
      i++
      while (i < len && sql[i] !== '"') {
        result.push(sql[i++])
      }
      if (i < len) {
        result.push(sql[i++])
      }
      continue
    }

    // Preserve parameter placeholders ($N)
    if (ch === '$') {
      result.push(ch)
      i++
      while (i < len && /\d/.test(sql[i])) {
        result.push(sql[i++])
      }
      continue
    }

    // Preserve @param placeholders
    if (ch === '@') {
      result.push(ch)
      i++
      while (i < len && /\w/.test(sql[i])) {
        result.push(sql[i++])
      }
      continue
    }

    // Preserve ? placeholders
    if (ch === '?') {
      result.push(ch)
      i++
      continue
    }

    // Collect word tokens (identifiers and keywords)
    if (/[a-zA-Z_]/.test(ch)) {
      let word = ''
      while (i < len && /\w/.test(sql[i])) {
        word += sql[i++]
      }

      // Check if it's a SQL keyword (case-insensitive)
      if (SQL_KEYWORDS.has(word.toUpperCase())) {
        result.push(word)
      } else {
        // Quote the identifier to preserve case
        result.push(`"${word}"`)
      }
      continue
    }

    // Everything else (operators, punctuation, numbers): pass through
    result.push(ch)
    i++
  }

  return result.join('')
}

export class PostgreSqlService implements IDatabaseService {
  /** Database type identifier */
  readonly type: DatabaseType = 'postgresql'

  private pool: Pool | null = null
  private config: PostgreSqlConfig

  constructor(config: PostgreSqlConfig) {
    this.config = config
  }

  /**
   * Connect to PostgreSQL database
   */
  async connect(): Promise<void> {
    if (this.pool) {
      log.warn('Already connected to PostgreSQL')
      throw new Error('Already connected to PostgreSQL')
    }

    try {
      this.pool = new Pool({
        host: this.config.host,
        port: this.config.port,
        user: this.config.user,
        password: this.config.password,
        database: this.config.database,
        max: this.config.maxPoolSize ?? 10,
        /**
         * Connection timeout in milliseconds.
         * Time to wait when connecting to PostgreSQL before failing.
         * Prevents hanging during network issues or server overload.
         */
        connectionTimeoutMillis: 10000,
        /**
         * PostgreSQL statement timeout in milliseconds.
         * Limits execution time for individual SQL statements.
         * Prevents long-running queries from blocking the connection pool.
         */
        statement_timeout: 30000,
        /**
         * Idle connection timeout in milliseconds.
         * Closes connections that have been idle for this duration.
         * Frees up pool resources and prevents stale connections.
         */
        idleTimeoutMillis: 30000,
        /**
         * Query timeout in milliseconds (pg driver level).
         * Fallback protection to abort queries that exceed this duration.
         * Should be longer than statement_timeout to allow PG to handle first.
         */
        query_timeout: 60000
      })

      // Test connection
      const client = await this.pool.connect()
      client.release()

      log.info('Connected to PostgreSQL', {
        host: this.config.host,
        port: this.config.port,
        database: this.config.database
      })
    } catch (error) {
      this.pool = null
      log.error('Failed to connect to PostgreSQL', {
        host: this.config.host,
        port: this.config.port,
        database: this.config.database,
        error
      })
      throw new Error(`Failed to connect to PostgreSQL: ${(error as Error).message}`)
    }
  }

  /**
   * Disconnect from PostgreSQL database
   */
  async disconnect(): Promise<void> {
    if (!this.pool) {
      return
    }

    try {
      await this.pool.end()
      this.pool = null
      log.info('Disconnected from PostgreSQL')
    } catch (error) {
      log.error('Failed to disconnect from PostgreSQL', { error })
      throw new Error(`Failed to disconnect from PostgreSQL: ${(error as Error).message}`)
    }
  }

  /**
   * Check if connected to PostgreSQL database
   */
  isConnected(): boolean {
    return this.pool !== null
  }

  /**
   * Execute a query and return results
   */
  async query(sql: string, params?: any[]): Promise<QueryResult> {
    if (!this.pool) {
      throw new Error('Not connected to PostgreSQL. Call connect() first.')
    }

    // Quote unquoted identifiers to preserve case for SSMA-migrated columns
    const preparedSql = prepareSql(sql)
    const sqlPreview = preparedSql.substring(0, 100)
    const paramCount = params?.length ?? 0

    try {
      const { result: queryResult } = await trackDuration(
        async () => {
          const result = await this.pool!.query(preparedSql, params)

          // Extract column names from fields
          const columns = result.fields ? result.fields.map((field) => field.name) : []

          // Result rows
          const rows = (result.rows as Record<string, unknown>[]) || []
          const rowCount = result.rowCount ?? rows.length

          return { rows, columns, rowCount }
        },
        { operationName: 'PostgreSqlService.query' }
      )

      log.debug('Query executed', { sqlPreview, rowCount: queryResult.rowCount, paramCount })
      return queryResult
    } catch (error) {
      log.error('PostgreSQL query failed', { sqlPreview, paramCount, error })
      throw new Error(`PostgreSQL query failed: ${(error as Error).message}`)
    }
  }

  /**
   * Execute multiple queries in a transaction
   */
  async transaction(queries: { sql: string; params?: any[] }[]): Promise<void> {
    if (!this.pool) {
      throw new Error('Not connected to PostgreSQL. Call connect() first.')
    }

    const queryCount = queries.length
    log.info('Transaction started', { queryCount })

    const client = await this.pool.connect()

    try {
      await client.query('BEGIN')

      for (let i = 0; i < queries.length; i++) {
        const { sql, params } = queries[i]
        const preparedSql = prepareSql(sql)
        await client.query(preparedSql, params)
        log.debug('Transaction query executed', {
          index: i,
          sqlPreview: preparedSql.substring(0, 100)
        })
      }

      await client.query('COMMIT')
      log.info('Transaction committed', { queryCount })
    } catch (error) {
      await client.query('ROLLBACK')
      log.warn('Transaction rolled back', { queryCount, error })
      throw new Error(`PostgreSQL transaction failed: ${(error as Error).message}`)
    } finally {
      client.release()
    }
  }
}