Add signature help for record struct patterns

This commit is contained in:
Lukas Wirth 2023-03-14 21:55:03 +01:00
parent 510e4b4029
commit 03a6ab0b62
5 changed files with 88 additions and 14 deletions

View file

@ -411,7 +411,7 @@ impl<'db, DB: HirDatabase> Semantics<'db, DB> {
self.imp.resolve_record_field(field) self.imp.resolve_record_field(field)
} }
pub fn resolve_record_pat_field(&self, field: &ast::RecordPatField) -> Option<Field> { pub fn resolve_record_pat_field(&self, field: &ast::RecordPatField) -> Option<(Field, Type)> {
self.imp.resolve_record_pat_field(field) self.imp.resolve_record_pat_field(field)
} }
@ -1201,7 +1201,7 @@ impl<'db> SemanticsImpl<'db> {
self.analyze(field.syntax())?.resolve_record_field(self.db, field) self.analyze(field.syntax())?.resolve_record_field(self.db, field)
} }
fn resolve_record_pat_field(&self, field: &ast::RecordPatField) -> Option<Field> { fn resolve_record_pat_field(&self, field: &ast::RecordPatField) -> Option<(Field, Type)> {
self.analyze(field.syntax())?.resolve_record_pat_field(self.db, field) self.analyze(field.syntax())?.resolve_record_pat_field(self.db, field)
} }

View file

