bendns' repos / rust-analyzer
Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/ide-completion/src/completions/vis.rs')
-rw-r--r--crates/ide-completion/src/completions/vis.rs52
1 files changed, 52 insertions, 0 deletions
diff --git a/crates/ide-completion/src/completions/vis.rs b/crates/ide-completion/src/completions/vis.rs
new file mode 100644
index 0000000000..7315a488b8
--- /dev/null
+++ b/crates/ide-completion/src/completions/vis.rs
@@ -0,0 +1,52 @@
+//! Completion for visibility specifiers.
+
+use hir::ScopeDef;
+
+use crate::{
+ context::{CompletionContext, PathCompletionCtx, PathKind, PathQualifierCtx},
+ Completions,
+};
+
+pub(crate) fn complete_vis(acc: &mut Completions, ctx: &CompletionContext) {
+ let (&is_absolute_path, qualifier, &has_in_token) = match &ctx.path_context {
+ Some(PathCompletionCtx {
+ kind: Some(PathKind::Vis { has_in_token }),
+ is_absolute_path,
+ qualifier,
+ ..
+ }) => (is_absolute_path, qualifier, has_in_token),
+ _ => return,
+ };
+
+ match qualifier {
+ Some(PathQualifierCtx { resolution, is_super_chain, .. }) => {
+ // Try completing next child module of the path that is still a parent of the current module
+ if let Some(hir::PathResolution::Def(hir::ModuleDef::Module(module))) = resolution {
+ let next_towards_current = ctx
+ .module
+ .path_to_root(ctx.db)
+ .into_iter()
+ .take_while(|it| it != module)
+ .last();
+ if let Some(next) = next_towards_current {
+ if let Some(name) = next.name(ctx.db) {
+ cov_mark::hit!(visibility_qualified);
+ acc.add_resolution(ctx, name, ScopeDef::ModuleDef(next.into()));
+ }
+ }
+ }
+
+ if *is_super_chain {
+ acc.add_keyword(ctx, "super::");
+ }
+ }
+ None if !is_absolute_path => {
+ if !has_in_token {
+ cov_mark::hit!(kw_completion_in);
+ acc.add_keyword(ctx, "in");
+ }
+ acc.add_nameref_keywords(ctx);
+ }
+ _ => {}
+ }
+}