src5-rs 2.0.0

Utility for generating SRC5 interfaces ids from Cairo traits
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
40
41

// Module for handling Cairo traits
use cairo_lang_compiler::db::RootDatabase;
use cairo_lang_syntax::node::kind::SyntaxKind;
use cairo_lang_syntax::node::SyntaxNode;

use super::cairo_function::get_functions_from_trait_body;
use super::cairo_function::CairoNonGenericFunction;
use super::utils::find_children;

#[derive(Debug)]
pub struct CairoNonGenericTrait {
    pub name: String,
    pub functions: Vec<CairoNonGenericFunction>,
}

pub fn get_non_generic_traits(
    db: &RootDatabase,
    syntax_tree: &SyntaxNode,
) -> Vec<CairoNonGenericTrait> {
    let mut no_generic_traits = Vec::new();
    for node in syntax_tree.descendants(db) {
        if SyntaxKind::ItemTrait == node.kind(db) {
            // Check if has no generic types
            if find_children(db, &node, SyntaxKind::OptionWrappedGenericParamListEmpty).is_some() {
                // Look up the Trait name
                let id_node = find_children(db, &node, SyntaxKind::TerminalIdentifier).unwrap();
                let trait_name = id_node.get_text_without_trivia(db);

                // Look up the Trait functions
                let trait_body = find_children(db, &node, SyntaxKind::TraitBody).unwrap();
                let functions = get_functions_from_trait_body(db, &trait_body);

                no_generic_traits.push(CairoNonGenericTrait {
                    name: trait_name,
                    functions,
                });
            }
        }
    }
    no_generic_traits
}