hemtt-preprocessor 1.0.0

A preprocessor library for hemtt
Documentation 
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

use std::collections::HashMap;

use hemtt_tokens::Token;
use strsim::levenshtein;

use crate::Definition;

/// `HashMap` of all current defines
pub type Defines = HashMap<String, (Token, Definition)>;

/// Helper functions for Defines
pub trait DefinitionLibrary {
    /// Find similar functions with a certain number of args
    ///
    /// Args can be Some(1) to only find macros that take 1 argument
    /// or None to check any number of arguments
    fn similar_function(&self, search: &str, args: Option<usize>) -> Vec<&str>;
}

impl DefinitionLibrary for Defines {
    fn similar_function(&self, search: &str, args: Option<usize>) -> Vec<&str> {
        let mut similar = self
            .iter()
            .filter(|(_, (_, def))| {
                let Definition::Function(func) = def else {
                    return false;
                };
                args.map_or(true, |args| func.parameters().len() == args)
            })
            .map(|(name, _)| (name.as_str(), levenshtein(name, search)))
            .collect::<Vec<_>>();
        similar.sort_by_key(|(_, v)| *v);
        similar.retain(|s| s.1 <= 3);
        if similar.len() > 3 {
            similar.truncate(3);
        }
        similar.into_iter().map(|(n, _)| n).collect::<Vec<_>>()
    }
}