Clean-up

src/buttons.rs

@@ -8,6 +8,7 @@ pub struct Button {
 pub struct Props {
     #[prop_or_default]
     pub title: String,
+    #[prop_or_default]
     pub onclick: Callback<MouseEvent>,
     pub children: html::Children,
 }
@@ -35,7 +36,9 @@ impl Component for Button {
 
     fn view(&self) -> Html {
         html! {
-            <button class="bp3-button" onclick={self.props.onclick.clone()}>{self.props.children.clone()}</button>
+            <button class="bp3-button" onclick={self.props.onclick.clone()}>
+                {self.props.children.clone()}
+            </button>
         }
     }
 }

src/forms/controls.rs

@@ -8,7 +8,9 @@ pub struct Switch {
 pub struct Props {
     #[prop_or_default]
     pub checked: bool,
+    #[prop_or_default]
     pub onclick: Callback<MouseEvent>,
+    #[prop_or_default]
     pub label: String,
 }
 

src/icon.rs

@@ -55,12 +55,9 @@ impl Component for Icon {
     }
 
     fn view(&self) -> Html {
-        let mut class = "bp3-icon ".to_string();
+        let mut class = Classes::from("bp3-icon");
-        class.push_str(self.props.class.as_str());
+        class.push(self.props.class.as_str());
-        class.push_str(" ");
+        class = class.extend(&self.props.intent);
-        if let Some(intent) = self.props.intent {
-            class.push_str(intent.into());
-        }
 
         let paths = if self.props.icon_size == SIZE_STANDARD {
             icon_svg_paths_16(self.props.icon)

src/lib.rs

@@ -8,6 +8,8 @@ pub mod forms;
 pub mod icon;
 pub mod tree;
 
+use yew::virtual_dom::Classes;
+
 #[cfg(feature = "dev")]
 #[macro_export]
 macro_rules! log {
@@ -32,9 +34,15 @@ pub enum Intent {
     Danger,
 }
 
-impl From<Intent> for &'static str {
+impl From<Intent> for Classes {
     fn from(intent: Intent) -> Self {
-        match intent {
+        Classes::from(intent.as_ref())
+    }
+}
+
+impl AsRef<str> for Intent {
+    fn as_ref(&self) -> &'static str {
+        match self {
             Intent::Primary => "bp3-intent-primary",
             Intent::Success => "bp3-intent-success",
             Intent::Warning => "bp3-intent-warning",

src/tree.rs

@@ -192,16 +192,16 @@ impl Component for TreeNode {
     }
 
     fn view(&self) -> Html {
-        let mut content_class = "bp3-tree-node-content ".to_string();
+        let mut content_class = Classes::from("bp3-tree-node-content");
-        content_class.push_str(&format!("bp3-tree-node-content-{}", self.props.depth));
+        content_class.push(&format!("bp3-tree-node-content-{}", self.props.depth));
 
         html! {
             <li class="bp3-tree-node">
                 <div class=content_class>
                     {
                         if self.props.has_caret {
-                            let mut class = "bp3-tree-node-caret ".to_string();
+                            let mut class = Classes::from("bp3-tree-node-caret");
-                            class.push_str(if self.props.is_expanded {
+                            class.push(if self.props.is_expanded {
                                 "bp3-tree-node-caret-open"
                             } else {
                                 "bp3-tree-node-caret-closed"
@@ -209,7 +209,7 @@ impl Component for TreeNode {
 
                             html! {
                                 <Icon
-                                    class=class
+                                    class=class.to_string()
                                     icon=IconName::ChevronRight
                                     onclick={
                                         if self.props.disabled {