Omit conditional elements without an else branch when present in children

Author: texodus

packages/yew-macro/src/html_tree/html_if.rs

@@ -1,6 +1,6 @@
 use boolinator::Boolinator;
 use proc_macro2::TokenStream;
-use quote::{quote_spanned, ToTokens};
+use quote::{quote, quote_spanned, ToTokens};
 use syn::buffer::Cursor;
 use syn::parse::{Parse, ParseStream};
 use syn::spanned::Spanned;
@@ -139,3 +139,33 @@ impl ToTokens for HtmlRootBracedOrIf {
         }
     }
 }
+
+pub struct HtmlIfIterItem<'a>(pub &'a HtmlIf);
+
+impl<'a> ToNodeIterator for HtmlIfIterItem<'a> {
+    fn to_node_iterator_stream(&self) -> Option<TokenStream> {
+        let HtmlIf {
+            if_token,
+            cond,
+            then_branch,
+            else_branch,
+            ..
+        } = &self.0;
+
+        if else_branch.is_none() {
+            let new_tokens = quote_spanned! {if_token.span()=>
+                #[allow(clippy::manual_map)]
+                if #cond {
+                    ::std::option::Option::Some(::std::convert::Into::into(#then_branch))
+                } else {
+                    ::std::option::Option::None
+                }
+            };
+
+            Some(new_tokens)
+        } else {
+            let new_tokens = self.0.to_node_iterator_stream().unwrap();
+            Some(quote! { ::std::option::Option::Some(#new_tokens) })
+        }
+    }
+}

packages/yew-macro/src/html_tree/mod.rs

@@ -237,23 +237,55 @@ impl HtmlChildrenTree {
 
     pub fn to_build_vec_token_stream(&self) -> TokenStream {
         let Self(children) = self;
-
         if self.only_single_node_children() {
             // optimize for the common case where all children are single nodes (only using literal
             // html).
-            let children_into = children
-                .iter()
-                .map(|child| quote_spanned! {child.span()=> ::std::convert::Into::into(#child) });
-            return quote! {
-                ::std::vec![#(#children_into),*]
-            };
+            let children_into = children.iter().map(|child| {
+                if let HtmlTree::If(if_clause) = child {
+                    html_if::HtmlIfIterItem(if_clause.as_ref())
+                        .to_node_iterator_stream()
+                        .unwrap()
+                } else {
+                    quote_spanned! {child.span()=> ::std::option::Option::Some(::std::convert::Into::into(#child)) }
+                }
+            });
+
+            return quote! {{
+                use ::std::iter::IntoIterator;
+                use ::std::iter::Iterator;
+                [#(#children_into),*].into_iter().filter(|x: &::std::option::Option<_>| x.is_some()).map(|x| x.unwrap()).collect::<::std::vec::Vec<_>>()
+            }};
         }
 
         let vec_ident = Ident::new("__yew_v", Span::mixed_site());
         let add_children_streams = children.iter().map(|child| {
             if let Some(node_iterator_stream) = child.to_node_iterator_stream() {
-                quote! {
-                    ::std::iter::Extend::extend(&mut #vec_ident, #node_iterator_stream);
+                if let HtmlTree::If(if_clause) = child {
+                    let tokens = html_if::HtmlIfIterItem(if_clause.as_ref())
+                        .to_node_iterator_stream()
+                        .unwrap();
+
+                    quote! {
+                        if let ::std::option::Option::Some(iter) = #tokens {
+                            ::std::iter::Extend::extend(&mut #vec_ident, iter)
+                        } else {
+                            #vec_ident
+                        }
+                    }
+                } else {
+                    quote! {
+                        ::std::iter::Extend::extend(&mut #vec_ident, #node_iterator_stream);
+                    }
+                }
+            } else if let HtmlTree::If(if_clause) = child {
+                let tokens = html_if::HtmlIfIterItem(if_clause.as_ref())
+                    .to_node_iterator_stream()
+                    .unwrap();
+
+                quote! {child.span()=>
+                    if let ::std::option::Option::Some(iter) = #tokens {
+                        #vec_ident.push(iter)
+                    }
                 }
             } else {
                 quote_spanned! {child.span()=>

packages/yew/src/dom_bundle/btag/mod.rs

@@ -1175,7 +1175,6 @@ mod tests_without_browser {
             },
             html! {
                 <div>
-                    <></>
                 </div>
             },
         );
@@ -1265,7 +1264,7 @@ mod tests_without_browser {
                     }
                 </div>
             },
-            html! { <div><></></div> },
+            html! { <div></div> },
         );
     }
 }

packages/yew/tests/children.rs

@@ -0,0 +1,61 @@
+#![cfg(not(target_arch = "wasm32"))]
+
+use yew::prelude::*;
+
+#[tokio::test]
+async fn conditional_children_are_omitted() {
+    #[derive(Properties, PartialEq)]
+    pub struct Props {
+        #[prop_or_default]
+        pub children: Children,
+    }
+
+    #[function_component]
+    pub fn Problem(p: &Props) -> Html {
+        html! {
+            <div class="container">
+                {
+                    for p.children.iter().enumerate().map(|(i, x)| {
+                        html! {
+                            <div>
+                                <span>{x}</span>
+                                <span>{i}</span>
+                            </div>
+                        }
+                    })
+                }
+            </div>
+        }
+    }
+
+    #[function_component]
+    fn App() -> Html {
+        let b = false;
+        html! {
+            <Problem>
+                <span>{ "A" }</span>
+                <span>{ "B" }</span>
+                if b {
+                    <span>{ "C" }</span>
+                }
+                <span>{ "D" }</span>
+            </Problem>
+        }
+    }
+
+    let mut s = String::new();
+    yew::ServerRenderer::<App>::new()
+        .render_to_string(&mut s)
+        .await;
+
+    assert_eq!(
+        s,
+        "<!--<[children::conditional_children_are_omitted::{{closure}}::App]>--><!\
+         --<[children::conditional_children_are_omitted::{{closure}}::Problem]>--><div \
+         class=\"container\"><div><span><span>A</span></span><span>0</span></\
+         div><div><span><span>B</span></span><span>1</span></div><div><span><span>D</span></\
+         span><span>2</span></div></div><!--</\
+         [children::conditional_children_are_omitted::{{closure}}::Problem]>--><!--</\
+         [children::conditional_children_are_omitted::{{closure}}::App]>-->"
+    );
+}