fix(completions): adjust range for later stmts

Author: juleswritescode

crates/pgls_lsp/src/handlers/completions.rs

@@ -20,10 +20,12 @@ pub fn get_completions(
     let doc = session.document(&url)?;
     let encoding = adapters::negotiated_encoding(session.client_capabilities().unwrap());
 
-    let completion_result = match session.workspace.get_completions(GetCompletionsParams {
-        path,
-        position: get_cursor_position(session, &url, params.text_document_position.position)?,
-    }) {
+    let position = get_cursor_position(session, &url, params.text_document_position.position)?;
+
+    let completion_result = match session
+        .workspace
+        .get_completions(GetCompletionsParams { path, position })
+    {
         Ok(result) => result,
         Err(e) => match e {
             WorkspaceError::DatabaseConnectionError(_) => {

crates/pgls_workspace/src/features/completions.rs

@@ -21,6 +21,18 @@ pub struct CompletionsResult {
     pub(crate) items: Vec<CompletionItem>,
 }
 
+impl CompletionsResult {
+    pub fn with_offset(mut items: Vec<CompletionItem>, offset: TextSize) -> Self {
+        for item in &mut items {
+            if let Some(text_edit) = &mut item.completion_text {
+                text_edit.range += offset;
+            }
+        }
+
+        Self { items }
+    }
+}
+
 impl IntoIterator for CompletionsResult {
     type Item = CompletionItem;
     type IntoIter = <Vec<CompletionItem> as IntoIterator>::IntoIter;

crates/pgls_workspace/src/workspace/server.rs

@@ -746,7 +746,7 @@ impl Workspace for WorkspaceServer {
                     text: id.content().to_string(),
                 });
 
-                Ok(CompletionsResult { items })
+                Ok(CompletionsResult::with_offset(items, range.start()))
             }
         }
     }

crates/pgls_workspace/src/workspace/server.tests.rs

@@ -793,3 +793,79 @@ async fn test_search_path_configuration(test_db: PgPool) {
         );
     }
 }
+
+#[sqlx::test(migrator = "pgls_test_utils::MIGRATIONS")]
+async fn test_multi_line_completions(test_db: PgPool) {
+    let setup = r#"
+        create schema auth;
+
+        create table auth.users (
+            id serial primary key,
+            email text not null
+        );
+    "#;
+
+    test_db.execute(setup).await.expect("setup sql failed");
+
+    let mut conf = PartialConfiguration::init();
+    conf.merge_with(PartialConfiguration {
+        db: Some(PartialDatabaseConfiguration {
+            database: Some(
+                test_db
+                    .connect_options()
+                    .get_database()
+                    .unwrap()
+                    .to_string(),
+            ),
+            ..Default::default()
+        }),
+        ..Default::default()
+    });
+
+    let workspace = get_test_workspace(Some(conf)).expect("Unable to create test workspace");
+
+    let path = PgLSPath::new("test.sql");
+
+    let content = r#"
+select * from auth.users;
+
+select * from auth.|;
+
+select * from auth.users;
+    "#
+    .trim();
+
+    let position = content
+        .find('|')
+        .map(|idx| pgls_text_size::TextSize::new(idx as u32))
+        .expect("Unable to find cursor position in test content");
+
+    let sanitized_content = content.replace('|', "");
+
+    workspace
+        .open_file(OpenFileParams {
+            path: path.clone(),
+            content: sanitized_content.into(),
+            version: 1,
+        })
+        .expect("Unable to open test file");
+
+    let completions = workspace
+        .get_completions(crate::workspace::GetCompletionsParams {
+            path: path.clone(),
+            position,
+        })
+        .expect("Unable to request completions");
+
+    assert_eq!(
+        completions.items.len(),
+        1,
+        "Expected one completion response"
+    );
+
+    assert_eq!(
+        completions.items[0].completion_text.as_ref().unwrap().range,
+        TextRange::new(position, position),
+        "Expected no syntax diagnostic"
+    );
+}