import type {
  ExternalGenerationJobStatusRecord,
  ExternalGenerationQueueOverview,
} from '../../../packages/shared/src/contracts/externalGeneration';
import type { ExternalGenerationQueueStatus } from '../CustomWorldGenerationView';

export function buildExternalGenerationQueueStatus(
  overview: ExternalGenerationQueueOverview | null,
  job: ExternalGenerationJobStatusRecord | null,
): ExternalGenerationQueueStatus | null {
  if (!overview && !job) {
    return null;
  }

  return {
    currentStatus: job?.status ?? null,
    currentProgress: job?.progress ?? null,
    pendingCount: overview?.pendingCount ?? null,
    runningCount: overview?.runningCount ?? null,
  };
}