From ed495121ddf508e9ba6b1eed051461e59ac4cfb9 Mon Sep 17 00:00:00 2001
From: Manuel Cillero <manuel@cillero.es>
Date: Sat, 18 Nov 2023 08:27:40 +0100
Subject: [PATCH] =?UTF-8?q?=F0=9F=A9=B9=20Fix=20component=20reference=20un?=
 =?UTF-8?q?wrap()=20issues?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pagetop-bootsier/src/lib.rs              | 242 ++++++++++++++---------
 pagetop-bulmix/src/lib.rs                | 218 ++++++++++++--------
 pagetop/src/core/component/definition.rs |   8 +-
 3 files changed, 293 insertions(+), 175 deletions(-)

diff --git a/pagetop-bootsier/src/lib.rs b/pagetop-bootsier/src/lib.rs
index 543bca46..4f165f11 100644
--- a/pagetop-bootsier/src/lib.rs
+++ b/pagetop-bootsier/src/lib.rs
@@ -84,102 +84,166 @@ impl ThemeTrait for Bootsier {
     fn before_prepare_component(&self, component: &mut dyn ComponentTrait, _cx: &mut Context) {
         match component.handle() {
             COMPONENT_BASE_ICON => {
-                let i = component_as_mut::<Icon>(component);
-                match i.font_size() {
-                    FontSize::ExtraLarge => {
-                        i.alter_classes(ClassesOp::Replace(i.font_size().to_string()), "fs-1");
-                    }
-                    FontSize::XxLarge => {
-                        i.alter_classes(ClassesOp::Replace(i.font_size().to_string()), "fs-2");
-                    }
-                    FontSize::XLarge => {
-                        i.alter_classes(ClassesOp::Replace(i.font_size().to_string()), "fs-3");
-                    }
-                    FontSize::Large => {
-                        i.alter_classes(ClassesOp::Replace(i.font_size().to_string()), "fs-4");
-                    }
-                    FontSize::Medium => {
-                        i.alter_classes(ClassesOp::Replace(i.font_size().to_string()), "fs-5");
-                    }
-                    _ => {}
-                };
+                if let Some(icon) = component_as_mut::<Icon>(component) {
+                    match icon.font_size() {
+                        FontSize::ExtraLarge => {
+                            icon.alter_classes(
+                                ClassesOp::Replace(icon.font_size().to_string()),
+                                "fs-1",
+                            );
+                        }
+                        FontSize::XxLarge => {
+                            icon.alter_classes(
+                                ClassesOp::Replace(icon.font_size().to_string()),
+                                "fs-2",
+                            );
+                        }
+                        FontSize::XLarge => {
+                            icon.alter_classes(
+                                ClassesOp::Replace(icon.font_size().to_string()),
+                                "fs-3",
+                            );
+                        }
+                        FontSize::Large => {
+                            icon.alter_classes(
+                                ClassesOp::Replace(icon.font_size().to_string()),
+                                "fs-4",
+                            );
+                        }
+                        FontSize::Medium => {
+                            icon.alter_classes(
+                                ClassesOp::Replace(icon.font_size().to_string()),
+                                "fs-5",
+                            );
+                        }
+                        _ => {}
+                    };
+                }
             }
             COMPONENT_BASE_BUTTON => {
-                let a = component_as_mut::<Button>(component);
-                match a.font_size() {
-                    FontSize::ExtraLarge => {
-                        a.alter_classes(ClassesOp::Replace(a.font_size().to_string()), "fs-1");
-                    }
-                    FontSize::XxLarge => {
-                        a.alter_classes(ClassesOp::Replace(a.font_size().to_string()), "fs-2");
-                    }
-                    FontSize::XLarge => {
-                        a.alter_classes(ClassesOp::Replace(a.font_size().to_string()), "fs-3");
-                    }
-                    FontSize::Large => {
-                        a.alter_classes(ClassesOp::Replace(a.font_size().to_string()), "fs-4");
-                    }
-                    FontSize::Medium => {
-                        a.alter_classes(ClassesOp::Replace(a.font_size().to_string()), "fs-5");
-                    }
-                    _ => {}
-                };
-                match a.button_type() {
-                    ButtonType::Link => {
-                        a.alter_classes(
-                            ClassesOp::Replace(a.button_type().to_string()),
-                            "btn btn-link",
-                        );
-                    }
-                    ButtonType::Primary => {
-                        a.alter_classes(
-                            ClassesOp::Replace(a.button_type().to_string()),
-                            "btn btn-primary",
-                        );
-                    }
+                if let Some(button) = component_as_mut::<Button>(component) {
+                    match button.font_size() {
+                        FontSize::ExtraLarge => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.font_size().to_string()),
+                                "fs-1",
+                            );
+                        }
+                        FontSize::XxLarge => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.font_size().to_string()),
+                                "fs-2",
+                            );
+                        }
+                        FontSize::XLarge => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.font_size().to_string()),
+                                "fs-3",
+                            );
+                        }
+                        FontSize::Large => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.font_size().to_string()),
+                                "fs-4",
+                            );
+                        }
+                        FontSize::Medium => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.font_size().to_string()),
+                                "fs-5",
+                            );
+                        }
+                        _ => {}
+                    };
+                    match button.button_type() {
+                        ButtonType::Link => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.button_type().to_string()),
+                                "btn btn-link",
+                            );
+                        }
+                        ButtonType::Primary => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.button_type().to_string()),
+                                "btn btn-primary",
+                            );
+                        }
+                    };
                 }
             }
             COMPONENT_BASE_HEADING => {
-                let h = component_as_mut::<Heading>(component);
-                match h.display() {
-                    HeadingDisplay::ExtraLarge => {
-                        h.alter_classes(ClassesOp::Replace(h.display().to_string()), "display-1");
-                    }
-                    HeadingDisplay::XxLarge => {
-                        h.alter_classes(ClassesOp::Replace(h.display().to_string()), "display-2");
-                    }
-                    HeadingDisplay::XLarge => {
-                        h.alter_classes(ClassesOp::Replace(h.display().to_string()), "display-3");
-                    }
-                    HeadingDisplay::Large => {
-                        h.alter_classes(ClassesOp::Replace(h.display().to_string()), "display-4");
-                    }
-                    HeadingDisplay::Medium => {
-                        h.alter_classes(ClassesOp::Replace(h.display().to_string()), "display-5");
-                    }
-                    _ => {}
-                };
+                if let Some(heading) = component_as_mut::<Heading>(component) {
+                    match heading.display() {
+                        HeadingDisplay::ExtraLarge => {
+                            heading.alter_classes(
+                                ClassesOp::Replace(heading.display().to_string()),
+                                "display-1",
+                            );
+                        }
+                        HeadingDisplay::XxLarge => {
+                            heading.alter_classes(
+                                ClassesOp::Replace(heading.display().to_string()),
+                                "display-2",
+                            );
+                        }
+                        HeadingDisplay::XLarge => {
+                            heading.alter_classes(
+                                ClassesOp::Replace(heading.display().to_string()),
+                                "display-3",
+                            );
+                        }
+                        HeadingDisplay::Large => {
+                            heading.alter_classes(
+                                ClassesOp::Replace(heading.display().to_string()),
+                                "display-4",
+                            );
+                        }
+                        HeadingDisplay::Medium => {
+                            heading.alter_classes(
+                                ClassesOp::Replace(heading.display().to_string()),
+                                "display-5",
+                            );
+                        }
+                        _ => {}
+                    };
+                }
             }
             COMPONENT_BASE_PARAGRAPH => {
-                let p = component_as_mut::<Paragraph>(component);
-                match p.font_size() {
-                    FontSize::ExtraLarge => {
-                        p.alter_classes(ClassesOp::Replace(p.font_size().to_string()), "fs-1");
-                    }
-                    FontSize::XxLarge => {
-                        p.alter_classes(ClassesOp::Replace(p.font_size().to_string()), "fs-2");
-                    }
-                    FontSize::XLarge => {
-                        p.alter_classes(ClassesOp::Replace(p.font_size().to_string()), "fs-3");
-                    }
-                    FontSize::Large => {
-                        p.alter_classes(ClassesOp::Replace(p.font_size().to_string()), "fs-4");
-                    }
-                    FontSize::Medium => {
-                        p.alter_classes(ClassesOp::Replace(p.font_size().to_string()), "fs-5");
-                    }
-                    _ => {}
-                };
+                if let Some(paragraph) = component_as_mut::<Paragraph>(component) {
+                    match paragraph.font_size() {
+                        FontSize::ExtraLarge => {
+                            paragraph.alter_classes(
+                                ClassesOp::Replace(paragraph.font_size().to_string()),
+                                "fs-1",
+                            );
+                        }
+                        FontSize::XxLarge => {
+                            paragraph.alter_classes(
+                                ClassesOp::Replace(paragraph.font_size().to_string()),
+                                "fs-2",
+                            );
+                        }
+                        FontSize::XLarge => {
+                            paragraph.alter_classes(
+                                ClassesOp::Replace(paragraph.font_size().to_string()),
+                                "fs-3",
+                            );
+                        }
+                        FontSize::Large => {
+                            paragraph.alter_classes(
+                                ClassesOp::Replace(paragraph.font_size().to_string()),
+                                "fs-4",
+                            );
+                        }
+                        FontSize::Medium => {
+                            paragraph.alter_classes(
+                                ClassesOp::Replace(paragraph.font_size().to_string()),
+                                "fs-5",
+                            );
+                        }
+                        _ => {}
+                    };
+                }
             }
             _ => {}
         }
