lb-rs 26.7.4

use super::{SearchFilter, SearchResult, build_descendants};
use crate::Lb;
use crate::model::file::File;
use nucleo::{
    Matcher, Nucleo,
    pattern::{CaseMatching, Normalization},
};
use std::cmp::Reverse;
use std::collections::{HashMap, HashSet};
use std::sync::Arc;
use uuid::Uuid;

/// Split a path into (parent_path, filename).
pub(crate) fn split_path(path: &str) -> (&str, &str) {
    // Strip trailing slash for folders
    let path = path.trim_end_matches('/');
    match path.rfind('/') {
        Some(idx) if idx > 0 => (&path[..idx], &path[idx + 1..]),
        Some(_) => ("/", &path[1..]), // root case: "/filename"
        None => ("", path),
    }
}

#[derive(Clone)]
struct PathEntry {
    file: File,
    path: String,
    filename: String,
    parent_path: String,
}

fn path_result(entry: &PathEntry, path_indices: Vec<u32>) -> SearchResult {
    SearchResult {
        id: entry.file.id,
        filename: entry.filename.clone(),
        parent_path: entry.parent_path.clone(),
        is_folder: entry.file.is_folder(),
        path_indices,
        path_matches: Vec::new(),
        content_matches: Vec::new(),
    }
}

pub struct PathSearcher {
    nucleo: Nucleo<PathEntry>,
    results: Vec<SearchResult>,
    submitted_query: String,
    descendants: HashMap<Uuid, Vec<Uuid>>,
    path_to_id: HashMap<String, Uuid>,
    filter_ids: Option<HashSet<Uuid>>,
}

impl PathSearcher {
    pub async fn new(lb: &Lb) -> Self {
        let files = lb.list_metadatas().await.unwrap_or_default();
        let mut paths = lb.list_paths_with_ids(None).await.unwrap_or_default();
        paths.retain(|(_, path)| path != "/");

        let notify = Arc::new(|| {});
        let nucleo: Nucleo<PathEntry> = Nucleo::new(nucleo::Config::DEFAULT, notify, None, 1);
        let injector = nucleo.injector();

        for (id, path) in &paths {
            if let Some(file) = files.iter().find(|f| f.id == *id) {
                let (parent_path, filename) = split_path(path);
                injector.push(
                    PathEntry {
                        file: file.clone(),
                        path: path.clone(),
                        filename: filename.to_string(),
                        parent_path: parent_path.to_string(),
                    },
                    |entry, cols| {
                        cols[0] = entry.path.as_str().into();
                    },
                );
            }
        }

        let descendants = build_descendants(&files);
        let path_to_id = paths.iter().map(|(id, path)| (path.clone(), *id)).collect();

        let mut searcher = Self {
            nucleo,
            results: Vec::new(),
            submitted_query: String::new(),
            descendants,
            path_to_id,
            filter_ids: None,
        };
        searcher.query("");
        searcher
    }

    /// Update the active filter and refresh results for the current query.
    pub fn update_filter(&mut self, filter: Option<SearchFilter>) {
        self.filter_ids = super::resolve_filter(filter, &self.path_to_id, &self.descendants);
        self.rebuild();
    }

    /// Update the search query. Results available via `results()`.
    pub fn query(&mut self, input: &str) {
        self.nucleo.pattern.reparse(
            0,
            input,
            CaseMatching::Smart,
            Normalization::Smart,
            input.starts_with(&self.submitted_query),
        );
        self.submitted_query = input.to_string();

        while self.nucleo.tick(10).running {}

        self.rebuild();
    }

    /// Rebuild `results` from the current query and filter without re-running the
    /// matcher. Shared by `query` (after a reparse) and `update_filter`.
    fn rebuild(&mut self) {
        self.results.clear();
        let snapshot = self.nucleo.snapshot();

        if self.submitted_query.is_empty() {
            let mut entries: Vec<&PathEntry> = (0..snapshot.matched_item_count())
                .filter_map(|i| snapshot.get_matched_item(i).map(|item| item.data))
                .filter(|e| {
                    self.filter_ids
                        .as_ref()
                        .is_none_or(|ids| ids.contains(&e.file.id))
                })
                .collect();
            entries.sort_by_key(|e| Reverse(e.file.last_modified));
            self.results.extend(
                entries
                    .into_iter()
                    .take(100)
                    .map(|e| path_result(e, Vec::new())),
            );
            return;
        }

        let mut matcher = Matcher::new(nucleo::Config::DEFAULT);

        for i in 0..snapshot.matched_item_count() {
            if self.results.len() >= 100 {
                break;
            }
            if let Some(item) = snapshot.get_matched_item(i) {
                if let Some(ids) = &self.filter_ids {
                    if !ids.contains(&item.data.file.id) {
                        continue;
                    }
                }

                let mut indices = Vec::new();
                self.nucleo.pattern.column_pattern(0).indices(
                    item.matcher_columns[0].slice(..),
                    &mut matcher,
                    &mut indices,
                );

                self.results.push(path_result(item.data, indices));
            }
        }
    }

    /// Get search results.
    pub fn results(&self) -> &[SearchResult] {
        &self.results
    }
}