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.rs | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/crates/ide_completion/src/config.rs b/crates/ide_completion/src/config.rs index c300ce887b..c659b4455a 100644 --- a/crates/ide_completion/src/config.rs +++ b/crates/ide_completion/src/config.rs @@ -6,6 +6,8 @@ use ide_db::helpers::{insert_use::InsertUseConfig, SnippetCap}; +use crate::snippet::Snippet; + #[derive(Clone, Debug, PartialEq, Eq)] pub struct CompletionConfig { pub enable_postfix_completions: bool, @@ -15,4 +17,18 @@ pub struct CompletionConfig { pub add_call_argument_snippets: bool, pub snippet_cap: Option<SnippetCap>, pub insert_use: InsertUseConfig, + 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)) + }) + } } |