yt-dlp 2.7.2

use std::cmp::Ordering;

use ordered_float::OrderedFloat;

use crate::Downloader;
use crate::model::Video;
use crate::model::format::{Format, FormatType};
use crate::model::selector::{
    AudioCodecPreference, AudioQuality, StoryboardQuality, ThumbnailQuality, VideoCodecPreference, VideoQuality,
    matches_audio_codec, matches_video_codec,
};
use crate::model::thumbnail::Thumbnail;

/// Trait for selecting video, audio, and storyboard formats from a Video.
pub trait VideoSelection {
    /// Returns the best video format available.
    ///
    /// Sorting criteria: quality → resolution height → fps → video bitrate (all descending).
    ///
    /// # Returns
    ///
    /// The highest-ranked video-only format, or `None` if no video formats exist.
    fn best_video_format(&self) -> Option<&Format>;

    /// Returns the best audio format available.
    ///
    /// Sorting criteria: quality → audio bitrate → sample rate → audio channels (all descending).
    ///
    /// # Returns
    ///
    /// The highest-ranked audio-only format, or `None` if no audio formats exist.
    fn best_audio_format(&self) -> Option<&Format>;

    /// Returns the worst video format available.
    ///
    /// Uses the same sorting criteria as [`best_video_format`] but returns the lowest-ranked.
    ///
    /// # Returns
    ///
    /// The lowest-ranked video-only format, or `None` if no video formats exist.
    fn worst_video_format(&self) -> Option<&Format>;

    /// Returns the worst audio format available.
    ///
    /// Uses the same sorting criteria as [`best_audio_format`] but returns the lowest-ranked.
    ///
    /// # Returns
    ///
    /// The lowest-ranked audio-only format, or `None` if no audio formats exist.
    fn worst_audio_format(&self) -> Option<&Format>;

    /// Compares two video formats for ordering.
    ///
    /// Sorting criteria: quality → resolution height → fps → video bitrate.
    ///
    /// # Arguments
    ///
    /// * `a` - First video format to compare.
    /// * `b` - Second video format to compare.
    ///
    /// # Returns
    ///
    /// An [`Ordering`] indicating how `a` and `b` compare.
    fn compare_video_formats(&self, a: &Format, b: &Format) -> Ordering;

    /// Compares two audio formats for ordering.
    ///
    /// Sorting criteria: quality → audio bitrate → sample rate → audio channels.
    ///
    /// # Arguments
    ///
    /// * `a` - First audio format to compare.
    /// * `b` - Second audio format to compare.
    ///
    /// # Returns
    ///
    /// An [`Ordering`] indicating how `a` and `b` compare.
    fn compare_audio_formats(&self, a: &Format, b: &Format) -> Ordering;

    /// Selects a video format based on quality and codec preferences.
    ///
    /// Filters by codec first (falling back to all formats if no codec matches),
    /// then selects based on the quality preset or custom target.
    ///
    /// # Arguments
    ///
    /// * `quality` - The desired video quality level.
    /// * `codec` - The preferred video codec.
    ///
    /// # Returns
    ///
    /// The best-matching video format, or `None` if no video formats exist.
    fn select_video_format(&self, quality: VideoQuality, codec: VideoCodecPreference) -> Option<&Format>;

    /// Selects an audio format based on quality and codec preferences.
    ///
    /// Filters by codec first (falling back to all formats if no codec matches),
    /// then selects based on the quality preset or custom bitrate target.
    ///
    /// # Arguments
    ///
    /// * `quality` - The desired audio quality level.
    /// * `codec` - The preferred audio codec.
    ///
    /// # Returns
    ///
    /// The best-matching audio format, or `None` if no audio formats exist.
    fn select_audio_format(&self, quality: AudioQuality, codec: AudioCodecPreference) -> Option<&Format>;

    /// Returns all storyboard formats, ordered from best to worst quality.
    ///
    /// Best quality = highest fragment count × largest resolution (width × height).
    ///
    /// # Returns
    ///
    /// A vector of storyboard format references in descending quality order.
    fn storyboard_formats(&self) -> Vec<&Format>;

    /// Returns the best storyboard format (most fragments, then highest resolution).
    ///
    /// # Returns
    ///
    /// The best storyboard format, or `None` if no storyboards are available.
    fn best_storyboard_format(&self) -> Option<&Format>;

    /// Returns the worst storyboard format (fewest fragments, then lowest resolution).
    ///
    /// # Returns
    ///
    /// The worst storyboard format, or `None` if no storyboards are available.
    fn worst_storyboard_format(&self) -> Option<&Format>;

