ktx/packages/context/src/connections/query-executor.ts

import type { KtxProjectConnectionConfig } from '../project/index.js';

export interface KtxSqlQueryExecutionInput {
  connectionId: string;
  projectDir?: string;
  connection: KtxProjectConnectionConfig | undefined;
  sql: string;
  maxRows?: number;
}

export interface KtxSqlQueryExecutionResult {
  headers: string[];
  rows: unknown[][];
  totalRows: number;
  command: string;
  rowCount: number | null;
}

export interface KtxSqlQueryExecutorPort {
  execute(input: KtxSqlQueryExecutionInput): Promise<KtxSqlQueryExecutionResult>;
}

export function normalizeQueryRows(rows: unknown[]): unknown[][] {
  return rows.map((row) => (Array.isArray(row) ? row : Object.values(row as Record<string, unknown>)));
}