Skip to content

Commit 60e7c1d

Browse files
committed
minor: address review comments
1 parent 7ec8434 commit 60e7c1d

File tree

4 files changed

+32
-37
lines changed

4 files changed

+32
-37
lines changed

crates/hir/src/lib.rs

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -346,6 +346,22 @@ impl ModuleDef {
346346
}
347347
}
348348

349+
impl HasVisibility for ModuleDef {
350+
fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
351+
match *self {
352+
ModuleDef::Module(it) => it.visibility(db),
353+
ModuleDef::Function(it) => it.visibility(db),
354+
ModuleDef::Adt(it) => it.visibility(db),
355+
ModuleDef::Const(it) => it.visibility(db),
356+
ModuleDef::Static(it) => it.visibility(db),
357+
ModuleDef::Trait(it) => it.visibility(db),
358+
ModuleDef::TypeAlias(it) => it.visibility(db),
359+
ModuleDef::Variant(it) => it.visibility(db),
360+
ModuleDef::BuiltinType(_) => Visibility::Public,
361+
}
362+
}
363+
}
364+
349365
impl Module {
350366
/// Name of this module.
351367
pub fn name(self, db: &dyn HirDatabase) -> Option<Name> {

crates/ide_assists/src/handlers/expand_glob_import.rs

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -809,22 +809,22 @@ fn baz(bar: Bar) {}
809809
",
810810
);
811811

812-
// check_assist_not_applicable(
813-
// expand_glob_import,
814-
// r"
815-
// mod foo {
816-
// mod bar {
817-
// pub mod baz {
818-
// pub struct Baz;
819-
// }
820-
// }
821-
// }
812+
check_assist_not_applicable(
813+
expand_glob_import,
814+
r"
815+
mod foo {
816+
mod bar {
817+
pub mod baz {
818+
pub struct Baz;
819+
}
820+
}
821+
}
822822
823-
// use foo::bar::baz::*$0;
823+
use foo::bar::baz::*$0;
824824
825-
// fn qux(baz: Baz) {}
826-
// ",
827-
// );
825+
fn qux(baz: Baz) {}
826+
",
827+
);
828828
}
829829

830830
#[test]

crates/ide_assists/src/handlers/fix_visibility.rs

Lines changed: 1 addition & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -46,18 +46,7 @@ fn add_vis_to_referenced_module_def(acc: &mut Assists, ctx: &AssistContext) -> O
4646
let current_module = ctx.sema.scope(path.syntax()).module()?;
4747
let target_module = def.module(ctx.db())?;
4848

49-
let vis = match def {
50-
hir::ModuleDef::Module(it) => it.visibility(ctx.db()),
51-
hir::ModuleDef::Function(it) => it.visibility(ctx.db()),
52-
hir::ModuleDef::Adt(it) => it.visibility(ctx.db()),
53-
hir::ModuleDef::Variant(it) => it.visibility(ctx.db()),
54-
hir::ModuleDef::Const(it) => it.visibility(ctx.db()),
55-
hir::ModuleDef::Static(it) => it.visibility(ctx.db()),
56-
hir::ModuleDef::Trait(it) => it.visibility(ctx.db()),
57-
hir::ModuleDef::TypeAlias(it) => it.visibility(ctx.db()),
58-
hir::ModuleDef::BuiltinType(_) => return None,
59-
};
60-
if vis.is_visible_from(ctx.db(), current_module.into()) {
49+
if def.visibility(ctx.db()).is_visible_from(ctx.db(), current_module.into()) {
6150
return None;
6251
};
6352

crates/ide_db/src/defs.rs

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -44,17 +44,7 @@ impl Definition {
4444
pub fn visibility(&self, db: &RootDatabase) -> Option<Visibility> {
4545
match self {
4646
Definition::Field(sf) => Some(sf.visibility(db)),
47-
Definition::ModuleDef(def) => match def {
48-
ModuleDef::Module(it) => Some(it.visibility(db)),
49-
ModuleDef::Function(it) => Some(it.visibility(db)),
50-
ModuleDef::Adt(it) => Some(it.visibility(db)),
51-
ModuleDef::Const(it) => Some(it.visibility(db)),
52-
ModuleDef::Static(it) => Some(it.visibility(db)),
53-
ModuleDef::Trait(it) => Some(it.visibility(db)),
54-
ModuleDef::TypeAlias(it) => Some(it.visibility(db)),
55-
ModuleDef::Variant(it) => Some(it.visibility(db)),
56-
ModuleDef::BuiltinType(_) => None,
57-
},
47+
Definition::ModuleDef(def) => Some(def.visibility(db)),
5848
Definition::Macro(_)
5949
| Definition::SelfType(_)
6050
| Definition::Local(_)

0 commit comments

Comments
 (0)