    /// Selects a storyboard format based on quality preference.
    ///
    /// # Arguments
    ///
    /// * `quality` - The desired storyboard quality.
    ///
    /// # Returns
    ///
    /// The matching storyboard format, or `None` if no storyboards are available.
    fn select_storyboard_format(&self, quality: StoryboardQuality) -> Option<&Format>;

    /// Selects a thumbnail based on quality preference.
    ///
    /// # Arguments
    ///
    /// * `quality` - The desired thumbnail quality.
    ///
    /// # Returns
    ///
    /// The matching thumbnail, or `None` if no thumbnails are available.
    fn select_thumbnail(&self, quality: ThumbnailQuality) -> Option<&Thumbnail>;
}

impl VideoSelection for Video {
    /// Returns the best video format available.
    /// Formats sorting : "quality", "video resolution", "fps", "video bitrate"
    fn best_video_format(&self) -> Option<&Format> {
        tracing::debug!(
            video_id = %self.id,
            format_count = self.formats.len(),
            "🧩 Selecting best video format"
        );

        self.formats
            .iter()
            .filter(|f| f.is_video())
            .max_by(|a, b| self.compare_video_formats(a, b))
    }

    /// Returns the best audio format available.
    /// Formats sorting : "quality", "audio bitrate", "sample rate", "audio channels"
    fn best_audio_format(&self) -> Option<&Format> {
        tracing::debug!(
            video_id = %self.id,
            format_count = self.formats.len(),
            "🧩 Selecting best audio format"
        );

        self.formats
            .iter()
            .filter(|f| f.is_audio())
            .max_by(|a, b| self.compare_audio_formats(a, b))
    }

    /// Returns the worst video format available.
    /// Formats sorting : "quality", "video resolution", "fps", "video bitrate"
    fn worst_video_format(&self) -> Option<&Format> {
        tracing::debug!(
            video_id = %self.id,
            format_count = self.formats.len(),
            "🧩 Selecting worst video format"
        );

        self.formats
            .iter()
            .filter(|f| f.is_video())
            .min_by(|a, b| self.compare_video_formats(a, b))
    }

    /// Returns the worst audio format available.
    /// Formats sorting : "quality", "audio bitrate", "sample rate", "audio channels"
    fn worst_audio_format(&self) -> Option<&Format> {
        tracing::debug!(
            video_id = %self.id,
            format_count = self.formats.len(),
            "🧩 Selecting worst audio format"
        );

        self.formats
            .iter()
            .filter(|f| f.is_audio())
            .min_by(|a, b| self.compare_audio_formats(a, b))
    }

    /// Compares two video formats.
    /// Formats sorting : "quality", "video resolution", "fps", "video bitrate"
    fn compare_video_formats(&self, a: &Format, b: &Format) -> Ordering {
        let a_quality = a.quality_info.quality.unwrap_or(OrderedFloat(0.0));
        let b_quality = b.quality_info.quality.unwrap_or(OrderedFloat(0.0));

        let cmp_quality = a_quality.cmp(&b_quality);
        if cmp_quality != Ordering::Equal {
            return cmp_quality;
        }

        let a_height = a.video_resolution.height.unwrap_or(0);
        let b_height = b.video_resolution.height.unwrap_or(0);

        let cmp_height = a_height.cmp(&b_height);
        if cmp_height != Ordering::Equal {
            return cmp_height;
        }

        let a_fps = a.video_resolution.fps.map(|f| *f).unwrap_or(0.0);
        let b_fps = b.video_resolution.fps.map(|f| *f).unwrap_or(0.0);

        let cmp_fps = OrderedFloat(a_fps).cmp(&OrderedFloat(b_fps));
        if cmp_fps != Ordering::Equal {
            return cmp_fps;
        }

        let a_vbr = a.rates_info.video_rate.map(|vr| *vr).unwrap_or(0.0);
        let b_vbr = b.rates_info.video_rate.map(|vr| *vr).unwrap_or(0.0);

        OrderedFloat(a_vbr).cmp(&OrderedFloat(b_vbr))
    }

