cool

Author: juleswritescode

crates/pgt_completions/src/context/context.rs

@@ -24,7 +24,7 @@ pub enum WrappingClause<'a> {
     Update,
     Delete,
     PolicyName,
-    ToRole,
+    ToRoleAssignment,
 }
 
 /// We can map a few nodes, such as the "update" node, to actual SQL clauses.
@@ -200,7 +200,7 @@ impl<'a> CompletionContext<'a> {
                 "policy_name" if policy_context.statement_kind != PolicyStmtKind::Create => {
                     Some(WrappingClause::PolicyName)
                 }
-                "policy_role" => Some(WrappingClause::ToRole),
+                "policy_role" => Some(WrappingClause::ToRoleAssignment),
                 "policy_table" => Some(WrappingClause::From),
                 _ => None,
             };

crates/pgt_workspace/src/workspace/server.rs

@@ -481,7 +481,6 @@ impl Workspace for WorkspaceServer {
             Some(pool) => pool,
             None => {
                 tracing::debug!("No connection to database. Skipping completions.");
-                tracing::warn!("No connection to database.");
                 return Ok(CompletionsResult::default());
             }
         };
@@ -490,11 +489,10 @@ impl Workspace for WorkspaceServer {
 
         match get_statement_for_completions(&parsed_doc, params.position) {
             None => {
-                tracing::warn!("No statement found.");
+                tracing::debug!("No statement found.");
                 Ok(CompletionsResult::default())
             }
-            Some((_id, range, content, cst)) => {
-                tracing::warn!("found matching statement, content: {}", content);
+            Some((id, range, content, cst)) => {
                 let position = params.position - range.start();
 
                 let items = pgt_completions::complete(pgt_completions::CompletionParams {
@@ -504,6 +502,12 @@ impl Workspace for WorkspaceServer {
                     text: content,
                 });
 
+                tracing::debug!(
+                    "Found {} completion items for statement with id {}",
+                    items.len(),
+                    id.raw()
+                );
+
                 Ok(CompletionsResult { items })
             }
         }