diff --git a/pagetop-bulmix/src/lib.rs b/pagetop-bulmix/src/lib.rs
index ba50dd68..369a9a97 100644
--- a/pagetop-bulmix/src/lib.rs
+++ b/pagetop-bulmix/src/lib.rs
@@ -33,91 +33,141 @@ impl ThemeTrait for Bulmix {
     fn before_prepare_component(&self, component: &mut dyn ComponentTrait, _cx: &mut Context) {
         match component.handle() {
             COMPONENT_BASE_ICON => {
-                let i = component_as_mut::<Icon>(component);
-                match i.font_size() {
-                    FontSize::ExtraLarge => {
-                        i.alter_classes(ClassesOp::Replace(i.font_size().to_string()), "is-size-1");
-                    }
-                    FontSize::XxLarge => {
-                        i.alter_classes(ClassesOp::Replace(i.font_size().to_string()), "is-size-2");
-                    }
-                    FontSize::XLarge => {
-                        i.alter_classes(ClassesOp::Replace(i.font_size().to_string()), "is-size-3");
-                    }
-                    FontSize::Large => {
-                        i.alter_classes(ClassesOp::Replace(i.font_size().to_string()), "is-size-4");
-                    }
-                    FontSize::Medium => {
-                        i.alter_classes(ClassesOp::Replace(i.font_size().to_string()), "is-size-5");
-                    }
-                    _ => {}
-                };
+                if let Some(icon) = component_as_mut::<Icon>(component) {
+                    match icon.font_size() {
+                        FontSize::ExtraLarge => {
+                            icon.alter_classes(
+                                ClassesOp::Replace(icon.font_size().to_string()),
+                                "is-size-1",
+                            );
+                        }
+                        FontSize::XxLarge => {
+                            icon.alter_classes(
+                                ClassesOp::Replace(icon.font_size().to_string()),
+                                "is-size-2",
+                            );
+                        }
+                        FontSize::XLarge => {
+                            icon.alter_classes(
+                                ClassesOp::Replace(icon.font_size().to_string()),
+                                "is-size-3",
+                            );
+                        }
+                        FontSize::Large => {
+                            icon.alter_classes(
+                                ClassesOp::Replace(icon.font_size().to_string()),
+                                "is-size-4",
+                            );
+                        }
+                        FontSize::Medium => {
+                            icon.alter_classes(
+                                ClassesOp::Replace(icon.font_size().to_string()),
+                                "is-size-5",
+                            );
+                        }
+                        _ => {}
+                    };
+                }
             }
             COMPONENT_BASE_BUTTON => {
-                let a = component_as_mut::<Button>(component);
-                match a.font_size() {
-                    FontSize::ExtraLarge => {
-                        a.alter_classes(ClassesOp::Replace(a.font_size().to_string()), "is-size-1");
-                    }
-                    FontSize::XxLarge => {
-                        a.alter_classes(ClassesOp::Replace(a.font_size().to_string()), "is-size-2");
-                    }
-                    FontSize::XLarge => {
-                        a.alter_classes(ClassesOp::Replace(a.font_size().to_string()), "is-size-3");
-                    }
-                    FontSize::Large => {
-                        a.alter_classes(ClassesOp::Replace(a.font_size().to_string()), "is-size-4");
-                    }
-                    FontSize::Medium => {
-                        a.alter_classes(ClassesOp::Replace(a.font_size().to_string()), "is-size-5");
-                    }
-                    _ => {}
-                };
-                match a.button_type() {
-                    ButtonType::Link => {
-                        a.alter_classes(
-                            ClassesOp::Replace(a.button_type().to_string()),
-                            "button is-text",
-                        );
-                    }
-                    ButtonType::Primary => {
-                        a.alter_classes(
-                            ClassesOp::Replace(a.button_type().to_string()),
-                            "button is-primary",
-                        );
-                    }
+                if let Some(button) = component_as_mut::<Button>(component) {
+                    match button.font_size() {
+                        FontSize::ExtraLarge => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.font_size().to_string()),
+                                "is-size-1",
+                            );
+                        }
+                        FontSize::XxLarge => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.font_size().to_string()),
+                                "is-size-2",
+                            );
+                        }
+                        FontSize::XLarge => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.font_size().to_string()),
+                                "is-size-3",
+                            );
+                        }
+                        FontSize::Large => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.font_size().to_string()),
+                                "is-size-4",
+                            );
+                        }
+                        FontSize::Medium => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.font_size().to_string()),
+                                "is-size-5",
+                            );
+                        }
+                        _ => {}
+                    };
+                    match button.button_type() {
+                        ButtonType::Link => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.button_type().to_string()),
+                                "button is-text",
+                            );
+                        }
+                        ButtonType::Primary => {
+                            button.alter_classes(
+                                ClassesOp::Replace(button.button_type().to_string()),
+                                "button is-primary",
+                            );
+                        }
+                    };
                 }
             }
             COMPONENT_BASE_HEADING => {
-                let h = component_as_mut::<Heading>(component);
-                match h.display() {
-                    HeadingDisplay::Subtitle => {
-                        h.alter_classes(ClassesOp::Replace(h.display().to_string()), "subtitle")
-                    }
-                    _ => h.alter_classes(ClassesOp::Add, "title"),
-                };
+                if let Some(heading) = component_as_mut::<Heading>(component) {
+                    match heading.display() {
+                        HeadingDisplay::Subtitle => heading.alter_classes(
+                            ClassesOp::Replace(heading.display().to_string()),
+                            "subtitle",
+                        ),
+                        _ => heading.alter_classes(ClassesOp::Add, "title"),
+                    };
+                }
             }
             COMPONENT_BASE_PARAGRAPH => {
-                let p = component_as_mut::<Paragraph>(component);
-                p.alter_classes(ClassesOp::Add, "block");
-                match p.font_size() {
-                    FontSize::ExtraLarge => {
-                        p.alter_classes(ClassesOp::Replace(p.font_size().to_string()), "is-size-1");
-                    }
-                    FontSize::XxLarge => {
-                        p.alter_classes(ClassesOp::Replace(p.font_size().to_string()), "is-size-2");
-                    }
-                    FontSize::XLarge => {
-                        p.alter_classes(ClassesOp::Replace(p.font_size().to_string()), "is-size-3");
-                    }
-                    FontSize::Large => {
-                        p.alter_classes(ClassesOp::Replace(p.font_size().to_string()), "is-size-4");
-                    }
-                    FontSize::Medium => {
-                        p.alter_classes(ClassesOp::Replace(p.font_size().to_string()), "is-size-5");
-                    }
-                    _ => {}
-                };
+                if let Some(paragraph) = component_as_mut::<Paragraph>(component) {
+                    paragraph.alter_classes(ClassesOp::Add, "block");
+                    match paragraph.font_size() {
+                        FontSize::ExtraLarge => {
+                            paragraph.alter_classes(
+                                ClassesOp::Replace(paragraph.font_size().to_string()),
+                                "is-size-1",
+                            );
+                        }
+                        FontSize::XxLarge => {
+                            paragraph.alter_classes(
+                                ClassesOp::Replace(paragraph.font_size().to_string()),
+                                "is-size-2",
+                            );
+                        }
+                        FontSize::XLarge => {
+                            paragraph.alter_classes(
+                                ClassesOp::Replace(paragraph.font_size().to_string()),
+                                "is-size-3",
+                            );
+                        }
+                        FontSize::Large => {
+                            paragraph.alter_classes(
+                                ClassesOp::Replace(paragraph.font_size().to_string()),
+                                "is-size-4",
+                            );
+                        }
+                        FontSize::Medium => {
+                            paragraph.alter_classes(
+                                ClassesOp::Replace(paragraph.font_size().to_string()),
+                                "is-size-5",
+                            );
+                        }
+                        _ => {}
+                    };
+                }
             }
             _ => {}
         }
