scrings 0.1.6

Semantic strings based on tree-sitter
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
42
43
44
45
46
47
48
49
50
51
52
53
54

use crate::parser::Parser;
use crate::tree::Tree;
use crate::visitor::LanguageVisitor;
use crate::error::Result;
use tree_sitter_javascript::LANGUAGE as javascript_language;

fn build_javascript_tree(source: &str) -> Result<Tree> {
    let mut parser = tree_sitter::Parser::new();
    parser.set_language(&javascript_language.into())?;

    let tree_sitter = parser.parse(source, None).unwrap();
    Ok(Tree::new(
        source.as_bytes(),
        tree_sitter,
    ))
}

#[derive(Default)]
pub struct Javascript;

impl Parser for Javascript {
    fn parse(&mut self, src: &str) -> Result<Option<(u64, String)>> {
        let tree = build_javascript_tree(src)?;

        let mut detection_rule = LanguageVisitor::new(|c| {
                matches!(c,
                    "function_declaration" |
                    "export_statement" | "debugger_statement" |
                    "declaration" |
                    "if_statement" | "switch_statement" |
                    "for_statement" | "for_in_statement" |
                    "while_statement" | "do_statement" |
                    "try_statement" | "with_statement"
                )
            }
        );

        tree.apply(&mut detection_rule)?;

        if detection_rule.is_matched {
            Ok(
                Some(
                    (
                        detection_rule.start.unwrap_or(0) as u64,
                        String::from(&src[detection_rule.start.unwrap_or(0)..detection_rule.end.unwrap_or(src.len())])
                    )
                )
            )
        }
        else {
            Ok(None)
        }
    }
}