ok

Author: juleswritescode

crates/pgt_completions/src/relevance/filtering.rs

@@ -99,6 +99,10 @@ impl CompletionFilter<'_> {
                     CompletionRelevanceData::Table(_) => match clause {
                         WrappingClause::From | WrappingClause::Update => true,
 
+                        WrappingClause::RevokeStatement => {
+                            ctx.matches_ancestor_history(&["revoke_on_table", "object_reference"])
+                        }
+
                         WrappingClause::Join { on_node: None } => true,
                         WrappingClause::Join { on_node: Some(on) } => ctx
                             .node_under_cursor
@@ -202,6 +206,12 @@ impl CompletionFilter<'_> {
                         | WrappingClause::Update
                         | WrappingClause::Delete => true,
 
+                        WrappingClause::RevokeStatement => {
+                            (ctx.matches_ancestor_history(&["revoke_on_table", "object_reference"])
+                                && ctx.schema_or_alias_name.is_none())
+                                || ctx.matches_ancestor_history(&["revoke_on_all"])
+                        }
+
                         WrappingClause::Where => {
                             ctx.before_cursor_matches_kind(&["keyword_and", "keyword_where"])
                         }

crates/pgt_hover/src/hovered_node.rs

@@ -32,7 +32,10 @@ impl HoveredNode {
         let under_cursor = ctx.node_under_cursor.as_ref()?;
 
         match under_cursor.kind() {
-            "identifier" if ctx.matches_ancestor_history(&["relation", "object_reference"]) => {
+            "identifier"
+                if ctx.matches_ancestor_history(&["relation", "object_reference"])
+                    || ctx.matches_ancestor_history(&["revoke_on_table", "object_reference"]) =>
+            {
                 let num_sibs = ctx.num_siblings();
                 if ctx.node_under_cursor_is_nth_child(1) && num_sibs > 0 {
                     return Some(HoveredNode::Schema(NodeIdentification::Name(node_content)));