// yt - A fully featured command line YouTube client
//
// Copyright (C) 2024 Benedikt Peetz <benedikt.peetz@b-peetz.de>
// SPDX-License-Identifier: GPL-3.0-or-later
//
// This file is part of Yt.
//
// You should have received a copy of the License along with this program.
// If not, see <https://www.gnu.org/licenses/gpl-3.0.txt>.

use anyhow::Result;
use futures::{stream::FuturesUnordered, TryStreamExt};
use nucleo_matcher::{
    pattern::{CaseMatching, Normalization, Pattern},
    Matcher,
};

use crate::{
    app::App,
    storage::video_database::{getters::get_videos, VideoStatus},
};

pub async fn query(app: &App, limit: Option<usize>, search_query: Option<String>) -> Result<()> {
    let all_videos = get_videos(app, VideoStatus::ALL, None).await?;

    // turn one video to a color display, to pre-warm the hash shrinking cache
    if let Some(val) = all_videos.get(0) {
        val.to_color_display(app).await?;
    }

    let limit = limit.unwrap_or(all_videos.len());

    let all_video_strings: Vec<String> = all_videos
        .into_iter()
        .take(limit)
        .map(|vid| vid.to_color_display_owned(app))
        .collect::<FuturesUnordered<_>>()
        .try_collect()
        .await?;

    if let Some(query) = search_query {
        let mut matcher = Matcher::new(nucleo_matcher::Config::DEFAULT.match_paths());

        let matches = Pattern::parse(
            &query.replace(' ', "\\ "),
            CaseMatching::Ignore,
            Normalization::Smart,
        )
        .match_list(all_video_strings, &mut matcher);

        matches
            .iter()
            .rev()
            .for_each(|(val, key)| println!("{} ({})", val, key));
    } else {
        println!("{}", all_video_strings.join("\n"))
    }

    Ok(())
}