feat(completions): consider invocations in score

Author: juleswritescode

crates/pg_completions/src/providers/functions.rs

@@ -1,7 +1,8 @@
 use pg_schema_cache::Function;
 
 use crate::{
-    builder::CompletionBuilder, context::CompletionContext, CompletionItem, CompletionItemKind,
+    builder::CompletionBuilder, context::CompletionContext, relevance::CompletionRelevanceData,
+    CompletionItem, CompletionItemKind,
 };
 
 pub fn complete_functions(ctx: &CompletionContext, builder: &mut CompletionBuilder) {
@@ -11,7 +12,7 @@ pub fn complete_functions(ctx: &CompletionContext, builder: &mut CompletionBuild
         .iter()
         .map(|foo| CompletionItem {
             label: foo.name.clone(),
-            score: get_score(ctx, foo),
+            score: CompletionRelevanceData::Function(foo).get_score(ctx),
             description: format!("Schema: {}", foo.schema),
             preselected: false,
             kind: CompletionItemKind::Function,
@@ -23,10 +24,6 @@ pub fn complete_functions(ctx: &CompletionContext, builder: &mut CompletionBuild
     }
 }
 
-fn get_score(ctx: &CompletionContext, foo: &Function) -> i32 {
-    0
-}
-
 #[cfg(test)]
 mod tests {
     use crate::{

crates/pg_completions/src/relevance.rs

@@ -32,6 +32,7 @@ impl<'a> CompletionRelevance<'a> {
         self.check_matches_schema(ctx);
         self.check_matches_query_input(ctx);
         self.check_if_catalog(ctx);
+        self.check_is_invocation(ctx);
 
         self.score
     }
@@ -59,6 +60,25 @@ impl<'a> CompletionRelevance<'a> {
         };
     }
 
+    fn check_is_invocation(&mut self, ctx: &CompletionContext) {
+        self.score += match self.data {
+            CompletionRelevanceData::Function(_) => {
+                if ctx.is_invocation {
+                    30
+                } else {
+                    -30
+                }
+            }
+            _ => {
+                if ctx.is_invocation {
+                    -10
+                } else {
+                    0
+                }
+            }
+        };
+    }
+
     fn check_matches_schema(&mut self, ctx: &CompletionContext) {
         let schema_name = match ctx.schema_name.as_ref() {
             None => return,