    /// Compares two audio formats.
    /// Formats sorting : "quality", "audio bitrate", "sample rate", "audio channels"
    fn compare_audio_formats(&self, a: &Format, b: &Format) -> Ordering {
        let a_quality = a.quality_info.quality.unwrap_or(OrderedFloat(0.0));
        let b_quality = b.quality_info.quality.unwrap_or(OrderedFloat(0.0));

        let cmp_quality = a_quality.cmp(&b_quality);
        if cmp_quality != Ordering::Equal {
            return cmp_quality;
        }

        let a_abr = a.rates_info.audio_rate.map(|ar| *ar).unwrap_or(0.0);
        let b_abr = b.rates_info.audio_rate.map(|ar| *ar).unwrap_or(0.0);

        let cmp_abr = OrderedFloat(a_abr).cmp(&OrderedFloat(b_abr));
        if cmp_abr != Ordering::Equal {
            return cmp_abr;
        }

        let a_asr = a.codec_info.asr.unwrap_or(0);
        let b_asr = b.codec_info.asr.unwrap_or(0);

        let cmp_asr = a_asr.cmp(&b_asr);
        if cmp_asr != Ordering::Equal {
            return cmp_asr;
        }

        let a_channels = a.codec_info.audio_channels.unwrap_or(0);
        let b_channels = b.codec_info.audio_channels.unwrap_or(0);

        a_channels.cmp(&b_channels)
    }

    /// Selects a video format based on quality preference and codec preference.
    fn select_video_format(&self, quality: VideoQuality, codec: VideoCodecPreference) -> Option<&Format> {
        tracing::debug!(
            video_id = %self.id,
            quality = ?quality,
            codec = ?codec,
            total_formats = self.formats.len(),
            "🧩 Selecting video format with preferences"
        );

        let video_formats: Vec<&Format> = self.formats.iter().filter(|f| f.is_video()).collect();
        if video_formats.is_empty() {
            return None;
        }

        // Single-pass codec filter; fall back to all video formats if none match
        let filtered: Vec<&Format>;
        let active: &[&Format] = if codec == VideoCodecPreference::Any {
            &video_formats
        } else {
            filtered = video_formats
                .iter()
                .copied()
                .filter(|f| {
                    f.codec_info
                        .video_codec
                        .as_ref()
                        .is_some_and(|c| matches_video_codec(c, &codec))
                })
                .collect();
            if filtered.is_empty() {
                tracing::warn!(
                    video_id = %self.id,
                    codec = ?codec,
                    "Requested video codec not available, falling back to all formats"
                );
                &video_formats
            } else {
                &filtered
            }
        };

        // Select based on quality preference
        match quality {
            VideoQuality::Best => active.iter().copied().max_by(|a, b| self.compare_video_formats(a, b)),
            VideoQuality::Worst => active.iter().copied().min_by(|a, b| self.compare_video_formats(a, b)),
            VideoQuality::High => select_closest_video_height(active.iter().copied(), 1080, self),
            VideoQuality::Medium => select_closest_video_height(active.iter().copied(), 720, self),
            VideoQuality::Low => select_closest_video_height(active.iter().copied(), 480, self),
            VideoQuality::CustomHeight(height) => select_closest_video_height(active.iter().copied(), height, self),
            VideoQuality::CustomWidth(width) => select_closest_video_width(active.iter().copied(), width, self),
        }
    }

    /// Selects an audio format based on quality preference and codec preference.
    fn select_audio_format(&self, quality: AudioQuality, codec: AudioCodecPreference) -> Option<&Format> {
        tracing::debug!(
            video_id = %self.id,
            quality = ?quality,
            codec = ?codec,
            total_formats = self.formats.len(),
            "🧩 Selecting audio format with preferences"
        );

        let audio_formats: Vec<&Format> = self.formats.iter().filter(|f| f.is_audio()).collect();
        if audio_formats.is_empty() {
            return None;
        }

        // Single-pass codec filter; fall back to all audio formats if none match
        let filtered: Vec<&Format>;
        let active: &[&Format] = if codec == AudioCodecPreference::Any {
            &audio_formats
        } else {
            filtered = audio_formats
                .iter()
                .copied()
                .filter(|f| {
                    f.codec_info
                        .audio_codec
                        .as_ref()
                        .is_some_and(|c| matches_audio_codec(c, &codec))
                })
                .collect();
            if filtered.is_empty() {
                tracing::warn!(
                    video_id = %self.id,
                    codec = ?codec,
                    "Requested audio codec not available, falling back to all formats"
                );
                &audio_formats
            } else {
                &filtered
            }
        };

        // Select based on quality preference
        match quality {
            AudioQuality::Best => active.iter().copied().max_by(|a, b| self.compare_audio_formats(a, b)),
            AudioQuality::Worst => active.iter().copied().min_by(|a, b| self.compare_audio_formats(a, b)),
            AudioQuality::High => select_closest_audio_bitrate(active.iter().copied(), 192, self),
            AudioQuality::Medium => select_closest_audio_bitrate(active.iter().copied(), 128, self),
            AudioQuality::Low => select_closest_audio_bitrate(active.iter().copied(), 96, self),
            AudioQuality::CustomBitrate(bitrate) => select_closest_audio_bitrate(active.iter().copied(), bitrate, self),
        }
    }

