Defines space kinds

mozilla · Mar 22, 2022 · c765293 · c765293
1 parent 8bac42f
commit c765293
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/getter.rs b/src/getter.rs
@@ -515,6 +515,9 @@ impl Getter for JavaCode {
 
         let typ = node.object().kind_id();
         match typ.into() {
+            ClassDeclaration => SpaceKind::Class,
+            MethodDeclaration | LambdaExpression => SpaceKind::Function,
+            PackageDeclaration | ModuleDeclaration => SpaceKind::Namespace,
             Program => SpaceKind::Unit,
             _ => SpaceKind::Unknown,
         }