diff options
Diffstat (limited to 'crates/ra_assists/src')
-rw-r--r-- | crates/ra_assists/src/handlers/extract_struct_from_enum_variant.rs | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/crates/ra_assists/src/handlers/extract_struct_from_enum_variant.rs b/crates/ra_assists/src/handlers/extract_struct_from_enum_variant.rs index 6e9f2d0fc..b4e19b3dc 100644 --- a/crates/ra_assists/src/handlers/extract_struct_from_enum_variant.rs +++ b/crates/ra_assists/src/handlers/extract_struct_from_enum_variant.rs | |||
@@ -6,7 +6,7 @@ use rustc_hash::FxHashSet; | |||
6 | use syntax::{ | 6 | use syntax::{ |
7 | algo::find_node_at_offset, | 7 | algo::find_node_at_offset, |
8 | ast::{self, ArgListOwner, AstNode, NameOwner, VisibilityOwner}, | 8 | ast::{self, ArgListOwner, AstNode, NameOwner, VisibilityOwner}, |
9 | SourceFile, SyntaxNode, TextRange, TextSize, | 9 | SourceFile, TextRange, TextSize, |
10 | }; | 10 | }; |
11 | 11 | ||
12 | use crate::{ | 12 | use crate::{ |
@@ -72,7 +72,7 @@ pub(crate) fn extract_struct_from_enum_variant( | |||
72 | } | 72 | } |
73 | extract_struct_def( | 73 | extract_struct_def( |
74 | builder, | 74 | builder, |
75 | enum_ast.syntax(), | 75 | &enum_ast, |
76 | &variant_name, | 76 | &variant_name, |
77 | &field_list.to_string(), | 77 | &field_list.to_string(), |
78 | start_offset, | 78 | start_offset, |
@@ -112,9 +112,10 @@ fn insert_import( | |||
112 | Some(()) | 112 | Some(()) |
113 | } | 113 | } |
114 | 114 | ||
115 | // FIXME: this should use strongly-typed `make`, rather than string manipulation1 | ||
115 | fn extract_struct_def( | 116 | fn extract_struct_def( |
116 | builder: &mut AssistBuilder, | 117 | builder: &mut AssistBuilder, |
117 | enum_ast: &SyntaxNode, | 118 | enum_: &ast::Enum, |
118 | variant_name: &str, | 119 | variant_name: &str, |
119 | variant_list: &str, | 120 | variant_list: &str, |
120 | start_offset: TextSize, | 121 | start_offset: TextSize, |
@@ -126,7 +127,7 @@ fn extract_struct_def( | |||
126 | } else { | 127 | } else { |
127 | "".to_string() | 128 | "".to_string() |
128 | }; | 129 | }; |
129 | let indent = if let Some(indent) = leading_indent(enum_ast) { | 130 | let indent = if let Some(indent) = leading_indent(enum_.syntax()) { |
130 | indent.to_string() | 131 | indent.to_string() |
131 | } else { | 132 | } else { |
132 | "".to_string() | 133 | "".to_string() |