crates/ide-completion/src/config.rs

   1 //! Settings for tweaking completion.
   2 //!
   3 //! The fun thing here is `SnippetCap` -- this type can only be created in this
   4 //! module, and we use to statically check that we only produce snippet
   5 //! completions if we are allowed to.
   6
   7 use ide_db::{imports::insert_use::InsertUseConfig, SnippetCap};
   8
   9 use crate::snippet::Snippet;
  10
  11 #[derive(Clone, Debug, PartialEq, Eq)]
  12 pub struct CompletionConfig {
  13     pub enable_postfix_completions: bool,
  14     pub enable_imports_on_the_fly: bool,
  15     pub enable_self_on_the_fly: bool,
  16     pub enable_private_editable: bool,
  17     pub callable: Option<CallableSnippets>,
  18     pub snippet_cap: Option<SnippetCap>,
  19     pub insert_use: InsertUseConfig,
  20     pub snippets: Vec<Snippet>,
  21 }
  22
  23 #[derive(Clone, Debug, PartialEq, Eq)]
  24 pub enum CallableSnippets {
  25     FillArguments,
  26     AddParentheses,
  27 }
  28
  29 impl CompletionConfig {
  30     pub fn postfix_snippets(&self) -> impl Iterator<Item = (&str, &Snippet)> {
  31         self.snippets
  32             .iter()
  33             .flat_map(|snip| snip.postfix_triggers.iter().map(move |trigger| (&**trigger, snip)))
  34     }
  35
  36     pub fn prefix_snippets(&self) -> impl Iterator<Item = (&str, &Snippet)> {
  37         self.snippets
  38             .iter()
  39             .flat_map(|snip| snip.prefix_triggers.iter().map(move |trigger| (&**trigger, snip)))
  40     }
  41 }