import { Condition } from '../conditions/index.js';
import { EntityKeyInput, DynamoEntity } from '../core/entity.js';
import { ObjectLikeZodType, EntitySchema } from '../core/index.js';
import { Select } from '@aws-sdk/client-dynamodb';
import { BaseConfig, BaseResult, BaseCommand, BasePaginatable } from './index.js';
import { QueryCommandInput, NativeAttributeValue, QueryCommandOutput } from '@aws-sdk/lib-dynamodb';

/**
 * Configuration for the Query command.
 *
 * @template Schema - The Zod schema defining the structure of the entity.
 */
type QueryConfig<Schema extends ObjectLikeZodType> = BaseConfig & EntityKeyInput<EntitySchema<Schema>> & {
    sortKeyCondition?: Condition;
    filter?: Condition;
    limit?: number;
    selectAttributes?: Select;
    consistent?: boolean;
    validationConcurrency?: number;
    reverseIndexScan?: boolean;
    exclusiveStartKey?: Partial<EntitySchema<Schema>>;
    pageSize?: number;
};
/**
 * Result of the Query command.
 *
 * @template Schema - The Zod schema defining the structure of the entity.
 */
type QueryResult<Schema extends ObjectLikeZodType> = BaseResult & {
    items: EntitySchema<Schema>[];
    count: number;
    scannedCount: number;
    lastEvaluatedKey?: Partial<EntitySchema<Schema>> | undefined;
};
/**
 * Command to retrieve multiple items by partition key with optional sort key conditions.
 *
 * @template Schema - The Zod schema defining the structure of the entity.
 *
 * @example
 * ```typescript
 * import { DynamoTable, DynamoEntity, key, Query, beginsWith } from 'dynamo-document-builder';
 *
 * const table = new DynamoTable({
 *   tableName: 'ExampleTable',
 *   documentClient,
 * });
 *
 * const todoEntity = new DynamoEntity({
 *   table,
 *   schema: z.object({
 *     userId: z.string(),
 *     todoId: z.string(),
 *     title: z.string(),
 *     isComplete: z.boolean(),
 *   }),
 *   partitionKey: todo => key('USER', todo.userId),
 *   sortKey: todo => key('TODO', todo.todoId),
 * });
 *
 * const queryCommand = new Query({
 *   key: { userId: 'user123' },
 *   sortKeyCondition: { SK: beginsWith('TODO#') },
 *   limit: 10,
 * });
 *
 * const { items, count } = await todoEntity.send(queryCommand);
 * ```
 */
declare class Query<Schema extends ObjectLikeZodType> implements BaseCommand<QueryResult<Schema>, Schema>, BasePaginatable<QueryResult<Schema>, Schema> {
    #private;
    constructor(config: QueryConfig<Schema>);
    buildCommandInput(entity: DynamoEntity<Schema>): QueryCommandInput;
    validateItems(entity: DynamoEntity<Schema>, items: Record<string, NativeAttributeValue>[] | undefined): Promise<EntitySchema<Schema>[]>;
    buildResult(items: EntitySchema<Schema>[], queryResult: QueryCommandOutput): QueryResult<Schema>;
    execute(entity: DynamoEntity<Schema>): Promise<QueryResult<Schema>>;
    executePaginated(entity: DynamoEntity<Schema>): AsyncGenerator<QueryResult<Schema>, void, unknown>;
}

export { Query };
export type { QueryConfig, QueryResult };