import { type Knex } from "knex";
import { type DatabaseSchemaExtend } from "../types/types";
import { type DBPreset } from "./db";
import { Puri } from "./puri";
import { type ColumnKeys, type IdType, type OmitInternalTypeKeys, type PuriTable } from "./puri.types";
import { type InsertOnlyOptions, type UBRef, type UpsertBuilder, type UpsertOptions } from "./upsert-builder";
type TableName<TSchema extends DatabaseSchemaExtend> = Extract<keyof TSchema, string>;
export type TransactionalOptions = {
    isolation?: Exclude<Knex.IsolationLevels, "snapshot">;
    dbPreset?: DBPreset;
    readOnly?: boolean;
};
export declare class PuriWrapper<TSchema extends DatabaseSchemaExtend = DatabaseSchemaExtend> {
    knex: Knex;
    upsertBuilder: UpsertBuilder;
    constructor(knex: Knex, upsertBuilder: UpsertBuilder);
    raw(sql: string): Knex.Raw;
    from<TTable extends keyof TSchema>(tableName: TTable): Puri<TSchema, Record<TTable, PuriTable<TSchema[TTable]>>, OmitInternalTypeKeys<PuriTable<TSchema[TTable]>>>;
    from<TTable extends keyof TSchema, TAlias extends string>(spec: {
        [K in TAlias]: TTable;
    }): Puri<TSchema, Record<TAlias, PuriTable<TSchema[TTable]>>, OmitInternalTypeKeys<PuriTable<TSchema[TTable]>>>;
    from<TAlias extends string, TSubResult>(spec: {
        [K in TAlias]: Puri<TSchema, any, TSubResult>;
    }): Puri<TSchema, Record<TAlias, PuriTable<TSubResult>>, OmitInternalTypeKeys<PuriTable<TSubResult>>>;
    table<TTable extends keyof TSchema>(tableName: TTable): Puri<TSchema, Record<TTable, PuriTable<TSchema[TTable]>>, OmitInternalTypeKeys<PuriTable<TSchema[TTable]>>>;
    table<TTable extends keyof TSchema, TAlias extends string>(spec: {
        [K in TAlias]: TTable;
    }): Puri<TSchema, Record<TAlias, PuriTable<TSchema[TTable]>>, OmitInternalTypeKeys<PuriTable<TSchema[TTable]>>>;
    table<TAlias extends string, TSubResult>(spec: {
        [K in TAlias]: Puri<TSchema, any, TSubResult>;
    }): Puri<TSchema, Record<TAlias, PuriTable<TSubResult>>, OmitInternalTypeKeys<PuriTable<TSubResult>>>;
    transaction<T>(callback: (trx: PuriTransactionWrapper) => Promise<T>, options?: TransactionalOptions): Promise<T>;
    ubRegister<TTable extends TableName<TSchema>>(tableName: TTable, row: Partial<{
        [K in ColumnKeys<TSchema[TTable]>]: TSchema[TTable][K] | UBRef;
    }>): UBRef;
    ubUpsert<TTable extends TableName<TSchema> & keyof DatabaseSchemaExtend>(tableName: TTable, options?: UpsertOptions<TTable>): Promise<IdType<DatabaseSchemaExtend, TTable>[]>;
    ubInsertOnly<TTable extends TableName<TSchema> & keyof DatabaseSchemaExtend>(tableName: TTable, options?: InsertOnlyOptions): Promise<IdType<DatabaseSchemaExtend, TTable>[]>;
    ubUpsertOrInsert<TTable extends TableName<TSchema> & keyof DatabaseSchemaExtend>(tableName: TTable, mode: "upsert" | "insert", options?: UpsertOptions<TTable>): Promise<IdType<DatabaseSchemaExtend, TTable>[]>;
    ubUpdateBatch(tableName: TableName<TSchema>, options?: {
        chunkSize?: number;
        where?: string | string[];
    }): Promise<void>;
    debugTransaction(): Promise<void>;
    private getTransactionInfo;
}
export declare class PuriTransactionWrapper extends PuriWrapper {
    trx: Knex.Transaction;
    upsertBuilder: UpsertBuilder;
    constructor(trx: Knex.Transaction, upsertBuilder: UpsertBuilder);
    rollback(): Promise<void>;
    commit(): Promise<void>;
}
export {};
//# sourceMappingURL=puri-wrapper.d.ts.map