wow!

Author: juleswritescode

crates/pgt_completions/src/providers/tables.rs

@@ -379,4 +379,28 @@ mod tests {
         )
         .await;
     }
+
+    #[tokio::test]
+    async fn suggests_tables_in_insert_into() {
+        let setup = r#"
+            create schema auth;
+
+            create table auth.users (
+                uid serial primary key,
+                name text not null,
+                email text unique not null
+            );
+        "#;
+
+        assert_complete_results(
+            format!("insert into {}", CURSOR_POS).as_str(),
+            vec![
+                CompletionAssertion::LabelAndKind("public".into(), CompletionItemKind::Schema),
+                CompletionAssertion::LabelAndKind("auth".into(), CompletionItemKind::Schema),
+                CompletionAssertion::LabelAndKind("users".into(), CompletionItemKind::Table),
+            ],
+            setup,
+        )
+        .await;
+    }
 }

crates/pgt_completions/src/relevance/filtering.rs

@@ -81,7 +81,7 @@ impl CompletionFilter<'_> {
                         WrappingClause::Insert => {
                             ctx.wrapping_node_kind
                                 .as_ref()
-                                .is_some_and(|n| n != &WrappingNode::List)
+                                .is_none_or(|n| n != &WrappingNode::List)
                                 && ctx.before_cursor_matches_kind(&["keyword_into"])
                         }
 
@@ -148,7 +148,7 @@ impl CompletionFilter<'_> {
                         WrappingClause::Insert => {
                             ctx.wrapping_node_kind
                                 .as_ref()
-                                .is_some_and(|n| n != &WrappingNode::List)
+                                .is_none_or(|n| n != &WrappingNode::List)
                                 && ctx.before_cursor_matches_kind(&["keyword_into"])
                         }