refactor: extract completions context into pgt_treesitter crate (#466)

I wanted to extract the treesitter-context that's used in completions,
so it can also be used for the on-hover feature.

I've also tried to clean up the APIs, detangled completion-context and
sanitization, and extracted the general test helpers into the
`pgt_test_utils` crate.

Author: juleswritescode

Cargo.lock

@@ -82,7 +82,7 @@ pgt_suppressions           = { path = "./crates/pgt_suppressions", version = "0.
 pgt_text_edit              = { path = "./crates/pgt_text_edit", version = "0.0.0" }
 pgt_text_size              = { path = "./crates/pgt_text_size", version = "0.0.0" }
 pgt_tokenizer              = { path = "./crates/pgt_tokenizer", version = "0.0.0" }
-pgt_treesitter_queries     = { path = "./crates/pgt_treesitter_queries", version = "0.0.0" }
+pgt_treesitter             = { path = "./crates/pgt_treesitter", version = "0.0.0" }
 pgt_typecheck              = { path = "./crates/pgt_typecheck", version = "0.0.0" }
 pgt_workspace              = { path = "./crates/pgt_workspace", version = "0.0.0" }
 pgt_workspace_macros       = { path = "./crates/pgt_workspace_macros", version = "0.0.0" }

Cargo.toml

@@ -14,18 +14,17 @@ version              = "0.0.0"
 [dependencies]
 async-std = "1.12.0"
 
-pgt_text_size.workspace = true
-
-
-fuzzy-matcher                    = "0.3.7"
-pgt_schema_cache.workspace       = true
-pgt_treesitter_queries.workspace = true
-schemars                         = { workspace = true, optional = true }
-serde                            = { workspace = true, features = ["derive"] }
-serde_json                       = { workspace = true }
-tracing                          = { workspace = true }
-tree-sitter.workspace            = true
-tree_sitter_sql.workspace        = true
+pgt_schema_cache.workspace = true
+pgt_text_size.workspace    = true
+pgt_treesitter.workspace   = true
+
+fuzzy-matcher             = "0.3.7"
+schemars                  = { workspace = true, optional = true }
+serde                     = { workspace = true, features = ["derive"] }
+serde_json                = { workspace = true }
+tracing                   = { workspace = true }
+tree-sitter.workspace     = true
+tree_sitter_sql.workspace = true
 
 sqlx.workspace = true
 

crates/pgt_completions/Cargo.toml

@@ -1,10 +1,11 @@
 use crate::{
     CompletionItemKind, CompletionText,
-    context::CompletionContext,
     item::CompletionItem,
     relevance::{filtering::CompletionFilter, scoring::CompletionScore},
 };
 
+use pgt_treesitter::TreesitterContext;
+
 pub(crate) struct PossibleCompletionItem<'a> {
     pub label: String,
     pub description: String,
@@ -17,11 +18,11 @@ pub(crate) struct PossibleCompletionItem<'a> {
 
 pub(crate) struct CompletionBuilder<'a> {
     items: Vec<PossibleCompletionItem<'a>>,
-    ctx: &'a CompletionContext<'a>,
+    ctx: &'a TreesitterContext<'a>,
 }
 
 impl<'a> CompletionBuilder<'a> {
-    pub fn new(ctx: &'a CompletionContext) -> Self {
+    pub fn new(ctx: &'a TreesitterContext) -> Self {
         CompletionBuilder { items: vec![], ctx }
     }
 

crates/pgt_completions/src/builder.rs

@@ -1,8 +1,9 @@
 use pgt_text_size::TextSize;
 
+use pgt_treesitter::{TreeSitterContextParams, context::TreesitterContext};
+
 use crate::{
     builder::CompletionBuilder,
-    context::CompletionContext,
     item::CompletionItem,
     providers::{
         complete_columns, complete_functions, complete_policies, complete_roles, complete_schemas,
@@ -28,16 +29,20 @@ pub struct CompletionParams<'a> {
 pub fn complete(params: CompletionParams) -> Vec<CompletionItem> {
     let sanitized_params = SanitizedCompletionParams::from(params);
 
-    let ctx = CompletionContext::new(&sanitized_params);
+    let ctx = TreesitterContext::new(TreeSitterContextParams {
+        position: sanitized_params.position,
+        text: &sanitized_params.text,
+        tree: &sanitized_params.tree,
+    });
 
     let mut builder = CompletionBuilder::new(&ctx);
 
-    complete_tables(&ctx, &mut builder);
-    complete_functions(&ctx, &mut builder);
-    complete_columns(&ctx, &mut builder);
-    complete_schemas(&ctx, &mut builder);
-    complete_policies(&ctx, &mut builder);
-    complete_roles(&ctx, &mut builder);
+    complete_tables(&ctx, sanitized_params.schema, &mut builder);
+    complete_functions(&ctx, sanitized_params.schema, &mut builder);
+    complete_columns(&ctx, sanitized_params.schema, &mut builder);
+    complete_schemas(&ctx, sanitized_params.schema, &mut builder);
+    complete_policies(&ctx, sanitized_params.schema, &mut builder);
+    complete_roles(&ctx, sanitized_params.schema, &mut builder);
 
     builder.finish()
 }

crates/pgt_completions/src/complete.rs

@@ -1,6 +1,5 @@
 mod builder;
 mod complete;
-mod context;
 mod item;
 mod providers;
 mod relevance;