    /// Returns all storyboard formats for this video, ordered from best to worst quality.
    ///
    /// Best quality = highest fragment count × largest resolution (width × height).
    fn storyboard_formats(&self) -> Vec<&Format> {
        tracing::debug!(
            video_id = %self.id,
            format_count = self.formats.len(),
            "🧩 Collecting storyboard formats"
        );

        let mut formats: Vec<&Format> = self
            .formats
            .iter()
            .filter(|f| f.format_type() == FormatType::Storyboard)
            .collect();

        // Sort descending: most fragments first, then largest resolution
        formats.sort_by(|a, b| {
            let a_frags = a.storyboard_info.fragments.as_ref().map_or(0, |v| v.len());
            let b_frags = b.storyboard_info.fragments.as_ref().map_or(0, |v| v.len());
            let a_area = a.video_resolution.width.unwrap_or(0) as u64 * a.video_resolution.height.unwrap_or(0) as u64;
            let b_area = b.video_resolution.width.unwrap_or(0) as u64 * b.video_resolution.height.unwrap_or(0) as u64;
            b_frags.cmp(&a_frags).then_with(|| b_area.cmp(&a_area))
        });

        formats
    }

    fn best_storyboard_format(&self) -> Option<&Format> {
        // Delegates to the model-level implementation on Video
        Video::best_storyboard_format(self)
    }

    fn worst_storyboard_format(&self) -> Option<&Format> {
        // Delegates to the model-level implementation on Video
        Video::worst_storyboard_format(self)
    }

    /// Selects a storyboard format based on quality preference.
    fn select_storyboard_format(&self, quality: StoryboardQuality) -> Option<&Format> {
        match quality {
            StoryboardQuality::Best => self.best_storyboard_format(),
            StoryboardQuality::Worst => self.worst_storyboard_format(),
        }
    }

    /// Selects a thumbnail based on quality preference.
    fn select_thumbnail(&self, quality: ThumbnailQuality) -> Option<&crate::model::thumbnail::Thumbnail> {
        tracing::debug!(
            video_id = %self.id,
            quality = ?quality,
            total_thumbnails = self.thumbnails.len(),
            "🧩 Selecting thumbnail with preferences"
        );

        match quality {
            ThumbnailQuality::Best => self.best_thumbnail(),
            ThumbnailQuality::Worst => self.worst_thumbnail(),
            ThumbnailQuality::MinimumResolution(width, height) => self.thumbnail_for_size(width, height),
        }
    }
}

/// Selects the video format with the closest height to the target
///
/// # Arguments
///
/// * `formats` - List of video formats to choose from
/// * `target_height` - Target height in pixels
/// * `video` - The video being processed (for quality comparisons)
///
/// # Returns
///
/// The format with the closest height to the target, or None if no formats available
fn select_closest_video_height<'a, I>(formats: I, target_height: u32, video: &Video) -> Option<&'a Format>
where
    I: Iterator<Item = &'a Format> + Clone,
{
    tracing::debug!(
        target_height = target_height,
        video_id = %video.id,
        "🧩 Selecting video format closest to target height"
    );

    let closest_above = formats
        .clone()
        .filter(|format| format.video_resolution.height.is_some_and(|h| h >= target_height))
        .min_by(|a, b| {
            let a_diff = a.video_resolution.height.unwrap_or(0).saturating_sub(target_height);
            let b_diff = b.video_resolution.height.unwrap_or(0).saturating_sub(target_height);

            // Compare difference then quality
            a_diff.cmp(&b_diff).then_with(|| video.compare_video_formats(a, b))
        });

    if let Some(closest) = closest_above {
        return Some(closest);
    }

    // If no format with height >= target, get the highest available
    formats.max_by(|a, b| {
        let a_height = a.video_resolution.height.unwrap_or(0);
        let b_height = b.video_resolution.height.unwrap_or(0);

        // Compare height then quality
        a_height.cmp(&b_height).then_with(|| video.compare_video_formats(a, b))
    })
}

