import type { DurationEntryData } from "./types.ts";

export const DURATION_ENTRY_TYPE = "compact-thinking-duration";

export function restoreDurationEntries(
  entries: Array<{
    type: string;
    customType?: string;
    data?: unknown;
  }>,
  completedDurations: Map<number, Map<number, number>>,
) {
  completedDurations.clear();
  for (const entry of entries) {
    if (
      entry.type !== "custom" ||
      entry.customType !== DURATION_ENTRY_TYPE ||
      !entry.data ||
      typeof entry.data !== "object"
    ) {
      continue;
    }

    const data = entry.data as Partial<DurationEntryData>;
    if (
      typeof data.messageTimestamp !== "number" ||
      !Number.isFinite(data.messageTimestamp) ||
      typeof data.contentIndex !== "number" ||
      !Number.isInteger(data.contentIndex) ||
      typeof data.durationMs !== "number" ||
      !Number.isFinite(data.durationMs) ||
      data.durationMs < 1
    ) {
      continue;
    }

    let durations = completedDurations.get(data.messageTimestamp);
    if (!durations) {
      durations = new Map();
      completedDurations.set(data.messageTimestamp, durations);
    }
    durations.set(data.contentIndex, data.durationMs);
  }
}