import type {
  MiniGameDraftGenerationKind,
  MiniGameDraftGenerationState,
} from '../../services/miniGameDraftGenerationProgress';
import type { SelectionStage } from './platformEntryTypes';

type MiniGameGenerationProgressTickStateMap = Partial<
  Record<MiniGameDraftGenerationKind, MiniGameDraftGenerationState | null>
>;

export function resolveMiniGameGenerationProgressTickState(
  selectionStage: SelectionStage,
  states: MiniGameGenerationProgressTickStateMap,
) {
  const stageKindMap: Partial<
    Record<SelectionStage, MiniGameDraftGenerationKind>
  > = {
    'puzzle-generating': 'puzzle',
    'big-fish-generating': 'big-fish',
    'square-hole-generating': 'square-hole',
    'match3d-generating': 'match3d',
    'baby-object-match-generating': 'baby-object-match',
    'jump-hop-generating': 'jump-hop',
    'puzzle-clear-generating': 'puzzle-clear',
    'wooden-fish-generating': 'wooden-fish',
  };
  const kind = stageKindMap[selectionStage];

  return kind ? (states[kind] ?? null) : null;
}