import type { QuerySyntax, StatisticsMode, TransactionMode } from '../store/query';
interface Position {
    row?: number;
    column?: number;
    file?: string;
}
/** source: https://github.com/ydb-platform/ydb/blob/main/ydb/public/api/protos/ydb_issue_message.proto */
export interface IssueMessage {
    position?: Position;
    end_position?: Position;
    message?: string;
    issue_code?: number;
    severity?: number;
    issues?: IssueMessage[];
}
/** incomplete */
interface TDqStageStats {
}
/** incomplete */
interface TKqpStatsCompile {
}
interface TDqTableStats {
    TablePath?: string;
    /** uint64 */
    ReadRows?: string | number;
    /** uint64 */
    ReadBytes?: string | number;
    /** uint64 */
    WriteRows?: string | number;
    /** uint64 */
    WriteBytes?: string | number;
    /** uint64 */
    EraseRows?: string | number;
    /** uint64 */
    EraseBytes?: string | number;
    AffectedPartitions?: number;
    Extra?: unknown;
}
/** source: https://github.com/ydb-platform/ydb/blob/main/ydb/library/yql/dq/actors/protos/dq_stats.proto */
interface TDqExecutionStats {
    /** uint64 */
    CpuTimeUs?: string | number;
    /** uint64 */
    DurationUs?: string | number;
    /** uint64 */
    ResultRows?: string | number;
    /** uint64 */
    ResultBytes?: string | number;
    Tables?: TDqTableStats[];
    /** uint64 */
    ExecuterCpuTimeUs?: string | number;
    /** uint64 */
    StartTimeMs?: string | number;
    /** uint64 */
    FinishTimeMs?: string | number;
    /** uint64 */
    FirstRowTimeMs?: string | number;
    Stages?: TDqStageStats[];
    TxPlansWithStats?: string[];
    Extra: unknown;
}
/** source: https://github.com/ydb-platform/ydb/blob/main/ydb/core/protos/kqp_stats.proto */
export interface TKqpStatsQuery {
    /** uint64 */
    DurationUs?: string | number;
    Compilation?: TKqpStatsCompile;
    /** uint64 */
    WorkerCpuTimeUs?: string | number;
    /** uint64 */
    ReadSetsCount?: string | number;
    /** uint64 */
    MaxShardProgramSize?: string | number;
    /** uint64 */
    MaxShardReplySize?: string | number;
    Executions?: TDqExecutionStats[];
}
export interface PlanMeta {
    version: string;
    type: 'script' | 'query';
}
export interface PlanTable {
    name: string;
    reads?: {
        type: string;
        lookup_by?: string[];
        scan_by?: string[];
        limit?: string;
        reverse?: boolean;
        columns?: string[];
    }[];
    writes?: {
        type: string;
        key?: string[];
        columns?: string[];
    }[];
}
interface PlanNodeStats {
    TotalCpuTimeUs?: number;
    TotalTasks?: number;
    TotalInputBytes?: number;
    TotalInputRows?: number;
    TotalOutputBytes?: number;
    TotalDurationMs?: number;
    TotalOutputRows?: number;
    ComputeNodes?: unknown[];
    NodesScanShards?: unknown[];
    UseLlvm?: unknown;
}
interface PlanNodeOperator {
    Name: string;
    Limit?: string;
    ReadLimit?: string;
    ReadColumns?: string[];
    ReadRanges?: string[];
    ReadRange?: string[];
    Table?: string;
    Iterator?: string;
}
interface CommonPlanNode {
    PlanNodeId?: number;
    'Node Type'?: string;
    PlanNodeType?: string;
}
export interface PlanNode extends CommonPlanNode {
    Operators?: PlanNodeOperator[];
    Tables?: string[];
    Plans?: PlanNode[];
    Stats?: PlanNodeStats;
    'CTE Name'?: string;
    'Subplan Name'?: string;
    'Parent Relationship'?: string;
}
export interface SimlifiedPlanOperatorOtherParams {
    [key: string]: string | number | string[] | Record<string, unknown> | undefined;
}
interface SimlifiedPlanOperator extends SimlifiedPlanOperatorOtherParams {
    'E-Rows'?: string;
    'A-Rows'?: number;
    'A-Cpu'?: number;
    Name: string;
    'E-Size'?: string;
    'E-Cost'?: string;
}
export interface SimplifiedNode extends CommonPlanNode {
    Plans?: SimplifiedNode[];
    Operators?: [SimlifiedPlanOperator];
}
export interface ScriptPlan {
    queries?: {
        Plan?: PlanNode;
        SimplifiedPlan?: SimplifiedNode;
        tables?: PlanTable[];
    }[];
    meta: PlanMeta;
}
export interface QueryPlan {
    Plan?: PlanNode;
    SimplifiedPlan?: SimplifiedNode;
    tables?: PlanTable[];
    meta: PlanMeta;
}
export type CellValue = string | number | null | undefined;
export type KeyValueRow<T = CellValue> = {
    [key: string]: T;
};
export type ArrayRow<T = CellValue> = Array<T>;
export interface ColumnType {
    name: string;
    type: string;
}
export declare const TracingLevelNumber: {
    off: number;
    toplevel: number;
    basic: number;
    detailed: number;
    diagnostic: number;
    trace: number;
};
/** undefined = 'none' */
export type Stats = StatisticsMode;
/** undefined = '60000' */
export type Timeout = number;
/** undefined = '15' */
export type TracingLevel = number;
/** undefined = 'execute' */
export type ExecuteActions = 'execute' | 'execute-scan' | 'execute-script' | 'execute-data' | 'execute-query' | 'execute-query-and-forget' | undefined;
export type ExplainActions = 'explain' | 'explain-scan' | 'explain-script' | 'explain-data' | 'explain-query' | 'explain-ast';
export type CancelActions = 'cancel-query';
export type Actions = ExecuteActions | ExplainActions | CancelActions;
export interface ErrorResponse {
    error?: IssueMessage;
    issues?: IssueMessage[] | null;
}
/**
 * meta.type = 'script'
 *
 * explain-script
 */
