it works!

Author: juleswritescode

crates/pgt_completions/src/context/context.rs

@@ -209,6 +209,8 @@ impl<'a> CompletionContext<'a> {
             ctx.gather_info_from_ts_queries();
         }
 
+        tracing::warn!("{:#?}", ctx.get_node_under_cursor_content());
+
         ctx
     }
 

crates/pgt_completions/src/providers/helper.rs

@@ -1,6 +1,6 @@
 use pgt_text_size::{TextRange, TextSize};
 
-use crate::{CompletionText, context::CompletionContext};
+use crate::{CompletionText, context::CompletionContext, remove_sanitized_token};
 
 pub(crate) fn find_matching_alias_for_table(
     ctx: &CompletionContext,
@@ -15,22 +15,18 @@ pub(crate) fn find_matching_alias_for_table(
 }
 
 pub(crate) fn get_range_to_replace(ctx: &CompletionContext) -> TextRange {
-    let start = ctx
-        .node_under_cursor
-        .as_ref()
-        .map(|n| n.start_byte())
-        .unwrap_or(0);
+    match ctx.node_under_cursor.as_ref() {
+        Some(node) => {
+            let content = ctx.get_node_under_cursor_content().unwrap_or("".into());
+            let length = remove_sanitized_token(content.as_str()).len();
 
-    let end = ctx
-        .get_node_under_cursor_content()
-        .unwrap_or("".into())
-        .len()
-        + start;
+            let start = node.start_byte();
+            let end = start + length;
 
-    TextRange::new(
-        TextSize::new(start.try_into().unwrap()),
-        end.try_into().unwrap(),
-    )
+            TextRange::new(start.try_into().unwrap(), end.try_into().unwrap())
+        }
+        None => TextRange::empty(TextSize::new(0)),
+    }
 }
 
 pub(crate) fn get_completion_text_with_schema_or_alias(

crates/pgt_completions/src/providers/policies.rs

@@ -1,3 +1,5 @@
+use pgt_text_size::{TextRange, TextSize};
+
 use crate::{
     CompletionItemKind, CompletionText,
     builder::{CompletionBuilder, PossibleCompletionItem},
@@ -10,9 +12,9 @@ use super::helper::get_range_to_replace;
 pub fn complete_policies<'a>(ctx: &CompletionContext<'a>, builder: &mut CompletionBuilder<'a>) {
     let available_policies = &ctx.schema_cache.policies;
 
-    let has_quotes = ctx
+    let surrounded_by_quotes = ctx
         .get_node_under_cursor_content()
-        .is_some_and(|c| c.starts_with('"') && c.ends_with('"'));
+        .is_some_and(|c| c.starts_with('"') && c.ends_with('"') && c != "\"\"");
 
     for pol in available_policies {
         let relevance = CompletionRelevanceData::Policy(pol);
@@ -23,13 +25,22 @@ pub fn complete_policies<'a>(ctx: &CompletionContext<'a>, builder: &mut Completi
             filter: CompletionFilter::from(relevance),
             description: format!("{}", pol.table_name),
             kind: CompletionItemKind::Policy,
-            completion_text: if !has_quotes {
+            completion_text: if !surrounded_by_quotes {
                 Some(CompletionText {
                     text: format!("\"{}\"", pol.name),
                     range: get_range_to_replace(ctx),
                 })
             } else {
-                None
+                let range = get_range_to_replace(ctx);
+                Some(CompletionText {
+                    text: pol.name.clone(),
+
+                    // trim the quotes.
+                    range: TextRange::new(
+                        range.start() + TextSize::new(1),
+                        range.end() - TextSize::new(1),
+                    ),
+                })
             },
         };
 

crates/pgt_completions/src/sanitization.rs

@@ -18,6 +18,10 @@ pub fn benchmark_sanitization(params: CompletionParams) -> String {
     params.text
 }
 
+pub(crate) fn remove_sanitized_token(it: &str) -> String {
+    it.replace(SANITIZED_TOKEN, "")
+}
+
 #[derive(PartialEq, Eq, Debug)]
 pub(crate) enum NodeText {
     Replaced,
@@ -157,6 +161,7 @@ fn cursor_on_a_dot(sql: &str, position: TextSize) -> bool {
 }
 
 fn cursor_between_double_quotes(sql: &str, position: TextSize) -> bool {
+    return false;
     let position: usize = position.into();
     let mut chars = sql.chars();
     chars.nth(position - 1).is_some_and(|c| c == '"') && chars.next().is_some_and(|c| c == '"')

crates/pgt_lsp/src/handlers/completions.rs

@@ -54,8 +54,6 @@ pub fn get_completions(
         })
         .collect();
 
-    tracing::warn!("{:#?}", items);
-
     Ok(lsp_types::CompletionResponse::Array(items))
 }