@@ -130,11 +180,15 @@ impl ThemeTrait for Bulmix {
     ) -> Option<Markup> {
         match component.handle() {
             COMPONENT_BASE_ICON => {
-                let icon = component_as_ref::<Icon>(component);
-                match icon.icon_name().get() {
-                    None => None,
-                    _ => Some(html! { span class="icon" { i class=[icon.classes().get()] {} } }),
+                if let Some(icon) = component_as_ref::<Icon>(component) {
+                    return match icon.icon_name().get() {
+                        None => None,
+                        _ => {
+                            Some(html! { span class="icon" { i class=[icon.classes().get()] {} } })
+                        }
+                    };
                 }
+                None
             }
             _ => None,
         }
diff --git a/pagetop/src/core/component/definition.rs b/pagetop/src/core/component/definition.rs
index e89942cc..da7a59be 100644
--- a/pagetop/src/core/component/definition.rs
+++ b/pagetop/src/core/component/definition.rs
@@ -97,10 +97,10 @@ impl<C: ComponentTrait> ComponentBase for C {
     }
 }
 
-pub fn component_as_ref<C: ComponentTrait>(component: &dyn ComponentTrait) -> &C {
-    component.as_ref_any().downcast_ref::<C>().unwrap()
+pub fn component_as_ref<C: ComponentTrait>(component: &dyn ComponentTrait) -> Option<&C> {
+    component.as_ref_any().downcast_ref::<C>()
 }
 
-pub fn component_as_mut<C: ComponentTrait>(component: &mut dyn ComponentTrait) -> &mut C {
-    component.as_mut_any().downcast_mut::<C>().unwrap()
+pub fn component_as_mut<C: ComponentTrait>(component: &mut dyn ComponentTrait) -> Option<&mut C> {
+    component.as_mut_any().downcast_mut::<C>()
 }