crates/hir/src/lib.rs

   1 //! HIR (previously known as descriptors) provides a high-level object oriented
   2 //! access to Rust code.
   3 //!
   4 //! The principal difference between HIR and syntax trees is that HIR is bound
   5 //! to a particular crate instance. That is, it has cfg flags and features
   6 //! applied. So, the relation between syntax and HIR is many-to-one.
   7 //!
   8 //! HIR is the public API of the all of the compiler logic above syntax trees.
   9 //! It is written in "OO" style. Each type is self contained (as in, it knows it's
  10 //! parents and full context). It should be "clean code".
  11 //!
  12 //! `hir_*` crates are the implementation of the compiler logic.
  13 //! They are written in "ECS" style, with relatively little abstractions.
  14 //! Many types are not self-contained, and explicitly use local indexes, arenas, etc.
  15 //!
  16 //! `hir` is what insulates the "we don't know how to actually write an incremental compiler"
  17 //! from the ide with completions, hovers, etc. It is a (soft, internal) boundary:
  18 //! <https://www.tedinski.com/2018/02/06/system-boundaries.html>.
  19
  20 #![recursion_limit = "512"]
  21
  22 mod semantics;
  23 mod source_analyzer;
  24
  25 mod from_id;
  26 mod attrs;
  27 mod has_source;
  28
  29 pub mod diagnostics;
  30 pub mod db;
  31
  32 mod display;
  33
  34 use std::{collections::HashMap, iter, ops::ControlFlow, sync::Arc};
  35
  36 use arrayvec::ArrayVec;
  37 use base_db::{CrateDisplayName, CrateId, CrateOrigin, Edition, FileId};
  38 use either::Either;
  39 use hir_def::{
  40     adt::{ReprKind, VariantData},
  41     body::{BodyDiagnostic, SyntheticSyntax},
  42     expr::{BindingAnnotation, LabelId, Pat, PatId},
  43     lang_item::LangItemTarget,
  44     nameres,
  45     per_ns::PerNs,
  46     resolver::{HasResolver, Resolver},
  47     AttrDefId, ConstId, ConstParamId, EnumId, FunctionId, GenericDefId, HasModule, LifetimeParamId,
  48     LocalEnumVariantId, LocalFieldId, StaticId, StructId, TypeAliasId, TypeParamId, UnionId,
  49 };
  50 use hir_expand::{name::name, MacroCallKind, MacroDefKind};
  51 use hir_ty::{
  52     autoderef,
  53     consteval::{eval_const, ComputedExpr, ConstEvalCtx, ConstEvalError, ConstExt},
  54     could_unify,
  55     diagnostics::BodyValidationDiagnostic,
  56     method_resolution::{self, TyFingerprint},
  57     primitive::UintTy,
  58     subst_prefix,
  59     traits::FnTrait,
  60     AliasEq, AliasTy, BoundVar, CallableDefId, CallableSig, Canonical, CanonicalVarKinds, Cast,
  61     DebruijnIndex, InEnvironment, Interner, QuantifiedWhereClause, Scalar, Solution, Substitution,
  62     TraitEnvironment, TraitRefExt, Ty, TyBuilder, TyDefId, TyExt, TyKind, TyVariableKind,
  63     WhereClause,
  64 };
  65 use itertools::Itertools;
  66 use nameres::diagnostics::DefDiagnosticKind;
  67 use once_cell::unsync::Lazy;
  68 use rustc_hash::FxHashSet;
  69 use stdx::{format_to, impl_from};
  70 use syntax::{
  71     ast::{self, HasAttrs as _, HasName},
  72     AstNode, AstPtr, SmolStr, SyntaxKind, SyntaxNodePtr,
  73 };
  74 use tt::{Ident, Leaf, Literal, TokenTree};
  75
  76 use crate::db::{DefDatabase, HirDatabase};
  77
  78 pub use crate::{
  79     attrs::{HasAttrs, Namespace},
  80     diagnostics::{
  81         AddReferenceHere, AnyDiagnostic, BreakOutsideOfLoop, InactiveCode, IncorrectCase,
  82         InvalidDeriveTarget, MacroError, MalformedDerive, MismatchedArgCount, MissingFields,
  83         MissingMatchArms, MissingOkOrSomeInTailExpr, MissingUnsafe, NoSuchField,
  84         RemoveThisSemicolon, ReplaceFilterMapNextWithFindMap, UnimplementedBuiltinMacro,
  85         UnresolvedExternCrate, UnresolvedImport, UnresolvedMacroCall, UnresolvedModule,
  86         UnresolvedProcMacro,
  87     },
  88     has_source::HasSource,
  89     semantics::{PathResolution, Semantics, SemanticsScope, TypeInfo},
  90 };
  91
  92 // Be careful with these re-exports.
  93 //
  94 // `hir` is the boundary between the compiler and the IDE. It should try hard to
  95 // isolate the compiler from the ide, to allow the two to be refactored
  96 // independently. Re-exporting something from the compiler is the sure way to
  97 // breach the boundary.
  98 //
  99 // Generally, a refactoring which *removes* a name from this list is a good
 100 // idea!
 101 pub use {
 102     cfg::{CfgAtom, CfgExpr, CfgOptions},
 103     hir_def::{
 104         adt::StructKind,
 105         attr::{Attr, Attrs, AttrsWithOwner, Documentation},
 106         builtin_attr::AttributeTemplate,
 107         find_path::PrefixKind,
 108         import_map,
 109         item_scope::ItemScope,
 110         item_tree::ItemTreeNode,
 111         nameres::{DefMap, ModuleData, ModuleOrigin, ModuleSource},
 112         path::{ModPath, PathKind},
 113         src::HasSource as DefHasSource, // xx: I don't like this shadowing of HasSource... :(
 114         type_ref::{Mutability, TypeRef},
 115         visibility::Visibility,
 116         AdtId,
 117         AssocItemId,
 118         AssocItemLoc,
 119         DefWithBodyId,
 120         ImplId,
 121         ItemContainerId,
 122         ItemLoc,
 123         Lookup,
 124         ModuleDefId,
 125         ModuleId,
 126         TraitId,
 127     },
 128     hir_expand::{
 129         name::{known, Name},
 130         ExpandResult, HirFileId, InFile, MacroDefId, MacroFile, Origin,
 131     },
 132     hir_ty::display::HirDisplay,
 133 };
 134
 135 // These are negative re-exports: pub using these names is forbidden, they
 136 // should remain private to hir internals.
 137 #[allow(unused)]
 138 use {
 139     hir_def::path::Path,
 140     hir_expand::{hygiene::Hygiene, name::AsName},
 141 };
 142
 143 /// hir::Crate describes a single crate. It's the main interface with which
 144 /// a crate's dependencies interact. Mostly, it should be just a proxy for the
 145 /// root module.
 146 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
 147 pub struct Crate {
 148     pub(crate) id: CrateId,
 149 }
 150
 151 #[derive(Debug)]
 152 pub struct CrateDependency {
 153     pub krate: Crate,
 154     pub name: Name,
 155 }
 156
 157 impl Crate {
 158     pub fn origin(self, db: &dyn HirDatabase) -> CrateOrigin {
 159         db.crate_graph()[self.id].origin.clone()
 160     }
 161
 162     pub fn dependencies(self, db: &dyn HirDatabase) -> Vec<CrateDependency> {
 163         db.crate_graph()[self.id]
 164             .dependencies
 165             .iter()
 166             .map(|dep| {
 167                 let krate = Crate { id: dep.crate_id };
 168                 let name = dep.as_name();
 169                 CrateDependency { krate, name }
 170             })
 171             .collect()
 172     }
 173
 174     pub fn reverse_dependencies(self, db: &dyn HirDatabase) -> Vec<Crate> {
 175         let crate_graph = db.crate_graph();
 176         crate_graph
 177             .iter()
 178             .filter(|&krate| {
 179                 crate_graph[krate].dependencies.iter().any(|it| it.crate_id == self.id)
 180             })
 181             .map(|id| Crate { id })
 182             .collect()
 183     }
 184
 185     pub fn transitive_reverse_dependencies(self, db: &dyn HirDatabase) -> Vec<Crate> {
 186         db.crate_graph().transitive_rev_deps(self.id).into_iter().map(|id| Crate { id }).collect()
 187     }
 188
 189     pub fn root_module(self, db: &dyn HirDatabase) -> Module {
 190         let def_map = db.crate_def_map(self.id);
 191         Module { id: def_map.module_id(def_map.root()) }
 192     }
 193
 194     pub fn root_file(self, db: &dyn HirDatabase) -> FileId {
 195         db.crate_graph()[self.id].root_file_id
 196     }
 197
 198     pub fn edition(self, db: &dyn HirDatabase) -> Edition {
 199         db.crate_graph()[self.id].edition
 200     }
 201
 202     pub fn version(self, db: &dyn HirDatabase) -> Option<String> {
 203         db.crate_graph()[self.id].version.clone()
 204     }
 205
 206     pub fn display_name(self, db: &dyn HirDatabase) -> Option<CrateDisplayName> {
 207         db.crate_graph()[self.id].display_name.clone()
 208     }
 209
 210     pub fn query_external_importables(
 211         self,
 212         db: &dyn DefDatabase,
 213         query: import_map::Query,
 214     ) -> impl Iterator<Item = Either<ModuleDef, MacroDef>> {
 215         let _p = profile::span("query_external_importables");
 216         import_map::search_dependencies(db, self.into(), query).into_iter().map(|item| {
 217             match ItemInNs::from(item) {
 218                 ItemInNs::Types(mod_id) | ItemInNs::Values(mod_id) => Either::Left(mod_id),
 219                 ItemInNs::Macros(mac_id) => Either::Right(mac_id),
 220             }
 221         })
 222     }
 223
 224     pub fn all(db: &dyn HirDatabase) -> Vec<Crate> {
 225         db.crate_graph().iter().map(|id| Crate { id }).collect()
 226     }
 227
 228     /// Try to get the root URL of the documentation of a crate.
 229     pub fn get_html_root_url(self: &Crate, db: &dyn HirDatabase) -> Option<String> {
 230         // Look for #![doc(html_root_url = "...")]
 231         let attrs = db.attrs(AttrDefId::ModuleId(self.root_module(db).into()));
 232         let doc_attr_q = attrs.by_key("doc");
 233
 234         if !doc_attr_q.exists() {
 235             return None;
 236         }
 237
 238         let doc_url = doc_attr_q.tt_values().map(|tt| {
 239             let name = tt.token_trees.iter()
 240                 .skip_while(|tt| !matches!(tt, TokenTree::Leaf(Leaf::Ident(Ident { text, ..} )) if text == "html_root_url"))
 241                 .nth(2);
 242
 243             match name {
 244                 Some(TokenTree::Leaf(Leaf::Literal(Literal{ref text, ..}))) => Some(text),
 245                 _ => None
 246             }
 247         }).flatten().next();
 248
 249         doc_url.map(|s| s.trim_matches('"').trim_end_matches('/').to_owned() + "/")
 250     }
 251
 252     pub fn cfg(&self, db: &dyn HirDatabase) -> CfgOptions {
 253         db.crate_graph()[self.id].cfg_options.clone()
 254     }
 255
 256     pub fn potential_cfg(&self, db: &dyn HirDatabase) -> CfgOptions {
 257         db.crate_graph()[self.id].potential_cfg_options.clone()
 258     }
 259 }
 260
 261 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
 262 pub struct Module {
 263     pub(crate) id: ModuleId,
 264 }
 265
 266 /// The defs which can be visible in the module.
 267 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
 268 pub enum ModuleDef {
 269     Module(Module),
 270     Function(Function),
 271     Adt(Adt),
 272     // Can't be directly declared, but can be imported.
 273     Variant(Variant),
 274     Const(Const),
 275     Static(Static),
 276     Trait(Trait),
 277     TypeAlias(TypeAlias),
 278     BuiltinType(BuiltinType),
 279 }
 280 impl_from!(
 281     Module,
 282     Function,
 283     Adt(Struct, Enum, Union),
 284     Variant,
 285     Const,
 286     Static,
 287     Trait,
 288     TypeAlias,
 289     BuiltinType
 290     for ModuleDef
 291 );
 292
 293 impl From<VariantDef> for ModuleDef {
 294     fn from(var: VariantDef) -> Self {
 295         match var {
 296             VariantDef::Struct(t) => Adt::from(t).into(),
 297             VariantDef::Union(t) => Adt::from(t).into(),
 298             VariantDef::Variant(t) => t.into(),
 299         }
 300     }
 301 }
 302
 303 impl ModuleDef {
 304     pub fn module(self, db: &dyn HirDatabase) -> Option<Module> {
 305         match self {
 306             ModuleDef::Module(it) => it.parent(db),
 307             ModuleDef::Function(it) => Some(it.module(db)),
 308             ModuleDef::Adt(it) => Some(it.module(db)),
 309             ModuleDef::Variant(it) => Some(it.module(db)),
 310             ModuleDef::Const(it) => Some(it.module(db)),
 311             ModuleDef::Static(it) => Some(it.module(db)),
 312             ModuleDef::Trait(it) => Some(it.module(db)),
 313             ModuleDef::TypeAlias(it) => Some(it.module(db)),
 314             ModuleDef::BuiltinType(_) => None,
 315         }
 316     }
 317
 318     pub fn canonical_path(&self, db: &dyn HirDatabase) -> Option<String> {
 319         let mut segments = vec![self.name(db)?];
 320         for m in self.module(db)?.path_to_root(db) {
 321             segments.extend(m.name(db))
 322         }
 323         segments.reverse();
 324         Some(segments.into_iter().join("::"))
 325     }
 326
 327     pub fn canonical_module_path(
 328         &self,
 329         db: &dyn HirDatabase,
 330     ) -> Option<impl Iterator<Item = Module>> {
 331         self.module(db).map(|it| it.path_to_root(db).into_iter().rev())
 332     }
 333
 334     pub fn name(self, db: &dyn HirDatabase) -> Option<Name> {
 335         let name = match self {
 336             ModuleDef::Module(it) => it.name(db)?,
 337             ModuleDef::Const(it) => it.name(db)?,
 338             ModuleDef::Adt(it) => it.name(db),
 339             ModuleDef::Trait(it) => it.name(db),
 340             ModuleDef::Function(it) => it.name(db),
 341             ModuleDef::Variant(it) => it.name(db),
 342             ModuleDef::TypeAlias(it) => it.name(db),
 343             ModuleDef::Static(it) => it.name(db),
 344             ModuleDef::BuiltinType(it) => it.name(),
 345         };
 346         Some(name)
 347     }
 348
 349     pub fn diagnostics(self, db: &dyn HirDatabase) -> Vec<AnyDiagnostic> {
 350         let id = match self {
 351             ModuleDef::Adt(it) => match it {
 352                 Adt::Struct(it) => it.id.into(),
 353                 Adt::Enum(it) => it.id.into(),
 354                 Adt::Union(it) => it.id.into(),
 355             },
 356             ModuleDef::Trait(it) => it.id.into(),
 357             ModuleDef::Function(it) => it.id.into(),
 358             ModuleDef::TypeAlias(it) => it.id.into(),
 359             ModuleDef::Module(it) => it.id.into(),
 360             ModuleDef::Const(it) => it.id.into(),
 361             ModuleDef::Static(it) => it.id.into(),
 362             _ => return Vec::new(),
 363         };
 364
 365         let module = match self.module(db) {
 366             Some(it) => it,
 367             None => return Vec::new(),
 368         };
 369
 370         let mut acc = Vec::new();
 371
 372         match self.as_def_with_body() {
 373             Some(def) => {
 374                 def.diagnostics(db, &mut acc);
 375             }
 376             None => {
 377                 for diag in hir_ty::diagnostics::incorrect_case(db, module.id.krate(), id) {
 378                     acc.push(diag.into())
 379                 }
 380             }
 381         }
 382
 383         acc
 384     }
 385
 386     pub fn as_def_with_body(self) -> Option<DefWithBody> {
 387         match self {
 388             ModuleDef::Function(it) => Some(it.into()),
 389             ModuleDef::Const(it) => Some(it.into()),
 390             ModuleDef::Static(it) => Some(it.into()),
 391
 392             ModuleDef::Module(_)
 393             | ModuleDef::Adt(_)
 394             | ModuleDef::Variant(_)
 395             | ModuleDef::Trait(_)
 396             | ModuleDef::TypeAlias(_)
 397             | ModuleDef::BuiltinType(_) => None,
 398         }
 399     }
 400
 401     pub fn attrs(&self, db: &dyn HirDatabase) -> Option<AttrsWithOwner> {
 402         Some(match self {
 403             ModuleDef::Module(it) => it.attrs(db),
 404             ModuleDef::Function(it) => it.attrs(db),
 405             ModuleDef::Adt(it) => it.attrs(db),
 406             ModuleDef::Variant(it) => it.attrs(db),
 407             ModuleDef::Const(it) => it.attrs(db),
 408             ModuleDef::Static(it) => it.attrs(db),
 409             ModuleDef::Trait(it) => it.attrs(db),
 410             ModuleDef::TypeAlias(it) => it.attrs(db),
 411             ModuleDef::BuiltinType(_) => return None,
 412         })
 413     }
 414 }
 415
 416 impl HasVisibility for ModuleDef {
 417     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
 418         match *self {
 419             ModuleDef::Module(it) => it.visibility(db),
 420             ModuleDef::Function(it) => it.visibility(db),
 421             ModuleDef::Adt(it) => it.visibility(db),
 422             ModuleDef::Const(it) => it.visibility(db),
 423             ModuleDef::Static(it) => it.visibility(db),
 424             ModuleDef::Trait(it) => it.visibility(db),
 425             ModuleDef::TypeAlias(it) => it.visibility(db),
 426             ModuleDef::Variant(it) => it.visibility(db),
 427             ModuleDef::BuiltinType(_) => Visibility::Public,
 428         }
 429     }
 430 }
 431
 432 impl Module {
 433     /// Name of this module.
 434     pub fn name(self, db: &dyn HirDatabase) -> Option<Name> {
 435         let def_map = self.id.def_map(db.upcast());
 436         let parent = def_map[self.id.local_id].parent?;
 437         def_map[parent].children.iter().find_map(|(name, module_id)| {
 438             if *module_id == self.id.local_id {
 439                 Some(name.clone())
 440             } else {
 441                 None
 442             }
 443         })
 444     }
 445
 446     /// Returns the crate this module is part of.
 447     pub fn krate(self) -> Crate {
 448         Crate { id: self.id.krate() }
 449     }
 450
 451     /// Topmost parent of this module. Every module has a `crate_root`, but some
 452     /// might be missing `krate`. This can happen if a module's file is not included
 453     /// in the module tree of any target in `Cargo.toml`.
 454     pub fn crate_root(self, db: &dyn HirDatabase) -> Module {
 455         let def_map = db.crate_def_map(self.id.krate());
 456         Module { id: def_map.module_id(def_map.root()) }
 457     }
 458
 459     /// Iterates over all child modules.
 460     pub fn children(self, db: &dyn HirDatabase) -> impl Iterator<Item = Module> {
 461         let def_map = self.id.def_map(db.upcast());
 462         let children = def_map[self.id.local_id]
 463             .children
 464             .iter()
 465             .map(|(_, module_id)| Module { id: def_map.module_id(*module_id) })
 466             .collect::<Vec<_>>();
 467         children.into_iter()
 468     }
 469
 470     /// Finds a parent module.
 471     pub fn parent(self, db: &dyn HirDatabase) -> Option<Module> {
 472         // FIXME: handle block expressions as modules (their parent is in a different DefMap)
 473         let def_map = self.id.def_map(db.upcast());
 474         let parent_id = def_map[self.id.local_id].parent?;
 475         Some(Module { id: def_map.module_id(parent_id) })
 476     }
 477
 478     pub fn path_to_root(self, db: &dyn HirDatabase) -> Vec<Module> {
 479         let mut res = vec![self];
 480         let mut curr = self;
 481         while let Some(next) = curr.parent(db) {
 482             res.push(next);
 483             curr = next
 484         }
 485         res
 486     }
 487
 488     /// Returns a `ModuleScope`: a set of items, visible in this module.
 489     pub fn scope(
 490         self,
 491         db: &dyn HirDatabase,
 492         visible_from: Option<Module>,
 493     ) -> Vec<(Name, ScopeDef)> {
 494         self.id.def_map(db.upcast())[self.id.local_id]
 495             .scope
 496             .entries()
 497             .filter_map(|(name, def)| {
 498                 if let Some(m) = visible_from {
 499                     let filtered =
 500                         def.filter_visibility(|vis| vis.is_visible_from(db.upcast(), m.id));
 501                     if filtered.is_none() && !def.is_none() {
 502                         None
 503                     } else {
 504                         Some((name, filtered))
 505                     }
 506                 } else {
 507                     Some((name, def))
 508                 }
 509             })
 510             .flat_map(|(name, def)| {
 511                 ScopeDef::all_items(def).into_iter().map(move |item| (name.clone(), item))
 512             })
 513             .collect()
 514     }
 515
 516     pub fn diagnostics(self, db: &dyn HirDatabase, acc: &mut Vec<AnyDiagnostic>) {
 517         let _p = profile::span("Module::diagnostics").detail(|| {
 518             format!("{:?}", self.name(db).map_or("<unknown>".into(), |name| name.to_string()))
 519         });
 520         let def_map = self.id.def_map(db.upcast());
 521         for diag in def_map.diagnostics() {
 522             if diag.in_module != self.id.local_id {
 523                 // FIXME: This is accidentally quadratic.
 524                 continue;
 525             }
 526             match &diag.kind {
 527                 DefDiagnosticKind::UnresolvedModule { ast: declaration, candidate } => {
 528                     let decl = declaration.to_node(db.upcast());
 529                     acc.push(
 530                         UnresolvedModule {
 531                             decl: InFile::new(declaration.file_id, AstPtr::new(&decl)),
 532                             candidate: candidate.clone(),
 533                         }
 534                         .into(),
 535                     )
 536                 }
 537                 DefDiagnosticKind::UnresolvedExternCrate { ast } => {
 538                     let item = ast.to_node(db.upcast());
 539                     acc.push(
 540                         UnresolvedExternCrate {
 541                             decl: InFile::new(ast.file_id, AstPtr::new(&item)),
 542                         }
 543                         .into(),
 544                     );
 545                 }
 546
 547                 DefDiagnosticKind::UnresolvedImport { id, index } => {
 548                     let file_id = id.file_id();
 549                     let item_tree = id.item_tree(db.upcast());
 550                     let import = &item_tree[id.value];
 551
 552                     let use_tree = import.use_tree_to_ast(db.upcast(), file_id, *index);
 553                     acc.push(
 554                         UnresolvedImport { decl: InFile::new(file_id, AstPtr::new(&use_tree)) }
 555                             .into(),
 556                     );
 557                 }
 558
 559                 DefDiagnosticKind::UnconfiguredCode { ast, cfg, opts } => {
 560                     let item = ast.to_node(db.upcast());
 561                     acc.push(
 562                         InactiveCode {
 563                             node: ast.with_value(AstPtr::new(&item).into()),
 564                             cfg: cfg.clone(),
 565                             opts: opts.clone(),
 566                         }
 567                         .into(),
 568                     );
 569                 }
 570
 571                 DefDiagnosticKind::UnresolvedProcMacro { ast } => {
 572                     let mut precise_location = None;
 573                     let (node, name) = match ast {
 574                         MacroCallKind::FnLike { ast_id, .. } => {
 575                             let node = ast_id.to_node(db.upcast());
 576                             (ast_id.with_value(SyntaxNodePtr::from(AstPtr::new(&node))), None)
 577                         }
 578                         MacroCallKind::Derive { ast_id, derive_name, .. } => {
 579                             let node = ast_id.to_node(db.upcast());
 580
 581                             // Compute the precise location of the macro name's token in the derive
 582                             // list.
 583                             // FIXME: This does not handle paths to the macro, but neither does the
 584                             // rest of r-a.
 585                             let derive_attrs =
 586                                 node.attrs().filter_map(|attr| match attr.as_simple_call() {
 587                                     Some((name, args)) if name == "derive" => Some(args),
 588                                     _ => None,
 589                                 });
 590                             'outer: for attr in derive_attrs {
 591                                 let tokens =
 592                                     attr.syntax().children_with_tokens().filter_map(|elem| {
 593                                         match elem {
 594                                             syntax::NodeOrToken::Node(_) => None,
 595                                             syntax::NodeOrToken::Token(tok) => Some(tok),
 596                                         }
 597                                     });
 598                                 for token in tokens {
 599                                     if token.kind() == SyntaxKind::IDENT
 600                                         && token.text() == &**derive_name
 601                                     {
 602                                         precise_location = Some(token.text_range());
 603                                         break 'outer;
 604                                     }
 605                                 }
 606                             }
 607
 608                             (
 609                                 ast_id.with_value(SyntaxNodePtr::from(AstPtr::new(&node))),
 610                                 Some(derive_name.clone()),
 611                             )
 612                         }
 613                         MacroCallKind::Attr { ast_id, invoc_attr_index, attr_name, .. } => {
 614                             let node = ast_id.to_node(db.upcast());
 615                             let attr =
 616                                 node.attrs().nth((*invoc_attr_index) as usize).unwrap_or_else(
 617                                     || panic!("cannot find attribute #{}", invoc_attr_index),
 618                                 );
 619                             (
 620                                 ast_id.with_value(SyntaxNodePtr::from(AstPtr::new(&attr))),
 621                                 Some(attr_name.clone()),
 622                             )
 623                         }
 624                     };
 625                     acc.push(
 626                         UnresolvedProcMacro {
 627                             node,
 628                             precise_location,
 629                             macro_name: name.map(Into::into),
 630                         }
 631                         .into(),
 632                     );
 633                 }
 634
 635                 DefDiagnosticKind::UnresolvedMacroCall { ast, path } => {
 636                     let node = ast.to_node(db.upcast());
 637                     acc.push(
 638                         UnresolvedMacroCall {
 639                             macro_call: InFile::new(ast.file_id, AstPtr::new(&node)),
 640                             path: path.clone(),
 641                         }
 642                         .into(),
 643                     );
 644                 }
 645
 646                 DefDiagnosticKind::MacroError { ast, message } => {
 647                     let node = match ast {
 648                         MacroCallKind::FnLike { ast_id, .. } => {
 649                             let node = ast_id.to_node(db.upcast());
 650                             ast_id.with_value(SyntaxNodePtr::from(AstPtr::new(&node)))
 651                         }
 652                         MacroCallKind::Derive { ast_id, .. }
 653                         | MacroCallKind::Attr { ast_id, .. } => {
 654                             // FIXME: point to the attribute instead, this creates very large diagnostics
 655                             let node = ast_id.to_node(db.upcast());
 656                             ast_id.with_value(SyntaxNodePtr::from(AstPtr::new(&node)))
 657                         }
 658                     };
 659                     acc.push(MacroError { node, message: message.clone() }.into());
 660                 }
 661
 662                 DefDiagnosticKind::UnimplementedBuiltinMacro { ast } => {
 663                     let node = ast.to_node(db.upcast());
 664                     // Must have a name, otherwise we wouldn't emit it.
 665                     let name = node.name().expect("unimplemented builtin macro with no name");
 666                     acc.push(
 667                         UnimplementedBuiltinMacro {
 668                             node: ast.with_value(SyntaxNodePtr::from(AstPtr::new(&name))),
 669                         }
 670                         .into(),
 671                     );
 672                 }
 673                 DefDiagnosticKind::InvalidDeriveTarget { ast, id } => {
 674                     let node = ast.to_node(db.upcast());
 675                     let derive = node.attrs().nth(*id as usize);
 676                     match derive {
 677                         Some(derive) => {
 678                             acc.push(
 679                                 InvalidDeriveTarget {
 680                                     node: ast.with_value(SyntaxNodePtr::from(AstPtr::new(&derive))),
 681                                 }
 682                                 .into(),
 683                             );
 684                         }
 685                         None => stdx::never!("derive diagnostic on item without derive attribute"),
 686                     }
 687                 }
 688                 DefDiagnosticKind::MalformedDerive { ast, id } => {
 689                     let node = ast.to_node(db.upcast());
 690                     let derive = node.attrs().nth(*id as usize);
 691                     match derive {
 692                         Some(derive) => {
 693                             acc.push(
 694                                 MalformedDerive {
 695                                     node: ast.with_value(SyntaxNodePtr::from(AstPtr::new(&derive))),
 696                                 }
 697                                 .into(),
 698                             );
 699                         }
 700                         None => stdx::never!("derive diagnostic on item without derive attribute"),
 701                     }
 702                 }
 703             }
 704         }
 705         for decl in self.declarations(db) {
 706             match decl {
 707                 ModuleDef::Module(m) => {
 708                     // Only add diagnostics from inline modules
 709                     if def_map[m.id.local_id].origin.is_inline() {
 710                         m.diagnostics(db, acc)
 711                     }
 712                 }
 713                 _ => acc.extend(decl.diagnostics(db)),
 714             }
 715         }
 716
 717         for impl_def in self.impl_defs(db) {
 718             for item in impl_def.items(db) {
 719                 let def: DefWithBody = match item {
 720                     AssocItem::Function(it) => it.into(),
 721                     AssocItem::Const(it) => it.into(),
 722                     AssocItem::TypeAlias(_) => continue,
 723                 };
 724
 725                 def.diagnostics(db, acc);
 726             }
 727         }
 728     }
 729
 730     pub fn declarations(self, db: &dyn HirDatabase) -> Vec<ModuleDef> {
 731         let def_map = self.id.def_map(db.upcast());
 732         let scope = &def_map[self.id.local_id].scope;
 733         scope
 734             .declarations()
 735             .map(ModuleDef::from)
 736             .chain(scope.unnamed_consts().map(|id| ModuleDef::Const(Const::from(id))))
 737             .collect()
 738     }
 739
 740     pub fn impl_defs(self, db: &dyn HirDatabase) -> Vec<Impl> {
 741         let def_map = self.id.def_map(db.upcast());
 742         def_map[self.id.local_id].scope.impls().map(Impl::from).collect()
 743     }
 744
 745     /// Finds a path that can be used to refer to the given item from within
 746     /// this module, if possible.
 747     pub fn find_use_path(self, db: &dyn DefDatabase, item: impl Into<ItemInNs>) -> Option<ModPath> {
 748         hir_def::find_path::find_path(db, item.into().into(), self.into())
 749     }
 750
 751     /// Finds a path that can be used to refer to the given item from within
 752     /// this module, if possible. This is used for returning import paths for use-statements.
 753     pub fn find_use_path_prefixed(
 754         self,
 755         db: &dyn DefDatabase,
 756         item: impl Into<ItemInNs>,
 757         prefix_kind: PrefixKind,
 758     ) -> Option<ModPath> {
 759         hir_def::find_path::find_path_prefixed(db, item.into().into(), self.into(), prefix_kind)
 760     }
 761 }
 762
 763 impl HasVisibility for Module {
 764     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
 765         let def_map = self.id.def_map(db.upcast());
 766         let module_data = &def_map[self.id.local_id];
 767         module_data.visibility
 768     }
 769 }
 770
 771 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
 772 pub struct Field {
 773     pub(crate) parent: VariantDef,
 774     pub(crate) id: LocalFieldId,
 775 }
 776
 777 #[derive(Debug, PartialEq, Eq)]
 778 pub enum FieldSource {
 779     Named(ast::RecordField),
 780     Pos(ast::TupleField),
 781 }
 782
 783 impl Field {
 784     pub fn name(&self, db: &dyn HirDatabase) -> Name {
 785         self.parent.variant_data(db).fields()[self.id].name.clone()
 786     }
 787
 788     /// Returns the type as in the signature of the struct (i.e., with
 789     /// placeholder types for type parameters). Only use this in the context of
 790     /// the field definition.
 791     pub fn ty(&self, db: &dyn HirDatabase) -> Type {
 792         let var_id = self.parent.into();
 793         let generic_def_id: GenericDefId = match self.parent {
 794             VariantDef::Struct(it) => it.id.into(),
 795             VariantDef::Union(it) => it.id.into(),
 796             VariantDef::Variant(it) => it.parent.id.into(),
 797         };
 798         let substs = TyBuilder::type_params_subst(db, generic_def_id);
 799         let ty = db.field_types(var_id)[self.id].clone().substitute(Interner, &substs);
 800         Type::new(db, self.parent.module(db).id.krate(), var_id, ty)
 801     }
 802
 803     pub fn parent_def(&self, _db: &dyn HirDatabase) -> VariantDef {
 804         self.parent
 805     }
 806 }
 807
 808 impl HasVisibility for Field {
 809     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
 810         let variant_data = self.parent.variant_data(db);
 811         let visibility = &variant_data.fields()[self.id].visibility;
 812         let parent_id: hir_def::VariantId = self.parent.into();
 813         visibility.resolve(db.upcast(), &parent_id.resolver(db.upcast()))
 814     }
 815 }
 816
 817 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
 818 pub struct Struct {
 819     pub(crate) id: StructId,
 820 }
 821
 822 impl Struct {
 823     pub fn module(self, db: &dyn HirDatabase) -> Module {
 824         Module { id: self.id.lookup(db.upcast()).container }
 825     }
 826
 827     pub fn name(self, db: &dyn HirDatabase) -> Name {
 828         db.struct_data(self.id).name.clone()
 829     }
 830
 831     pub fn fields(self, db: &dyn HirDatabase) -> Vec<Field> {
 832         db.struct_data(self.id)
 833             .variant_data
 834             .fields()
 835             .iter()
 836             .map(|(id, _)| Field { parent: self.into(), id })
 837             .collect()
 838     }
 839
 840     pub fn ty(self, db: &dyn HirDatabase) -> Type {
 841         Type::from_def(db, self.id.lookup(db.upcast()).container.krate(), self.id)
 842     }
 843
 844     pub fn repr(self, db: &dyn HirDatabase) -> Option<ReprKind> {
 845         db.struct_data(self.id).repr.clone()
 846     }
 847
 848     pub fn kind(self, db: &dyn HirDatabase) -> StructKind {
 849         self.variant_data(db).kind()
 850     }
 851
 852     fn variant_data(self, db: &dyn HirDatabase) -> Arc<VariantData> {
 853         db.struct_data(self.id).variant_data.clone()
 854     }
 855 }
 856
 857 impl HasVisibility for Struct {
 858     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
 859         db.struct_data(self.id).visibility.resolve(db.upcast(), &self.id.resolver(db.upcast()))
 860     }
 861 }
 862
 863 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
 864 pub struct Union {
 865     pub(crate) id: UnionId,
 866 }
 867
 868 impl Union {
 869     pub fn name(self, db: &dyn HirDatabase) -> Name {
 870         db.union_data(self.id).name.clone()
 871     }
 872
 873     pub fn module(self, db: &dyn HirDatabase) -> Module {
 874         Module { id: self.id.lookup(db.upcast()).container }
 875     }
 876
 877     pub fn ty(self, db: &dyn HirDatabase) -> Type {
 878         Type::from_def(db, self.id.lookup(db.upcast()).container.krate(), self.id)
 879     }
 880
 881     pub fn fields(self, db: &dyn HirDatabase) -> Vec<Field> {
 882         db.union_data(self.id)
 883             .variant_data
 884             .fields()
 885             .iter()
 886             .map(|(id, _)| Field { parent: self.into(), id })
 887             .collect()
 888     }
 889
 890     fn variant_data(self, db: &dyn HirDatabase) -> Arc<VariantData> {
 891         db.union_data(self.id).variant_data.clone()
 892     }
 893 }
 894
 895 impl HasVisibility for Union {
 896     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
 897         db.union_data(self.id).visibility.resolve(db.upcast(), &self.id.resolver(db.upcast()))
 898     }
 899 }
 900
 901 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
 902 pub struct Enum {
 903     pub(crate) id: EnumId,
 904 }
 905
 906 impl Enum {
 907     pub fn module(self, db: &dyn HirDatabase) -> Module {
 908         Module { id: self.id.lookup(db.upcast()).container }
 909     }
 910
 911     pub fn name(self, db: &dyn HirDatabase) -> Name {
 912         db.enum_data(self.id).name.clone()
 913     }
 914
 915     pub fn variants(self, db: &dyn HirDatabase) -> Vec<Variant> {
 916         db.enum_data(self.id).variants.iter().map(|(id, _)| Variant { parent: self, id }).collect()
 917     }
 918
 919     pub fn ty(self, db: &dyn HirDatabase) -> Type {
 920         Type::from_def(db, self.id.lookup(db.upcast()).container.krate(), self.id)
 921     }
 922 }
 923
 924 impl HasVisibility for Enum {
 925     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
 926         db.enum_data(self.id).visibility.resolve(db.upcast(), &self.id.resolver(db.upcast()))
 927     }
 928 }
 929
 930 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
 931 pub struct Variant {
 932     pub(crate) parent: Enum,
 933     pub(crate) id: LocalEnumVariantId,
 934 }
 935
 936 impl Variant {
 937     pub fn module(self, db: &dyn HirDatabase) -> Module {
 938         self.parent.module(db)
 939     }
 940
 941     pub fn parent_enum(self, _db: &dyn HirDatabase) -> Enum {
 942         self.parent
 943     }
 944
 945     pub fn name(self, db: &dyn HirDatabase) -> Name {
 946         db.enum_data(self.parent.id).variants[self.id].name.clone()
 947     }
 948
 949     pub fn fields(self, db: &dyn HirDatabase) -> Vec<Field> {
 950         self.variant_data(db)
 951             .fields()
 952             .iter()
 953             .map(|(id, _)| Field { parent: self.into(), id })
 954             .collect()
 955     }
 956
 957     pub fn kind(self, db: &dyn HirDatabase) -> StructKind {
 958         self.variant_data(db).kind()
 959     }
 960
 961     pub(crate) fn variant_data(self, db: &dyn HirDatabase) -> Arc<VariantData> {
 962         db.enum_data(self.parent.id).variants[self.id].variant_data.clone()
 963     }
 964 }
 965
 966 /// Variants inherit visibility from the parent enum.
 967 impl HasVisibility for Variant {
 968     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
 969         self.parent_enum(db).visibility(db)
 970     }
 971 }
 972
 973 /// A Data Type
 974 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
 975 pub enum Adt {
 976     Struct(Struct),
 977     Union(Union),
 978     Enum(Enum),
 979 }
 980 impl_from!(Struct, Union, Enum for Adt);
 981
 982 impl Adt {
 983     pub fn has_non_default_type_params(self, db: &dyn HirDatabase) -> bool {
 984         let subst = db.generic_defaults(self.into());
 985         subst.iter().any(|ty| ty.skip_binders().is_unknown())
 986     }
 987
 988     /// Turns this ADT into a type. Any type parameters of the ADT will be
 989     /// turned into unknown types, which is good for e.g. finding the most
 990     /// general set of completions, but will not look very nice when printed.
 991     pub fn ty(self, db: &dyn HirDatabase) -> Type {
 992         let id = AdtId::from(self);
 993         Type::from_def(db, id.module(db.upcast()).krate(), id)
 994     }
 995
 996     pub fn module(self, db: &dyn HirDatabase) -> Module {
 997         match self {
 998             Adt::Struct(s) => s.module(db),
 999             Adt::Union(s) => s.module(db),
1000             Adt::Enum(e) => e.module(db),
1001         }
1002     }
1003
1004     pub fn name(self, db: &dyn HirDatabase) -> Name {
1005         match self {
1006             Adt::Struct(s) => s.name(db),
1007             Adt::Union(u) => u.name(db),
1008             Adt::Enum(e) => e.name(db),
1009         }
1010     }
1011 }
1012
1013 impl HasVisibility for Adt {
1014     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
1015         match self {
1016             Adt::Struct(it) => it.visibility(db),
1017             Adt::Union(it) => it.visibility(db),
1018             Adt::Enum(it) => it.visibility(db),
1019         }
1020     }
1021 }
1022
1023 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
1024 pub enum VariantDef {
1025     Struct(Struct),
1026     Union(Union),
1027     Variant(Variant),
1028 }
1029 impl_from!(Struct, Union, Variant for VariantDef);
1030
1031 impl VariantDef {
1032     pub fn fields(self, db: &dyn HirDatabase) -> Vec<Field> {
1033         match self {
1034             VariantDef::Struct(it) => it.fields(db),
1035             VariantDef::Union(it) => it.fields(db),
1036             VariantDef::Variant(it) => it.fields(db),
1037         }
1038     }
1039
1040     pub fn module(self, db: &dyn HirDatabase) -> Module {
1041         match self {
1042             VariantDef::Struct(it) => it.module(db),
1043             VariantDef::Union(it) => it.module(db),
1044             VariantDef::Variant(it) => it.module(db),
1045         }
1046     }
1047
1048     pub fn name(&self, db: &dyn HirDatabase) -> Name {
1049         match self {
1050             VariantDef::Struct(s) => s.name(db),
1051             VariantDef::Union(u) => u.name(db),
1052             VariantDef::Variant(e) => e.name(db),
1053         }
1054     }
1055
1056     pub(crate) fn variant_data(self, db: &dyn HirDatabase) -> Arc<VariantData> {
1057         match self {
1058             VariantDef::Struct(it) => it.variant_data(db),
1059             VariantDef::Union(it) => it.variant_data(db),
1060             VariantDef::Variant(it) => it.variant_data(db),
1061         }
1062     }
1063 }
1064
1065 /// The defs which have a body.
1066 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
1067 pub enum DefWithBody {
1068     Function(Function),
1069     Static(Static),
1070     Const(Const),
1071 }
1072 impl_from!(Function, Const, Static for DefWithBody);
1073
1074 impl DefWithBody {
1075     pub fn module(self, db: &dyn HirDatabase) -> Module {
1076         match self {
1077             DefWithBody::Const(c) => c.module(db),
1078             DefWithBody::Function(f) => f.module(db),
1079             DefWithBody::Static(s) => s.module(db),
1080         }
1081     }
1082
1083     pub fn name(self, db: &dyn HirDatabase) -> Option<Name> {
1084         match self {
1085             DefWithBody::Function(f) => Some(f.name(db)),
1086             DefWithBody::Static(s) => Some(s.name(db)),
1087             DefWithBody::Const(c) => c.name(db),
1088         }
1089     }
1090
1091     /// Returns the type this def's body has to evaluate to.
1092     pub fn body_type(self, db: &dyn HirDatabase) -> Type {
1093         match self {
1094             DefWithBody::Function(it) => it.ret_type(db),
1095             DefWithBody::Static(it) => it.ty(db),
1096             DefWithBody::Const(it) => it.ty(db),
1097         }
1098     }
1099
1100     pub fn diagnostics(self, db: &dyn HirDatabase, acc: &mut Vec<AnyDiagnostic>) {
1101         let krate = self.module(db).id.krate();
1102
1103         let source_map = db.body_with_source_map(self.into()).1;
1104         for diag in source_map.diagnostics() {
1105             match diag {
1106                 BodyDiagnostic::InactiveCode { node, cfg, opts } => acc.push(
1107                     InactiveCode { node: node.clone(), cfg: cfg.clone(), opts: opts.clone() }
1108                         .into(),
1109                 ),
1110                 BodyDiagnostic::MacroError { node, message } => acc.push(
1111                     MacroError {
1112                         node: node.clone().map(|it| it.into()),
1113                         message: message.to_string(),
1114                     }
1115                     .into(),
1116                 ),
1117                 BodyDiagnostic::UnresolvedProcMacro { node } => acc.push(
1118                     UnresolvedProcMacro {
1119                         node: node.clone().map(|it| it.into()),
1120                         precise_location: None,
1121                         macro_name: None,
1122                     }
1123                     .into(),
1124                 ),
1125                 BodyDiagnostic::UnresolvedMacroCall { node, path } => acc.push(
1126                     UnresolvedMacroCall { macro_call: node.clone(), path: path.clone() }.into(),
1127                 ),
1128             }
1129         }
1130
1131         let infer = db.infer(self.into());
1132         let source_map = Lazy::new(|| db.body_with_source_map(self.into()).1);
1133         for d in &infer.diagnostics {
1134             match d {
1135                 hir_ty::InferenceDiagnostic::NoSuchField { expr } => {
1136                     let field = source_map.field_syntax(*expr);
1137                     acc.push(NoSuchField { field }.into())
1138                 }
1139                 hir_ty::InferenceDiagnostic::BreakOutsideOfLoop { expr } => {
1140                     let expr = source_map
1141                         .expr_syntax(*expr)
1142                         .expect("break outside of loop in synthetic syntax");
1143                     acc.push(BreakOutsideOfLoop { expr }.into())
1144                 }
1145             }
1146         }
1147
1148         for expr in hir_ty::diagnostics::missing_unsafe(db, self.into()) {
1149             match source_map.expr_syntax(expr) {
1150                 Ok(expr) => acc.push(MissingUnsafe { expr }.into()),
1151                 Err(SyntheticSyntax) => {
1152                     // FIXME: Here and eslwhere in this file, the `expr` was
1153                     // desugared, report or assert that this doesn't happen.
1154                 }
1155             }
1156         }
1157
1158         for diagnostic in BodyValidationDiagnostic::collect(db, self.into()) {
1159             match diagnostic {
1160                 BodyValidationDiagnostic::RecordMissingFields {
1161                     record,
1162                     variant,
1163                     missed_fields,
1164                 } => {
1165                     let variant_data = variant.variant_data(db.upcast());
1166                     let missed_fields = missed_fields
1167                         .into_iter()
1168                         .map(|idx| variant_data.fields()[idx].name.clone())
1169                         .collect();
1170
1171                     match record {
1172                         Either::Left(record_expr) => match source_map.expr_syntax(record_expr) {
1173                             Ok(source_ptr) => {
1174                                 let root = source_ptr.file_syntax(db.upcast());
1175                                 if let ast::Expr::RecordExpr(record_expr) =
1176                                     &source_ptr.value.to_node(&root)
1177                                 {
1178                                     if record_expr.record_expr_field_list().is_some() {
1179                                         acc.push(
1180                                             MissingFields {
1181                                                 file: source_ptr.file_id,
1182                                                 field_list_parent: Either::Left(AstPtr::new(
1183                                                     record_expr,
1184                                                 )),
1185                                                 field_list_parent_path: record_expr
1186                                                     .path()
1187                                                     .map(|path| AstPtr::new(&path)),
1188                                                 missed_fields,
1189                                             }
1190                                             .into(),
1191                                         )
1192                                     }
1193                                 }
1194                             }
1195                             Err(SyntheticSyntax) => (),
1196                         },
1197                         Either::Right(record_pat) => match source_map.pat_syntax(record_pat) {
1198                             Ok(source_ptr) => {
1199                                 if let Some(expr) = source_ptr.value.as_ref().left() {
1200                                     let root = source_ptr.file_syntax(db.upcast());
1201                                     if let ast::Pat::RecordPat(record_pat) = expr.to_node(&root) {
1202                                         if record_pat.record_pat_field_list().is_some() {
1203                                             acc.push(
1204                                                 MissingFields {
1205                                                     file: source_ptr.file_id,
1206                                                     field_list_parent: Either::Right(AstPtr::new(
1207                                                         &record_pat,
1208                                                     )),
1209                                                     field_list_parent_path: record_pat
1210                                                         .path()
1211                                                         .map(|path| AstPtr::new(&path)),
1212                                                     missed_fields,
1213                                                 }
1214                                                 .into(),
1215                                             )
1216                                         }
1217                                     }
1218                                 }
1219                             }
1220                             Err(SyntheticSyntax) => (),
1221                         },
1222                     }
1223                 }
1224                 BodyValidationDiagnostic::ReplaceFilterMapNextWithFindMap { method_call_expr } => {
1225                     if let Ok(next_source_ptr) = source_map.expr_syntax(method_call_expr) {
1226                         acc.push(
1227                             ReplaceFilterMapNextWithFindMap {
1228                                 file: next_source_ptr.file_id,
1229                                 next_expr: next_source_ptr.value,
1230                             }
1231                             .into(),
1232                         );
1233                     }
1234                 }
1235                 BodyValidationDiagnostic::MismatchedArgCount { call_expr, expected, found } => {
1236                     match source_map.expr_syntax(call_expr) {
1237                         Ok(source_ptr) => acc.push(
1238                             MismatchedArgCount { call_expr: source_ptr, expected, found }.into(),
1239                         ),
1240                         Err(SyntheticSyntax) => (),
1241                     }
1242                 }
1243                 BodyValidationDiagnostic::RemoveThisSemicolon { expr } => {
1244                     match source_map.expr_syntax(expr) {
1245                         Ok(expr) => acc.push(RemoveThisSemicolon { expr }.into()),
1246                         Err(SyntheticSyntax) => (),
1247                     }
1248                 }
1249                 BodyValidationDiagnostic::MissingOkOrSomeInTailExpr { expr, required } => {
1250                     match source_map.expr_syntax(expr) {
1251                         Ok(expr) => acc.push(
1252                             MissingOkOrSomeInTailExpr {
1253                                 expr,
1254                                 required,
1255                                 expected: self.body_type(db),
1256                             }
1257                             .into(),
1258                         ),
1259                         Err(SyntheticSyntax) => (),
1260                     }
1261                 }
1262                 BodyValidationDiagnostic::MissingMatchArms { match_expr } => {
1263                     match source_map.expr_syntax(match_expr) {
1264                         Ok(source_ptr) => {
1265                             let root = source_ptr.file_syntax(db.upcast());
1266                             if let ast::Expr::MatchExpr(match_expr) =
1267                                 &source_ptr.value.to_node(&root)
1268                             {
1269                                 if let (Some(match_expr), Some(arms)) =
1270                                     (match_expr.expr(), match_expr.match_arm_list())
1271                                 {
1272                                     acc.push(
1273                                         MissingMatchArms {
1274                                             file: source_ptr.file_id,
1275                                             match_expr: AstPtr::new(&match_expr),
1276                                             arms: AstPtr::new(&arms),
1277                                         }
1278                                         .into(),
1279                                     )
1280                                 }
1281                             }
1282                         }
1283                         Err(SyntheticSyntax) => (),
1284                     }
1285                 }
1286                 BodyValidationDiagnostic::AddReferenceHere { arg_expr, mutability } => {
1287                     match source_map.expr_syntax(arg_expr) {
1288                         Ok(expr) => acc.push(AddReferenceHere { expr, mutability }.into()),
1289                         Err(SyntheticSyntax) => (),
1290                     }
1291                 }
1292             }
1293         }
1294
1295         let def: ModuleDef = match self {
1296             DefWithBody::Function(it) => it.into(),
1297             DefWithBody::Static(it) => it.into(),
1298             DefWithBody::Const(it) => it.into(),
1299         };
1300         for diag in hir_ty::diagnostics::incorrect_case(db, krate, def.into()) {
1301             acc.push(diag.into())
1302         }
1303     }
1304 }
1305
1306 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
1307 pub struct Function {
1308     pub(crate) id: FunctionId,
1309 }
1310
1311 impl Function {
1312     pub fn module(self, db: &dyn HirDatabase) -> Module {
1313         self.id.lookup(db.upcast()).module(db.upcast()).into()
1314     }
1315
1316     pub fn name(self, db: &dyn HirDatabase) -> Name {
1317         db.function_data(self.id).name.clone()
1318     }
1319
1320     /// Get this function's return type
1321     pub fn ret_type(self, db: &dyn HirDatabase) -> Type {
1322         let resolver = self.id.resolver(db.upcast());
1323         let krate = self.id.lookup(db.upcast()).container.module(db.upcast()).krate();
1324         let ret_type = &db.function_data(self.id).ret_type;
1325         let ctx = hir_ty::TyLoweringContext::new(db, &resolver);
1326         let ty = ctx.lower_ty(ret_type);
1327         Type::new_with_resolver_inner(db, krate, &resolver, ty)
1328     }
1329
1330     pub fn self_param(self, db: &dyn HirDatabase) -> Option<SelfParam> {
1331         if !db.function_data(self.id).has_self_param() {
1332             return None;
1333         }
1334         Some(SelfParam { func: self.id })
1335     }
1336
1337     pub fn assoc_fn_params(self, db: &dyn HirDatabase) -> Vec<Param> {
1338         let resolver = self.id.resolver(db.upcast());
1339         let krate = self.id.lookup(db.upcast()).container.module(db.upcast()).krate();
1340         let ctx = hir_ty::TyLoweringContext::new(db, &resolver);
1341         let environment = db.trait_environment(self.id.into());
1342         db.function_data(self.id)
1343             .params
1344             .iter()
1345             .enumerate()
1346             .map(|(idx, (_, type_ref))| {
1347                 let ty = Type { krate, env: environment.clone(), ty: ctx.lower_ty(type_ref) };
1348                 Param { func: self, ty, idx }
1349             })
1350             .collect()
1351     }
1352
1353     pub fn method_params(self, db: &dyn HirDatabase) -> Option<Vec<Param>> {
1354         if self.self_param(db).is_none() {
1355             return None;
1356         }
1357         let mut res = self.assoc_fn_params(db);
1358         res.remove(0);
1359         Some(res)
1360     }
1361
1362     pub fn is_unsafe(self, db: &dyn HirDatabase) -> bool {
1363         db.function_data(self.id).is_unsafe()
1364     }
1365
1366     pub fn is_const(self, db: &dyn HirDatabase) -> bool {
1367         db.function_data(self.id).is_const()
1368     }
1369
1370     pub fn is_async(self, db: &dyn HirDatabase) -> bool {
1371         db.function_data(self.id).is_async()
1372     }
1373
1374     /// Whether this function declaration has a definition.
1375     ///
1376     /// This is false in the case of required (not provided) trait methods.
1377     pub fn has_body(self, db: &dyn HirDatabase) -> bool {
1378         db.function_data(self.id).has_body()
1379     }
1380
1381     /// A textual representation of the HIR of this function for debugging purposes.
1382     pub fn debug_hir(self, db: &dyn HirDatabase) -> String {
1383         let body = db.body(self.id.into());
1384
1385         let mut result = String::new();
1386         format_to!(result, "HIR expressions in the body of `{}`:\n", self.name(db));
1387         for (id, expr) in body.exprs.iter() {
1388             format_to!(result, "{:?}: {:?}\n", id, expr);
1389         }
1390
1391         result
1392     }
1393 }
1394
1395 // Note: logically, this belongs to `hir_ty`, but we are not using it there yet.
1396 pub enum Access {
1397     Shared,
1398     Exclusive,
1399     Owned,
1400 }
1401
1402 impl From<hir_ty::Mutability> for Access {
1403     fn from(mutability: hir_ty::Mutability) -> Access {
1404         match mutability {
1405             hir_ty::Mutability::Not => Access::Shared,
1406             hir_ty::Mutability::Mut => Access::Exclusive,
1407         }
1408     }
1409 }
1410
1411 #[derive(Clone, Debug)]
1412 pub struct Param {
1413     func: Function,
1414     /// The index in parameter list, including self parameter.
1415     idx: usize,
1416     ty: Type,
1417 }
1418
1419 impl Param {
1420     pub fn ty(&self) -> &Type {
1421         &self.ty
1422     }
1423
1424     pub fn name(&self, db: &dyn HirDatabase) -> Option<Name> {
1425         db.function_data(self.func.id).params[self.idx].0.clone()
1426     }
1427
1428     pub fn as_local(&self, db: &dyn HirDatabase) -> Local {
1429         let parent = DefWithBodyId::FunctionId(self.func.into());
1430         let body = db.body(parent);
1431         Local { parent, pat_id: body.params[self.idx] }
1432     }
1433
1434     pub fn pattern_source(&self, db: &dyn HirDatabase) -> Option<ast::Pat> {
1435         self.source(db).and_then(|p| p.value.pat())
1436     }
1437
1438     pub fn source(&self, db: &dyn HirDatabase) -> Option<InFile<ast::Param>> {
1439         let InFile { file_id, value } = self.func.source(db)?;
1440         let params = value.param_list()?;
1441         if params.self_param().is_some() {
1442             params.params().nth(self.idx.checked_sub(1)?)
1443         } else {
1444             params.params().nth(self.idx)
1445         }
1446         .map(|value| InFile { file_id, value })
1447     }
1448 }
1449
1450 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
1451 pub struct SelfParam {
1452     func: FunctionId,
1453 }
1454
1455 impl SelfParam {
1456     pub fn access(self, db: &dyn HirDatabase) -> Access {
1457         let func_data = db.function_data(self.func);
1458         func_data
1459             .params
1460             .first()
1461             .map(|(_, param)| match &**param {
1462                 TypeRef::Reference(.., mutability) => match mutability {
1463                     hir_def::type_ref::Mutability::Shared => Access::Shared,
1464                     hir_def::type_ref::Mutability::Mut => Access::Exclusive,
1465                 },
1466                 _ => Access::Owned,
1467             })
1468             .unwrap_or(Access::Owned)
1469     }
1470
1471     pub fn display(self, db: &dyn HirDatabase) -> &'static str {
1472         match self.access(db) {
1473             Access::Shared => "&self",
1474             Access::Exclusive => "&mut self",
1475             Access::Owned => "self",
1476         }
1477     }
1478
1479     pub fn source(&self, db: &dyn HirDatabase) -> Option<InFile<ast::SelfParam>> {
1480         let InFile { file_id, value } = Function::from(self.func).source(db)?;
1481         value
1482             .param_list()
1483             .and_then(|params| params.self_param())
1484             .map(|value| InFile { file_id, value })
1485     }
1486
1487     pub fn ty(&self, db: &dyn HirDatabase) -> Type {
1488         let resolver = self.func.resolver(db.upcast());
1489         let krate = self.func.lookup(db.upcast()).container.module(db.upcast()).krate();
1490         let ctx = hir_ty::TyLoweringContext::new(db, &resolver);
1491         let environment = db.trait_environment(self.func.into());
1492
1493         Type {
1494             krate,
1495             env: environment.clone(),
1496             ty: ctx.lower_ty(&db.function_data(self.func).params[0].1),
1497         }
1498     }
1499 }
1500
1501 impl HasVisibility for Function {
1502     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
1503         let function_data = db.function_data(self.id);
1504         let visibility = &function_data.visibility;
1505         visibility.resolve(db.upcast(), &self.id.resolver(db.upcast()))
1506     }
1507 }
1508
1509 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
1510 pub struct Const {
1511     pub(crate) id: ConstId,
1512 }
1513
1514 impl Const {
1515     pub fn module(self, db: &dyn HirDatabase) -> Module {
1516         Module { id: self.id.lookup(db.upcast()).module(db.upcast()) }
1517     }
1518
1519     pub fn name(self, db: &dyn HirDatabase) -> Option<Name> {
1520         db.const_data(self.id).name.clone()
1521     }
1522
1523     pub fn value(self, db: &dyn HirDatabase) -> Option<ast::Expr> {
1524         self.source(db)?.value.body()
1525     }
1526
1527     pub fn ty(self, db: &dyn HirDatabase) -> Type {
1528         let data = db.const_data(self.id);
1529         let resolver = self.id.resolver(db.upcast());
1530         let krate = self.id.lookup(db.upcast()).container.krate(db);
1531         let ctx = hir_ty::TyLoweringContext::new(db, &resolver);
1532         let ty = ctx.lower_ty(&data.type_ref);
1533         Type::new_with_resolver_inner(db, krate.id, &resolver, ty)
1534     }
1535
1536     pub fn eval(self, db: &dyn HirDatabase) -> Result<ComputedExpr, ConstEvalError> {
1537         let body = db.body(self.id.into());
1538         let root = &body.exprs[body.body_expr];
1539         let infer = db.infer_query(self.id.into());
1540         let infer = infer.as_ref();
1541         let result = eval_const(
1542             root,
1543             ConstEvalCtx {
1544                 exprs: &body.exprs,
1545                 pats: &body.pats,
1546                 local_data: HashMap::default(),
1547                 infer,
1548             },
1549         );
1550         result
1551     }
1552 }
1553
1554 impl HasVisibility for Const {
1555     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
1556         let function_data = db.const_data(self.id);
1557         let visibility = &function_data.visibility;
1558         visibility.resolve(db.upcast(), &self.id.resolver(db.upcast()))
1559     }
1560 }
1561
1562 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
1563 pub struct Static {
1564     pub(crate) id: StaticId,
1565 }
1566
1567 impl Static {
1568     pub fn module(self, db: &dyn HirDatabase) -> Module {
1569         Module { id: self.id.lookup(db.upcast()).module(db.upcast()) }
1570     }
1571
1572     pub fn name(self, db: &dyn HirDatabase) -> Name {
1573         db.static_data(self.id).name.clone()
1574     }
1575
1576     pub fn is_mut(self, db: &dyn HirDatabase) -> bool {
1577         db.static_data(self.id).mutable
1578     }
1579
1580     pub fn value(self, db: &dyn HirDatabase) -> Option<ast::Expr> {
1581         self.source(db)?.value.body()
1582     }
1583
1584     pub fn ty(self, db: &dyn HirDatabase) -> Type {
1585         let data = db.static_data(self.id);
1586         let resolver = self.id.resolver(db.upcast());
1587         let krate = self.id.lookup(db.upcast()).container.module(db.upcast()).krate();
1588         let ctx = hir_ty::TyLoweringContext::new(db, &resolver);
1589         let ty = ctx.lower_ty(&data.type_ref);
1590         Type::new_with_resolver_inner(db, krate, &resolver, ty)
1591     }
1592 }
1593
1594 impl HasVisibility for Static {
1595     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
1596         db.static_data(self.id).visibility.resolve(db.upcast(), &self.id.resolver(db.upcast()))
1597     }
1598 }
1599
1600 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
1601 pub struct Trait {
1602     pub(crate) id: TraitId,
1603 }
1604
1605 impl Trait {
1606     pub fn module(self, db: &dyn HirDatabase) -> Module {
1607         Module { id: self.id.lookup(db.upcast()).container }
1608     }
1609
1610     pub fn name(self, db: &dyn HirDatabase) -> Name {
1611         db.trait_data(self.id).name.clone()
1612     }
1613
1614     pub fn items(self, db: &dyn HirDatabase) -> Vec<AssocItem> {
1615         db.trait_data(self.id).items.iter().map(|(_name, it)| (*it).into()).collect()
1616     }
1617
1618     pub fn is_auto(self, db: &dyn HirDatabase) -> bool {
1619         db.trait_data(self.id).is_auto
1620     }
1621
1622     pub fn is_unsafe(&self, db: &dyn HirDatabase) -> bool {
1623         db.trait_data(self.id).is_unsafe
1624     }
1625 }
1626
1627 impl HasVisibility for Trait {
1628     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
1629         db.trait_data(self.id).visibility.resolve(db.upcast(), &self.id.resolver(db.upcast()))
1630     }
1631 }
1632
1633 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
1634 pub struct TypeAlias {
1635     pub(crate) id: TypeAliasId,
1636 }
1637
1638 impl TypeAlias {
1639     pub fn has_non_default_type_params(self, db: &dyn HirDatabase) -> bool {
1640         let subst = db.generic_defaults(self.id.into());
1641         subst.iter().any(|ty| ty.skip_binders().is_unknown())
1642     }
1643
1644     pub fn module(self, db: &dyn HirDatabase) -> Module {
1645         Module { id: self.id.lookup(db.upcast()).module(db.upcast()) }
1646     }
1647
1648     pub fn type_ref(self, db: &dyn HirDatabase) -> Option<TypeRef> {
1649         db.type_alias_data(self.id).type_ref.as_deref().cloned()
1650     }
1651
1652     pub fn ty(self, db: &dyn HirDatabase) -> Type {
1653         Type::from_def(db, self.id.lookup(db.upcast()).module(db.upcast()).krate(), self.id)
1654     }
1655
1656     pub fn name(self, db: &dyn HirDatabase) -> Name {
1657         db.type_alias_data(self.id).name.clone()
1658     }
1659 }
1660
1661 impl HasVisibility for TypeAlias {
1662     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
1663         let function_data = db.type_alias_data(self.id);
1664         let visibility = &function_data.visibility;
1665         visibility.resolve(db.upcast(), &self.id.resolver(db.upcast()))
1666     }
1667 }
1668
1669 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
1670 pub struct BuiltinType {
1671     pub(crate) inner: hir_def::builtin_type::BuiltinType,
1672 }
1673
1674 impl BuiltinType {
1675     pub fn str() -> BuiltinType {
1676         BuiltinType { inner: hir_def::builtin_type::BuiltinType::Str }
1677     }
1678
1679     pub fn ty(self, db: &dyn HirDatabase, module: Module) -> Type {
1680         let resolver = module.id.resolver(db.upcast());
1681         Type::new_with_resolver(db, &resolver, TyBuilder::builtin(self.inner))
1682             .expect("crate not present in resolver")
1683     }
1684
1685     pub fn name(self) -> Name {
1686         self.inner.as_name()
1687     }
1688
1689     pub fn is_int(&self) -> bool {
1690         matches!(self.inner, hir_def::builtin_type::BuiltinType::Int(_))
1691     }
1692
1693     pub fn is_uint(&self) -> bool {
1694         matches!(self.inner, hir_def::builtin_type::BuiltinType::Uint(_))
1695     }
1696
1697     pub fn is_float(&self) -> bool {
1698         matches!(self.inner, hir_def::builtin_type::BuiltinType::Float(_))
1699     }
1700
1701     pub fn is_char(&self) -> bool {
1702         matches!(self.inner, hir_def::builtin_type::BuiltinType::Char)
1703     }
1704
1705     pub fn is_str(&self) -> bool {
1706         matches!(self.inner, hir_def::builtin_type::BuiltinType::Str)
1707     }
1708 }
1709
1710 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
1711 pub enum MacroKind {
1712     /// `macro_rules!` or Macros 2.0 macro.
1713     Declarative,
1714     /// A built-in or custom derive.
1715     Derive,
1716     /// A built-in function-like macro.
1717     BuiltIn,
1718     /// A procedural attribute macro.
1719     Attr,
1720     /// A function-like procedural macro.
1721     ProcMacro,
1722 }
1723
1724 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
1725 pub struct MacroDef {
1726     pub(crate) id: MacroDefId,
1727 }
1728
1729 impl MacroDef {
1730     /// FIXME: right now, this just returns the root module of the crate that
1731     /// defines this macro. The reasons for this is that macros are expanded
1732     /// early, in `hir_expand`, where modules simply do not exist yet.
1733     pub fn module(self, db: &dyn HirDatabase) -> Option<Module> {
1734         let krate = self.id.krate;
1735         let def_map = db.crate_def_map(krate);
1736         let module_id = def_map.root();
1737         Some(Module { id: def_map.module_id(module_id) })
1738     }
1739
1740     /// XXX: this parses the file
1741     pub fn name(self, db: &dyn HirDatabase) -> Option<Name> {
1742         match self.source(db)?.value {
1743             Either::Left(it) => it.name().map(|it| it.as_name()),
1744             Either::Right(_) => {
1745                 let krate = self.id.krate;
1746                 let def_map = db.crate_def_map(krate);
1747                 let (_, name) = def_map.exported_proc_macros().find(|&(id, _)| id == self.id)?;
1748                 Some(name)
1749             }
1750         }
1751     }
1752
1753     pub fn kind(&self) -> MacroKind {
1754         match self.id.kind {
1755             MacroDefKind::Declarative(_) => MacroKind::Declarative,
1756             MacroDefKind::BuiltIn(_, _) | MacroDefKind::BuiltInEager(_, _) => MacroKind::BuiltIn,
1757             MacroDefKind::BuiltInDerive(_, _) => MacroKind::Derive,
1758             MacroDefKind::BuiltInAttr(_, _) => MacroKind::Attr,
1759             MacroDefKind::ProcMacro(_, base_db::ProcMacroKind::CustomDerive, _) => {
1760                 MacroKind::Derive
1761             }
1762             MacroDefKind::ProcMacro(_, base_db::ProcMacroKind::Attr, _) => MacroKind::Attr,
1763             MacroDefKind::ProcMacro(_, base_db::ProcMacroKind::FuncLike, _) => MacroKind::ProcMacro,
1764         }
1765     }
1766
1767     pub fn is_fn_like(&self) -> bool {
1768         match self.kind() {
1769             MacroKind::Declarative | MacroKind::BuiltIn | MacroKind::ProcMacro => true,
1770             MacroKind::Attr | MacroKind::Derive => false,
1771         }
1772     }
1773
1774     pub fn is_attr(&self) -> bool {
1775         matches!(self.kind(), MacroKind::Attr)
1776     }
1777 }
1778
1779 #[derive(Clone, Copy, PartialEq, Eq, Debug, Hash)]
1780 pub enum ItemInNs {
1781     Types(ModuleDef),
1782     Values(ModuleDef),
1783     Macros(MacroDef),
1784 }
1785
1786 impl From<MacroDef> for ItemInNs {
1787     fn from(it: MacroDef) -> Self {
1788         Self::Macros(it)
1789     }
1790 }
1791
1792 impl From<ModuleDef> for ItemInNs {
1793     fn from(module_def: ModuleDef) -> Self {
1794         match module_def {
1795             ModuleDef::Static(_) | ModuleDef::Const(_) | ModuleDef::Function(_) => {
1796                 ItemInNs::Values(module_def)
1797             }
1798             _ => ItemInNs::Types(module_def),
1799         }
1800     }
1801 }
1802
1803 impl ItemInNs {
1804     pub fn as_module_def(self) -> Option<ModuleDef> {
1805         match self {
1806             ItemInNs::Types(id) | ItemInNs::Values(id) => Some(id),
1807             ItemInNs::Macros(_) => None,
1808         }
1809     }
1810
1811     /// Returns the crate defining this item (or `None` if `self` is built-in).
1812     pub fn krate(&self, db: &dyn HirDatabase) -> Option<Crate> {
1813         match self {
1814             ItemInNs::Types(did) | ItemInNs::Values(did) => did.module(db).map(|m| m.krate()),
1815             ItemInNs::Macros(id) => id.module(db).map(|m| m.krate()),
1816         }
1817     }
1818
1819     pub fn attrs(&self, db: &dyn HirDatabase) -> Option<AttrsWithOwner> {
1820         match self {
1821             ItemInNs::Types(it) | ItemInNs::Values(it) => it.attrs(db),
1822             ItemInNs::Macros(it) => Some(it.attrs(db)),
1823         }
1824     }
1825 }
1826
1827 /// Invariant: `inner.as_assoc_item(db).is_some()`
1828 /// We do not actively enforce this invariant.
1829 #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
1830 pub enum AssocItem {
1831     Function(Function),
1832     Const(Const),
1833     TypeAlias(TypeAlias),
1834 }
1835 #[derive(Debug)]
1836 pub enum AssocItemContainer {
1837     Trait(Trait),
1838     Impl(Impl),
1839 }
1840 pub trait AsAssocItem {
1841     fn as_assoc_item(self, db: &dyn HirDatabase) -> Option<AssocItem>;
1842 }
1843
1844 impl AsAssocItem for Function {
1845     fn as_assoc_item(self, db: &dyn HirDatabase) -> Option<AssocItem> {
1846         as_assoc_item(db, AssocItem::Function, self.id)
1847     }
1848 }
1849 impl AsAssocItem for Const {
1850     fn as_assoc_item(self, db: &dyn HirDatabase) -> Option<AssocItem> {
1851         as_assoc_item(db, AssocItem::Const, self.id)
1852     }
1853 }
1854 impl AsAssocItem for TypeAlias {
1855     fn as_assoc_item(self, db: &dyn HirDatabase) -> Option<AssocItem> {
1856         as_assoc_item(db, AssocItem::TypeAlias, self.id)
1857     }
1858 }
1859 impl AsAssocItem for ModuleDef {
1860     fn as_assoc_item(self, db: &dyn HirDatabase) -> Option<AssocItem> {
1861         match self {
1862             ModuleDef::Function(it) => it.as_assoc_item(db),
1863             ModuleDef::Const(it) => it.as_assoc_item(db),
1864             ModuleDef::TypeAlias(it) => it.as_assoc_item(db),
1865             _ => None,
1866         }
1867     }
1868 }
1869 fn as_assoc_item<ID, DEF, CTOR, AST>(db: &dyn HirDatabase, ctor: CTOR, id: ID) -> Option<AssocItem>
1870 where
1871     ID: Lookup<Data = AssocItemLoc<AST>>,
1872     DEF: From<ID>,
1873     CTOR: FnOnce(DEF) -> AssocItem,
1874     AST: ItemTreeNode,
1875 {
1876     match id.lookup(db.upcast()).container {
1877         ItemContainerId::TraitId(_) | ItemContainerId::ImplId(_) => Some(ctor(DEF::from(id))),
1878         ItemContainerId::ModuleId(_) | ItemContainerId::ExternBlockId(_) => None,
1879     }
1880 }
1881
1882 impl AssocItem {
1883     pub fn name(self, db: &dyn HirDatabase) -> Option<Name> {
1884         match self {
1885             AssocItem::Function(it) => Some(it.name(db)),
1886             AssocItem::Const(it) => it.name(db),
1887             AssocItem::TypeAlias(it) => Some(it.name(db)),
1888         }
1889     }
1890     pub fn module(self, db: &dyn HirDatabase) -> Module {
1891         match self {
1892             AssocItem::Function(f) => f.module(db),
1893             AssocItem::Const(c) => c.module(db),
1894             AssocItem::TypeAlias(t) => t.module(db),
1895         }
1896     }
1897     pub fn container(self, db: &dyn HirDatabase) -> AssocItemContainer {
1898         let container = match self {
1899             AssocItem::Function(it) => it.id.lookup(db.upcast()).container,
1900             AssocItem::Const(it) => it.id.lookup(db.upcast()).container,
1901             AssocItem::TypeAlias(it) => it.id.lookup(db.upcast()).container,
1902         };
1903         match container {
1904             ItemContainerId::TraitId(id) => AssocItemContainer::Trait(id.into()),
1905             ItemContainerId::ImplId(id) => AssocItemContainer::Impl(id.into()),
1906             ItemContainerId::ModuleId(_) | ItemContainerId::ExternBlockId(_) => {
1907                 panic!("invalid AssocItem")
1908             }
1909         }
1910     }
1911
1912     pub fn containing_trait(self, db: &dyn HirDatabase) -> Option<Trait> {
1913         match self.container(db) {
1914             AssocItemContainer::Trait(t) => Some(t),
1915             _ => None,
1916         }
1917     }
1918
1919     pub fn containing_trait_impl(self, db: &dyn HirDatabase) -> Option<Trait> {
1920         match self.container(db) {
1921             AssocItemContainer::Impl(i) => i.trait_(db),
1922             _ => None,
1923         }
1924     }
1925
1926     pub fn containing_trait_or_trait_impl(self, db: &dyn HirDatabase) -> Option<Trait> {
1927         match self.container(db) {
1928             AssocItemContainer::Trait(t) => Some(t),
1929             AssocItemContainer::Impl(i) => i.trait_(db),
1930         }
1931     }
1932 }
1933
1934 impl HasVisibility for AssocItem {
1935     fn visibility(&self, db: &dyn HirDatabase) -> Visibility {
1936         match self {
1937             AssocItem::Function(f) => f.visibility(db),
1938             AssocItem::Const(c) => c.visibility(db),
1939             AssocItem::TypeAlias(t) => t.visibility(db),
1940         }
1941     }
1942 }
1943
1944 impl From<AssocItem> for ModuleDef {
1945     fn from(assoc: AssocItem) -> Self {
1946         match assoc {
1947             AssocItem::Function(it) => ModuleDef::Function(it),
1948             AssocItem::Const(it) => ModuleDef::Const(it),
1949             AssocItem::TypeAlias(it) => ModuleDef::TypeAlias(it),
1950         }
1951     }
1952 }
1953
1954 #[derive(Clone, Copy, PartialEq, Eq, Debug, Hash)]
1955 pub enum GenericDef {
1956     Function(Function),
1957     Adt(Adt),
1958     Trait(Trait),
1959     TypeAlias(TypeAlias),
1960     Impl(Impl),
1961     // enum variants cannot have generics themselves, but their parent enums
1962     // can, and this makes some code easier to write
1963     Variant(Variant),
1964     // consts can have type parameters from their parents (i.e. associated consts of traits)
1965     Const(Const),
1966 }
1967 impl_from!(
1968     Function,
1969     Adt(Struct, Enum, Union),
1970     Trait,
1971     TypeAlias,
1972     Impl,
1973     Variant,
1974     Const
1975     for GenericDef
1976 );
1977
1978 impl GenericDef {
1979     pub fn params(self, db: &dyn HirDatabase) -> Vec<GenericParam> {
1980         let generics = db.generic_params(self.into());
1981         let ty_params = generics
1982             .types
1983             .iter()
1984             .map(|(local_id, _)| TypeParam { id: TypeParamId { parent: self.into(), local_id } })
1985             .map(GenericParam::TypeParam);
1986         let lt_params = generics
1987             .lifetimes
1988             .iter()
1989             .map(|(local_id, _)| LifetimeParam {
1990                 id: LifetimeParamId { parent: self.into(), local_id },
1991             })
1992             .map(GenericParam::LifetimeParam);
1993         let const_params = generics
1994             .consts
1995             .iter()
1996             .map(|(local_id, _)| ConstParam { id: ConstParamId { parent: self.into(), local_id } })
1997             .map(GenericParam::ConstParam);
1998         ty_params.chain(lt_params).chain(const_params).collect()
1999     }
2000
2001     pub fn type_params(self, db: &dyn HirDatabase) -> Vec<TypeParam> {
2002         let generics = db.generic_params(self.into());
2003         generics
2004             .types
2005             .iter()
2006             .map(|(local_id, _)| TypeParam { id: TypeParamId { parent: self.into(), local_id } })
2007             .collect()
2008     }
2009 }
2010
2011 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
2012 pub struct Local {
2013     pub(crate) parent: DefWithBodyId,
2014     pub(crate) pat_id: PatId,
2015 }
2016
2017 impl Local {
2018     pub fn is_param(self, db: &dyn HirDatabase) -> bool {
2019         let src = self.source(db);
2020         match src.value {
2021             Either::Left(bind_pat) => {
2022                 bind_pat.syntax().ancestors().any(|it| ast::Param::can_cast(it.kind()))
2023             }
2024             Either::Right(_self_param) => true,
2025         }
2026     }
2027
2028     pub fn as_self_param(self, db: &dyn HirDatabase) -> Option<SelfParam> {
2029         match self.parent {
2030             DefWithBodyId::FunctionId(func) if self.is_self(db) => Some(SelfParam { func }),
2031             _ => None,
2032         }
2033     }
2034
2035     // FIXME: why is this an option? It shouldn't be?
2036     pub fn name(self, db: &dyn HirDatabase) -> Option<Name> {
2037         let body = db.body(self.parent);
2038         match &body[self.pat_id] {
2039             Pat::Bind { name, .. } => Some(name.clone()),
2040             _ => None,
2041         }
2042     }
2043
2044     pub fn is_self(self, db: &dyn HirDatabase) -> bool {
2045         self.name(db) == Some(name![self])
2046     }
2047
2048     pub fn is_mut(self, db: &dyn HirDatabase) -> bool {
2049         let body = db.body(self.parent);
2050         matches!(&body[self.pat_id], Pat::Bind { mode: BindingAnnotation::Mutable, .. })
2051     }
2052
2053     pub fn is_ref(self, db: &dyn HirDatabase) -> bool {
2054         let body = db.body(self.parent);
2055         matches!(
2056             &body[self.pat_id],
2057             Pat::Bind { mode: BindingAnnotation::Ref | BindingAnnotation::RefMut, .. }
2058         )
2059     }
2060
2061     pub fn parent(self, _db: &dyn HirDatabase) -> DefWithBody {
2062         self.parent.into()
2063     }
2064
2065     pub fn module(self, db: &dyn HirDatabase) -> Module {
2066         self.parent(db).module(db)
2067     }
2068
2069     pub fn ty(self, db: &dyn HirDatabase) -> Type {
2070         let def = self.parent;
2071         let infer = db.infer(def);
2072         let ty = infer[self.pat_id].clone();
2073         let krate = def.module(db.upcast()).krate();
2074         Type::new(db, krate, def, ty)
2075     }
2076
2077     pub fn source(self, db: &dyn HirDatabase) -> InFile<Either<ast::IdentPat, ast::SelfParam>> {
2078         let (_body, source_map) = db.body_with_source_map(self.parent);
2079         let src = source_map.pat_syntax(self.pat_id).unwrap(); // Hmm...
2080         let root = src.file_syntax(db.upcast());
2081         src.map(|ast| {
2082             ast.map_left(|it| it.cast().unwrap().to_node(&root)).map_right(|it| it.to_node(&root))
2083         })
2084     }
2085 }
2086
2087 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
2088 pub struct BuiltinAttr(usize);
2089
2090 impl BuiltinAttr {
2091     pub(crate) fn by_name(name: &str) -> Option<Self> {
2092         // FIXME: def maps registered attrs?
2093         hir_def::builtin_attr::find_builtin_attr_idx(name).map(Self)
2094     }
2095
2096     pub fn name(&self, _: &dyn HirDatabase) -> &str {
2097         // FIXME: Return a `Name` here
2098         hir_def::builtin_attr::INERT_ATTRIBUTES[self.0].name
2099     }
2100
2101     pub fn template(&self, _: &dyn HirDatabase) -> AttributeTemplate {
2102         hir_def::builtin_attr::INERT_ATTRIBUTES[self.0].template
2103     }
2104 }
2105
2106 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
2107 pub struct ToolModule(usize);
2108
2109 impl ToolModule {
2110     pub(crate) fn by_name(name: &str) -> Option<Self> {
2111         // FIXME: def maps registered tools
2112         hir_def::builtin_attr::TOOL_MODULES.iter().position(|&tool| tool == name).map(Self)
2113     }
2114
2115     pub fn name(&self, _: &dyn HirDatabase) -> &str {
2116         // FIXME: Return a `Name` here
2117         hir_def::builtin_attr::TOOL_MODULES[self.0]
2118     }
2119 }
2120
2121 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
2122 pub struct Label {
2123     pub(crate) parent: DefWithBodyId,
2124     pub(crate) label_id: LabelId,
2125 }
2126
2127 impl Label {
2128     pub fn module(self, db: &dyn HirDatabase) -> Module {
2129         self.parent(db).module(db)
2130     }
2131
2132     pub fn parent(self, _db: &dyn HirDatabase) -> DefWithBody {
2133         self.parent.into()
2134     }
2135
2136     pub fn name(self, db: &dyn HirDatabase) -> Name {
2137         let body = db.body(self.parent);
2138         body[self.label_id].name.clone()
2139     }
2140
2141     pub fn source(self, db: &dyn HirDatabase) -> InFile<ast::Label> {
2142         let (_body, source_map) = db.body_with_source_map(self.parent);
2143         let src = source_map.label_syntax(self.label_id);
2144         let root = src.file_syntax(db.upcast());
2145         src.map(|ast| ast.to_node(&root))
2146     }
2147 }
2148
2149 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
2150 pub enum GenericParam {
2151     TypeParam(TypeParam),
2152     LifetimeParam(LifetimeParam),
2153     ConstParam(ConstParam),
2154 }
2155 impl_from!(TypeParam, LifetimeParam, ConstParam for GenericParam);
2156
2157 impl GenericParam {
2158     pub fn module(self, db: &dyn HirDatabase) -> Module {
2159         match self {
2160             GenericParam::TypeParam(it) => it.module(db),
2161             GenericParam::LifetimeParam(it) => it.module(db),
2162             GenericParam::ConstParam(it) => it.module(db),
2163         }
2164     }
2165
2166     pub fn name(self, db: &dyn HirDatabase) -> Name {
2167         match self {
2168             GenericParam::TypeParam(it) => it.name(db),
2169             GenericParam::LifetimeParam(it) => it.name(db),
2170             GenericParam::ConstParam(it) => it.name(db),
2171         }
2172     }
2173 }
2174
2175 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
2176 pub struct TypeParam {
2177     pub(crate) id: TypeParamId,
2178 }
2179
2180 impl TypeParam {
2181     pub fn name(self, db: &dyn HirDatabase) -> Name {
2182         let params = db.generic_params(self.id.parent);
2183         params.types[self.id.local_id].name.clone().unwrap_or_else(Name::missing)
2184     }
2185
2186     pub fn module(self, db: &dyn HirDatabase) -> Module {
2187         self.id.parent.module(db.upcast()).into()
2188     }
2189
2190     pub fn ty(self, db: &dyn HirDatabase) -> Type {
2191         let resolver = self.id.parent.resolver(db.upcast());
2192         let krate = self.id.parent.module(db.upcast()).krate();
2193         let ty = TyKind::Placeholder(hir_ty::to_placeholder_idx(db, self.id)).intern(Interner);
2194         Type::new_with_resolver_inner(db, krate, &resolver, ty)
2195     }
2196
2197     pub fn trait_bounds(self, db: &dyn HirDatabase) -> Vec<Trait> {
2198         db.generic_predicates_for_param(self.id, None)
2199             .iter()
2200             .filter_map(|pred| match &pred.skip_binders().skip_binders() {
2201                 hir_ty::WhereClause::Implemented(trait_ref) => {
2202                     Some(Trait::from(trait_ref.hir_trait_id()))
2203                 }
2204                 _ => None,
2205             })
2206             .collect()
2207     }
2208
2209     pub fn default(self, db: &dyn HirDatabase) -> Option<Type> {
2210         let params = db.generic_defaults(self.id.parent);
2211         let local_idx = hir_ty::param_idx(db, self.id)?;
2212         let resolver = self.id.parent.resolver(db.upcast());
2213         let krate = self.id.parent.module(db.upcast()).krate();
2214         let ty = params.get(local_idx)?.clone();
2215         let subst = TyBuilder::type_params_subst(db, self.id.parent);
2216         let ty = ty.substitute(Interner, &subst_prefix(&subst, local_idx));
2217         Some(Type::new_with_resolver_inner(db, krate, &resolver, ty))
2218     }
2219 }
2220
2221 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
2222 pub struct LifetimeParam {
2223     pub(crate) id: LifetimeParamId,
2224 }
2225
2226 impl LifetimeParam {
2227     pub fn name(self, db: &dyn HirDatabase) -> Name {
2228         let params = db.generic_params(self.id.parent);
2229         params.lifetimes[self.id.local_id].name.clone()
2230     }
2231
2232     pub fn module(self, db: &dyn HirDatabase) -> Module {
2233         self.id.parent.module(db.upcast()).into()
2234     }
2235
2236     pub fn parent(self, _db: &dyn HirDatabase) -> GenericDef {
2237         self.id.parent.into()
2238     }
2239 }
2240
2241 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
2242 pub struct ConstParam {
2243     pub(crate) id: ConstParamId,
2244 }
2245
2246 impl ConstParam {
2247     pub fn name(self, db: &dyn HirDatabase) -> Name {
2248         let params = db.generic_params(self.id.parent);
2249         params.consts[self.id.local_id].name.clone()
2250     }
2251
2252     pub fn module(self, db: &dyn HirDatabase) -> Module {
2253         self.id.parent.module(db.upcast()).into()
2254     }
2255
2256     pub fn parent(self, _db: &dyn HirDatabase) -> GenericDef {
2257         self.id.parent.into()
2258     }
2259
2260     pub fn ty(self, db: &dyn HirDatabase) -> Type {
2261         let def = self.id.parent;
2262         let krate = def.module(db.upcast()).krate();
2263         Type::new(db, krate, def, db.const_param_ty(self.id))
2264     }
2265 }
2266
2267 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
2268 pub struct Impl {
2269     pub(crate) id: ImplId,
2270 }
2271
2272 impl Impl {
2273     pub fn all_in_crate(db: &dyn HirDatabase, krate: Crate) -> Vec<Impl> {
2274         let inherent = db.inherent_impls_in_crate(krate.id);
2275         let trait_ = db.trait_impls_in_crate(krate.id);
2276
2277         inherent.all_impls().chain(trait_.all_impls()).map(Self::from).collect()
2278     }
2279
2280     pub fn all_for_type(db: &dyn HirDatabase, Type { krate, ty, .. }: Type) -> Vec<Impl> {
2281         let def_crates = match method_resolution::def_crates(db, &ty, krate) {
2282             Some(def_crates) => def_crates,
2283             None => return Vec::new(),
2284         };
2285
2286         let filter = |impl_def: &Impl| {
2287             let self_ty = impl_def.self_ty(db);
2288             let rref = self_ty.remove_ref();
2289             ty.equals_ctor(rref.as_ref().map_or(&self_ty.ty, |it| &it.ty))
2290         };
2291
2292         let fp = TyFingerprint::for_inherent_impl(&ty);
2293         let fp = match fp {
2294             Some(fp) => fp,
2295             None => return Vec::new(),
2296         };
2297
2298         let mut all = Vec::new();
2299         def_crates.iter().for_each(|&id| {
2300             all.extend(
2301                 db.inherent_impls_in_crate(id)
2302                     .for_self_ty(&ty)
2303                     .iter()
2304                     .cloned()
2305                     .map(Self::from)
2306                     .filter(filter),
2307             )
2308         });
2309         for id in def_crates
2310             .iter()
2311             .flat_map(|&id| Crate { id }.transitive_reverse_dependencies(db))
2312             .map(|Crate { id }| id)
2313             .chain(def_crates.iter().copied())
2314             .unique()
2315         {
2316             all.extend(
2317                 db.trait_impls_in_crate(id)
2318                     .for_self_ty_without_blanket_impls(fp)
2319                     .map(Self::from)
2320                     .filter(filter),
2321             );
2322         }
2323         all
2324     }
2325
2326     pub fn all_for_trait(db: &dyn HirDatabase, trait_: Trait) -> Vec<Impl> {
2327         let krate = trait_.module(db).krate();
2328         let mut all = Vec::new();
2329         for Crate { id } in krate.transitive_reverse_dependencies(db).into_iter() {
2330             let impls = db.trait_impls_in_crate(id);
2331             all.extend(impls.for_trait(trait_.id).map(Self::from))
2332         }
2333         all
2334     }
2335
2336     // FIXME: the return type is wrong. This should be a hir version of
2337     // `TraitRef` (to account for parameters and qualifiers)
2338     pub fn trait_(self, db: &dyn HirDatabase) -> Option<Trait> {
2339         let trait_ref = db.impl_trait(self.id)?.skip_binders().clone();
2340         let id = hir_ty::from_chalk_trait_id(trait_ref.trait_id);
2341         Some(Trait { id })
2342     }
2343
2344     pub fn self_ty(self, db: &dyn HirDatabase) -> Type {
2345         let impl_data = db.impl_data(self.id);
2346         let resolver = self.id.resolver(db.upcast());
2347         let krate = self.id.lookup(db.upcast()).container.krate();
2348         let ctx = hir_ty::TyLoweringContext::new(db, &resolver);
2349         let ty = ctx.lower_ty(&impl_data.self_ty);
2350         Type::new_with_resolver_inner(db, krate, &resolver, ty)
2351     }
2352
2353     pub fn items(self, db: &dyn HirDatabase) -> Vec<AssocItem> {
2354         db.impl_data(self.id).items.iter().map(|it| (*it).into()).collect()
2355     }
2356
2357     pub fn is_negative(self, db: &dyn HirDatabase) -> bool {
2358         db.impl_data(self.id).is_negative
2359     }
2360
2361     pub fn module(self, db: &dyn HirDatabase) -> Module {
2362         self.id.lookup(db.upcast()).container.into()
2363     }
2364
2365     pub fn is_builtin_derive(self, db: &dyn HirDatabase) -> Option<InFile<ast::Attr>> {
2366         let src = self.source(db)?;
2367         let item = src.file_id.is_builtin_derive(db.upcast())?;
2368         let hygenic = hir_expand::hygiene::Hygiene::new(db.upcast(), item.file_id);
2369
2370         // FIXME: handle `cfg_attr`
2371         let attr = item
2372             .value
2373             .attrs()
2374             .filter_map(|it| {
2375                 let path = ModPath::from_src(db.upcast(), it.path()?, &hygenic)?;
2376                 if path.as_ident()?.to_smol_str() == "derive" {
2377                     Some(it)
2378                 } else {
2379                     None
2380                 }
2381             })
2382             .last()?;
2383
2384         Some(item.with_value(attr))
2385     }
2386 }
2387
2388 #[derive(Clone, PartialEq, Eq, Debug)]
2389 pub struct Type {
2390     krate: CrateId,
2391     env: Arc<TraitEnvironment>,
2392     ty: Ty,
2393 }
2394
2395 impl Type {
2396     pub(crate) fn new_with_resolver(
2397         db: &dyn HirDatabase,
2398         resolver: &Resolver,
2399         ty: Ty,
2400     ) -> Option<Type> {
2401         let krate = resolver.krate()?;
2402         Some(Type::new_with_resolver_inner(db, krate, resolver, ty))
2403     }
2404     pub(crate) fn new_with_resolver_inner(
2405         db: &dyn HirDatabase,
2406         krate: CrateId,
2407         resolver: &Resolver,
2408         ty: Ty,
2409     ) -> Type {
2410         let environment = resolver
2411             .generic_def()
2412             .map_or_else(|| Arc::new(TraitEnvironment::empty(krate)), |d| db.trait_environment(d));
2413         Type { krate, env: environment, ty }
2414     }
2415
2416     fn new(db: &dyn HirDatabase, krate: CrateId, lexical_env: impl HasResolver, ty: Ty) -> Type {
2417         let resolver = lexical_env.resolver(db.upcast());
2418         let environment = resolver
2419             .generic_def()
2420             .map_or_else(|| Arc::new(TraitEnvironment::empty(krate)), |d| db.trait_environment(d));
2421         Type { krate, env: environment, ty }
2422     }
2423
2424     fn from_def(
2425         db: &dyn HirDatabase,
2426         krate: CrateId,
2427         def: impl HasResolver + Into<TyDefId>,
2428     ) -> Type {
2429         let ty = TyBuilder::def_ty(db, def.into()).fill_with_unknown().build();
2430         Type::new(db, krate, def, ty)
2431     }
2432
2433     pub fn new_slice(ty: Type) -> Type {
2434         Type { krate: ty.krate, env: ty.env, ty: TyBuilder::slice(ty.ty) }
2435     }
2436
2437     pub fn is_unit(&self) -> bool {
2438         matches!(self.ty.kind(Interner), TyKind::Tuple(0, ..))
2439     }
2440
2441     pub fn is_bool(&self) -> bool {
2442         matches!(self.ty.kind(Interner), TyKind::Scalar(Scalar::Bool))
2443     }
2444
2445     pub fn is_never(&self) -> bool {
2446         matches!(self.ty.kind(Interner), TyKind::Never)
2447     }
2448
2449     pub fn is_mutable_reference(&self) -> bool {
2450         matches!(self.ty.kind(Interner), TyKind::Ref(hir_ty::Mutability::Mut, ..))
2451     }
2452
2453     pub fn is_reference(&self) -> bool {
2454         matches!(self.ty.kind(Interner), TyKind::Ref(..))
2455     }
2456
2457     pub fn is_usize(&self) -> bool {
2458         matches!(self.ty.kind(Interner), TyKind::Scalar(Scalar::Uint(UintTy::Usize)))
2459     }
2460
2461     pub fn remove_ref(&self) -> Option<Type> {
2462         match &self.ty.kind(Interner) {
2463             TyKind::Ref(.., ty) => Some(self.derived(ty.clone())),
2464             _ => None,
2465         }
2466     }
2467
2468     pub fn strip_references(&self) -> Type {
2469         self.derived(self.ty.strip_references().clone())
2470     }
2471
2472     pub fn is_unknown(&self) -> bool {
2473         self.ty.is_unknown()
2474     }
2475
2476     /// Checks that particular type `ty` implements `std::future::Future`.
2477     /// This function is used in `.await` syntax completion.
2478     pub fn impls_future(&self, db: &dyn HirDatabase) -> bool {
2479         // No special case for the type of async block, since Chalk can figure it out.
2480
2481         let krate = self.krate;
2482
2483         let std_future_trait =
2484             db.lang_item(krate, SmolStr::new_inline("future_trait")).and_then(|it| it.as_trait());
2485         let std_future_trait = match std_future_trait {
2486             Some(it) => it,
2487             None => return false,
2488         };
2489
2490         let canonical_ty =
2491             Canonical { value: self.ty.clone(), binders: CanonicalVarKinds::empty(Interner) };
2492         method_resolution::implements_trait(
2493             &canonical_ty,
2494             db,
2495             self.env.clone(),
2496             krate,
2497             std_future_trait,
2498         )
2499     }
2500
2501     /// Checks that particular type `ty` implements `std::ops::FnOnce`.
2502     ///
2503     /// This function can be used to check if a particular type is callable, since FnOnce is a
2504     /// supertrait of Fn and FnMut, so all callable types implements at least FnOnce.
2505     pub fn impls_fnonce(&self, db: &dyn HirDatabase) -> bool {
2506         let krate = self.krate;
2507
2508         let fnonce_trait = match FnTrait::FnOnce.get_id(db, krate) {
2509             Some(it) => it,
2510             None => return false,
2511         };
2512
2513         let canonical_ty =
2514             Canonical { value: self.ty.clone(), binders: CanonicalVarKinds::empty(Interner) };
2515         method_resolution::implements_trait_unique(
2516             &canonical_ty,
2517             db,
2518             self.env.clone(),
2519             krate,
2520             fnonce_trait,
2521         )
2522     }
2523
2524     pub fn impls_trait(&self, db: &dyn HirDatabase, trait_: Trait, args: &[Type]) -> bool {
2525         let trait_ref = TyBuilder::trait_ref(db, trait_.id)
2526             .push(self.ty.clone())
2527             .fill(args.iter().map(|t| t.ty.clone()))
2528             .build();
2529
2530         let goal = Canonical {
2531             value: hir_ty::InEnvironment::new(&self.env.env, trait_ref.cast(Interner)),
2532             binders: CanonicalVarKinds::empty(Interner),
2533         };
2534
2535         db.trait_solve(self.krate, goal).is_some()
2536     }
2537
2538     pub fn normalize_trait_assoc_type(
2539         &self,
2540         db: &dyn HirDatabase,
2541         args: &[Type],
2542         alias: TypeAlias,
2543     ) -> Option<Type> {
2544         let projection = TyBuilder::assoc_type_projection(db, alias.id)
2545             .push(self.ty.clone())
2546             .fill(args.iter().map(|t| t.ty.clone()))
2547             .build();
2548         let goal = hir_ty::make_canonical(
2549             InEnvironment::new(
2550                 &self.env.env,
2551                 AliasEq {
2552                     alias: AliasTy::Projection(projection),
2553                     ty: TyKind::BoundVar(BoundVar::new(DebruijnIndex::INNERMOST, 0))
2554                         .intern(Interner),
2555                 }
2556                 .cast(Interner),
2557             ),
2558             [TyVariableKind::General].into_iter(),
2559         );
2560
2561         match db.trait_solve(self.krate, goal)? {
2562             Solution::Unique(s) => s
2563                 .value
2564                 .subst
2565                 .as_slice(Interner)
2566                 .first()
2567                 .map(|ty| self.derived(ty.assert_ty_ref(Interner).clone())),
2568             Solution::Ambig(_) => None,
2569         }
2570     }
2571
2572     pub fn is_copy(&self, db: &dyn HirDatabase) -> bool {
2573         let lang_item = db.lang_item(self.krate, SmolStr::new_inline("copy"));
2574         let copy_trait = match lang_item {
2575             Some(LangItemTarget::TraitId(it)) => it,
2576             _ => return false,
2577         };
2578         self.impls_trait(db, copy_trait.into(), &[])
2579     }
2580
2581     pub fn as_callable(&self, db: &dyn HirDatabase) -> Option<Callable> {
2582         let def = self.ty.callable_def(db);
2583
2584         let sig = self.ty.callable_sig(db)?;
2585         Some(Callable { ty: self.clone(), sig, def, is_bound_method: false })
2586     }
2587
2588     pub fn is_closure(&self) -> bool {
2589         matches!(&self.ty.kind(Interner), TyKind::Closure { .. })
2590     }
2591
2592     pub fn is_fn(&self) -> bool {
2593         matches!(&self.ty.kind(Interner), TyKind::FnDef(..) | TyKind::Function { .. })
2594     }
2595
2596     pub fn is_array(&self) -> bool {
2597         matches!(&self.ty.kind(Interner), TyKind::Array(..))
2598     }
2599
2600     pub fn is_packed(&self, db: &dyn HirDatabase) -> bool {
2601         let adt_id = match *self.ty.kind(Interner) {
2602             TyKind::Adt(hir_ty::AdtId(adt_id), ..) => adt_id,
2603             _ => return false,
2604         };
2605
2606         let adt = adt_id.into();
2607         match adt {
2608             Adt::Struct(s) => matches!(s.repr(db), Some(ReprKind::Packed)),
2609             _ => false,
2610         }
2611     }
2612
2613     pub fn is_raw_ptr(&self) -> bool {
2614         matches!(&self.ty.kind(Interner), TyKind::Raw(..))
2615     }
2616
2617     pub fn contains_unknown(&self) -> bool {
2618         return go(&self.ty);
2619
2620         fn go(ty: &Ty) -> bool {
2621             match ty.kind(Interner) {
2622                 TyKind::Error => true,
2623
2624                 TyKind::Adt(_, substs)
2625                 | TyKind::AssociatedType(_, substs)
2626                 | TyKind::Tuple(_, substs)
2627                 | TyKind::OpaqueType(_, substs)
2628                 | TyKind::FnDef(_, substs)
2629                 | TyKind::Closure(_, substs) => {
2630                     substs.iter(Interner).filter_map(|a| a.ty(Interner)).any(go)
2631                 }
2632
2633                 TyKind::Array(_ty, len) if len.is_unknown() => true,
2634                 TyKind::Array(ty, _)
2635                 | TyKind::Slice(ty)
2636                 | TyKind::Raw(_, ty)
2637                 | TyKind::Ref(_, _, ty) => go(ty),
2638
2639                 TyKind::Scalar(_)
2640                 | TyKind::Str
2641                 | TyKind::Never
2642                 | TyKind::Placeholder(_)
2643                 | TyKind::BoundVar(_)
2644                 | TyKind::InferenceVar(_, _)
2645                 | TyKind::Dyn(_)
2646                 | TyKind::Function(_)
2647                 | TyKind::Alias(_)
2648                 | TyKind::Foreign(_)
2649                 | TyKind::Generator(..)
2650                 | TyKind::GeneratorWitness(..) => false,
2651             }
2652         }
2653     }
2654
2655     pub fn fields(&self, db: &dyn HirDatabase) -> Vec<(Field, Type)> {
2656         let (variant_id, substs) = match self.ty.kind(Interner) {
2657             TyKind::Adt(hir_ty::AdtId(AdtId::StructId(s)), substs) => ((*s).into(), substs),
2658             TyKind::Adt(hir_ty::AdtId(AdtId::UnionId(u)), substs) => ((*u).into(), substs),
2659             _ => return Vec::new(),
2660         };
2661
2662         db.field_types(variant_id)
2663             .iter()
2664             .map(|(local_id, ty)| {
2665                 let def = Field { parent: variant_id.into(), id: local_id };
2666                 let ty = ty.clone().substitute(Interner, substs);
2667                 (def, self.derived(ty))
2668             })
2669             .collect()
2670     }
2671
2672     pub fn tuple_fields(&self, _db: &dyn HirDatabase) -> Vec<Type> {
2673         if let TyKind::Tuple(_, substs) = &self.ty.kind(Interner) {
2674             substs
2675                 .iter(Interner)
2676                 .map(|ty| self.derived(ty.assert_ty_ref(Interner).clone()))
2677                 .collect()
2678         } else {
2679             Vec::new()
2680         }
2681     }
2682
2683     pub fn autoderef<'a>(&'a self, db: &'a dyn HirDatabase) -> impl Iterator<Item = Type> + 'a {
2684         self.autoderef_(db).map(move |ty| self.derived(ty))
2685     }
2686
2687     pub fn autoderef_<'a>(&'a self, db: &'a dyn HirDatabase) -> impl Iterator<Item = Ty> + 'a {
2688         // There should be no inference vars in types passed here
2689         let canonical = hir_ty::replace_errors_with_variables(&self.ty);
2690         let environment = self.env.env.clone();
2691         let ty = InEnvironment { goal: canonical, environment };
2692         autoderef(db, Some(self.krate), ty).map(|canonical| canonical.value)
2693     }
2694
2695     // This would be nicer if it just returned an iterator, but that runs into
2696     // lifetime problems, because we need to borrow temp `CrateImplDefs`.
2697     pub fn iterate_assoc_items<T>(
2698         &self,
2699         db: &dyn HirDatabase,
2700         krate: Crate,
2701         mut callback: impl FnMut(AssocItem) -> Option<T>,
2702     ) -> Option<T> {
2703         let mut slot = None;
2704         self.iterate_assoc_items_dyn(db, krate, &mut |assoc_item_id| {
2705             slot = callback(assoc_item_id.into());
2706             slot.is_some()
2707         });
2708         slot
2709     }
2710
2711     fn iterate_assoc_items_dyn(
2712         &self,
2713         db: &dyn HirDatabase,
2714         krate: Crate,
2715         callback: &mut dyn FnMut(AssocItemId) -> bool,
2716     ) {
2717         let def_crates = match method_resolution::def_crates(db, &self.ty, krate.id) {
2718             Some(it) => it,
2719             None => return,
2720         };
2721         for krate in def_crates {
2722             let impls = db.inherent_impls_in_crate(krate);
2723
2724             for impl_def in impls.for_self_ty(&self.ty) {
2725                 for &item in db.impl_data(*impl_def).items.iter() {
2726                     if callback(item) {
2727                         return;
2728                     }
2729                 }
2730             }
2731         }
2732     }
2733
2734     pub fn type_arguments(&self) -> impl Iterator<Item = Type> + '_ {
2735         self.ty
2736             .strip_references()
2737             .as_adt()
2738             .into_iter()
2739             .flat_map(|(_, substs)| substs.iter(Interner))
2740             .filter_map(|arg| arg.ty(Interner).cloned())
2741             .map(move |ty| self.derived(ty))
2742     }
2743
2744     pub fn iterate_method_candidates<T>(
2745         &self,
2746         db: &dyn HirDatabase,
2747         krate: Crate,
2748         traits_in_scope: &FxHashSet<TraitId>,
2749         name: Option<&Name>,
2750         mut callback: impl FnMut(Type, Function) -> Option<T>,
2751     ) -> Option<T> {
2752         let _p = profile::span("iterate_method_candidates");
2753         let mut slot = None;
2754
2755         self.iterate_method_candidates_dyn(
2756             db,
2757             krate,
2758             traits_in_scope,
2759             name,
2760             &mut |ty, assoc_item_id| {
2761                 if let AssocItemId::FunctionId(func) = assoc_item_id {
2762                     if let Some(res) = callback(self.derived(ty.clone()), func.into()) {
2763                         slot = Some(res);
2764                         return ControlFlow::Break(());
2765                     }
2766                 }
2767                 ControlFlow::Continue(())
2768             },
2769         );
2770         slot
2771     }
2772
2773     fn iterate_method_candidates_dyn(
2774         &self,
2775         db: &dyn HirDatabase,
2776         krate: Crate,
2777         traits_in_scope: &FxHashSet<TraitId>,
2778         name: Option<&Name>,
2779         callback: &mut dyn FnMut(&Ty, AssocItemId) -> ControlFlow<()>,
2780     ) {
2781         // There should be no inference vars in types passed here
2782         let canonical = hir_ty::replace_errors_with_variables(&self.ty);
2783
2784         let env = self.env.clone();
2785         let krate = krate.id;
2786
2787         method_resolution::iterate_method_candidates_dyn(
2788             &canonical,
2789             db,
2790             env,
2791             krate,
2792             traits_in_scope,
2793             None,
2794             name,
2795             method_resolution::LookupMode::MethodCall,
2796             &mut |ty, id| callback(&ty.value, id),
2797         );
2798     }
2799
2800     pub fn iterate_path_candidates<T>(
2801         &self,
2802         db: &dyn HirDatabase,
2803         krate: Crate,
2804         traits_in_scope: &FxHashSet<TraitId>,
2805         name: Option<&Name>,
2806         mut callback: impl FnMut(Type, AssocItem) -> Option<T>,
2807     ) -> Option<T> {
2808         let _p = profile::span("iterate_path_candidates");
2809         let mut slot = None;
2810         self.iterate_path_candidates_dyn(
2811             db,
2812             krate,
2813             traits_in_scope,
2814             name,
2815             &mut |ty, assoc_item_id| {
2816                 if let Some(res) = callback(self.derived(ty.clone()), assoc_item_id.into()) {
2817                     slot = Some(res);
2818                     return ControlFlow::Break(());
2819                 }
2820                 ControlFlow::Continue(())
2821             },
2822         );
2823         slot
2824     }
2825
2826     fn iterate_path_candidates_dyn(
2827         &self,
2828         db: &dyn HirDatabase,
2829         krate: Crate,
2830         traits_in_scope: &FxHashSet<TraitId>,
2831         name: Option<&Name>,
2832         callback: &mut dyn FnMut(&Ty, AssocItemId) -> ControlFlow<()>,
2833     ) {
2834         let canonical = hir_ty::replace_errors_with_variables(&self.ty);
2835
2836         let env = self.env.clone();
2837         let krate = krate.id;
2838
2839         method_resolution::iterate_method_candidates_dyn(
2840             &canonical,
2841             db,
2842             env,
2843             krate,
2844             traits_in_scope,
2845             None,
2846             name,
2847             method_resolution::LookupMode::Path,
2848             &mut |ty, id| callback(&ty.value, id),
2849         );
2850     }
2851
2852     pub fn as_adt(&self) -> Option<Adt> {
2853         let (adt, _subst) = self.ty.as_adt()?;
2854         Some(adt.into())
2855     }
2856
2857     pub fn as_builtin(&self) -> Option<BuiltinType> {
2858         self.ty.as_builtin().map(|inner| BuiltinType { inner })
2859     }
2860
2861     pub fn as_dyn_trait(&self) -> Option<Trait> {
2862         self.ty.dyn_trait().map(Into::into)
2863     }
2864
2865     /// If a type can be represented as `dyn Trait`, returns all traits accessible via this type,
2866     /// or an empty iterator otherwise.
2867     pub fn applicable_inherent_traits<'a>(
2868         &'a self,
2869         db: &'a dyn HirDatabase,
2870     ) -> impl Iterator<Item = Trait> + 'a {
2871         let _p = profile::span("applicable_inherent_traits");
2872         self.autoderef_(db)
2873             .filter_map(|ty| ty.dyn_trait())
2874             .flat_map(move |dyn_trait_id| hir_ty::all_super_traits(db.upcast(), dyn_trait_id))
2875             .map(Trait::from)
2876     }
2877
2878     pub fn env_traits<'a>(&'a self, db: &'a dyn HirDatabase) -> impl Iterator<Item = Trait> + 'a {
2879         let _p = profile::span("env_traits");
2880         self.autoderef_(db)
2881             .filter(|ty| matches!(ty.kind(Interner), TyKind::Placeholder(_)))
2882             .flat_map(|ty| {
2883                 self.env
2884                     .traits_in_scope_from_clauses(ty)
2885                     .flat_map(|t| hir_ty::all_super_traits(db.upcast(), t))
2886             })
2887             .map(Trait::from)
2888     }
2889
2890     pub fn as_impl_traits(&self, db: &dyn HirDatabase) -> Option<impl Iterator<Item = Trait>> {
2891         self.ty.impl_trait_bounds(db).map(|it| {
2892             it.into_iter().filter_map(|pred| match pred.skip_binders() {
2893                 hir_ty::WhereClause::Implemented(trait_ref) => {
2894                     Some(Trait::from(trait_ref.hir_trait_id()))
2895                 }
2896                 _ => None,
2897             })
2898         })
2899     }
2900
2901     pub fn as_associated_type_parent_trait(&self, db: &dyn HirDatabase) -> Option<Trait> {
2902         self.ty.associated_type_parent_trait(db).map(Into::into)
2903     }
2904
2905     fn derived(&self, ty: Ty) -> Type {
2906         Type { krate: self.krate, env: self.env.clone(), ty }
2907     }
2908
2909     pub fn walk(&self, db: &dyn HirDatabase, mut cb: impl FnMut(Type)) {
2910         // TypeWalk::walk for a Ty at first visits parameters and only after that the Ty itself.
2911         // We need a different order here.
2912
2913         fn walk_substs(
2914             db: &dyn HirDatabase,
2915             type_: &Type,
2916             substs: &Substitution,
2917             cb: &mut impl FnMut(Type),
2918         ) {
2919             for ty in substs.iter(Interner).filter_map(|a| a.ty(Interner)) {
2920                 walk_type(db, &type_.derived(ty.clone()), cb);
2921             }
2922         }
2923
2924         fn walk_bounds(
2925             db: &dyn HirDatabase,
2926             type_: &Type,
2927             bounds: &[QuantifiedWhereClause],
2928             cb: &mut impl FnMut(Type),
2929         ) {
2930             for pred in bounds {
2931                 if let WhereClause::Implemented(trait_ref) = pred.skip_binders() {
2932                     cb(type_.clone());
2933                     // skip the self type. it's likely the type we just got the bounds from
2934                     for ty in
2935                         trait_ref.substitution.iter(Interner).skip(1).filter_map(|a| a.ty(Interner))
2936                     {
2937                         walk_type(db, &type_.derived(ty.clone()), cb);
2938                     }
2939                 }
2940             }
2941         }
2942
2943         fn walk_type(db: &dyn HirDatabase, type_: &Type, cb: &mut impl FnMut(Type)) {
2944             let ty = type_.ty.strip_references();
2945             match ty.kind(Interner) {
2946                 TyKind::Adt(_, substs) => {
2947                     cb(type_.derived(ty.clone()));
2948                     walk_substs(db, type_, substs, cb);
2949                 }
2950                 TyKind::AssociatedType(_, substs) => {
2951                     if ty.associated_type_parent_trait(db).is_some() {
2952                         cb(type_.derived(ty.clone()));
2953                     }
2954                     walk_substs(db, type_, substs, cb);
2955                 }
2956                 TyKind::OpaqueType(_, subst) => {
2957                     if let Some(bounds) = ty.impl_trait_bounds(db) {
2958                         walk_bounds(db, &type_.derived(ty.clone()), &bounds, cb);
2959                     }
2960
2961                     walk_substs(db, type_, subst, cb);
2962                 }
2963                 TyKind::Alias(AliasTy::Opaque(opaque_ty)) => {
2964                     if let Some(bounds) = ty.impl_trait_bounds(db) {
2965                         walk_bounds(db, &type_.derived(ty.clone()), &bounds, cb);
2966                     }
2967
2968                     walk_substs(db, type_, &opaque_ty.substitution, cb);
2969                 }
2970                 TyKind::Placeholder(_) => {
2971                     if let Some(bounds) = ty.impl_trait_bounds(db) {
2972                         walk_bounds(db, &type_.derived(ty.clone()), &bounds, cb);
2973                     }
2974                 }
2975                 TyKind::Dyn(bounds) => {
2976                     walk_bounds(
2977                         db,
2978                         &type_.derived(ty.clone()),
2979                         bounds.bounds.skip_binders().interned(),
2980                         cb,
2981                     );
2982                 }
2983
2984                 TyKind::Ref(_, _, ty)
2985                 | TyKind::Raw(_, ty)
2986                 | TyKind::Array(ty, _)
2987                 | TyKind::Slice(ty) => {
2988                     walk_type(db, &type_.derived(ty.clone()), cb);
2989                 }
2990
2991                 TyKind::FnDef(_, substs)
2992                 | TyKind::Tuple(_, substs)
2993                 | TyKind::Closure(.., substs) => {
2994                     walk_substs(db, type_, substs, cb);
2995                 }
2996                 TyKind::Function(hir_ty::FnPointer { substitution, .. }) => {
2997                     walk_substs(db, type_, &substitution.0, cb);
2998                 }
2999
3000                 _ => {}
3001             }
3002         }
3003
3004         walk_type(db, self, &mut cb);
3005     }
3006
3007     pub fn could_unify_with(&self, db: &dyn HirDatabase, other: &Type) -> bool {
3008         let tys = hir_ty::replace_errors_with_variables(&(self.ty.clone(), other.ty.clone()));
3009         could_unify(db, self.env.clone(), &tys)
3010     }
3011 }
3012
3013 // FIXME: closures
3014 #[derive(Debug)]
3015 pub struct Callable {
3016     ty: Type,
3017     sig: CallableSig,
3018     def: Option<CallableDefId>,
3019     pub(crate) is_bound_method: bool,
3020 }
3021
3022 pub enum CallableKind {
3023     Function(Function),
3024     TupleStruct(Struct),
3025     TupleEnumVariant(Variant),
3026     Closure,
3027 }
3028
3029 impl Callable {
3030     pub fn kind(&self) -> CallableKind {
3031         match self.def {
3032             Some(CallableDefId::FunctionId(it)) => CallableKind::Function(it.into()),
3033             Some(CallableDefId::StructId(it)) => CallableKind::TupleStruct(it.into()),
3034             Some(CallableDefId::EnumVariantId(it)) => CallableKind::TupleEnumVariant(it.into()),
3035             None => CallableKind::Closure,
3036         }
3037     }
3038     pub fn receiver_param(&self, db: &dyn HirDatabase) -> Option<ast::SelfParam> {
3039         let func = match self.def {
3040             Some(CallableDefId::FunctionId(it)) if self.is_bound_method => it,
3041             _ => return None,
3042         };
3043         let src = func.lookup(db.upcast()).source(db.upcast());
3044         let param_list = src.value.param_list()?;
3045         param_list.self_param()
3046     }
3047     pub fn n_params(&self) -> usize {
3048         self.sig.params().len() - if self.is_bound_method { 1 } else { 0 }
3049     }
3050     pub fn params(
3051         &self,
3052         db: &dyn HirDatabase,
3053     ) -> Vec<(Option<Either<ast::SelfParam, ast::Pat>>, Type)> {
3054         let types = self
3055             .sig
3056             .params()
3057             .iter()
3058             .skip(if self.is_bound_method { 1 } else { 0 })
3059             .map(|ty| self.ty.derived(ty.clone()));
3060         let patterns = match self.def {
3061             Some(CallableDefId::FunctionId(func)) => {
3062                 let src = func.lookup(db.upcast()).source(db.upcast());
3063                 src.value.param_list().map(|param_list| {
3064                     param_list
3065                         .self_param()
3066                         .map(|it| Some(Either::Left(it)))
3067                         .filter(|_| !self.is_bound_method)
3068                         .into_iter()
3069                         .chain(param_list.params().map(|it| it.pat().map(Either::Right)))
3070                 })
3071             }
3072             _ => None,
3073         };
3074         patterns.into_iter().flatten().chain(iter::repeat(None)).zip(types).collect()
3075     }
3076     pub fn return_type(&self) -> Type {
3077         self.ty.derived(self.sig.ret().clone())
3078     }
3079 }
3080
3081 /// For IDE only
3082 #[derive(Copy, Clone, Debug, PartialEq, Eq, Hash)]
3083 pub enum ScopeDef {
3084     ModuleDef(ModuleDef),
3085     MacroDef(MacroDef),
3086     GenericParam(GenericParam),
3087     ImplSelfType(Impl),
3088     AdtSelfType(Adt),
3089     Local(Local),
3090     Label(Label),
3091     Unknown,
3092 }
3093
3094 impl ScopeDef {
3095     pub fn all_items(def: PerNs) -> ArrayVec<Self, 3> {
3096         let mut items = ArrayVec::new();
3097
3098         match (def.take_types(), def.take_values()) {
3099             (Some(m1), None) => items.push(ScopeDef::ModuleDef(m1.into())),
3100             (None, Some(m2)) => items.push(ScopeDef::ModuleDef(m2.into())),
3101             (Some(m1), Some(m2)) => {
3102                 // Some items, like unit structs and enum variants, are
3103                 // returned as both a type and a value. Here we want
3104                 // to de-duplicate them.
3105                 if m1 != m2 {
3106                     items.push(ScopeDef::ModuleDef(m1.into()));
3107                     items.push(ScopeDef::ModuleDef(m2.into()));
3108                 } else {
3109                     items.push(ScopeDef::ModuleDef(m1.into()));
3110                 }
3111             }
3112             (None, None) => {}
3113         };
3114
3115         if let Some(macro_def_id) = def.take_macros() {
3116             items.push(ScopeDef::MacroDef(macro_def_id.into()));
3117         }
3118
3119         if items.is_empty() {
3120             items.push(ScopeDef::Unknown);
3121         }
3122
3123         items
3124     }
3125
3126     pub fn attrs(&self, db: &dyn HirDatabase) -> Option<AttrsWithOwner> {
3127         match self {
3128             ScopeDef::ModuleDef(it) => it.attrs(db),
3129             ScopeDef::MacroDef(it) => Some(it.attrs(db)),
3130             ScopeDef::GenericParam(it) => Some(it.attrs(db)),
3131             ScopeDef::ImplSelfType(_)
3132             | ScopeDef::AdtSelfType(_)
3133             | ScopeDef::Local(_)
3134             | ScopeDef::Label(_)
3135             | ScopeDef::Unknown => None,
3136         }
3137     }
3138
3139     pub fn krate(&self, db: &dyn HirDatabase) -> Option<Crate> {
3140         match self {
3141             ScopeDef::ModuleDef(it) => it.module(db).map(|m| m.krate()),
3142             ScopeDef::MacroDef(it) => it.module(db).map(|m| m.krate()),
3143             ScopeDef::GenericParam(it) => Some(it.module(db).krate()),
3144             ScopeDef::ImplSelfType(_) => None,
3145             ScopeDef::AdtSelfType(it) => Some(it.module(db).krate()),
3146             ScopeDef::Local(it) => Some(it.module(db).krate()),
3147             ScopeDef::Label(it) => Some(it.module(db).krate()),
3148             ScopeDef::Unknown => None,
3149         }
3150     }
3151 }
3152
3153 impl From<ItemInNs> for ScopeDef {
3154     fn from(item: ItemInNs) -> Self {
3155         match item {
3156             ItemInNs::Types(id) => ScopeDef::ModuleDef(id),
3157             ItemInNs::Values(id) => ScopeDef::ModuleDef(id),
3158             ItemInNs::Macros(id) => ScopeDef::MacroDef(id),
3159         }
3160     }
3161 }
3162
3163 pub trait HasVisibility {
3164     fn visibility(&self, db: &dyn HirDatabase) -> Visibility;
3165     fn is_visible_from(&self, db: &dyn HirDatabase, module: Module) -> bool {
3166         let vis = self.visibility(db);
3167         vis.is_visible_from(db.upcast(), module.id)
3168     }
3169 }
3170
3171 /// Trait for obtaining the defining crate of an item.
3172 pub trait HasCrate {
3173     fn krate(&self, db: &dyn HirDatabase) -> Crate;
3174 }
3175
3176 impl<T: hir_def::HasModule> HasCrate for T {
3177     fn krate(&self, db: &dyn HirDatabase) -> Crate {
3178         self.module(db.upcast()).krate().into()
3179     }
3180 }
3181
3182 impl HasCrate for AssocItem {
3183     fn krate(&self, db: &dyn HirDatabase) -> Crate {
3184         self.module(db).krate()
3185     }
3186 }
3187
3188 impl HasCrate for Field {
3189     fn krate(&self, db: &dyn HirDatabase) -> Crate {
3190         self.parent_def(db).module(db).krate()
3191     }
3192 }
3193
3194 impl HasCrate for Function {
3195     fn krate(&self, db: &dyn HirDatabase) -> Crate {
3196         self.module(db).krate()
3197     }
3198 }
3199
3200 impl HasCrate for Const {
3201     fn krate(&self, db: &dyn HirDatabase) -> Crate {
3202         self.module(db).krate()
3203     }
3204 }
3205
3206 impl HasCrate for TypeAlias {
3207     fn krate(&self, db: &dyn HirDatabase) -> Crate {
3208         self.module(db).krate()
3209     }
3210 }
3211
3212 impl HasCrate for Type {
3213     fn krate(&self, _db: &dyn HirDatabase) -> Crate {
3214         self.krate.into()
3215     }
3216 }