import type { Readable } from "node:stream";

export type VoiceCaptureEntry = {
  generation: number;
  stream: Readable;
};

export type VoiceCaptureFinalizeTimer = {
  generation: number;
  timer: ReturnType<typeof setTimeout>;
};

export type VoiceCaptureState = {
  activeSpeakers: Set<string>;
  activeCaptureStreams: Map<string, VoiceCaptureEntry>;
  captureFinalizeTimers: Map<string, VoiceCaptureFinalizeTimer>;
  captureGenerations: Map<string, number>;
};

export function createVoiceCaptureState(): VoiceCaptureState {
  return {
    activeSpeakers: new Set(),
    activeCaptureStreams: new Map(),
    captureFinalizeTimers: new Map(),
    captureGenerations: new Map(),
  };
}

export function stopVoiceCaptureState(state: VoiceCaptureState): void {
  for (const { timer } of state.captureFinalizeTimers.values()) {
    clearTimeout(timer);
  }
  state.captureFinalizeTimers.clear();
  for (const { stream } of state.activeCaptureStreams.values()) {
    stream.destroy();
  }
  state.activeCaptureStreams.clear();
  state.captureGenerations.clear();
  state.activeSpeakers.clear();
}

export function getActiveVoiceCapture(
  state: VoiceCaptureState,
  userId: string,
): VoiceCaptureEntry | undefined {
  return state.activeCaptureStreams.get(userId);
}

export function isVoiceCaptureActive(state: VoiceCaptureState, userId: string): boolean {
  return state.activeSpeakers.has(userId);
}

export function clearVoiceCaptureFinalizeTimer(
  state: VoiceCaptureState,
  userId: string,
  generation?: number,
): boolean {
  const scheduled = state.captureFinalizeTimers.get(userId);
  if (!scheduled || (generation !== undefined && scheduled.generation !== generation)) {
    return false;
  }
  clearTimeout(scheduled.timer);
  state.captureFinalizeTimers.delete(userId);
  return true;
}

export function beginVoiceCapture(
  state: VoiceCaptureState,
  userId: string,
  stream: Readable,
): number {
  const generation = (state.captureGenerations.get(userId) ?? 0) + 1;
  state.captureGenerations.set(userId, generation);
  state.activeSpeakers.add(userId);
  state.activeCaptureStreams.set(userId, { generation, stream });
  clearVoiceCaptureFinalizeTimer(state, userId, generation);
  return generation;
}

export function finishVoiceCapture(
  state: VoiceCaptureState,
  userId: string,
  generation: number,
): boolean {
  clearVoiceCaptureFinalizeTimer(state, userId, generation);
  const activeCapture = state.activeCaptureStreams.get(userId);
  if (activeCapture?.generation !== generation) {
    return false;
  }
  state.activeCaptureStreams.delete(userId);
  state.activeSpeakers.delete(userId);
  return true;
}

export function scheduleVoiceCaptureFinalize(params: {
  state: VoiceCaptureState;
  userId: string;
  delayMs: number;
  onFinalize?: (capture: VoiceCaptureEntry) => void;
}): boolean {
  const { state, userId, delayMs, onFinalize } = params;
  const capture = state.activeCaptureStreams.get(userId);
  if (!capture) {
    return false;
  }
  clearVoiceCaptureFinalizeTimer(state, userId, capture.generation);
  const timer = setTimeout(() => {
    const activeCapture = state.activeCaptureStreams.get(userId);
    if (!activeCapture || activeCapture.generation !== capture.generation) {
      return;
    }
    state.captureFinalizeTimers.delete(userId);
    state.activeCaptureStreams.delete(userId);
    state.activeSpeakers.delete(userId);
    onFinalize?.(activeCapture);
    activeCapture.stream.destroy();
  }, delayMs);
  state.captureFinalizeTimers.set(userId, { generation: capture.generation, timer });
  return true;
}
