]> git.lizzy.rs Git - rust.git/commitdiff
allow any field type
authorDomantas Jadenkus <djadenkus@gmail.com>
Sat, 13 Feb 2021 12:29:26 +0000 (14:29 +0200)
committerDomantas Jadenkus <djadenkus@gmail.com>
Sat, 13 Feb 2021 12:29:26 +0000 (14:29 +0200)
crates/assists/src/handlers/generate_from_impl_for_enum.rs

index 71897e6ebbcb5b588e06b34915d629e3e562e0ca..ac9806368f52749de7cd234e3f7c45c8b4d5ca49 100644 (file)
@@ -34,10 +34,6 @@ pub(crate) fn generate_from_impl_for_enum(acc: &mut Assists, ctx: &AssistContext
         return None;
     }
     let field_type = field_list.fields().next()?.ty()?;
-    let path = match field_type {
-        ast::Type::PathType(it) => it,
-        _ => return None,
-    };
 
     if existing_from_impl(&ctx.sema, &variant).is_some() {
         mark::hit!(test_add_from_impl_already_exists);
@@ -59,7 +55,7 @@ fn from(v: {0}) -> Self {{
         Self::{2}(v)
     }}
 }}"#,
-                path.syntax(),
+                field_type.syntax(),
                 enum_name,
                 variant_name
             );
@@ -195,6 +191,21 @@ fn from(v: String) -> Self {
 
 pub trait From<T> {
     fn from(T) -> Self;
+}"#,
+        );
+    }
+
+    #[test]
+    fn test_add_from_impl_static_str() {
+        check_assist(
+            generate_from_impl_for_enum,
+            "enum A { $0One(&'static str) }",
+            r#"enum A { One(&'static str) }
+
+impl From<&'static str> for A {
+    fn from(v: &'static str) -> Self {
+        Self::One(v)
+    }
 }"#,
         );
     }