crates/ide_completion/src/completions/pattern.rs

   1 //! Completes constants and paths in unqualified patterns.
   2
   3 use crate::{
   4     context::{PatternContext, PatternRefutability},
   5     CompletionContext, Completions,
   6 };
   7
   8 /// Completes constants and paths in unqualified patterns.
   9 pub(crate) fn complete_pattern(acc: &mut Completions, ctx: &CompletionContext) {
  10     let refutable = match ctx.pattern_ctx {
  11         Some(PatternContext { refutability, .. }) if ctx.path_context.is_none() => {
  12             refutability == PatternRefutability::Refutable
  13         }
  14         _ => return,
  15     };
  16
  17     if refutable {
  18         if let Some(hir::Adt::Enum(e)) =
  19             ctx.expected_type.as_ref().and_then(|ty| ty.strip_references().as_adt())
  20         {
  21             super::enum_variants_with_paths(acc, ctx, e, |acc, ctx, variant, path| {
  22                 acc.add_qualified_variant_pat(ctx, variant, path.clone());
  23                 acc.add_qualified_enum_variant(ctx, variant, path);
  24             });
  25         }
  26     }
  27
  28     // FIXME: ideally, we should look at the type we are matching against and
  29     // suggest variants + auto-imports
  30     ctx.process_all_names(&mut |name, res| {
  31         let add_resolution = match &res {
  32             hir::ScopeDef::ModuleDef(def) => match def {
  33                 hir::ModuleDef::Adt(hir::Adt::Struct(strukt)) => {
  34                     acc.add_struct_pat(ctx, *strukt, Some(name.clone()));
  35                     true
  36                 }
  37                 hir::ModuleDef::Variant(variant) if refutable => {
  38                     acc.add_variant_pat(ctx, *variant, Some(name.clone()));
  39                     true
  40                 }
  41                 hir::ModuleDef::Adt(hir::Adt::Enum(..))
  42                 | hir::ModuleDef::Variant(..)
  43                 | hir::ModuleDef::Const(..)
  44                 | hir::ModuleDef::Module(..) => refutable,
  45                 _ => false,
  46             },
  47             hir::ScopeDef::MacroDef(mac) => mac.is_fn_like(),
  48             hir::ScopeDef::ImplSelfType(impl_) => match impl_.self_ty(ctx.db).as_adt() {
  49                 Some(hir::Adt::Struct(strukt)) => {
  50                     acc.add_struct_pat(ctx, strukt, Some(name.clone()));
  51                     true
  52                 }
  53                 Some(hir::Adt::Enum(_)) => refutable,
  54                 _ => true,
  55             },
  56             _ => false,
  57         };
  58         if add_resolution {
  59             acc.add_resolution(ctx, name, &res);
  60         }
  61     });
  62 }