mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add params_display and ty_display

Browse Source
This commit is contained in:
Josh Mcguigan 2021-03-12 13:46:40 -08:00
parent 53bb46fa85
commit d5f0f58e63
1 changed files with 17 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
crates/ide_completion/src/render/function.rs
View File

@ -60,7 +60,19 @@ impl<'a> FunctionRender<'a> {
} }
fn detail(&self) -> String { fn detail(&self) -> String {
let params = if let Some(self_param) = self.func.self_param(self.ctx.db()) { let ret_ty = self.func.ret_type(self.ctx.db());
let ret = if ret_ty.is_unit() {
// Omit the return type if it is the unit type
String::new()
} else {
format!(" {}", self.ty_display())
};
format!("fn({}){}", self.params_display(), ret)
}
fn params_display(&self) -> String {
if let Some(self_param) = self.func.self_param(self.ctx.db()) {
let params = self let params = self
.func .func
.assoc_fn_params(self.ctx.db()) .assoc_fn_params(self.ctx.db())
@ -77,17 +89,13 @@ impl<'a> FunctionRender<'a> {
.map(|p| p.ty().display(self.ctx.db()).to_string()) .map(|p| p.ty().display(self.ctx.db()).to_string())
.join(", "); .join(", ");
params params
}; }
}
fn ty_display(&self) -> String {
let ret_ty = self.func.ret_type(self.ctx.db()); let ret_ty = self.func.ret_type(self.ctx.db());
let ret = if ret_ty.is_unit() {
// Omit the `-> ()` for unit return types
String::new()
} else {
format!(" -> {}", ret_ty.display(self.ctx.db()))
};
format!("fn({}){}", params, ret) format!("-> {}", ret_ty.display(self.ctx.db()))
} }
fn add_arg(&self, arg: &str, ty: &Type) -> String { fn add_arg(&self, arg: &str, ty: &Type) -> String {