changie

Author: juleswritescode

crates/pgls_treesitter/src/context/ancestors.rs

@@ -1,9 +1,4 @@
-use std::{
-    collections::{HashMap, HashSet},
-    process::Child,
-};
-
-use crate::TreesitterContext;
+use std::collections::{HashMap, HashSet};
 
 #[derive(Debug)]
 pub struct Scope {

crates/pgls_treesitter/src/context/mod.rs

@@ -153,7 +153,6 @@ impl<'a> TreesitterContext<'a> {
         ctx.gather_tree_context();
         ctx.gather_info_from_ts_queries();
 
-        println!("{:#?}", ctx);
         ctx
     }
 
@@ -296,6 +295,8 @@ impl<'a> TreesitterContext<'a> {
         let parent_node_kind = parent_node.kind();
         let current_node_kind = current_node.kind();
 
+        self.scope_tracker.register(current_node, self.position);
+
         // prevent infinite recursion – this can happen with ERROR nodes
         if current_node_kind == parent_node_kind && ["ERROR", "program"].contains(&parent_node_kind)
         {
@@ -375,7 +376,6 @@ impl<'a> TreesitterContext<'a> {
             return;
         }
 
-        self.scope_tracker.register(current_node, self.position);
         cursor.goto_first_child_for_byte(self.position);
         self.gather_context_from_node(cursor, current_node);
     }