From a6dc7cf36de40f6b2a5b1dfee86430848f94ce42 Mon Sep 17 00:00:00 2001
From: Arnaud <arnaud.golfouse@free.fr>
Date: Mon, 4 Jan 2021 18:58:42 +0100
Subject: Make it possible to retrieve `hir::Function`'s return type

This is done by adding a `ret_type` method to `hir::Function`.
I followed `assoc_fn_params` convention by creating a new `RetType` type,
that contains the actual return type accessible via a `ty` method.
---
 crates/hir/src/code_model.rs | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

(limited to 'crates/hir/src/code_model.rs')

diff --git a/crates/hir/src/code_model.rs b/crates/hir/src/code_model.rs
index 071e553a8..e783e0aba 100644
--- a/crates/hir/src/code_model.rs
+++ b/crates/hir/src/code_model.rs
@@ -743,6 +743,18 @@ impl Function {
         db.function_data(self.id).name.clone()
     }
 
+    pub fn ret_type(self, db: &dyn HirDatabase) -> RetType {
+        let resolver = self.id.resolver(db.upcast());
+        let ret_type = &db.function_data(self.id).ret_type;
+        let ctx = hir_ty::TyLoweringContext::new(db, &resolver);
+        let environment = TraitEnvironment::lower(db, &resolver);
+        let ty = Type {
+            krate: self.id.lookup(db.upcast()).container.module(db.upcast()).krate,
+            ty: InEnvironment { value: Ty::from_hir_ext(&ctx, ret_type).0, environment },
+        };
+        RetType { ty }
+    }
+
     pub fn self_param(self, db: &dyn HirDatabase) -> Option<SelfParam> {
         if !db.function_data(self.id).has_self_param {
             return None;
@@ -826,6 +838,17 @@ impl From<Mutability> for Access {
     }
 }
 
+#[derive(Debug)]
+pub struct RetType {
+    ty: Type,
+}
+
+impl RetType {
+    pub fn ty(&self) -> &Type {
+        &self.ty
+    }
+}
+
 #[derive(Debug)]
 pub struct Param {
     ty: Type,
-- 
cgit v1.2.3