11use tree_sitter:: Node ;
22
3- use crate :: enums :: NodeKind ;
3+ use crate :: spaces :: SpaceKind ;
44use crate :: traits:: Search ;
55
66use 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
2828impl 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
4040impl 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
8787impl 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
134134impl 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
181181impl 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
228228impl 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}
0 commit comments