prj-cli 1.1.0

A CLI tool for managing projects on your local machine
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use nucleo::{
    Matcher, Utf32Str,
    pattern::{CaseMatching, Normalization, Pattern},
};

pub struct FuzzyMatcher {
    matcher: Matcher,
}

pub struct FuzzyMatch {
    pub index: usize,
    pub score: u32,
}

impl FuzzyMatcher {
    pub fn new() -> Self {
        Self {
            matcher: Matcher::new(nucleo::Config::DEFAULT),
        }
    }

    /// Filter items by query, returning matching indices sorted by score (best first).
    pub fn filter(&mut self, query: &str, items: &[String]) -> Vec<FuzzyMatch> {
        if query.is_empty() {
            return items
                .iter()
                .enumerate()
                .map(|(i, _)| FuzzyMatch { index: i, score: 0 })
                .collect();
        }

        let pattern = Pattern::new(
            query,
            CaseMatching::Smart,
            Normalization::Smart,
            nucleo::pattern::AtomKind::Fuzzy,
        );

        let mut results: Vec<FuzzyMatch> = items
            .iter()
            .enumerate()
            .filter_map(|(i, item)| {
                let mut buf = Vec::new();
                let haystack = Utf32Str::new(item, &mut buf);
                let score = pattern.score(haystack, &mut self.matcher)?;
                Some(FuzzyMatch { index: i, score })
            })
            .collect();

        results.sort_by(|a, b| b.score.cmp(&a.score));
        results
    }
}