1 files changed, 31 insertions, 3 deletions
diff --git a/web/src/widget/button.rs b/web/src/widget/button.rs
index 1f117d82..8ccda107 100644
--- a/web/src/widget/button.rs
+++ b/web/src/widget/button.rs
@@ -1,14 +1,42 @@
-use crate::{Element, Widget};
+use crate::{Bus, Element, Widget};
+use dodrio::bumpalo;
 
 pub use iced::button::{Class, State};
 
 pub type Button<'a, Message> = iced::Button<'a, Message>;
 
-impl<'a, Message> Widget<Message> for Button<'a, Message> {}
+impl<'a, Message> Widget<Message> for Button<'a, Message>
+where
+    Message: 'static + Copy,
+{
+    fn node<'b>(
+        &self,
+        bump: &'b bumpalo::Bump,
+        bus: &Bus<Message>,
+    ) -> dodrio::Node<'b> {
+        use dodrio::builder::*;
+
+        let label = bumpalo::format!(in bump, "{}", self.label);
+
+        let mut node = button(bump).children(vec![text(label.into_bump_str())]);
+
+        if let Some(on_press) = self.on_press {
+            let event_bus = bus.clone();
+
+            node = node.on("click", move |root, vdom, _event| {
+                event_bus.publish(on_press, root);
+
+                vdom.schedule_render();
+            });
+        }
+
+        node.finish()
+    }
+}
 
 impl<'a, Message> From<Button<'a, Message>> for Element<'a, Message>
 where
-    Message: 'static,
+    Message: 'static + Copy,
 {
     fn from(button: Button<'a, Message>) -> Element<'a, Message> {
         Element::new(button)