import { Stream, Writable } from 'stream';

import { RecordResult } from './record-result';
import { Record } from './record';
import { Parsable } from './record-stream';

export interface BatchInfo {
    id: string;
    jobId: string;
    state: string;
    stateMessage: string;
    numberRecordsProcessed: string;
    numberRecordsFailed: string;
    totalProcessingTime: string;
}

export interface BatchResultInfo {
    id: string;
    batchId?: string | undefined;
    jobId?: string | undefined;
    success?: boolean | undefined;
    errors?: string[] | undefined;
}

export class Batch extends Writable {
    check(callback?: (batchInfo: BatchInfo) => void): Promise<BatchInfo>;
    execute(input?: Record[] | Stream | string, callback?: (err: Error, result: RecordResult[] | BatchResultInfo[]) => void): Batch;
    poll(interval: number, timeout: number): void;
    retrieve(callback?: (batchInfo: BatchInfo) => void): Promise<RecordResult[] | BatchResultInfo[]>;
    result(resultId: string): Parsable<any>;
    then(): Promise<any>;
    thenAll(callback: (data: any) => void): void;
}