Skip to content

Commit bf10b83

Browse files
authored
Clean up src (#164)
* Merge the enums file into metrics * Remove an unused file * Change confusing names * Create a directory for output functions * Change wrong structures names
1 parent b30c5a3 commit bf10b83

37 files changed

Lines changed: 95 additions & 102 deletions

src/enums.rs

Lines changed: 0 additions & 29 deletions
This file was deleted.

src/error.rs

Lines changed: 0 additions & 3 deletions
This file was deleted.

src/getter.rs

Lines changed: 41 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
use tree_sitter::Node;
22

3-
use crate::enums::NodeKind;
3+
use crate::spaces::SpaceKind;
44
use crate::traits::Search;
55

66
use crate::*;
@@ -20,25 +20,25 @@ pub trait Getter {
2020
}
2121
}
2222

23-
fn get_kind(_node: &Node) -> NodeKind {
24-
NodeKind::Unknown
23+
fn get_space_kind(_node: &Node) -> SpaceKind {
24+
SpaceKind::Unknown
2525
}
2626
}
2727

2828
impl Getter for PythonCode {
29-
fn get_kind(node: &Node) -> NodeKind {
29+
fn get_space_kind(node: &Node) -> SpaceKind {
3030
let typ = node.kind_id();
3131
match typ.into() {
32-
Python::FunctionDefinition => NodeKind::Function,
33-
Python::ClassDefinition => NodeKind::Class,
34-
Python::Module => NodeKind::Unit,
35-
_ => NodeKind::Unknown,
32+
Python::FunctionDefinition => SpaceKind::Function,
33+
Python::ClassDefinition => SpaceKind::Class,
34+
Python::Module => SpaceKind::Unit,
35+
_ => SpaceKind::Unknown,
3636
}
3737
}
3838
}
3939

4040
impl Getter for MozjsCode {
41-
fn get_kind(node: &Node) -> NodeKind {
41+
fn get_space_kind(node: &Node) -> SpaceKind {
4242
use Mozjs::*;
4343

4444
let typ = node.kind_id();
@@ -48,10 +48,10 @@ impl Getter for MozjsCode {
4848
| GeneratorFunction
4949
| FunctionDeclaration
5050
| GeneratorFunctionDeclaration
51-
| ArrowFunction => NodeKind::Function,
52-
Class | ClassDeclaration => NodeKind::Class,
53-
Program => NodeKind::Unit,
54-
_ => NodeKind::Unknown,
51+
| ArrowFunction => SpaceKind::Function,
52+
Class | ClassDeclaration => SpaceKind::Class,
53+
Program => SpaceKind::Unit,
54+
_ => SpaceKind::Unknown,
5555
}
5656
}
5757

@@ -85,7 +85,7 @@ impl Getter for MozjsCode {
8585
}
8686

8787
impl Getter for JavascriptCode {
88-
fn get_kind(node: &Node) -> NodeKind {
88+
fn get_space_kind(node: &Node) -> SpaceKind {
8989
use Javascript::*;
9090

9191
let typ = node.kind_id();
@@ -95,10 +95,10 @@ impl Getter for JavascriptCode {
9595
| GeneratorFunction
9696
| FunctionDeclaration
9797
| GeneratorFunctionDeclaration
98-
| ArrowFunction => NodeKind::Function,
99-
Class | ClassDeclaration => NodeKind::Class,
100-
Program => NodeKind::Unit,
101-
_ => NodeKind::Unknown,
98+
| ArrowFunction => SpaceKind::Function,
99+
Class | ClassDeclaration => SpaceKind::Class,
100+
Program => SpaceKind::Unit,
101+
_ => SpaceKind::Unknown,
102102
}
103103
}
104104

@@ -132,7 +132,7 @@ impl Getter for JavascriptCode {
132132
}
133133

134134
impl Getter for TypescriptCode {
135-
fn get_kind(node: &Node) -> NodeKind {
135+
fn get_space_kind(node: &Node) -> SpaceKind {
136136
use Typescript::*;
137137

138138
let typ = node.kind_id();
@@ -142,10 +142,10 @@ impl Getter for TypescriptCode {
142142
| GeneratorFunction
143143
| FunctionDeclaration
144144
| GeneratorFunctionDeclaration
145-
| ArrowFunction => NodeKind::Function,
146-
Class | ClassDeclaration => NodeKind::Class,
147-
Program => NodeKind::Unit,
148-
_ => NodeKind::Unknown,
145+
| ArrowFunction => SpaceKind::Function,
146+
Class | ClassDeclaration => SpaceKind::Class,
147+
Program => SpaceKind::Unit,
148+
_ => SpaceKind::Unknown,
149149
}
150150
}
151151

@@ -179,7 +179,7 @@ impl Getter for TypescriptCode {
179179
}
180180

181181
impl Getter for TsxCode {
182-
fn get_kind(node: &Node) -> NodeKind {
182+
fn get_space_kind(node: &Node) -> SpaceKind {
183183
use Tsx::*;
184184

185185
let typ = node.kind_id();
@@ -189,10 +189,10 @@ impl Getter for TsxCode {
189189
| GeneratorFunction
190190
| FunctionDeclaration
191191
| GeneratorFunctionDeclaration
192-
| ArrowFunction => NodeKind::Function,
193-
Class | ClassDeclaration => NodeKind::Class,
194-
Program => NodeKind::Unit,
195-
_ => NodeKind::Unknown,
192+
| ArrowFunction => SpaceKind::Function,
193+
Class | ClassDeclaration => SpaceKind::Class,
194+
Program => SpaceKind::Unit,
195+
_ => SpaceKind::Unknown,
196196
}
197197
}
198198

@@ -226,16 +226,16 @@ impl Getter for TsxCode {
226226
}
227227

228228
impl Getter for RustCode {
229-
fn get_kind(node: &Node) -> NodeKind {
229+
fn get_space_kind(node: &Node) -> SpaceKind {
230230
use Rust::*;
231231

232232
let typ = node.kind_id();
233233
match typ.into() {
234-
FunctionItem | ClosureExpression => NodeKind::Function,
235-
TraitItem => NodeKind::Trait,
236-
ImplItem => NodeKind::Impl,
237-
SourceFile => NodeKind::Unit,
238-
_ => NodeKind::Unknown,
234+
FunctionItem | ClosureExpression => SpaceKind::Function,
235+
TraitItem => SpaceKind::Trait,
236+
ImplItem => SpaceKind::Impl,
237+
SourceFile => SpaceKind::Unit,
238+
_ => SpaceKind::Unknown,
239239
}
240240
}
241241
}
@@ -285,17 +285,17 @@ impl Getter for CppCode {
285285
None
286286
}
287287

288-
fn get_kind(node: &Node) -> NodeKind {
288+
fn get_space_kind(node: &Node) -> SpaceKind {
289289
use Cpp::*;
290290

291291
let typ = node.kind_id();
292292
match typ.into() {
293-
FunctionDefinition | FunctionDefinition2 | FunctionDefinition3 => NodeKind::Function,
294-
StructSpecifier => NodeKind::Struct,
295-
ClassSpecifier => NodeKind::Class,
296-
NamespaceDefinition => NodeKind::Namespace,
297-
TranslationUnit => NodeKind::Unit,
298-
_ => NodeKind::Unknown,
293+
FunctionDefinition | FunctionDefinition2 | FunctionDefinition3 => SpaceKind::Function,
294+
StructSpecifier => SpaceKind::Struct,
295+
ClassSpecifier => SpaceKind::Class,
296+
NamespaceDefinition => SpaceKind::Namespace,
297+
TranslationUnit => SpaceKind::Unit,
298+
_ => SpaceKind::Unknown,
299299
}
300300
}
301301
}
File renamed without changes.

0 commit comments

Comments
 (0)