lil cleanup of text_document_completion.rs

prisma · Apr 24, 2024 · 8aa631a · 8aa631a
1 parent 461fac7
commit 8aa631a
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 13 deletions.
diff --git a/prisma-fmt/src/text_document_completion.rs b/prisma-fmt/src/text_document_completion.rs
@@ -13,6 +13,8 @@ use std::sync::Arc;
 use crate::position_to_offset;
 
 mod datasource;
+mod multi_schema;
+mod referential_actions;
 
 pub(crate) fn empty_completion_list() -> CompletionList {
     CompletionList {
@@ -97,13 +99,7 @@ fn push_ast_completions(ctx: CompletionContext<'_>, completion_list: &mut Comple
             ast::ModelPosition::Field(_, ast::FieldPosition::Attribute("relation", _, Some(attr_name))),
         ) if attr_name == "onDelete" || attr_name == "onUpdate" => {
             for referential_action in ctx.connector().referential_actions().iter() {
-                completion_list.items.push(CompletionItem {
-                    label: referential_action.as_str().to_owned(),
-                    kind: Some(CompletionItemKind::ENUM),
-                    // what is the difference between detail and documentation?
-                    detail: Some(referential_action.documentation().to_owned()),
-                    ..Default::default()
-                });
+                referential_actions::referential_action_completion(completion_list, referential_action)
             }
         }
 
@@ -196,12 +192,7 @@ fn push_namespaces(ctx: CompletionContext<'_>, completion_list: &mut CompletionL
             namespace.to_string()
         };
 
-        completion_list.items.push(CompletionItem {
-            label: String::from(namespace),
-            insert_text: Some(insert_text),
-            kind: Some(CompletionItemKind::PROPERTY),
-            ..Default::default()
-        })
+        multi_schema::schema_namespace_completion(completion_list, namespace, insert_text);
     }
 }
 

diff --git a/prisma-fmt/src/text_document_completion/multi_schema.rs b/prisma-fmt/src/text_document_completion/multi_schema.rs
@@ -0,0 +1,14 @@
+use lsp_types::{CompletionItem, CompletionItemKind, CompletionList};
+
+pub(super) fn schema_namespace_completion(
+    completion_list: &mut CompletionList,
+    namespace: &String,
+    insert_text: String,
+) {
+    completion_list.items.push(CompletionItem {
+        label: String::from(namespace),
+        insert_text: Some(insert_text),
+        kind: Some(CompletionItemKind::PROPERTY),
+        ..Default::default()
+    })
+}
diff --git a/prisma-fmt/src/text_document_completion/referential_actions.rs b/prisma-fmt/src/text_document_completion/referential_actions.rs
@@ -0,0 +1,15 @@
+use lsp_types::{CompletionItem, CompletionItemKind, CompletionList};
+use psl::parser_database::ReferentialAction;
+
+pub(super) fn referential_action_completion(
+    completion_list: &mut CompletionList,
+    referential_action: ReferentialAction,
+) {
+    completion_list.items.push(CompletionItem {
+        label: referential_action.as_str().to_owned(),
+        kind: Some(CompletionItemKind::ENUM),
+        // ? (@tomhoule) what is the difference between detail and documentation?
+        detail: Some(referential_action.documentation().to_owned()),
+        ..Default::default()
+    })
+}