feat: infer self & Self HIR types

crates/mun_codegen/src/ir/body.rs

@@ -574,6 +574,7 @@
             .expect("unknown path")
             .0
         {
+            ValueNs::ImplSelf(_) => unimplemented!("no support for self types"),
             ValueNs::LocalBinding(pat) => {
                 if let Some(param) = self.pat_to_param.get(&pat) {
                     *param
@@ -618,6 +619,7 @@
             .expect("unknown path")
             .0
         {
+            ValueNs::ImplSelf(_) => unimplemented!("no support for self types"),
             ValueNs::LocalBinding(pat) => *self
                 .pat_to_local
                 .get(&pat)

crates/mun_hir/Cargo.toml

@@ -25,6 +25,7 @@ once_cell = { version = "1.19.0", default-features = false }
 rustc-hash = { version = "1.1", default-features = false }
 salsa = { version = "0.16.1", default-features = false }
 smallvec = { version = "1.11.2", features = ["union"], default-features = false }
+bitflags = { version = "2.5.0", default-features = false }
 
 [dev-dependencies]
 mun_test = { path = "../mun_test" }

crates/mun_hir/src/code_model.rs

@@ -12,7 +12,7 @@ pub use self::{
     function::{Function, FunctionData},
     module::{Module, ModuleDef},
     package::Package,
-    r#impl::{AssocItem, Impl, ImplData},
+    r#impl::{AssocItem, ImplData},
     r#struct::{Field, Struct, StructData, StructKind, StructMemoryKind},
     src::HasSource,
     type_alias::{TypeAlias, TypeAliasData},

crates/mun_hir/src/code_model/function.rs

@@ -7,6 +7,7 @@ use crate::{
     expr::{validator::ExprValidator, BodySourceMap},
     has_module::HasModule,
     ids::{FunctionId, Lookup},
+    item_tree::FunctionFlags,
     name_resolution::Namespace,
     resolve::HasResolver,
     type_ref::{LocalTypeRefId, TypeRefMap, TypeRefSourceMap},
@@ -34,7 +35,7 @@ pub struct FunctionData {
     ret_type: LocalTypeRefId,
     type_ref_map: TypeRefMap,
     type_ref_source_map: TypeRefSourceMap,
-    is_extern: bool,
+    flags: FunctionFlags,
 }
 
 impl FunctionData {
@@ -68,7 +69,7 @@ impl FunctionData {
             ret_type,
             type_ref_map,
             type_ref_source_map,
-            is_extern: func.is_extern,
+            flags: func.flags,
             visibility: item_tree[func.visibility].clone(),
         })
     }
@@ -99,7 +100,7 @@ impl FunctionData {
 
     /// Returns true if this function is an extern function.
     pub fn is_extern(&self) -> bool {
-        self.is_extern
+        self.flags.is_extern()
     }
 }
 
@@ -168,7 +169,7 @@ impl Function {
     }
 
     pub fn is_extern(self, db: &dyn HirDatabase) -> bool {
-        db.fn_data(self.id).is_extern
+        db.fn_data(self.id).flags.is_extern()
     }
 
     pub(crate) fn body_source_map(self, db: &dyn HirDatabase) -> Arc<BodySourceMap> {

crates/mun_hir/src/code_model/module.rs

@@ -1,4 +1,4 @@
-use super::{Function, Package, Struct, TypeAlias};
+use super::{r#impl::Impl, AssocItem, Function, Package, Struct, TypeAlias};
 use crate::{
     ids::{ItemDefinitionId, ModuleId},
     primitive_type::PrimitiveType,
@@ -81,7 +81,7 @@ impl Module {
         let package_defs = db.package_defs(self.id.package);
         package_defs.add_diagnostics(db.upcast(), self.id.local_id, sink);
 
-        // Add diagnostics from impls
+        // Add diagnostics from inherent impls
         let inherent_impls = db.inherent_impls_in_package(self.id.package);
         inherent_impls.add_module_diagnostics(db, self.id.local_id, sink);
 
@@ -102,6 +102,15 @@ impl Module {
                 _ => (),
             }
         }
+
+        // Add diagnostics from impls
+        for item in self.impls(db) {
+            for associated_item in item.items(db) {
+                let AssocItem::Function(fun) = associated_item;
+
+                fun.diagnostics(db, sink);
+            }
+        }
     }
 
     /// Returns all the child modules of this module
@@ -141,6 +150,14 @@ impl Module {
         )
         .collect()
     }
+
+    pub fn impls(self, db: &dyn HirDatabase) -> Vec<Impl> {
+        let package_defs = db.package_defs(self.id.package);
+        package_defs.modules[self.id.local_id]
+            .impls()
+            .map(Impl::from)
+            .collect()
+    }
 }
 
 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]

crates/mun_hir/src/db.rs

@@ -155,6 +155,9 @@ pub trait HirDatabase: DefDatabase + Upcast<dyn DefDatabase> {
     #[salsa::invoke(crate::ty::type_for_def)]
     fn type_for_def(&self, def: TypableDef, ns: Namespace) -> Ty;
 
+    #[salsa::invoke(crate::ty::type_for_impl_self)]
+    fn type_for_impl_self(&self, def: ImplId) -> Ty;
+
     #[salsa::invoke(InherentImpls::inherent_impls_in_package_query)]
     fn inherent_impls_in_package(&self, package: PackageId) -> Arc<InherentImpls>;
 }

crates/mun_hir/src/expr.rs

@@ -16,7 +16,7 @@
     diagnostics::DiagnosticSink,
     ids::{DefWithBodyId, Lookup},
     in_file::InFile,
-    name::AsName,
+    name::{name, AsName},
     primitive_type::{PrimitiveFloat, PrimitiveInt},
     type_ref::{LocalTypeRefId, TypeRef, TypeRefMap, TypeRefMapBuilder, TypeRefSourceMap},
     DefDatabase, FileId, HirDatabase, Name, Path,
@@ -46,6 +46,7 @@
     ///
     /// If this `Body` is for the body of a constant, this will just be empty.
     params: Vec<(PatId, LocalTypeRefId)>,
+    self_param: Option<(PatId, LocalTypeRefId)>,
     /// The `ExprId` of the actual body expression.
     body_expr: ExprId,
     ret_type: LocalTypeRefId,
@@ -82,6 +83,10 @@
         &self.params
     }
 
+    pub fn self_param(&self) -> Option<&(PatId, LocalTypeRefId)> {
+        self.self_param.as_ref()
+    }
+
     pub fn body_expr(&self) -> ExprId {
         self.body_expr
     }
@@ -147,7 +152,7 @@
 type ExprPtr = Either<AstPtr<ast::Expr>, AstPtr<ast::RecordField>>;
 type ExprSource = InFile<ExprPtr>;
 
-type PatPtr = AstPtr<ast::Pat>; //Either<AstPtr<ast::Pat>, AstPtr<ast::SelfParam>>;
+type PatPtr = Either<AstPtr<ast::Pat>, AstPtr<ast::SelfParam>>;
 type PatSource = InFile<PatPtr>;
 
 type RecordPtr = AstPtr<ast::RecordField>;
@@ -189,7 +194,7 @@
     }
 
     pub(crate) fn node_pat(&self, node: &ast::Pat) -> Option<PatId> {
-        self.pat_map.get(&AstPtr::new(node)).cloned()
+        self.pat_map.get(&Either::Left(AstPtr::new(node))).cloned()
     }
 
     pub fn type_refs(&self) -> &TypeRefSourceMap {
@@ -466,6 +471,7 @@
     pats: Arena<Pat>,
     source_map: BodySourceMap,
     params: Vec<(PatId, LocalTypeRefId)>,
+    self_param: Option<(PatId, LocalTypeRefId)>,
     body_expr: Option<ExprId>,
     ret_type: Option<LocalTypeRefId>,
     type_ref_builder: TypeRefMapBuilder,
@@ -482,6 +488,7 @@
             pats: Arena::default(),
             source_map: BodySourceMap::default(),
             params: Vec::new(),
+            self_param: None,
             body_expr: None,
             ret_type: None,
             type_ref_builder: TypeRefMap::builder(),
@@ -525,6 +532,16 @@
 
     fn collect_fn_body(&mut self, node: &ast::FunctionDef) {
         if let Some(param_list) = node.param_list() {
+            if let Some(self_param) = param_list.self_param() {
+                let self_pat = self.alloc_pat(
+                    Pat::Bind { name: name![self] },
+                    Either::Right(AstPtr::new(&self_param)),
+                );
+
+                let self_type = self.type_ref_builder.alloc_self();
+                self.self_param = Some((self_pat, self_type));
+            }
+
             for param in param_list.params() {
                 let pat = if let Some(pat) = param.pat() {
                     pat
@@ -895,7 +912,7 @@
             ast::PatKind::PlaceholderPat(_) => Pat::Wild,
         };
         let ptr = AstPtr::new(&pat);
-        self.alloc_pat(pattern, ptr)
+        self.alloc_pat(pattern, Either::Left(ptr))
     }
 
     fn collect_return(&mut self, expr: ast::ReturnExpr) -> ExprId {
@@ -930,6 +947,7 @@
             exprs: self.exprs,
             pats: self.pats,
             params: self.params,
+            self_param: self.self_param,
             body_expr: self.body_expr.expect("A body should have been collected"),
             type_refs,
             ret_type: self

crates/mun_hir/src/expr/scope.rs

@@ -52,6 +52,9 @@ impl ExprScopes {
             scope_by_expr: FxHashMap::default(),
         };
         let root = scopes.root_scope();
+        if let Some(self_param) = body.self_param {
+            scopes.add_bindings(body, root, self_param.0);
+        }
         scopes.add_params_bindings(body, root, body.params().iter().map(|p| &p.0));
         compute_expr_scopes(body.body_expr(), body, &mut scopes, root);
         scopes

crates/mun_hir/src/expr/validator/uninitialized_access.rs

@@ -21,6 +21,10 @@ impl<'d> ExprValidator<'d> {
 
         // Add all parameter patterns to the set of initialized patterns (they must have
         // been initialized)
+        if let Some((pat, _)) = self.body.self_param {
+            initialized_patterns.insert(pat);
+        }
+
         for (pat, _) in self.body.params.iter() {
             initialized_patterns.insert(*pat);
         }

crates/mun_hir/src/item_tree.rs

@@ -301,11 +301,38 @@
 pub struct Function {
     pub name: Name,
     pub visibility: RawVisibilityId,
-    pub is_extern: bool,
     pub types: TypeRefMap,
     pub params: IdRange<Param>,
     pub ret_type: LocalTypeRefId,
     pub ast_id: FileAstId<ast::FunctionDef>,
+    pub(crate) flags: FunctionFlags,
+}
+
+bitflags::bitflags! {
+    #[doc = "Flags that are used to store additional information about a function"]
+    #[derive(Debug, Clone, Copy, Eq, PartialEq, Default)]
+    pub(crate) struct FunctionFlags: u8 {
+        const HAS_SELF_PARAM = 1 << 0;
+        const HAS_BODY = 1 << 1;
+        const IS_EXTERN = 1 << 2;
+    }
+}
+
+impl FunctionFlags {
+    /// Whether the function has a self parameter.
+    pub fn has_self_param(self) -> bool {
+        self.contains(Self::HAS_SELF_PARAM)
+    }
+
+    /// Whether the function has a body.
+    pub fn has_body(self) -> bool {
+        self.contains(Self::HAS_BODY)
+    }
+
+    /// Whether the function is extern.
+    pub fn is_extern(self) -> bool {
+        self.contains(Self::IS_EXTERN)
+    }
 }
 
 #[derive(Debug, Clone, Eq, PartialEq)]
@@ -317,6 +344,7 @@
 #[derive(Debug, Clone, Eq, PartialEq)]
 pub enum ParamAstId {
     Param(FileAstId<ast::Param>),
+    SelfParam(FileAstId<ast::SelfParam>),
 }
 
 #[derive(Debug, Clone, Eq, PartialEq)]

crates/mun_hir/src/item_tree/lower.rs

@@ -9,9 +9,9 @@
 use smallvec::SmallVec;
 
 use super::{
-    diagnostics, AssociatedItem, Field, Fields, Function, IdRange, Impl, ItemTree, ItemTreeData,
-    ItemTreeNode, ItemVisibilities, LocalItemTreeId, ModItem, Param, ParamAstId, RawVisibilityId,
-    Struct, TypeAlias,
+    diagnostics, AssociatedItem, Field, Fields, Function, FunctionFlags, IdRange, Impl, ItemTree,
+    ItemTreeData, ItemTreeNode, ItemVisibilities, LocalItemTreeId, ModItem, Param, ParamAstId,
+    RawVisibilityId, Struct, TypeAlias,
 };
 use crate::{
     item_tree::Import,
@@ -158,7 +158,21 @@
 
         // Lower all the params
         let start_param_idx = self.next_param_idx();
+        let mut has_self_param = false;
         if let Some(param_list) = func.param_list() {
+            if let Some(self_param) = param_list.self_param() {
+                let ast_id = self.source_ast_id_map.ast_id(&self_param);
+                let type_ref = match self_param.ascribed_type().as_ref() {
+                    Some(type_ref) => types.alloc_from_node(type_ref),
+                    None => types.alloc_self(),
+                };
+                self.data.params.alloc(Param {
+                    type_ref,
+                    ast_id: ParamAstId::SelfParam(ast_id),
+                });
+                has_self_param = true;
+            }
+
             for param in param_list.params() {
                 let ast_id = self.source_ast_id_map.ast_id(&param);
                 let type_ref = types.alloc_from_node_opt(param.ascribed_type().as_ref());
@@ -177,18 +191,28 @@
             Some(ty) => types.alloc_from_node(&ty),
         };
 
-        let is_extern = func.is_extern();
-
         let (types, _types_source_map) = types.finish();
         let ast_id = self.source_ast_id_map.ast_id(func);
+
+        let mut flags = FunctionFlags::default();
+        if func.is_extern() {
+            flags |= FunctionFlags::IS_EXTERN;
+        }
+        if func.body().is_some() {
+            flags |= FunctionFlags::HAS_BODY;
+        }
+        if has_self_param {
+            flags |= FunctionFlags::HAS_SELF_PARAM;
+        }
+
         let res = Function {
             name,
             visibility,
-            is_extern,
             types,
             params,
             ret_type,
             ast_id,
+            flags,
         };
 
         Some(self.data.functions.alloc(res).into())

crates/mun_hir/src/item_tree/pretty.rs

@@ -167,21 +167,29 @@
         let Function {
             name,
             visibility,
-            is_extern,
             types,
             params,
             ret_type,
             ast_id: _,
+            flags,
         } = &self.tree[it];
         self.print_visibility(*visibility)?;
-        if *is_extern {
+        if flags.is_extern() {
             write!(self, "extern ")?;
         }
         write!(self, "fn {name}")?;
         write!(self, "(")?;
         if !params.is_empty() {
             self.indented(|this| {
-                for param in params.clone() {
+                let mut params = params.clone();
+                if flags.has_self_param() {
+                    // Skip self parameter
+                    params.next();
+
+                    write!(this, "self")?;
+                }
+
+                for param in params {
                     let Param {
                         type_ref,
                         ast_id: _,