import type { AnyBatchOperationStrategy } from './batching-types';
import { bettercontactBatchStrategies } from './bettercontact-batching';
import { DEFAULT_PLAY_RUNTIME_BATCH_STRATEGIES } from './default-batch-strategies';
import { fullenrichBatchStrategies } from './fullenrich-batching';
import { opensosdataBatchStrategies } from './opensosdata-batching';

export const PLAY_RUNTIME_BATCH_OPERATION_REGISTRY: Record<
  string,
  AnyBatchOperationStrategy
> = {
  ...DEFAULT_PLAY_RUNTIME_BATCH_STRATEGIES,
  ...bettercontactBatchStrategies,
  ...fullenrichBatchStrategies,
  ...opensosdataBatchStrategies,
};

export function getPlayRuntimeBatchStrategy(
  operation: string | null | undefined,
): AnyBatchOperationStrategy | null {
  if (!operation) {
    return null;
  }
  return PLAY_RUNTIME_BATCH_OPERATION_REGISTRY[operation] ?? null;
}