export interface ExplainScriptResponse {
    plan?: ScriptPlan;
}
/**
 * meta.type = 'query'
 *
 * explain, explain-scan, explain-data, explain-query, explain-ast
 */
export interface ExplainQueryResponse {
    ast?: string;
    plan?: QueryPlan;
}
export type GenericExplainResponse<Action extends ExplainActions> = Action extends 'explain-script' ? ExplainScriptResponse : ExplainQueryResponse;
type SchemaResult = {
    rows?: ArrayRow[] | null;
    columns?: ColumnType[];
    truncated?: boolean;
}[];
/**
 * meta.type = 'query'
 *
 * execute-scan, execute-data, execute-query
 */
export type ExecuteQueryResponse = {
    plan?: QueryPlan;
    ast?: string;
    stats?: TKqpStatsQuery;
    result?: SchemaResult;
};
/**
 * meta.type = 'script'
 *
 * execute, execute-script
 */
export type ExecuteScriptResponse = {
    plan?: ScriptPlan;
    ast?: string;
    stats?: TKqpStatsQuery;
    result?: SchemaResult;
};
export type ExecuteQueryAndForgetResponse = {
    version?: number;
    status?: string;
    message?: string;
};
export type GenericExecuteResponse<Action extends ExecuteActions> = Action extends 'execute-scan' | 'execute-data' | 'execute-query' ? ExecuteQueryResponse : Action extends 'execute-query-and-forget' ? ExecuteQueryAndForgetResponse : ExecuteScriptResponse;
export type CancelResponse = {
    stats?: TKqpStatsQuery;
};
export interface SendQueryParams<Action extends Actions> {
    query?: string;
    database?: string;
    action?: Action;
    syntax?: QuerySyntax;
    stats?: Stats;
    tracingLevel?: TracingLevel;
    transaction_mode?: TransactionMode;
    timeout?: Timeout;
    query_id?: string;
    limit_rows?: number;
    internal_call?: boolean;
    base64?: boolean;
    resource_pool?: string;
    'forget-after'?: number;
}
export interface StreamQueryParams<Action extends Actions> extends SendQueryParams<Action> {
    output_chunk_max_size?: number;
    concurrent_results?: boolean;
}
export type QueryAPIResponseByAction<Action extends Actions> = Action extends ExplainActions ? GenericExplainResponse<Action> : Action extends ExecuteActions ? GenericExecuteResponse<Action> : Action extends CancelActions ? CancelResponse : never;
type QueryAPIResponseMeta = {
    _meta?: {
        traceId?: string;
    };
};
export type QueryAPIResponse<Action extends Actions> = QueryAPIResponseByAction<Action> & QueryAPIResponseMeta;
export type ExplainResponse = ExplainQueryResponse | ExplainScriptResponse;
export type ExecuteResponse = ExecuteQueryResponse | ExecuteScriptResponse;
export {};