import { type Pool, type PoolConfig } from 'pg';
import { CockroachDatabase } from "../cockroach-core/db.js";
import { CockroachDialect } from "../cockroach-core/dialect.js";
import { entityKind } from "../entity.js";
import type { Logger } from "../logger.js";
import { type RelationalSchemaConfig, type TablesRelationalConfig } from "../relations.js";
import { type DrizzleConfig } from "../utils.js";
import type { NodeCockroachClient, NodeCockroachQueryResultHKT } from "./session.js";
import { NodeCockroachSession } from "./session.js";
export interface CockroachDriverOptions {
    logger?: Logger;
}
export declare class NodeCockroachDriver {
    private client;
    private dialect;
    private options;
    static readonly [entityKind]: string;
    constructor(client: NodeCockroachClient, dialect: CockroachDialect, options?: CockroachDriverOptions);
    createSession(schema: RelationalSchemaConfig<TablesRelationalConfig> | undefined): NodeCockroachSession<Record<string, unknown>, TablesRelationalConfig>;
}
export declare class NodeCockroachDatabase<TSchema extends Record<string, unknown> = Record<string, never>> extends CockroachDatabase<NodeCockroachQueryResultHKT, TSchema> {
    static readonly [entityKind]: string;
}
export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>, TClient extends NodeCockroachClient = Pool>(...params: [
    TClient | string
] | [
    TClient | string,
    DrizzleConfig<TSchema>
] | [
    (DrizzleConfig<TSchema> & ({
        connection: string | PoolConfig;
    } | {
        client: TClient;
    }))
]): NodeCockroachDatabase<TSchema> & {
    $client: TClient;
};
export declare namespace drizzle {
    function mock<TSchema extends Record<string, unknown> = Record<string, never>>(config?: DrizzleConfig<TSchema>): NodeCockroachDatabase<TSchema> & {
        $client: '$client is not available on drizzle.mock()';
    };
}
