import type {
  WorkReceipt,
  WorkReceiptBatchClaimCommand,
  WorkReceiptBatchCompleteCommand,
  WorkReceiptBatchFailCommand,
  WorkReceiptBatchHeartbeatCommand,
  WorkReceiptClaim,
  WorkReceiptCommand,
  WorkReceiptFailureKind,
  WorkReceiptGetCommand,
} from './work-receipts';
import type { ToolExecutionFailureV1 } from '../tool-execution-error';

export type WorkReceiptBatchGetCommand = {
  playName: string;
  keys: string[];
};

export type WorkReceiptCompleteCommand = WorkReceiptCommand & {
  output: unknown;
};

export type WorkReceiptFailCommand = WorkReceiptCommand & {
  error: string;
  errorPayload?: ToolExecutionFailureV1 | null;
  failureKind?: WorkReceiptFailureKind | null;
};

export type WorkReceiptReleaseCommand = WorkReceiptCommand;

export type WorkReceiptSkipCommand = WorkReceiptCommand & {
  output?: unknown | null;
};

export type ProviderRequestStatus =
  | 'pending'
  | 'running'
  | 'completed'
  | 'failed';

export type ProviderRequestRecord = {
  key: string;
  status: ProviderRequestStatus;
  receiptKey: string;
  providerIdempotencyKey: string;
  runId: string;
  provider: string;
  operation: string;
  method: string;
  endpoint: string;
  bodyHash: string;
  output?: unknown;
  error?: string | null;
  createdAt?: string | null;
  updatedAt?: string | null;
};

export type ProviderRequestClaim =
  | {
      disposition: 'claimed';
      record: ProviderRequestRecord;
    }
  | {
      disposition: 'reused';
      record: ProviderRequestRecord;
    }
  | {
      disposition: 'running';
      record: ProviderRequestRecord;
    }
  | {
      disposition: 'failed';
      record: ProviderRequestRecord;
    };

export type ProviderRequestClaimCommand = {
  key: string;
  receiptKey: string;
  providerIdempotencyKey: string;
  runId: string;
  provider: string;
  operation: string;
  method: string;
  endpoint: string;
  bodyHash: string;
};

export type ProviderRequestCompleteCommand = {
  key: string;
  runId: string;
  output?: unknown;
};

export type ProviderRequestFailCommand = {
  key: string;
  runId: string;
  error: string;
};

export interface ExecutionLedgerStore {
  getReceipt(input: WorkReceiptGetCommand): Promise<WorkReceipt | null>;
  getReceipts(
    input: WorkReceiptBatchGetCommand,
  ): Promise<Array<WorkReceipt | null>>;
  claimReceipt(input: WorkReceiptCommand): Promise<WorkReceiptClaim>;
  claimReceipts(
    input: WorkReceiptBatchClaimCommand,
  ): Promise<WorkReceiptClaim[]>;
  completeReceipt(
    input: WorkReceiptCompleteCommand,
  ): Promise<WorkReceipt | null>;
  completeReceipts(
    input: WorkReceiptBatchCompleteCommand,
  ): Promise<Array<WorkReceipt | null>>;
  failReceipt(input: WorkReceiptFailCommand): Promise<WorkReceipt | null>;
  failReceipts(
    input: WorkReceiptBatchFailCommand,
  ): Promise<Array<WorkReceipt | null>>;
  releaseReceipt(input: WorkReceiptReleaseCommand): Promise<WorkReceipt | null>;
  skipReceipt(input: WorkReceiptSkipCommand): Promise<WorkReceipt | null>;
  heartbeatReceipts(
    input: WorkReceiptBatchHeartbeatCommand,
  ): Promise<Array<WorkReceipt | null>>;
  claimProviderRequest(
    input: ProviderRequestClaimCommand,
  ): Promise<ProviderRequestClaim>;
  completeProviderRequest(
    input: ProviderRequestCompleteCommand,
  ): Promise<ProviderRequestRecord | null>;
  failProviderRequest(
    input: ProviderRequestFailCommand,
  ): Promise<ProviderRequestRecord | null>;
}