/// Selects the video format with the closest width to the target
///
/// # Arguments
///
/// * `formats` - List of video formats to choose from
/// * `target_width` - Target width in pixels
/// * `video` - The video being processed (for quality comparisons)
///
/// # Returns
///
/// The format with the closest width to the target, or None if no formats available
fn select_closest_video_width<'a, I>(formats: I, target_width: u32, video: &Video) -> Option<&'a Format>
where
    I: Iterator<Item = &'a Format> + Clone,
{
    tracing::debug!(
        target_width = target_width,
        video_id = %video.id,
        "🧩 Selecting video format closest to target width"
    );

    let closest_above = formats
        .clone()
        .filter(|format| format.video_resolution.width.is_some_and(|w| w >= target_width))
        .min_by(|a, b| {
            let a_diff = a.video_resolution.width.unwrap_or(0).saturating_sub(target_width);
            let b_diff = b.video_resolution.width.unwrap_or(0).saturating_sub(target_width);

            // Compare difference then quality
            a_diff.cmp(&b_diff).then_with(|| video.compare_video_formats(a, b))
        });

    if let Some(closest) = closest_above {
        return Some(closest);
    }

    // If no format with width >= target, get the highest available
    formats.max_by(|a, b| {
        let a_width = a.video_resolution.width.unwrap_or(0);
        let b_width = b.video_resolution.width.unwrap_or(0);

        // Compare width then quality
        a_width.cmp(&b_width).then_with(|| video.compare_video_formats(a, b))
    })
}

/// Selects the audio format with the closest bitrate to the target
///
/// # Arguments
///
/// * `formats` - List of audio formats to choose from
/// * `target_bitrate` - Target bitrate in kbps
/// * `video` - The video being processed (for quality comparisons)
///
/// # Returns
///
/// The format with the closest bitrate to the target, or None if no formats available
fn select_closest_audio_bitrate<'a, I>(formats: I, target_bitrate: u32, video: &Video) -> Option<&'a Format>
where
    I: Iterator<Item = &'a Format> + Clone,
{
    tracing::debug!(
        target_bitrate = target_bitrate,
        video_id = %video.id,
        "🧩 Selecting audio format closest to target bitrate"
    );

    let target_float = OrderedFloat(target_bitrate as f64);

    let closest_above = formats
        .clone()
        .filter(|format| format.rates_info.audio_rate.is_some_and(|r| r >= target_float))
        .min_by(|a, b| {
            let a_rate = a.rates_info.audio_rate.unwrap_or(OrderedFloat(0.0));
            let b_rate = b.rates_info.audio_rate.unwrap_or(OrderedFloat(0.0));

            let a_diff = (a_rate.0 - target_bitrate as f64).abs();
            let b_diff = (b_rate.0 - target_bitrate as f64).abs();

            // Compare bitrate difference then quality
            OrderedFloat(a_diff)
                .partial_cmp(&OrderedFloat(b_diff))
                .unwrap_or(Ordering::Equal)
                .then_with(|| video.compare_audio_formats(a, b))
        });

    if let Some(closest) = closest_above {
        return Some(closest);
    }

    // If no format with bitrate >= target, get the highest available
    formats.max_by(|a, b| {
        let a_rate = a.rates_info.audio_rate.unwrap_or(OrderedFloat(0.0));
        let b_rate = b.rates_info.audio_rate.unwrap_or(OrderedFloat(0.0));

        // Compare bitrate then quality
        a_rate
            .partial_cmp(&b_rate)
            .unwrap_or(Ordering::Equal)
            .then_with(|| video.compare_audio_formats(a, b))
    })
}

impl Downloader {
    /// Lists all available subtitle and automatic caption languages for a video.
    ///
    /// Returns a deduplicated list of language codes from both `subtitles` and
    /// `automatic_captions`. Languages present in both are listed only once.
    ///
    /// # Arguments
    ///
    /// * `video` - The video to list subtitle languages for.
    ///
    /// # Returns
    ///
    /// A sorted vector of unique language codes.
    pub fn list_subtitle_languages(&self, video: &Video) -> Vec<String> {
        let mut languages: Vec<String> = video
            .subtitles
            .keys()
            .chain(video.automatic_captions.keys())
            .cloned()
            .collect();
        languages.sort();
        languages.dedup();

        tracing::debug!(
            video_id = %video.id,
            language_count = languages.len(),
            languages = ?languages,
            "💬 Listing subtitle/caption languages"
        );

        languages
    }

    /// Checks if a video has subtitles or automatic captions in a specific language.
    ///
    /// # Arguments
    ///
    /// * `video` - The video to check.
    /// * `language_code` - The language code to check for (e.g., "en", "fr").
    ///
    /// # Returns
    ///
    /// `true` if subtitles or automatic captions are available in the specified language.
    pub fn has_subtitle_language(&self, video: &Video, language_code: &str) -> bool {
        video.subtitles.contains_key(language_code) || video.automatic_captions.contains_key(language_code)
    }
}