@ -441,14 +441,17 @@ impl SourceAnalyzer {
&self, &self,
db: &dyn HirDatabase, db: &dyn HirDatabase,
field: &ast::RecordPatField, field: &ast::RecordPatField,
) -> Option<Field> { ) -> Option<(Field, Type)> {
let field_name = field.field_name()?.as_name(); let field_name = field.field_name()?.as_name();
let record_pat = ast::RecordPat::cast(field.syntax().parent().and_then(|p| p.parent())?)?; let record_pat = ast::RecordPat::cast(field.syntax().parent().and_then(|p| p.parent())?)?;
let pat_id = self.pat_id(&record_pat.into())?; let pat_id = self.pat_id(&record_pat.into())?;
let variant = self.infer.as_ref()?.variant_resolution_for_pat(pat_id)?; let variant = self.infer.as_ref()?.variant_resolution_for_pat(pat_id)?;
let variant_data = variant.variant_data(db.upcast()); let variant_data = variant.variant_data(db.upcast());
let field = FieldId { parent: variant, local_id: variant_data.field(&field_name)? }; let field = FieldId { parent: variant, local_id: variant_data.field(&field_name)? };
Some(field.into()) let (_, subst) = self.infer.as_ref()?.type_of_pat.get(pat_id)?.as_adt()?;
let field_ty =
db.field_types(variant).get(field.local_id)?.clone().substitute(Interner, subst);
Some((field.into(), Type::new_with_resolver(db, &self.resolver, field_ty)))
} }
pub(crate) fn resolve_macro_call( pub(crate) fn resolve_macro_call(

View file

@ -327,7 +327,7 @@ impl NameClass {
let pat_parent = ident_pat.syntax().parent(); let pat_parent = ident_pat.syntax().parent();
if let Some(record_pat_field) = pat_parent.and_then(ast::RecordPatField::cast) { if let Some(record_pat_field) = pat_parent.and_then(ast::RecordPatField::cast) {
if record_pat_field.name_ref().is_none() { if record_pat_field.name_ref().is_none() {
if let Some(field) = sema.resolve_record_pat_field(&record_pat_field) { if let Some((field, _)) = sema.resolve_record_pat_field(&record_pat_field) {
return Some(NameClass::PatFieldShorthand { return Some(NameClass::PatFieldShorthand {
local_def: local, local_def: local,
field_ref: field, field_ref: field,
@ -483,6 +483,13 @@ impl NameRefClass {
}, },
ast::RecordPatField(record_pat_field) => { ast::RecordPatField(record_pat_field) => {
sema.resolve_record_pat_field(&record_pat_field) sema.resolve_record_pat_field(&record_pat_field)
.map(|(field, ..)|field)
.map(Definition::Field)
.map(NameRefClass::Definition)
},
ast::RecordExprField(record_expr_field) => {
sema.resolve_record_field(&record_expr_field)
.map(|(field, ..)|field)
.map(Definition::Field) .map(Definition::Field)
.map(NameRefClass::Definition) .map(NameRefClass::Definition)
}, },

View file

@ -55,7 +55,7 @@ pub(crate) fn goto_type_definition(
ty ty
} else { } else {
let record_field = ast::RecordPatField::for_field_name_ref(&it)?; let record_field = ast::RecordPatField::for_field_name_ref(&it)?;
sema.resolve_record_pat_field(&record_field)?.ty(db) sema.resolve_record_pat_field(&record_field)?.1
} }
}, },
_ => return None, _ => return None,

View file

@ -16,7 +16,7 @@ use stdx::format_to;
use syntax::{ use syntax::{
algo, algo,
ast::{self, HasArgList}, ast::{self, HasArgList},
match_ast, AstNode, Direction, SyntaxToken, TextRange, TextSize, match_ast, AstNode, Direction, SyntaxElementChildren, SyntaxToken, TextRange, TextSize,
}; };
use crate::RootDatabase; use crate::RootDatabase;
@ -102,6 +102,14 @@ pub(crate) fn signature_help(db: &RootDatabase, position: FilePosition) -> Optio
} }
return signature_help_for_record_lit(&sema, record, token); return signature_help_for_record_lit(&sema, record, token);
}, },
ast::RecordPat(record) => {
let cursor_outside = record.record_pat_field_list().and_then(|list| list.r_curly_token()).as_ref() == Some(&token);
if cursor_outside {
continue;
}
return signature_help_for_record_pat(&sema, record, token);
},
ast::TupleStructPat(tuple_pat) => {},
_ => (), _ => (),
} }
} }
@ -346,10 +354,27 @@ fn signature_help_for_record_lit(
record: ast::RecordExpr, record: ast::RecordExpr,
token: SyntaxToken, token: SyntaxToken,
) -> Option<SignatureHelp> { ) -> Option<SignatureHelp> {
let active_parameter = record signature_help_for_record_(
.record_expr_field_list()? sema,
.syntax() record.record_expr_field_list()?.syntax().children_with_tokens(),
.children_with_tokens() &record.path()?,
record
.record_expr_field_list()?
.fields()
.filter_map(|field| sema.resolve_record_field(&field))
.map(|(field, _, ty)| (field, ty)),
token,
)
}
fn signature_help_for_record_(
sema: &Semantics<'_, RootDatabase>,
field_list_children: SyntaxElementChildren,
path: &ast::Path,
fields2: impl Iterator<Item = (hir::Field, hir::Type)>,
token: SyntaxToken,
) -> Option<SignatureHelp> {
let active_parameter = field_list_children
.filter_map(syntax::NodeOrToken::into_token) .filter_map(syntax::NodeOrToken::into_token)
.filter(|t| t.kind() == syntax::T![,]) .filter(|t| t.kind() == syntax::T![,])
.take_while(|t| t.text_range().start() <= token.text_range().start()) .take_while(|t| t.text_range().start() <= token.text_range().start())
@ -365,7 +390,7 @@ fn signature_help_for_record_lit(
let fields; let fields;
let db = sema.db; let db = sema.db;
let path_res = sema.resolve_path(&record.path()?)?; let path_res = sema.resolve_path(path)?;
if let PathResolution::Def(ModuleDef::Variant(variant)) = path_res { if let PathResolution::Def(ModuleDef::Variant(variant)) = path_res {
fields = variant.fields(db); fields = variant.fields(db);
let en = variant.parent_enum(db); let en = variant.parent_enum(db);
@ -397,8 +422,7 @@ fn signature_help_for_record_lit(
let mut fields = let mut fields =
fields.into_iter().map(|field| (field.name(db), Some(field))).collect::<FxIndexMap<_, _>>(); fields.into_iter().map(|field| (field.name(db), Some(field))).collect::<FxIndexMap<_, _>>();
let mut buf = String::new(); let mut buf = String::new();
for field in record.record_expr_field_list()?.fields() { for (field, ty) in fields2 {
let Some((field, _, ty)) = sema.resolve_record_field(&field) else { continue };
let name = field.name(db); let name = field.name(db);
format_to!(buf, "{name}: {}", ty.display_truncated(db, Some(20))); format_to!(buf, "{name}: {}", ty.display_truncated(db, Some(20)));
res.push_record_field(&buf); res.push_record_field(&buf);
@ -418,6 +442,23 @@ fn signature_help_for_record_lit(
Some(res) Some(res)
} }
fn signature_help_for_record_pat(
sema: &Semantics<'_, RootDatabase>,
record: ast::RecordPat,
token: SyntaxToken,
) -> Option<SignatureHelp> {
signature_help_for_record_(
sema,
record.record_pat_field_list()?.syntax().children_with_tokens(),
&record.path()?,
record
.record_pat_field_list()?
.fields()
.filter_map(|field| sema.resolve_record_pat_field(&field)),
token,
)
}
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use std::iter; use std::iter;
@ -1550,6 +1591,29 @@ impl S {
); );
} }
#[test]
fn record_pat() {
check(
r#"
struct Strukt<T, U = ()> {
t: T,
u: U,
unit: (),
}
fn f() {
let Strukt {
u: 0,
$0
}
}
"#,
expect![[r#"
struct Strukt { u: i32, t: T, unit: () }
------ ^^^^ --------
"#]],
);
}
#[test] #[test]
fn test_enum_in_nested_method_in_lambda() { fn test_enum_in_nested_method_in_lambda() {
check( check(