Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/ide_completion/src/completions/attribute/derive.rs')
| -rw-r--r-- | crates/ide_completion/src/completions/attribute/derive.rs | 196 |
1 files changed, 92 insertions, 104 deletions
diff --git a/crates/ide_completion/src/completions/attribute/derive.rs b/crates/ide_completion/src/completions/attribute/derive.rs index 64f6e3989d..1edc92d5d6 100644 --- a/crates/ide_completion/src/completions/attribute/derive.rs +++ b/crates/ide_completion/src/completions/attribute/derive.rs @@ -1,123 +1,111 @@ //! Completion for derives -use hir::{HasAttrs, Macro, MacroKind}; -use ide_db::{ - imports::{import_assets::ImportAssets, insert_use::ImportScope}, - SymbolKind, -}; +use hir::{HasAttrs, ScopeDef}; +use ide_db::SymbolKind; use itertools::Itertools; -use rustc_hash::FxHashSet; -use syntax::{ast, SmolStr, SyntaxKind}; +use syntax::SmolStr; use crate::{ - completions::flyimport::compute_fuzzy_completion_order_key, context::CompletionContext, - item::CompletionItem, Completions, ImportEdit, + context::{CompletionContext, PathCompletionCtx, PathKind, PathQualifierCtx}, + item::CompletionItem, + Completions, }; -pub(super) fn complete_derive(acc: &mut Completions, ctx: &CompletionContext, attr: &ast::Attr) { +pub(crate) fn complete_derive(acc: &mut Completions, ctx: &CompletionContext) { + let (qualifier, is_absolute_path) = match ctx.path_context { + Some(PathCompletionCtx { + kind: Some(PathKind::Derive), + ref qualifier, + is_absolute_path, + .. + }) => (qualifier, is_absolute_path), + _ => return, + }; + let core = ctx.famous_defs().core(); - let existing_derives: FxHashSet<_> = - ctx.sema.resolve_derive_macro(attr).into_iter().flatten().flatten().collect(); - for (name, mac) in get_derives_in_scope(ctx) { - if existing_derives.contains(&mac) { - continue; - } + match qualifier { + Some(PathQualifierCtx { resolution, is_super_chain, .. }) => { + if *is_super_chain { + acc.add_keyword(ctx, "super::"); + } - let name = name.to_smol_str(); - let (label, lookup) = match (core, mac.module(ctx.db).krate()) { - // show derive dependencies for `core`/`std` derives - (Some(core), mac_krate) if core == mac_krate => { - if let Some(derive_completion) = DEFAULT_DERIVE_DEPENDENCIES - .iter() - .find(|derive_completion| derive_completion.label == name) - { - let mut components = vec![derive_completion.label]; - components.extend(derive_completion.dependencies.iter().filter( - |&&dependency| { - !existing_derives - .iter() - .map(|it| it.name(ctx.db)) - .any(|it| it.to_smol_str() == dependency) - }, - )); - let lookup = components.join(", "); - let label = Itertools::intersperse(components.into_iter().rev(), ", "); - (SmolStr::from_iter(label), Some(lookup)) - } else { - (name, None) + let module = match resolution { + Some(hir::PathResolution::Def(hir::ModuleDef::Module(it))) => it, + _ => return, + }; + + for (name, def) in module.scope(ctx.db, ctx.module) { + let add_def = match def { + ScopeDef::ModuleDef(hir::ModuleDef::Macro(mac)) => { + !ctx.existing_derives.contains(&mac) && mac.is_derive(ctx.db) + } + ScopeDef::ModuleDef(hir::ModuleDef::Module(_)) => true, + _ => false, + }; + if add_def { + acc.add_resolution(ctx, name, def); } } - _ => (name, None), - }; - - let mut item = CompletionItem::new(SymbolKind::Derive, ctx.source_range(), label); - if let Some(docs) = mac.docs(ctx.db) { - item.documentation(docs); + return; } - if let Some(lookup) = lookup { - item.lookup_by(lookup); - } - item.add_to(acc); - } + None if is_absolute_path => acc.add_crate_roots(ctx), + // only show modules in a fresh UseTree + None => { + ctx.process_all_names(&mut |name, def| { + let mac = match def { + ScopeDef::ModuleDef(hir::ModuleDef::Macro(mac)) + if !ctx.existing_derives.contains(&mac) && mac.is_derive(ctx.db) => + { + mac + } + ScopeDef::ModuleDef(hir::ModuleDef::Module(_)) => { + return acc.add_resolution(ctx, name, def); + } + _ => return, + }; - flyimport_derive(acc, ctx); -} + match (core, mac.module(ctx.db).krate()) { + // show derive dependencies for `core`/`std` derives + (Some(core), mac_krate) if core == mac_krate && qualifier.is_none() => {} + _ => return acc.add_resolution(ctx, name, def), + }; -fn get_derives_in_scope(ctx: &CompletionContext) -> Vec<(hir::Name, Macro)> { - let mut result = Vec::default(); - ctx.process_all_names(&mut |name, scope_def| { - if let hir::ScopeDef::ModuleDef(hir::ModuleDef::Macro(mac)) = scope_def { - if mac.kind(ctx.db) == hir::MacroKind::Derive { - result.push((name, mac)); - } - } - }); - result -} + let name_ = name.to_smol_str(); + let find = DEFAULT_DERIVE_DEPENDENCIES + .iter() + .find(|derive_completion| derive_completion.label == name_); -fn flyimport_derive(acc: &mut Completions, ctx: &CompletionContext) -> Option<()> { - if ctx.token.kind() != SyntaxKind::IDENT { - return None; - }; - let potential_import_name = ctx.token.to_string(); - let module = ctx.module?; - let parent = ctx.token.parent()?; - let user_input_lowercased = potential_import_name.to_lowercase(); - let import_assets = ImportAssets::for_fuzzy_path( - module, - None, - potential_import_name, - &ctx.sema, - parent.clone(), - )?; - let import_scope = ImportScope::find_insert_use_container(&parent, &ctx.sema)?; - acc.add_all( - import_assets - .search_for_imports(&ctx.sema, ctx.config.insert_use.prefix_kind) - .into_iter() - .filter_map(|import| match import.original_item { - hir::ItemInNs::Macros(mac) => Some((import, mac)), - _ => None, - }) - .filter(|&(_, mac)| mac.kind(ctx.db) == MacroKind::Derive) - .filter(|&(_, mac)| !ctx.is_item_hidden(&hir::ItemInNs::Macros(mac))) - .sorted_by_key(|(import, _)| { - compute_fuzzy_completion_order_key(&import.import_path, &user_input_lowercased) - }) - .filter_map(|(import, mac)| { - let mut item = CompletionItem::new( - SymbolKind::Derive, - ctx.source_range(), - mac.name(ctx.db).to_smol_str(), - ); - item.add_import(ImportEdit { import, scope: import_scope.clone() }); - if let Some(docs) = mac.docs(ctx.db) { - item.documentation(docs); + match find { + Some(derive_completion) => { + let mut components = vec![derive_completion.label]; + components.extend(derive_completion.dependencies.iter().filter( + |&&dependency| { + !ctx.existing_derives + .iter() + .map(|it| it.name(ctx.db)) + .any(|it| it.to_smol_str() == dependency) + }, + )); + let lookup = components.join(", "); + let label = Itertools::intersperse(components.into_iter().rev(), ", "); + + let mut item = CompletionItem::new( + SymbolKind::Derive, + ctx.source_range(), + SmolStr::from_iter(label), + ); + if let Some(docs) = mac.docs(ctx.db) { + item.documentation(docs); + } + item.lookup_by(lookup); + item.add_to(acc); + } + None => acc.add_resolution(ctx, name, def), } - Some(item.build()) - }), - ); - Some(()) + }); + acc.add_nameref_keywords(ctx); + } + } } struct DeriveDependencies { |