diff options
Diffstat (limited to 'crates/ide/src/diagnostics/fixes/create_field.rs')
-rw-r--r-- | crates/ide/src/diagnostics/fixes/create_field.rs | 157 |
1 files changed, 157 insertions, 0 deletions
diff --git a/crates/ide/src/diagnostics/fixes/create_field.rs b/crates/ide/src/diagnostics/fixes/create_field.rs new file mode 100644 index 000000000..24e0fda52 --- /dev/null +++ b/crates/ide/src/diagnostics/fixes/create_field.rs | |||
@@ -0,0 +1,157 @@ | |||
1 | use hir::{db::AstDatabase, diagnostics::NoSuchField, HasSource, HirDisplay, Semantics}; | ||
2 | use ide_db::{base_db::FileId, source_change::SourceChange, RootDatabase}; | ||
3 | use syntax::{ | ||
4 | ast::{self, edit::IndentLevel, make}, | ||
5 | AstNode, | ||
6 | }; | ||
7 | use text_edit::TextEdit; | ||
8 | |||
9 | use crate::{ | ||
10 | diagnostics::{fix, DiagnosticWithFix}, | ||
11 | Assist, AssistResolveStrategy, | ||
12 | }; | ||
13 | |||
14 | impl DiagnosticWithFix for NoSuchField { | ||
15 | fn fix( | ||
16 | &self, | ||
17 | sema: &Semantics<RootDatabase>, | ||
18 | _resolve: &AssistResolveStrategy, | ||
19 | ) -> Option<Assist> { | ||
20 | let root = sema.db.parse_or_expand(self.file)?; | ||
21 | missing_record_expr_field_fix( | ||
22 | &sema, | ||
23 | self.file.original_file(sema.db), | ||
24 | &self.field.to_node(&root), | ||
25 | ) | ||
26 | } | ||
27 | } | ||
28 | |||
29 | fn missing_record_expr_field_fix( | ||
30 | sema: &Semantics<RootDatabase>, | ||
31 | usage_file_id: FileId, | ||
32 | record_expr_field: &ast::RecordExprField, | ||
33 | ) -> Option<Assist> { | ||
34 | let record_lit = ast::RecordExpr::cast(record_expr_field.syntax().parent()?.parent()?)?; | ||
35 | let def_id = sema.resolve_variant(record_lit)?; | ||
36 | let module; | ||
37 | let def_file_id; | ||
38 | let record_fields = match def_id { | ||
39 | hir::VariantDef::Struct(s) => { | ||
40 | module = s.module(sema.db); | ||
41 | let source = s.source(sema.db)?; | ||
42 | def_file_id = source.file_id; | ||
43 | let fields = source.value.field_list()?; | ||
44 | record_field_list(fields)? | ||
45 | } | ||
46 | hir::VariantDef::Union(u) => { | ||
47 | module = u.module(sema.db); | ||
48 | let source = u.source(sema.db)?; | ||
49 | def_file_id = source.file_id; | ||
50 | source.value.record_field_list()? | ||
51 | } | ||
52 | hir::VariantDef::Variant(e) => { | ||
53 | module = e.module(sema.db); | ||
54 | let source = e.source(sema.db)?; | ||
55 | def_file_id = source.file_id; | ||
56 | let fields = source.value.field_list()?; | ||
57 | record_field_list(fields)? | ||
58 | } | ||
59 | }; | ||
60 | let def_file_id = def_file_id.original_file(sema.db); | ||
61 | |||
62 | let new_field_type = sema.type_of_expr(&record_expr_field.expr()?)?; | ||
63 | if new_field_type.is_unknown() { | ||
64 | return None; | ||
65 | } | ||
66 | let new_field = make::record_field( | ||
67 | None, | ||
68 | make::name(&record_expr_field.field_name()?.text()), | ||
69 | make::ty(&new_field_type.display_source_code(sema.db, module.into()).ok()?), | ||
70 | ); | ||
71 | |||
72 | let last_field = record_fields.fields().last()?; | ||
73 | let last_field_syntax = last_field.syntax(); | ||
74 | let indent = IndentLevel::from_node(last_field_syntax); | ||
75 | |||
76 | let mut new_field = new_field.to_string(); | ||
77 | if usage_file_id != def_file_id { | ||
78 | new_field = format!("pub(crate) {}", new_field); | ||
79 | } | ||
80 | new_field = format!("\n{}{}", indent, new_field); | ||
81 | |||
82 | let needs_comma = !last_field_syntax.to_string().ends_with(','); | ||
83 | if needs_comma { | ||
84 | new_field = format!(",{}", new_field); | ||
85 | } | ||
86 | |||
87 | let source_change = SourceChange::from_text_edit( | ||
88 | def_file_id, | ||
89 | TextEdit::insert(last_field_syntax.text_range().end(), new_field), | ||
90 | ); | ||
91 | |||
92 | return Some(fix( | ||
93 | "create_field", | ||
94 | "Create field", | ||
95 | source_change, | ||
96 | record_expr_field.syntax().text_range(), | ||
97 | )); | ||
98 | |||
99 | fn record_field_list(field_def_list: ast::FieldList) -> Option<ast::RecordFieldList> { | ||
100 | match field_def_list { | ||
101 | ast::FieldList::RecordFieldList(it) => Some(it), | ||
102 | ast::FieldList::TupleFieldList(_) => None, | ||
103 | } | ||
104 | } | ||
105 | } | ||
106 | |||
107 | #[cfg(test)] | ||
108 | mod tests { | ||
109 | use crate::diagnostics::tests::check_fix; | ||
110 | |||
111 | #[test] | ||
112 | fn test_add_field_from_usage() { | ||
113 | check_fix( | ||
114 | r" | ||
115 | fn main() { | ||
116 | Foo { bar: 3, baz$0: false}; | ||
117 | } | ||
118 | struct Foo { | ||
119 | bar: i32 | ||
120 | } | ||
121 | ", | ||
122 | r" | ||
123 | fn main() { | ||
124 | Foo { bar: 3, baz: false}; | ||
125 | } | ||
126 | struct Foo { | ||
127 | bar: i32, | ||
128 | baz: bool | ||
129 | } | ||
130 | ", | ||
131 | ) | ||
132 | } | ||
133 | |||
134 | #[test] | ||
135 | fn test_add_field_in_other_file_from_usage() { | ||
136 | check_fix( | ||
137 | r#" | ||
138 | //- /main.rs | ||
139 | mod foo; | ||
140 | |||
141 | fn main() { | ||
142 | foo::Foo { bar: 3, $0baz: false}; | ||
143 | } | ||
144 | //- /foo.rs | ||
145 | struct Foo { | ||
146 | bar: i32 | ||
147 | } | ||
148 | "#, | ||
149 | r#" | ||
150 | struct Foo { | ||
151 | bar: i32, | ||
152 | pub(crate) baz: bool | ||
153 | } | ||
154 | "#, | ||
155 | ) | ||
156 | } | ||
157 | } | ||