//! Completion for derives
-use hir::HasAttrs;
+use hir::{HasAttrs, MacroDef, MacroKind};
+use ide_db::{
+ helpers::{import_assets::ImportAssets, insert_use::ImportScope},
+ SymbolKind,
+};
use itertools::Itertools;
-use rustc_hash::FxHashMap;
-use syntax::ast;
+use rustc_hash::FxHashSet;
+use syntax::{ast, SmolStr, SyntaxKind};
use crate::{
- context::CompletionContext,
- item::{CompletionItem, CompletionItemKind, CompletionKind},
- Completions,
+ completions::flyimport::compute_fuzzy_completion_order_key, context::CompletionContext,
+ item::CompletionItem, Completions, ImportEdit,
};
-pub(super) fn complete_derive(
- acc: &mut Completions,
- ctx: &CompletionContext,
- derive_input: ast::TokenTree,
-) {
- if let Some(existing_derives) = super::parse_comma_sep_input(derive_input) {
- for (derive, docs) in get_derive_names_in_scope(ctx) {
- let (label, lookup) = if let Some(derive_completion) = DEFAULT_DERIVE_COMPLETIONS
- .iter()
- .find(|derive_completion| derive_completion.label == derive)
- {
- let mut components = vec![derive_completion.label];
- components.extend(
- derive_completion
- .dependencies
- .iter()
- .filter(|&&dependency| !existing_derives.contains(dependency)),
- );
- let lookup = components.join(", ");
- let label = components.iter().rev().join(", ");
- (label, Some(lookup))
- } else {
- (derive, None)
- };
- let mut item =
- CompletionItem::new(CompletionKind::Attribute, ctx.source_range(), label);
- item.kind(CompletionItemKind::Attribute);
- if let Some(docs) = docs {
- item.documentation(docs);
- }
- if let Some(lookup) = lookup {
- item.lookup_by(lookup);
+pub(super) fn complete_derive(acc: &mut Completions, ctx: &CompletionContext, attr: &ast::Attr) {
+ 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;
+ }
+
+ let name = name.to_smol_str();
+ let (label, lookup) = match core.zip(mac.module(ctx.db).map(|it| it.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()
+ .filter_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)
+ }
}
- item.add_to(acc);
+ _ => (name, None),
+ };
+
+ let mut item = CompletionItem::new(SymbolKind::Derive, ctx.source_range(), label);
+ if let Some(docs) = mac.docs(ctx.db) {
+ item.documentation(docs);
}
+ if let Some(lookup) = lookup {
+ item.lookup_by(lookup);
+ }
+ item.add_to(acc);
}
+
+ flyimport_derive(acc, ctx);
}
-fn get_derive_names_in_scope(
- ctx: &CompletionContext,
-) -> FxHashMap<String, Option<hir::Documentation>> {
- let mut result = FxHashMap::default();
- ctx.scope.process_all_names(&mut |name, scope_def| {
+fn get_derives_in_scope(ctx: &CompletionContext) -> Vec<(hir::Name, MacroDef)> {
+ let mut result = Vec::default();
+ ctx.process_all_names(&mut |name, scope_def| {
if let hir::ScopeDef::MacroDef(mac) = scope_def {
if mac.kind() == hir::MacroKind::Derive {
- result.insert(name.to_string(), mac.docs(ctx.db));
+ result.push((name, mac));
}
}
});
result
}
+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() == 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);
+ }
+ Some(item.build())
+ }),
+ );
+ Some(())
+}
+
struct DeriveDependencies {
label: &'static str,
dependencies: &'static [&'static str],
/// Standard Rust derives that have dependencies
/// (the dependencies are needed so that the main derive don't break the compilation when added)
-const DEFAULT_DERIVE_COMPLETIONS: &[DeriveDependencies] = &[
+const DEFAULT_DERIVE_DEPENDENCIES: &[DeriveDependencies] = &[
DeriveDependencies { label: "Copy", dependencies: &["Clone"] },
DeriveDependencies { label: "Eq", dependencies: &["PartialEq"] },
DeriveDependencies { label: "Ord", dependencies: &["PartialOrd", "Eq", "PartialEq"] },
DeriveDependencies { label: "PartialOrd", dependencies: &["PartialEq"] },
];
-
-#[cfg(test)]
-mod tests {
- use expect_test::{expect, Expect};
-
- use crate::{test_utils::completion_list, CompletionKind};
-
- fn check(ra_fixture: &str, expect: Expect) {
- let actual = completion_list(ra_fixture, CompletionKind::Attribute);
- expect.assert_eq(&actual);
- }
-
- #[test]
- fn no_completion_for_incorrect_derive() {
- check(r#"#[derive{$0)] struct Test;"#, expect![[]])
- }
-
- #[test]
- #[ignore] // FIXME: Fixtures cant test proc-macros/derives yet as we cant specify them in fixtures
- fn empty_derive() {
- check(
- r#"#[derive($0)] struct Test;"#,
- expect![[r#"
- at Clone
- at Clone, Copy
- at Debug
- at Default
- at Hash
- at PartialEq
- at PartialEq, Eq
- at PartialEq, PartialOrd
- at PartialEq, Eq, PartialOrd, Ord
- "#]],
- );
- }
-
- #[test]
- #[ignore] // FIXME: Fixtures cant test proc-macros/derives yet as we cant specify them in fixtures
- fn derive_with_input() {
- check(
- r#"#[derive(serde::Serialize, PartialEq, $0)] struct Test;"#,
- expect![[r#"
- at Clone
- at Clone, Copy
- at Debug
- at Default
- at Hash
- at Eq
- at PartialOrd
- at Eq, PartialOrd, Ord
- "#]],
- )
- }
-
- #[test]
- #[ignore] // FIXME: Fixtures cant test proc-macros/derives yet as we cant specify them in fixtures
- fn derive_with_input2() {
- check(
- r#"#[derive($0 serde::Serialize, PartialEq)] struct Test;"#,
- expect![[r#"
- at Clone
- at Clone, Copy
- at Debug
- at Default
- at Hash
- at Eq
- at PartialOrd
- at Eq, PartialOrd, Ord
- "#]],
- )
- }
-}