rust/crates/ra_hir/src/code_model_api.rs

791 lines
23 KiB
Rust
Raw Normal View History

2019-01-08 06:19:37 -06:00
use std::sync::Arc;
use ra_db::{CrateId, SourceRootId, Edition};
2019-03-23 10:35:14 -05:00
use ra_syntax::{ast::self, TreeArc};
2019-01-04 15:02:05 -06:00
2019-01-08 06:27:00 -06:00
use crate::{
Name, Ty, HirFileId, Either,
HirDatabase, DefDatabase,
2019-02-24 08:12:10 -06:00
type_ref::TypeRef,
2019-03-16 10:57:53 -05:00
nameres::{ModuleScope, Namespace, ImportId, CrateModuleId},
2019-03-02 06:14:37 -06:00
expr::{Body, BodySourceMap},
ty::{ TraitRef, InferenceResult},
2019-01-25 11:32:34 -06:00
adt::{EnumVariantId, StructFieldId, VariantDef},
generics::HasGenericParams,
docs::{Documentation, Docs, docs_from_ast},
2019-03-26 11:15:39 -05:00
ids::{FunctionId, StructId, EnumId, AstItemDef, ConstId, StaticId, TraitId, TypeAliasId},
impl_block::ImplBlock,
2019-01-19 14:23:26 -06:00
resolve::Resolver,
2019-03-23 12:41:59 -05:00
diagnostics::DiagnosticSink,
2019-03-24 11:36:15 -05:00
traits::{TraitItem, TraitData},
2019-01-08 06:27:00 -06:00
};
2019-01-04 15:02:05 -06:00
/// hir::Crate describes a single crate. It's the main interface with which
/// a crate's dependencies interact. Mostly, it should be just a proxy for the
2019-01-04 15:02:05 -06:00
/// root module.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
2019-01-04 15:02:05 -06:00
pub struct Crate {
2019-01-30 12:18:21 -06:00
pub(crate) crate_id: CrateId,
2019-01-04 15:02:05 -06:00
}
#[derive(Debug)]
pub struct CrateDependency {
pub krate: Crate,
pub name: Name,
}
impl Crate {
2019-01-06 04:41:12 -06:00
pub fn crate_id(&self) -> CrateId {
self.crate_id
}
2019-02-11 16:11:12 -06:00
pub fn dependencies(&self, db: &impl DefDatabase) -> Vec<CrateDependency> {
2019-01-15 09:33:26 -06:00
self.dependencies_impl(db)
2019-01-04 15:02:05 -06:00
}
2019-02-11 16:11:12 -06:00
pub fn root_module(&self, db: &impl DefDatabase) -> Option<Module> {
2019-01-04 15:02:05 -06:00
self.root_module_impl(db)
}
pub fn edition(&self, db: &impl DefDatabase) -> Edition {
2019-02-11 16:11:12 -06:00
let crate_graph = db.crate_graph();
crate_graph.edition(self.crate_id)
}
2019-03-23 02:53:48 -05:00
// FIXME: should this be in source_binder?
pub fn source_root_crates(db: &impl DefDatabase, source_root: SourceRootId) -> Vec<Crate> {
let crate_ids = db.source_root_crates(source_root);
crate_ids.iter().map(|&crate_id| Crate { crate_id }).collect()
}
2019-01-04 15:02:05 -06:00
}
2019-01-04 16:37:40 -06:00
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
2019-01-04 16:37:40 -06:00
pub struct Module {
2019-01-30 13:23:14 -06:00
pub(crate) krate: Crate,
2019-03-16 10:57:53 -05:00
pub(crate) module_id: CrateModuleId,
}
/// The defs which can be visible in the module.
2019-03-14 03:25:51 -05:00
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum ModuleDef {
Module(Module),
2019-01-24 06:28:50 -06:00
Function(Function),
2019-01-24 08:54:18 -06:00
Struct(Struct),
2019-01-24 09:56:38 -06:00
Enum(Enum),
2019-01-24 16:32:47 -06:00
// Can't be directly declared, but can be imported.
2019-01-24 14:32:41 -06:00
EnumVariant(EnumVariant),
2019-01-24 15:50:08 -06:00
Const(Const),
Static(Static),
2019-01-24 16:31:32 -06:00
Trait(Trait),
2019-02-24 14:36:49 -06:00
TypeAlias(TypeAlias),
}
impl_froms!(
ModuleDef: Module,
Function,
Struct,
Enum,
EnumVariant,
Const,
Static,
Trait,
TypeAlias
);
2019-01-24 08:54:18 -06:00
2019-01-06 06:58:45 -06:00
pub enum ModuleSource {
SourceFile(TreeArc<ast::SourceFile>),
Module(TreeArc<ast::Module>),
2019-01-06 06:58:45 -06:00
}
2019-01-04 16:37:40 -06:00
impl Module {
2019-01-06 07:10:25 -06:00
/// Name of this module.
2019-01-15 09:26:29 -06:00
pub fn name(&self, db: &impl HirDatabase) -> Option<Name> {
2019-01-06 06:58:45 -06:00
self.name_impl(db)
}
2019-01-06 07:10:25 -06:00
/// Returns a node which defines this module. That is, a file or a `mod foo {}` with items.
pub fn definition_source(&self, db: &impl DefDatabase) -> (HirFileId, ModuleSource) {
2019-01-08 16:38:51 -06:00
self.definition_source_impl(db)
2019-01-06 06:58:45 -06:00
}
2019-01-06 07:10:25 -06:00
/// Returns a node which declares this module, either a `mod foo;` or a `mod foo {}`.
/// `None` for the crate root.
2019-01-06 06:58:45 -06:00
pub fn declaration_source(
&self,
db: &impl HirDatabase,
) -> Option<(HirFileId, TreeArc<ast::Module>)> {
2019-01-06 06:58:45 -06:00
self.declaration_source_impl(db)
2019-01-06 04:41:12 -06:00
}
2019-01-18 07:36:56 -06:00
/// Returns the syntax of the last path segment corresponding to this import
2019-04-10 02:12:54 -05:00
pub fn import_source(
&self,
db: &impl HirDatabase,
import: ImportId,
) -> Either<TreeArc<ast::UseTree>, TreeArc<ast::ExternCrateItem>> {
let (file_id, source) = self.definition_source(db);
let (_, source_map) = db.raw_items_with_source_map(file_id);
source_map.get(&source, import)
2019-01-18 07:36:56 -06:00
}
2019-01-04 16:37:40 -06:00
/// Returns the crate this module is part of.
pub fn krate(&self, _db: &impl DefDatabase) -> Option<Crate> {
2019-01-30 13:23:14 -06:00
Some(self.krate)
2019-01-04 16:37:40 -06:00
}
2019-01-06 07:10:25 -06:00
/// Topmost parent of this module. Every module has a `crate_root`, but some
/// might be missing `krate`. This can happen if a module's file is not included
2019-02-11 10:18:27 -06:00
/// in the module tree of any target in `Cargo.toml`.
pub fn crate_root(&self, db: &impl DefDatabase) -> Module {
2019-01-04 16:37:40 -06:00
self.crate_root_impl(db)
}
2019-01-04 16:37:40 -06:00
/// Finds a child module with the specified name.
2019-01-15 09:13:11 -06:00
pub fn child(&self, db: &impl HirDatabase, name: &Name) -> Option<Module> {
2019-01-04 16:37:40 -06:00
self.child_impl(db, name)
}
/// Iterates over all child modules.
pub fn children(&self, db: &impl DefDatabase) -> impl Iterator<Item = Module> {
self.children_impl(db)
}
2019-01-06 05:05:03 -06:00
/// Finds a parent module.
pub fn parent(&self, db: &impl DefDatabase) -> Option<Module> {
2019-01-06 05:05:03 -06:00
self.parent_impl(db)
}
pub fn path_to_root(&self, db: &impl HirDatabase) -> Vec<Module> {
2019-01-06 06:58:45 -06:00
let mut res = vec![self.clone()];
let mut curr = self.clone();
while let Some(next) = curr.parent(db) {
2019-01-06 06:58:45 -06:00
res.push(next.clone());
curr = next
}
res
2019-01-06 06:58:45 -06:00
}
2019-01-06 06:16:21 -06:00
/// Returns a `ModuleScope`: a set of items, visible in this module.
2019-01-15 10:15:01 -06:00
pub fn scope(&self, db: &impl HirDatabase) -> ModuleScope {
2019-03-14 04:54:03 -05:00
db.crate_def_map(self.krate)[self.module_id].scope.clone()
2019-01-06 06:16:21 -06:00
}
2019-03-23 12:41:59 -05:00
pub fn diagnostics(&self, db: &impl HirDatabase, sink: &mut DiagnosticSink) {
2019-03-23 10:35:14 -05:00
db.crate_def_map(self.krate).add_diagnostics(db, self.module_id, sink);
2019-03-24 02:21:36 -05:00
for decl in self.declarations(db) {
match decl {
crate::ModuleDef::Function(f) => f.diagnostics(db, sink),
crate::ModuleDef::Module(f) => f.diagnostics(db, sink),
_ => (),
}
}
for impl_block in self.impl_blocks(db) {
for item in impl_block.items(db) {
match item {
crate::ImplItem::Method(f) => f.diagnostics(db, sink),
_ => (),
}
}
}
2019-01-06 06:58:45 -06:00
}
2019-01-19 14:23:26 -06:00
pub(crate) fn resolver(&self, db: &impl DefDatabase) -> Resolver {
2019-03-14 04:54:03 -05:00
let def_map = db.crate_def_map(self.krate);
Resolver::default().push_module_scope(def_map, self.module_id)
2019-01-19 14:23:26 -06:00
}
pub fn declarations(self, db: &impl DefDatabase) -> Vec<ModuleDef> {
2019-03-14 05:14:54 -05:00
let def_map = db.crate_def_map(self.krate);
def_map[self.module_id]
.scope
.entries()
.filter_map(|(_name, res)| if res.import.is_none() { Some(res.def) } else { None })
.flat_map(|per_ns| {
per_ns.take_types().into_iter().chain(per_ns.take_values().into_iter())
})
.collect()
}
pub fn impl_blocks(self, db: &impl HirDatabase) -> Vec<ImplBlock> {
let module_impl_blocks = db.impls_in_module(self);
module_impl_blocks
.impls
.iter()
.map(|(impl_id, _)| ImplBlock::from_id(self, impl_id))
.collect()
}
2019-01-04 16:37:40 -06:00
}
2019-01-08 06:19:37 -06:00
2019-01-25 11:51:36 -06:00
impl Docs for Module {
fn docs(&self, db: &impl HirDatabase) -> Option<Documentation> {
2019-02-08 05:49:43 -06:00
self.declaration_source(db).and_then(|it| docs_from_ast(&*it.1))
2019-01-25 11:51:36 -06:00
}
}
2019-01-25 05:21:14 -06:00
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
2019-01-08 06:27:00 -06:00
pub struct StructField {
2019-01-25 11:32:34 -06:00
pub(crate) parent: VariantDef,
2019-01-25 05:21:14 -06:00
pub(crate) id: StructFieldId,
2019-01-08 06:27:00 -06:00
}
2019-01-25 11:32:34 -06:00
#[derive(Debug)]
pub enum FieldSource {
Named(TreeArc<ast::NamedFieldDef>),
Pos(TreeArc<ast::PosFieldDef>),
2019-01-25 11:32:34 -06:00
}
2019-01-08 06:27:00 -06:00
impl StructField {
2019-01-25 05:21:14 -06:00
pub fn name(&self, db: &impl HirDatabase) -> Name {
2019-02-08 05:49:43 -06:00
self.parent.variant_data(db).fields().unwrap()[self.id].name.clone()
2019-01-08 06:27:00 -06:00
}
pub fn source(&self, db: &impl DefDatabase) -> (HirFileId, FieldSource) {
2019-01-25 11:32:34 -06:00
self.source_impl(db)
}
2019-01-25 05:21:14 -06:00
pub fn ty(&self, db: &impl HirDatabase) -> Ty {
db.type_for_field(*self)
}
pub fn parent_def(&self, _db: &impl HirDatabase) -> VariantDef {
self.parent
2019-01-08 06:32:27 -06:00
}
}
2019-01-25 13:29:56 -06:00
impl Docs for StructField {
fn docs(&self, db: &impl HirDatabase) -> Option<Documentation> {
match self.source(db).1 {
FieldSource::Named(named) => docs_from_ast(&*named),
FieldSource::Pos(..) => return None,
}
}
}
2019-01-24 08:54:18 -06:00
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
2019-01-08 06:19:37 -06:00
pub struct Struct {
2019-01-24 08:54:18 -06:00
pub(crate) id: StructId,
2019-01-08 06:19:37 -06:00
}
impl Struct {
pub fn source(&self, db: &impl DefDatabase) -> (HirFileId, TreeArc<ast::StructDef>) {
2019-01-24 15:02:18 -06:00
self.id.source(db)
}
2019-01-24 08:54:18 -06:00
pub fn module(&self, db: &impl HirDatabase) -> Module {
2019-01-24 15:02:18 -06:00
self.id.module(db)
2019-01-08 06:19:37 -06:00
}
2019-01-15 09:43:25 -06:00
pub fn name(&self, db: &impl HirDatabase) -> Option<Name> {
2019-01-24 08:54:18 -06:00
db.struct_data(*self).name.clone()
2019-01-08 06:22:57 -06:00
}
2019-01-08 06:23:56 -06:00
2019-01-15 09:43:25 -06:00
pub fn fields(&self, db: &impl HirDatabase) -> Vec<StructField> {
2019-01-24 08:54:18 -06:00
db.struct_data(*self)
2019-01-09 09:46:02 -06:00
.variant_data
.fields()
2019-01-25 05:21:14 -06:00
.into_iter()
.flat_map(|it| it.iter())
2019-02-08 05:49:43 -06:00
.map(|(id, _)| StructField { parent: (*self).into(), id })
2019-01-15 09:43:25 -06:00
.collect()
2019-01-08 06:23:56 -06:00
}
2019-01-25 05:21:14 -06:00
pub fn field(&self, db: &impl HirDatabase, name: &Name) -> Option<StructField> {
db.struct_data(*self)
.variant_data
.fields()
.into_iter()
.flat_map(|it| it.iter())
.find(|(_id, data)| data.name == *name)
2019-02-08 05:49:43 -06:00
.map(|(id, _)| StructField { parent: (*self).into(), id })
2019-01-25 05:21:14 -06:00
}
pub fn ty(&self, db: &impl HirDatabase) -> Ty {
db.type_for_def((*self).into(), Namespace::Types)
}
pub fn constructor_ty(&self, db: &impl HirDatabase) -> Ty {
db.type_for_def((*self).into(), Namespace::Values)
}
2019-01-26 15:52:04 -06:00
2019-03-23 02:53:48 -05:00
// FIXME move to a more general type
2019-01-26 15:52:04 -06:00
/// Builds a resolver for type references inside this struct.
2019-04-13 03:02:23 -05:00
pub(crate) fn resolver(&self, db: &impl HirDatabase) -> Resolver {
2019-01-26 15:52:04 -06:00
// take the outer scope...
let r = self.module(db).resolver(db);
// ...and add generic params, if present
let p = self.generic_params(db);
2019-02-08 05:49:43 -06:00
let r = if !p.params.is_empty() { r.push_generic_params_scope(p) } else { r };
2019-01-26 15:52:04 -06:00
r
}
2019-01-08 06:22:57 -06:00
}
impl Docs for Struct {
fn docs(&self, db: &impl HirDatabase) -> Option<Documentation> {
docs_from_ast(&*self.source(db).1)
}
}
2019-01-24 09:56:38 -06:00
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
2019-01-08 06:22:57 -06:00
pub struct Enum {
2019-01-24 09:56:38 -06:00
pub(crate) id: EnumId,
2019-01-08 06:22:57 -06:00
}
impl Enum {
pub fn source(&self, db: &impl DefDatabase) -> (HirFileId, TreeArc<ast::EnumDef>) {
2019-01-24 15:02:18 -06:00
self.id.source(db)
}
2019-01-24 09:56:38 -06:00
pub fn module(&self, db: &impl HirDatabase) -> Module {
2019-01-24 15:02:18 -06:00
self.id.module(db)
2019-01-08 06:22:57 -06:00
}
2019-01-15 09:43:25 -06:00
pub fn name(&self, db: &impl HirDatabase) -> Option<Name> {
2019-01-24 09:56:38 -06:00
db.enum_data(*self).name.clone()
2019-01-08 06:22:57 -06:00
}
pub fn variants(&self, db: &impl DefDatabase) -> Vec<EnumVariant> {
2019-01-25 03:41:23 -06:00
db.enum_data(*self)
.variants
.iter()
.map(|(id, _)| EnumVariant { parent: *self, id })
.collect()
}
pub fn variant(&self, db: &impl DefDatabase, name: &Name) -> Option<EnumVariant> {
2019-01-25 03:41:23 -06:00
db.enum_data(*self)
.variants
.iter()
.find(|(_id, data)| data.name.as_ref() == Some(name))
.map(|(id, _)| EnumVariant { parent: *self, id })
2019-01-08 06:19:37 -06:00
}
pub fn ty(&self, db: &impl HirDatabase) -> Ty {
db.type_for_def((*self).into(), Namespace::Types)
}
2019-01-26 15:52:04 -06:00
2019-03-23 02:53:48 -05:00
// FIXME: move to a more general type
2019-01-26 15:52:04 -06:00
/// Builds a resolver for type references inside this struct.
2019-04-13 03:02:23 -05:00
pub(crate) fn resolver(&self, db: &impl HirDatabase) -> Resolver {
2019-01-26 15:52:04 -06:00
// take the outer scope...
let r = self.module(db).resolver(db);
// ...and add generic params, if present
let p = self.generic_params(db);
2019-02-08 05:49:43 -06:00
let r = if !p.params.is_empty() { r.push_generic_params_scope(p) } else { r };
2019-01-26 15:52:04 -06:00
r
}
2019-01-08 06:19:37 -06:00
}
2019-01-08 11:11:13 -06:00
impl Docs for Enum {
fn docs(&self, db: &impl HirDatabase) -> Option<Documentation> {
docs_from_ast(&*self.source(db).1)
}
}
2019-01-24 14:32:41 -06:00
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub struct EnumVariant {
2019-01-25 02:35:38 -06:00
pub(crate) parent: Enum,
2019-01-25 03:41:23 -06:00
pub(crate) id: EnumVariantId,
}
impl EnumVariant {
pub fn source(&self, db: &impl DefDatabase) -> (HirFileId, TreeArc<ast::EnumVariant>) {
2019-01-25 02:35:38 -06:00
self.source_impl(db)
2019-01-24 15:02:18 -06:00
}
2019-01-24 14:32:41 -06:00
pub fn module(&self, db: &impl HirDatabase) -> Module {
2019-01-25 02:35:38 -06:00
self.parent.module(db)
}
pub fn parent_enum(&self, _db: &impl DefDatabase) -> Enum {
2019-01-25 02:35:38 -06:00
self.parent
}
pub fn name(&self, db: &impl DefDatabase) -> Option<Name> {
2019-01-25 03:41:23 -06:00
db.enum_data(self.parent).variants[self.id].name.clone()
}
pub fn fields(&self, db: &impl HirDatabase) -> Vec<StructField> {
self.variant_data(db)
.fields()
2019-01-25 05:21:14 -06:00
.into_iter()
.flat_map(|it| it.iter())
2019-02-08 05:49:43 -06:00
.map(|(id, _)| StructField { parent: (*self).into(), id })
.collect()
}
2019-01-25 05:21:14 -06:00
pub fn field(&self, db: &impl HirDatabase, name: &Name) -> Option<StructField> {
self.variant_data(db)
.fields()
.into_iter()
.flat_map(|it| it.iter())
.find(|(_id, data)| data.name == *name)
2019-02-08 05:49:43 -06:00
.map(|(id, _)| StructField { parent: (*self).into(), id })
2019-01-25 05:21:14 -06:00
}
}
impl Docs for EnumVariant {
fn docs(&self, db: &impl HirDatabase) -> Option<Documentation> {
docs_from_ast(&*self.source(db).1)
}
}
2019-03-30 05:50:00 -05:00
/// The defs which have a body.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum DefWithBody {
Function(Function),
2019-03-30 05:50:00 -05:00
Const(Const),
Static(Static),
}
impl_froms!(DefWithBody: Function, Const, Static);
2019-03-30 05:50:00 -05:00
impl DefWithBody {
2019-03-30 05:50:00 -05:00
pub fn infer(&self, db: &impl HirDatabase) -> Arc<InferenceResult> {
db.infer(*self)
}
pub fn body(&self, db: &impl HirDatabase) -> Arc<Body> {
db.body_hir(*self)
}
pub fn body_source_map(&self, db: &impl HirDatabase) -> Arc<BodySourceMap> {
2019-04-11 08:39:56 -05:00
db.body_with_source_map(*self).1
}
2019-03-30 05:50:00 -05:00
/// Builds a resolver for code inside this item.
2019-04-13 03:02:23 -05:00
pub(crate) fn resolver(&self, db: &impl HirDatabase) -> Resolver {
match *self {
DefWithBody::Const(ref c) => c.resolver(db),
DefWithBody::Function(ref f) => f.resolver(db),
DefWithBody::Static(ref s) => s.resolver(db),
}
2019-03-30 05:50:00 -05:00
}
}
2019-01-24 06:28:50 -06:00
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
2019-01-08 11:11:13 -06:00
pub struct Function {
2019-01-24 06:28:50 -06:00
pub(crate) id: FunctionId,
2019-01-08 11:11:13 -06:00
}
/// The declared signature of a function.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct FnSignature {
pub(crate) name: Name,
2019-01-26 05:06:41 -06:00
pub(crate) params: Vec<TypeRef>,
2019-01-08 11:11:13 -06:00
pub(crate) ret_type: TypeRef,
2019-01-12 14:58:16 -06:00
/// True if the first param is `self`. This is relevant to decide whether this
/// can be called as a method.
2019-01-12 14:58:16 -06:00
pub(crate) has_self_param: bool,
2019-01-08 11:11:13 -06:00
}
impl FnSignature {
pub fn name(&self) -> &Name {
&self.name
}
2019-01-26 05:06:41 -06:00
pub fn params(&self) -> &[TypeRef] {
&self.params
2019-01-08 11:11:13 -06:00
}
pub fn ret_type(&self) -> &TypeRef {
&self.ret_type
}
/// True if the first arg is `self`. This is relevant to decide whether this
/// can be called as a method.
2019-01-12 14:58:16 -06:00
pub fn has_self_param(&self) -> bool {
self.has_self_param
}
2019-01-08 11:11:13 -06:00
}
impl Function {
pub fn source(&self, db: &impl DefDatabase) -> (HirFileId, TreeArc<ast::FnDef>) {
2019-01-24 15:02:18 -06:00
self.id.source(db)
}
pub fn module(&self, db: &impl DefDatabase) -> Module {
2019-01-24 15:02:18 -06:00
self.id.module(db)
2019-01-08 11:11:13 -06:00
}
pub fn name(&self, db: &impl HirDatabase) -> Name {
self.signature(db).name.clone()
}
2019-04-10 03:15:55 -05:00
pub(crate) fn body_source_map(&self, db: &impl HirDatabase) -> Arc<BodySourceMap> {
db.body_with_source_map((*self).into()).1
2019-01-08 11:11:13 -06:00
}
2019-01-19 14:23:26 -06:00
pub fn body(&self, db: &impl HirDatabase) -> Arc<Body> {
db.body_hir((*self).into())
2019-01-19 14:23:26 -06:00
}
pub fn ty(&self, db: &impl HirDatabase) -> Ty {
db.type_for_def((*self).into(), Namespace::Values)
}
2019-01-08 11:11:13 -06:00
pub fn signature(&self, db: &impl HirDatabase) -> Arc<FnSignature> {
2019-01-24 06:28:50 -06:00
db.fn_signature(*self)
2019-01-08 11:11:13 -06:00
}
2019-01-15 11:54:18 -06:00
pub fn infer(&self, db: &impl HirDatabase) -> Arc<InferenceResult> {
db.infer((*self).into())
2019-01-08 11:11:13 -06:00
}
2019-02-16 15:06:23 -06:00
/// The containing impl block, if this is a method.
pub fn impl_block(&self, db: &impl DefDatabase) -> Option<ImplBlock> {
2019-02-16 15:06:23 -06:00
let module_impls = db.impls_in_module(self.module(db));
ImplBlock::containing(module_impls, (*self).into())
}
/// The containing trait, if this is a trait method definition.
pub fn parent_trait(&self, db: &impl DefDatabase) -> Option<Trait> {
db.trait_items_index(self.module(db)).get_parent_trait((*self).into())
}
pub fn container(&self, db: &impl DefDatabase) -> Option<Container> {
if let Some(impl_block) = self.impl_block(db) {
Some(impl_block.into())
} else if let Some(trait_) = self.parent_trait(db) {
Some(trait_.into())
} else {
None
}
}
2019-03-23 02:53:48 -05:00
// FIXME: move to a more general type for 'body-having' items
2019-01-23 16:08:41 -06:00
/// Builds a resolver for code inside this item.
2019-04-13 03:02:23 -05:00
pub(crate) fn resolver(&self, db: &impl HirDatabase) -> Resolver {
2019-01-23 16:08:41 -06:00
// take the outer scope...
let r = self.container(db).map_or_else(|| self.module(db).resolver(db), |c| c.resolver(db));
2019-01-23 16:08:41 -06:00
// ...and add generic params, if present
let p = self.generic_params(db);
2019-02-08 05:49:43 -06:00
let r = if !p.params.is_empty() { r.push_generic_params_scope(p) } else { r };
2019-01-23 16:08:41 -06:00
r
}
2019-03-21 14:13:11 -05:00
2019-03-23 12:41:59 -05:00
pub fn diagnostics(&self, db: &impl HirDatabase, sink: &mut DiagnosticSink) {
2019-03-23 10:35:14 -05:00
self.infer(db).add_diagnostics(db, *self, sink);
2019-03-21 14:13:11 -05:00
}
}
2019-01-22 07:55:05 -06:00
impl Docs for Function {
fn docs(&self, db: &impl HirDatabase) -> Option<Documentation> {
docs_from_ast(&*self.source(db).1)
2019-01-22 07:55:05 -06:00
}
2019-01-08 11:11:13 -06:00
}
2019-01-11 11:28:10 -06:00
2019-01-24 15:50:08 -06:00
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
2019-01-11 11:28:10 -06:00
pub struct Const {
2019-01-24 15:50:08 -06:00
pub(crate) id: ConstId,
2019-01-11 11:28:10 -06:00
}
2019-01-11 12:02:12 -06:00
impl Const {
pub fn source(&self, db: &impl DefDatabase) -> (HirFileId, TreeArc<ast::ConstDef>) {
2019-01-24 15:50:08 -06:00
self.id.source(db)
2019-01-11 12:02:12 -06:00
}
2019-02-16 15:06:23 -06:00
pub fn module(&self, db: &impl DefDatabase) -> Module {
2019-02-16 15:06:23 -06:00
self.id.module(db)
}
2019-02-25 01:27:47 -06:00
pub fn signature(&self, db: &impl HirDatabase) -> Arc<ConstSignature> {
db.const_signature(*self)
}
pub fn infer(&self, db: &impl HirDatabase) -> Arc<InferenceResult> {
db.infer((*self).into())
}
2019-02-16 15:06:23 -06:00
/// The containing impl block, if this is a method.
pub fn impl_block(&self, db: &impl DefDatabase) -> Option<ImplBlock> {
2019-02-16 15:06:23 -06:00
let module_impls = db.impls_in_module(self.module(db));
ImplBlock::containing(module_impls, (*self).into())
}
2019-02-25 01:27:47 -06:00
2019-03-23 02:53:48 -05:00
// FIXME: move to a more general type for 'body-having' items
2019-02-25 01:27:47 -06:00
/// Builds a resolver for code inside this item.
2019-04-13 03:02:23 -05:00
pub(crate) fn resolver(&self, db: &impl HirDatabase) -> Resolver {
2019-02-25 01:27:47 -06:00
// take the outer scope...
let r = self
.impl_block(db)
.map(|ib| ib.resolver(db))
.unwrap_or_else(|| self.module(db).resolver(db));
r
}
2019-01-11 12:02:12 -06:00
}
impl Docs for Const {
fn docs(&self, db: &impl HirDatabase) -> Option<Documentation> {
docs_from_ast(&*self.source(db).1)
}
}
2019-02-25 01:27:47 -06:00
/// The declared signature of a const.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct ConstSignature {
pub(crate) name: Name,
pub(crate) type_ref: TypeRef,
}
impl ConstSignature {
pub fn name(&self) -> &Name {
&self.name
}
pub fn type_ref(&self) -> &TypeRef {
&self.type_ref
}
}
2019-01-24 15:50:08 -06:00
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
2019-01-11 11:28:10 -06:00
pub struct Static {
2019-01-24 15:50:08 -06:00
pub(crate) id: StaticId,
2019-01-11 11:28:10 -06:00
}
2019-01-11 12:02:12 -06:00
impl Static {
pub fn source(&self, db: &impl DefDatabase) -> (HirFileId, TreeArc<ast::StaticDef>) {
2019-01-24 15:50:08 -06:00
self.id.source(db)
2019-01-11 12:02:12 -06:00
}
2019-02-16 15:06:23 -06:00
pub fn module(&self, db: &impl DefDatabase) -> Module {
2019-02-16 15:06:23 -06:00
self.id.module(db)
}
2019-02-25 02:21:01 -06:00
pub fn signature(&self, db: &impl HirDatabase) -> Arc<ConstSignature> {
db.static_signature(*self)
}
/// Builds a resolver for code inside this item.
2019-04-13 03:02:23 -05:00
pub(crate) fn resolver(&self, db: &impl HirDatabase) -> Resolver {
2019-02-25 02:21:01 -06:00
// take the outer scope...
self.module(db).resolver(db)
}
pub fn infer(&self, db: &impl HirDatabase) -> Arc<InferenceResult> {
db.infer((*self).into())
}
2019-01-11 12:02:12 -06:00
}
impl Docs for Static {
fn docs(&self, db: &impl HirDatabase) -> Option<Documentation> {
docs_from_ast(&*self.source(db).1)
}
}
2019-01-24 16:31:32 -06:00
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
2019-01-11 11:28:10 -06:00
pub struct Trait {
2019-01-24 16:31:32 -06:00
pub(crate) id: TraitId,
2019-01-11 11:28:10 -06:00
}
2019-01-11 12:02:12 -06:00
impl Trait {
pub fn source(&self, db: &impl DefDatabase) -> (HirFileId, TreeArc<ast::TraitDef>) {
2019-01-24 16:31:32 -06:00
self.id.source(db)
2019-01-11 12:02:12 -06:00
}
pub fn module(&self, db: &impl DefDatabase) -> Module {
2019-02-16 15:06:23 -06:00
self.id.module(db)
}
2019-03-24 11:36:15 -05:00
pub fn name(self, db: &impl DefDatabase) -> Option<Name> {
self.trait_data(db).name().clone()
}
pub fn items(self, db: &impl DefDatabase) -> Vec<TraitItem> {
self.trait_data(db).items().to_vec()
}
pub(crate) fn trait_data(self, db: &impl DefDatabase) -> Arc<TraitData> {
db.trait_data(self)
}
pub fn trait_ref(self, db: &impl HirDatabase) -> TraitRef {
TraitRef::for_trait(db, self)
}
pub(crate) fn resolver(&self, db: &impl DefDatabase) -> Resolver {
let r = self.module(db).resolver(db);
// add generic params, if present
let p = self.generic_params(db);
let r = if !p.params.is_empty() { r.push_generic_params_scope(p) } else { r };
r
}
2019-01-11 12:02:12 -06:00
}
impl Docs for Trait {
fn docs(&self, db: &impl HirDatabase) -> Option<Documentation> {
docs_from_ast(&*self.source(db).1)
}
}
2019-01-24 16:31:32 -06:00
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
2019-02-24 14:36:49 -06:00
pub struct TypeAlias {
2019-03-26 11:15:39 -05:00
pub(crate) id: TypeAliasId,
2019-01-11 11:28:10 -06:00
}
2019-01-11 12:02:12 -06:00
2019-02-24 14:36:49 -06:00
impl TypeAlias {
pub fn source(&self, db: &impl DefDatabase) -> (HirFileId, TreeArc<ast::TypeAliasDef>) {
2019-01-24 16:31:32 -06:00
self.id.source(db)
}
2019-02-16 15:06:23 -06:00
pub fn module(&self, db: &impl DefDatabase) -> Module {
2019-02-16 15:06:23 -06:00
self.id.module(db)
}
/// The containing impl block, if this is a method.
pub fn impl_block(&self, db: &impl DefDatabase) -> Option<ImplBlock> {
2019-02-16 15:06:23 -06:00
let module_impls = db.impls_in_module(self.module(db));
ImplBlock::containing(module_impls, (*self).into())
}
2019-02-24 10:25:41 -06:00
/// The containing trait, if this is a trait method definition.
pub fn parent_trait(&self, db: &impl DefDatabase) -> Option<Trait> {
db.trait_items_index(self.module(db)).get_parent_trait((*self).into())
}
pub fn container(&self, db: &impl DefDatabase) -> Option<Container> {
if let Some(impl_block) = self.impl_block(db) {
Some(impl_block.into())
} else if let Some(trait_) = self.parent_trait(db) {
Some(trait_.into())
} else {
None
}
}
pub fn type_ref(self, db: &impl DefDatabase) -> Arc<TypeRef> {
2019-02-24 10:25:41 -06:00
db.type_alias_ref(self)
}
/// Builds a resolver for the type references in this type alias.
2019-04-13 03:02:23 -05:00
pub(crate) fn resolver(&self, db: &impl HirDatabase) -> Resolver {
2019-02-24 10:25:41 -06:00
// take the outer scope...
let r = self
.impl_block(db)
.map(|ib| ib.resolver(db))
.unwrap_or_else(|| self.module(db).resolver(db));
// ...and add generic params, if present
let p = self.generic_params(db);
let r = if !p.params.is_empty() { r.push_generic_params_scope(p) } else { r };
r
}
2019-01-11 12:02:12 -06:00
}
2019-02-24 14:36:49 -06:00
impl Docs for TypeAlias {
fn docs(&self, db: &impl HirDatabase) -> Option<Documentation> {
docs_from_ast(&*self.source(db).1)
}
}
pub enum Container {
Trait(Trait),
ImplBlock(ImplBlock),
}
impl_froms!(Container: Trait, ImplBlock);
impl Container {
pub(crate) fn resolver(&self, db: &impl DefDatabase) -> Resolver {
match self {
Container::Trait(trait_) => trait_.resolver(db),
Container::ImplBlock(impl_block) => impl_block.resolver(db),
}
}
}