import type { MultiTrackType } from '@/types/multitrack'

export interface SegmentTrackPresentation {
  backgroundColor: string
  backgroundColorStrong: string
  borderColor: string
  textColor: string
  textClassName: string
  titleBackgroundColor: string | null
  waveformColor: string | null
  showThumbnail: boolean
  showWaveform: boolean
  waveformSource: 'decoded-media' | null
}

const VIDEO_PRESENTATION: SegmentTrackPresentation = {
  backgroundColor: 'var(--multitrack-video-bg)',
  backgroundColorStrong: 'var(--multitrack-video-bg-strong)',
  borderColor: 'var(--multitrack-video-border)',
  textColor: 'var(--multitrack-video-text)',
  textClassName: 'text-[10px]',
  titleBackgroundColor: null,
  waveformColor: 'var(--multitrack-video-waveform)',
  showThumbnail: true,
  showWaveform: true,
  waveformSource: 'decoded-media',
}

const AUDIO_PRESENTATION: SegmentTrackPresentation = {
  backgroundColor: 'var(--multitrack-audio-bg)',
  backgroundColorStrong: 'var(--multitrack-audio-bg-strong)',
  borderColor: 'var(--multitrack-audio-border)',
  textColor: 'var(--multitrack-audio-text)',
  textClassName: 'text-[10px]',
  titleBackgroundColor: 'var(--multitrack-audio-title-bg)',
  waveformColor: 'var(--multitrack-audio-waveform)',
  showThumbnail: false,
  showWaveform: true,
  waveformSource: 'decoded-media',
}

const TASK_PRESENTATION: SegmentTrackPresentation = {
  backgroundColor: 'var(--multitrack-task-bg)',
  backgroundColorStrong: 'var(--multitrack-task-bg-strong)',
  borderColor: 'var(--multitrack-task-border)',
  textColor: 'var(--multitrack-task-text)',
  textClassName: 'text-[8px]',
  titleBackgroundColor: null,
  waveformColor: null,
  showThumbnail: false,
  showWaveform: false,
  waveformSource: null,
}

const SUBTITLE_PRESENTATION: SegmentTrackPresentation = {
  backgroundColor: 'var(--multitrack-subtitle-bg)',
  backgroundColorStrong: 'var(--multitrack-subtitle-bg-strong)',
  borderColor: 'var(--multitrack-subtitle-border)',
  textColor: 'var(--multitrack-subtitle-text)',
  textClassName: 'text-[8px]',
  titleBackgroundColor: null,
  waveformColor: null,
  showThumbnail: false,
  showWaveform: false,
  waveformSource: null,
}

export function getSegmentTrackPresentation(trackType: MultiTrackType): SegmentTrackPresentation {
  if (trackType === 'video') return VIDEO_PRESENTATION
  if (trackType === 'audio') return AUDIO_PRESENTATION
  if (trackType === 'subtitle') return SUBTITLE_PRESENTATION
  return TASK_PRESENTATION
}