fm/modes/menu/

search.rs

use std::path::PathBuf;

use anyhow::Result;

use crate::app::Tab;
use crate::modes::{CaseDependantRegex, Display, FileInfo, Go, IndexToIndex, To, ToPath, Tree};

/// The current search term.
/// it records the regex used, the matched paths and where we are in those pathes.
/// The pathes are refreshed every time we jump to another match, allowing the
/// display to stay updated.
pub struct Search {
    pub regex: CaseDependantRegex,
    pub paths: Vec<PathBuf>,
    pub index: usize,
}

impl std::fmt::Display for Search {
    fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
        if self.is_empty() {
            write!(f, "")
        } else {
            write!(
                f,
                " Searched: {regex} - {pos} / {len} ",
                regex = self.regex,
                pos = self.index + 1 - self.paths.is_empty() as usize,
                len = self.paths.len()
            )
        }
    }
}

impl Search {
    pub fn empty() -> Self {
        Self {
            regex: CaseDependantRegex::new("").unwrap(),
            paths: vec![],
            index: 0,
        }
    }

    pub fn new(searched: &str) -> Result<Self> {
        Ok(Self {
            regex: CaseDependantRegex::new(searched)?,
            paths: vec![],
            index: 0,
        })
    }

    pub fn clone_with_regex(&self) -> Self {
        Self {
            regex: self.regex.clone(),
            paths: vec![],
            index: 0,
        }
    }

    pub fn is_empty(&self) -> bool {
        self.regex.is_empty()
    }

    pub fn reset_paths(&mut self) {
        self.paths = vec![];
        self.index = 0;
    }

    pub fn select_next(&mut self) -> Option<PathBuf> {
        if !self.paths.is_empty() && !self.regex.to_string().is_empty() {
            self.index = (self.index + 1) % self.paths.len();
            return Some(self.paths[self.index].to_owned());
        }
        None
    }

    pub fn execute_search(&mut self, tab: &mut Tab) -> Result<()> {
        match tab.display_mode {
            Display::Tree => {
                self.tree(&mut tab.tree);
            }
            Display::Directory => {
                self.directory(tab);
            }
            _ => (),
        };
        Ok(())
    }

    /// Search in current directory for an file whose name contains `searched_name`,
    /// from a starting position `next_index`.
    /// We search forward from that position and start again from top if nothing is found.
    /// We move the selection to the first matching file.
    #[inline]
    fn directory(&mut self, tab: &mut Tab) {
        let current_index = tab.directory.index;
        let mut next_index = current_index;
        let mut found = false;
        for (index, file) in tab.directory.enumerate().skip(current_index) {
            if self.regex.is_match(&file.filename) {
                (next_index, found) = self.set_found(index, file, next_index, found);
            }
        }
        for (index, file) in tab.directory.enumerate().take(current_index) {
            if self.regex.is_match(&file.filename) {
                (next_index, found) = self.set_found(index, file, next_index, found);
            }
        }
        tab.go_to_index(next_index);
    }

    #[inline]
    fn set_found(
        &mut self,
        index: usize,
        file: &FileInfo,
        mut next_index: usize,
        mut found: bool,
    ) -> (usize, bool) {
        if !found {
            next_index = index;
            self.index = self.paths.len();
            found = true;
        }
        self.paths.push(file.path.to_path_buf());

        (next_index, found)
    }

    pub fn directory_search_next<'a>(
        &mut self,
        files: impl Iterator<Item = &'a FileInfo>,
    ) -> Option<PathBuf> {
        let (paths, Some(next_index), Some(next_path)) = self.directory_update_search(files) else {
            return None;
        };
        self.set_index_paths(next_index, paths);
        Some(next_path)
    }

    fn directory_update_search<'a>(
        &self,
        files: impl std::iter::Iterator<Item = &'a FileInfo>,
    ) -> (Vec<PathBuf>, Option<usize>, Option<PathBuf>) {
        let mut paths = vec![];
        let mut next_index = None;
        let mut next_path = None;

        for file in files {
            if self.regex.is_match(&file.filename) {
                if next_index.is_none() {
                    (next_index, next_path) = self.found_first_match(file)
                }
                paths.push(file.path.to_path_buf());
            }
        }
        (paths, next_index, next_path)
    }

    fn found_first_match(&self, file: &FileInfo) -> (Option<usize>, Option<PathBuf>) {
        (Some(self.paths.len()), Some(file.path.to_path_buf()))
    }

    pub fn set_index_paths(&mut self, index: usize, paths: Vec<PathBuf>) {
        self.index = index;
        self.paths = paths;
    }

    pub fn tree(&mut self, tree: &mut Tree) {
        if let Some(path) = &self.tree_find_next_path(tree) {
            tree.go(To::Path(path));
        }
    }

    fn tree_find_next_path(&mut self, tree: &mut Tree) -> Option<PathBuf> {
        if let Some(path) = self.select_next() {
            return Some(path);
        }
        self.tree_search_again(tree)
    }

    fn tree_search_again(&mut self, tree: &mut Tree) -> Option<PathBuf> {
        let mut next_path = None;
        for line in tree.index_to_index() {
            let Some(filename) = line.path.file_name() else {
                continue;
            };
            if self.regex.is_match(&filename.to_string_lossy()) {
                let match_path = line.path.to_path_buf();
                if next_path.is_none() {
                    self.index = self.paths.len();
                    next_path = Some(match_path.clone());
                }
                self.paths.push(match_path);
            }
        }
        next_path
    }

    #[inline]
    pub fn matches_from(&self, content: &[impl ToPath]) -> Vec<String> {
        content
            .iter()
            .filter_map(|e| e.to_path().file_name())
            .map(|s| s.to_string_lossy().to_string())
            .filter(|p| self.regex.is_match(p))
            .collect()
    }

    #[inline]
    pub fn is_match(&self, filename: &str) -> bool {
        !self.is_empty() && self.regex.is_match(filename)
    }
}