Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/ide_completion/src/config.rs')
-rw-r--r--crates/ide_completion/src/config.rs16
1 files changed, 14 insertions, 2 deletions
diff --git a/crates/ide_completion/src/config.rs b/crates/ide_completion/src/config.rs
index bf1dc125c7..c659b4455a 100644
--- a/crates/ide_completion/src/config.rs
+++ b/crates/ide_completion/src/config.rs
@@ -6,7 +6,7 @@
use ide_db::helpers::{insert_use::InsertUseConfig, SnippetCap};
-use crate::snippet::{PostfixSnippet, Snippet};
+use crate::snippet::Snippet;
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct CompletionConfig {
@@ -17,6 +17,18 @@ pub struct CompletionConfig {
pub add_call_argument_snippets: bool,
pub snippet_cap: Option<SnippetCap>,
pub insert_use: InsertUseConfig,
- pub postfix_snippets: Vec<PostfixSnippet>,
pub snippets: Vec<Snippet>,
}
+
+impl CompletionConfig {
+ pub fn postfix_snippets(&self) -> impl Iterator<Item = (&str, &Snippet)> {
+ self.snippets.iter().flat_map(|snip| {
+ snip.postfix_triggers.iter().map(move |trigger| (trigger.as_str(), snip))
+ })
+ }
+ pub fn prefix_snippets(&self) -> impl Iterator<Item = (&str, &Snippet)> {
+ self.snippets.iter().flat_map(|snip| {
+ snip.prefix_triggers.iter().map(move |trigger| (trigger.as_str(), snip))
+ })
+ }
+}