review: addresses near-cli-rs#26 (comment)

dj8yfo · Feb 10, 2025 · 9bedd97 · 9bedd97
1 parent aad7bda
commit 9bedd97
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/interactive-clap-derive/src/derives/interactive_clap/common_methods/choose_variant.rs b/interactive-clap-derive/src/derives/interactive_clap/common_methods/choose_variant.rs
@@ -35,8 +35,12 @@ pub fn fn_choose_variant(
             if attr.path.is_ident("strum_discriminants") {
                 for attr_token in attr.tokens.clone() {
                     if let proc_macro2::TokenTree::Group(group) = attr_token {
-                        let group_stream_no_whitespace =
-                            group.stream().to_string().replace(" ", "");
+                        let group_stream_no_whitespace = group
+                            .stream()
+                            .to_string()
+                            .split_whitespace()
+                            .collect::<Vec<_>>()
+                            .join("");
                         dbg_cond!(&group_stream_no_whitespace);
                         if &group_stream_no_whitespace == "derive(EnumMessage,EnumIter)" {
                             ast_attrs.push("strum_discriminants");