1 use stdx::to_lower_snake_case;
2 use syntax::ast::VisibilityOwner;
3 use syntax::ast::{self, AstNode, NameOwner};
6 utils::{add_method_to_adt, find_struct_impl},
7 AssistContext, AssistId, AssistKind, Assists,
10 // Assist: generate_enum_is_method
12 // Generate an `is_` method for an enum variant.
30 // /// Returns `true` if the version is [`Minor`].
31 // fn is_minor(&self) -> bool {
32 // matches!(self, Self::Minor)
36 pub(crate) fn generate_enum_is_method(acc: &mut Assists, ctx: &AssistContext) -> Option<()> {
37 let variant = ctx.find_node_at_offset::<ast::Variant>()?;
38 let variant_name = variant.name()?;
39 let parent_enum = ast::Adt::Enum(variant.parent_enum());
40 let pattern_suffix = match variant.kind() {
41 ast::StructKind::Record(_) => " { .. }",
42 ast::StructKind::Tuple(_) => "(..)",
43 ast::StructKind::Unit => "",
46 let enum_lowercase_name = to_lower_snake_case(&parent_enum.name()?.to_string());
47 let fn_name = format!("is_{}", &to_lower_snake_case(&variant_name.text()));
49 // Return early if we've found an existing new fn
50 let impl_def = find_struct_impl(ctx, &parent_enum, &fn_name)?;
52 let target = variant.syntax().text_range();
54 AssistId("generate_enum_is_method", AssistKind::Generate),
55 "Generate an `is_` method for an enum variant",
58 let vis = parent_enum.visibility().map_or(String::new(), |v| format!("{} ", v));
60 " /// Returns `true` if the {} is [`{}`].
61 {}fn {}(&self) -> bool {{
62 matches!(self, Self::{}{})
64 enum_lowercase_name, variant_name, vis, fn_name, variant_name, pattern_suffix,
67 add_method_to_adt(builder, &parent_enum, impl_def, &method);
74 use crate::tests::{check_assist, check_assist_not_applicable};
79 fn test_generate_enum_is_from_variant() {
81 generate_enum_is_method,
95 /// Returns `true` if the variant is [`Minor`].
96 fn is_minor(&self) -> bool {
97 matches!(self, Self::Minor)
104 fn test_generate_enum_is_already_implemented() {
105 check_assist_not_applicable(
106 generate_enum_is_method,
115 fn is_minor(&self) -> bool {
116 matches!(self, Self::Minor)
123 fn test_generate_enum_is_from_tuple_variant() {
125 generate_enum_is_method,
139 /// Returns `true` if the variant is [`Minor`].
140 fn is_minor(&self) -> bool {
141 matches!(self, Self::Minor(..))
148 fn test_generate_enum_is_from_record_variant() {
150 generate_enum_is_method,
154 Minor { foo: i32 }$0,
164 /// Returns `true` if the variant is [`Minor`].
165 fn is_minor(&self) -> bool {
166 matches!(self, Self::Minor { .. })
173 fn test_generate_enum_is_from_variant_with_one_variant() {
175 generate_enum_is_method,
176 r#"enum Variant { Undefi$0ned }"#,
178 enum Variant { Undefined }
181 /// Returns `true` if the variant is [`Undefined`].
182 fn is_undefined(&self) -> bool {
183 matches!(self, Self::Undefined)
190 fn test_generate_enum_is_from_variant_with_visibility_marker() {
192 generate_enum_is_method,
194 pub(crate) enum Variant {
199 r#"pub(crate) enum Variant {
206 /// Returns `true` if the variant is [`Minor`].
207 pub(crate) fn is_minor(&self) -> bool {
208 matches!(self, Self::Minor)
215 fn test_multiple_generate_enum_is_from_variant() {
217 generate_enum_is_method,
226 /// Returns `true` if the variant is [`Minor`].
227 fn is_minor(&self) -> bool {
228 matches!(self, Self::Minor)
238 /// Returns `true` if the variant is [`Minor`].
239 fn is_minor(&self) -> bool {
240 matches!(self, Self::Minor)
243 /// Returns `true` if the variant is [`Major`].
244 fn is_major(&self) -> bool {
245 matches!(self, Self::Major)