-
Notifications
You must be signed in to change notification settings - Fork 1
/
rule.rs
119 lines (111 loc) · 3.55 KB
/
rule.rs
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
use crate::extractor::Extractor;
/*
tree-sitter query syntax
https://tree-sitter.github.io/tree-sitter/using-parsers#query-syntax
*/
pub struct Rule {
// which symbols has been used (possibly imported) in this file
pub(crate) import_grammar: &'static str,
// which symbols has been exported from this file
pub(crate) export_grammar: &'static str,
}
pub fn get_rule(extractor_type: &Extractor) -> Rule {
match extractor_type {
Extractor::Rust => Rule {
import_grammar: r#"
(identifier) @variable_name
(call_expression
function: (identifier) @function)
(call_expression
function: (field_expression
field: (field_identifier) @function.method))
(call_expression
function: (scoped_identifier
"::"
name: (identifier) @function))
"#,
export_grammar: r#"
(function_item name: (identifier) @exported_symbol)
(function_signature_item name: (identifier) @exported_symbol)
(generic_function
function: (identifier) @exported_symbol)
(generic_function
function: (scoped_identifier
name: (identifier) @exported_symbol))
"#,
},
Extractor::TypeScript => Rule {
import_grammar: r#"
(identifier) @variable_name
"#,
export_grammar: r#"
(export_statement (function_declaration name: (identifier) @exported_symbol))
(export_statement (arrow_function (identifier) @exported_symbol))
(export_statement (generator_function_declaration name: (identifier) @exported_symbol))
(method_definition name: (property_identifier) @exported_symbol)
(export_statement (type_alias_declaration name: (type_identifier) @exported_symbol))
(export_statement (interface_declaration name: (type_identifier) @exported_symbol))
(export_statement (class_declaration name: (type_identifier) @exported_symbol))
(export_specifier (identifier) @exported_symbol)
"#,
},
Extractor::Go => Rule {
import_grammar: r#"
(identifier) @variable_name
"#,
export_grammar: r#"
(function_declaration name: (identifier) @exported_symbol)
(method_declaration name: (field_identifier) @exported_symbol)
(type_alias name: (type_identifier) @exported_symbol)
(type_spec name: (type_identifier) @exported_symbol)
(const_spec name: (identifier) @exported_symbol)
(var_spec name: (identifier) @exported_symbol)
"#,
},
Extractor::Python => Rule {
import_grammar: r#"
(identifier) @variable_name
"#,
export_grammar: r#"
(function_definition name: (identifier) @exported_symbol)
(class_definition name: (identifier) @exported_symbol)
"#,
},
Extractor::JavaScript => Rule {
import_grammar: r#"
(identifier) @variable_name
"#,
export_grammar: r#"
(function_declaration name: (identifier) @exported_symbol)
(class_declaration name: (identifier) @exported_symbol)
"#,
},
Extractor::Java => Rule {
import_grammar: r#"
((identifier) @variable_name)
"#,
// todo: not enough maybe
export_grammar: r#"
(class_declaration name: (identifier) @exported_symbol)
"#,
},
Extractor::Kotlin => Rule {
import_grammar: r#"
(identifier (simple_identifier) @variable_name)
"#,
export_grammar: r#"
(class_declaration (type_identifier) @exported_symbol)
(function_declaration (simple_identifier) @exported_symbol)
"#,
},
Extractor::Swift => Rule {
import_grammar: r#"
((simple_identifier) @exported_symbol)
"#,
// TODO: not enough
export_grammar: r#"
(function_declaration (simple_identifier) @method)
"#,
},
}
}