From f3528088cd4e9f37fe8e8fa691fdefc0fcb0b8d4 Mon Sep 17 00:00:00 2001
From: Alexandre Bury <alexandre.bury@gmail.com>
Date: Wed, 11 Oct 2017 15:07:25 -0700
Subject: [PATCH] Add mouse support to Dialog and Button

---
 src/views/button.rs |  23 +++-
 src/views/dialog.rs | 328 +++++++++++++++++++++++++++++---------------
 2 files changed, 238 insertions(+), 113 deletions(-)

diff --git a/src/views/button.rs b/src/views/button.rs
index 591d300..6232e70 100644
--- a/src/views/button.rs
+++ b/src/views/button.rs
@@ -28,7 +28,8 @@ pub struct Button {
 impl Button {
     /// Creates a new button with the given content and callback.
     pub fn new<F, S: Into<String>>(label: S, cb: F) -> Self
-        where F: Fn(&mut Cursive) + 'static
+    where
+        F: Fn(&mut Cursive) + 'static,
     {
         Button {
             label: label.into(),
@@ -41,7 +42,8 @@ impl Button {
     ///
     /// Replaces the previous callback.
     pub fn set_callback<F>(&mut self, cb: F)
-        where F: Fn(&mut Cursive) + 'static
+    where
+        F: Fn(&mut Cursive) + 'static,
     {
         self.callback = Callback::from_fn(cb);
     }
@@ -74,11 +76,14 @@ impl Button {
     pub fn is_enabled(&self) -> bool {
         self.enabled
     }
+
+    fn req_size(&self) -> Vec2 {
+        Vec2::new(2 + self.label.width(), 1)
+    }
 }
 
 impl View for Button {
     fn draw(&self, printer: &Printer) {
-
         if printer.size.x == 0 {
             return;
         }
@@ -101,15 +106,25 @@ impl View for Button {
 
     fn required_size(&mut self, _: Vec2) -> Vec2 {
         // Meh. Fixed size we are.
-        Vec2::new(2 + self.label.width(), 1)
+        self.req_size()
     }
 
     fn on_event(&mut self, event: Event) -> EventResult {
+        eprintln!("{:?}", event);
+        eprintln!("{:?}", self.req_size());
         match event {
             // 10 is the ascii code for '\n', that is the return key
             Event::Key(Key::Enter) => {
                 EventResult::Consumed(Some(self.callback.clone()))
             }
+            Event::Mouse {
+                event: MouseEvent::Release(MouseButton::Left),
+                position,
+                offset,
+            } if position.fits_in_rect(offset, self.req_size()) =>
+            {
+                EventResult::Consumed(Some(self.callback.clone()))
+            }
             _ => EventResult::Ignored,
         }
     }
diff --git a/src/views/dialog.rs b/src/views/dialog.rs
index 1747171..8f53330 100644
--- a/src/views/dialog.rs
+++ b/src/views/dialog.rs
@@ -1,5 +1,3 @@
-
-
 use Cursive;
 use Printer;
 use With;
@@ -7,9 +5,9 @@ use align::*;
 use direction::Direction;
 use event::*;
 use std::any::Any;
+use std::cell::Cell;
 use std::cmp::max;
 use theme::ColorStyle;
-
 use unicode_width::UnicodeWidthStr;
 use vec::{Vec2, Vec4};
 use view::{Selector, View};
@@ -21,6 +19,23 @@ enum Focus {
     Button(usize),
 }
 
+struct ChildButton {
+    button: SizedView<Button>,
+    offset: Cell<Vec2>,
+}
+
+impl ChildButton {
+    pub fn new<F, S: Into<String>>(label: S, cb: F) -> Self
+    where
+        F: Fn(&mut Cursive) + 'static,
+    {
+        ChildButton {
+            button: SizedView::new(Button::new(label, cb)),
+            offset: Cell::new(Vec2::zero()),
+        }
+    }
+}
+
 /// Popup-like view with a main content, and optional buttons under it.
 ///
 /// # Examples
@@ -31,16 +46,26 @@ enum Focus {
 ///                     .button("Ok", |s| s.quit());
 /// ```
 pub struct Dialog {
+    // Possibly empty title.
     title: String,
-    content: Box<View>,
 
-    buttons: Vec<SizedView<Button>>,
+    // The actual inner view.
+    content: SizedView<Box<View>>,
 
+    // Optional list of buttons under the main view.
+    // Include the top-left corner.
+    buttons: Vec<ChildButton>,
+
+    // Padding around the inner view.
     padding: Vec4,
+
+    // Borders around everything.
     borders: Vec4,
 
+    // The current element in focus
     focus: Focus,
 
+    // How to align the buttons under the view.
     align: Align,
 }
 
@@ -57,7 +82,7 @@ impl Dialog {
     /// Creates a new `Dialog` with the given content.
     pub fn around<V: View + 'static>(view: V) -> Self {
         Dialog {
-            content: Box::new(view),
+            content: SizedView::new(Box::new(view)),
             buttons: Vec::new(),
             title: String::new(),
             focus: Focus::Content,
@@ -78,7 +103,7 @@ impl Dialog {
     ///
     /// Previous content will be dropped.
     pub fn set_content<V: View + 'static>(&mut self, view: V) {
-        self.content = Box::new(view);
+        self.content = SizedView::new(Box::new(view));
     }
 
     /// Convenient method to create a dialog with a simple text content.
@@ -97,9 +122,10 @@ impl Dialog {
     ///
     /// Consumes and returns self for easy chaining.
     pub fn button<F, S: Into<String>>(mut self, label: S, cb: F) -> Self
-        where F: Fn(&mut Cursive) + 'static
+    where
+        F: Fn(&mut Cursive) + 'static,
     {
-        self.buttons.push(SizedView::new(Button::new(label, cb)));
+        self.buttons.push(ChildButton::new(label, cb));
 
         self
     }
@@ -113,6 +139,9 @@ impl Dialog {
         self
     }
 
+    /*
+     * Commented out because currently un-implemented.
+     *
     /// Sets the vertical alignment for the buttons, if any.
     ///
     /// Only works if the buttons are as a column to the right of the dialog.
@@ -121,6 +150,7 @@ impl Dialog {
 
         self
     }
+    */
 
     /// Shortcut method to add a button that will dismiss the dialog.
     pub fn dismiss_button<S: Into<String>>(self, label: S) -> Self {
@@ -169,65 +199,148 @@ impl Dialog {
         self.padding.right = padding;
         self
     }
-}
 
-impl View for Dialog {
-    fn draw(&self, printer: &Printer) {
 
-        // This will be the buttons_height used by the buttons.
+    // Private methods
+
+    // An event is received while the content is in focus
+    fn on_event_content(&mut self, event: Event) -> EventResult {
+        match self.content.on_event(event.relativized((1, 1))) {
+            EventResult::Ignored if !self.buttons.is_empty() => {
+                match event {
+                    Event::Key(Key::Down) |
+                    Event::Key(Key::Tab) |
+                    Event::Shift(Key::Tab) => {
+                        // Default to leftmost button when going down.
+                        self.focus = Focus::Button(0);
+                        EventResult::Consumed(None)
+                    }
+                    _ => EventResult::Ignored,
+                }
+            }
+            res => res,
+        }
+    }
+
+    // An event is received while a button is in focus
+    fn on_event_button(
+        &mut self, event: Event, button_id: usize
+    ) -> EventResult {
+        let result = {
+            let button = &mut self.buttons[button_id];
+            button
+                .button
+                .on_event(event.relativized(button.offset.get()))
+        };
+        match result {
+            EventResult::Ignored => {
+                match event {
+                    // Up goes back to the content
+                    Event::Key(Key::Up) => {
+                        if self.content.take_focus(Direction::down()) {
+                            self.focus = Focus::Content;
+                            EventResult::Consumed(None)
+                        } else {
+                            EventResult::Ignored
+                        }
+                    }
+                    Event::Shift(Key::Tab) => {
+                        if self.content.take_focus(Direction::back()) {
+                            self.focus = Focus::Content;
+                            EventResult::Consumed(None)
+                        } else {
+                            EventResult::Ignored
+                        }
+                    }
+                    Event::Key(Key::Tab) => {
+                        if self.content.take_focus(Direction::front()) {
+                            self.focus = Focus::Content;
+                            EventResult::Consumed(None)
+                        } else {
+                            EventResult::Ignored
+                        }
+                    }
+                    // Left and Right move to other buttons
+                    Event::Key(Key::Right)
+                        if button_id + 1 < self.buttons.len() =>
+                    {
+                        self.focus = Focus::Button(button_id + 1);
+                        EventResult::Consumed(None)
+                    }
+                    Event::Key(Key::Left) if button_id > 0 => {
+                        self.focus = Focus::Button(button_id - 1);
+                        EventResult::Consumed(None)
+                    }
+                    _ => EventResult::Ignored,
+                }
+            }
+            res => res,
+        }
+    }
+
+    fn draw_buttons(&self, printer: &Printer) -> Option<usize> {
         let mut buttons_height = 0;
         // Current horizontal position of the next button we'll draw.
 
         // Sum of the sizes + len-1 for margins
         let width = self.buttons
             .iter()
-            .map(|button| button.size.x)
-            .fold(0, |a, b| a + b) +
-                    self.buttons.len().saturating_sub(1);
+            .map(|button| button.button.size.x)
+            .fold(0, |a, b| a + b)
+            + self.buttons.len().saturating_sub(1);
         let overhead = self.padding + self.borders;
         if printer.size.x < overhead.horizontal() {
-            return;
+            return None;
         }
-        let mut offset = overhead.left +
-                         self.align.h.get_offset(width,
-                                                 printer.size.x -
-                                                 overhead.horizontal());
+        let mut offset = overhead.left
+            + self.align
+                .h
+                .get_offset(width, printer.size.x - overhead.horizontal());
 
         let overhead_bottom = self.padding.bottom + self.borders.bottom + 1;
 
         let y = match printer.size.y.checked_sub(overhead_bottom) {
             Some(y) => y,
-            None => return,
+            None => return None,
         };
 
         for (i, button) in self.buttons.iter().enumerate() {
-            let size = button.size;
+            let size = button.button.size;
             // Add some special effect to the focused button
-            button.draw(&printer.sub_printer(Vec2::new(offset, y),
-                                             size,
-                                             self.focus == Focus::Button(i)));
+            let position = Vec2::new(offset, y);
+            button.offset.set(position);
+            button.button.draw(&printer.sub_printer(
+                position,
+                size,
+                self.focus == Focus::Button(i),
+            ));
             // Keep 1 blank between two buttons
             offset += size.x + 1;
             // Also keep 1 blank above the buttons
             buttons_height = max(buttons_height, size.y + 1);
         }
 
+        Some(buttons_height)
+    }
+
+    fn draw_content(&self, printer: &Printer, buttons_height: usize) {
         // What do we have left?
-        let taken = Vec2::new(0, buttons_height) + self.borders.combined() +
-                    self.padding.combined();
+        let taken = Vec2::new(0, buttons_height) + self.borders.combined()
+            + self.padding.combined();
 
         let inner_size = match printer.size.checked_sub(taken) {
             Some(s) => s,
             None => return,
         };
 
-        self.content.draw(&printer.sub_printer(self.borders.top_left() +
-                                               self.padding.top_left(),
-                                               inner_size,
-                                               self.focus == Focus::Content));
-
-        printer.print_box(Vec2::new(0, 0), printer.size, false);
+        self.content.draw(&printer.sub_printer(
+            self.borders.top_left() + self.padding.top_left(),
+            inner_size,
+            self.focus == Focus::Content,
+        ));
+    }
 
+    fn draw_title(&self, printer: &Printer) {
         if !self.title.is_empty() {
             let len = self.title.width();
             if len + 4 > printer.size.x {
@@ -239,10 +352,61 @@ impl View for Dialog {
                 printer.print((x + len, 0), " ├");
             });
 
-            printer.with_color(ColorStyle::TitlePrimary,
-                               |p| p.print((x, 0), &self.title));
+            printer.with_color(
+                ColorStyle::TitlePrimary,
+                |p| p.print((x, 0), &self.title),
+            );
         }
+    }
 
+    fn check_focus_grab(&mut self, event: &Event) {
+        if let &Event::Mouse {
+            offset,
+            position,
+            event,
+        } = event
+        {
+            if !event.grabs_focus() {
+                return;
+            }
+
+            let position = match position.checked_sub(offset) {
+                None => return,
+                Some(pos) => pos,
+            };
+
+            eprintln!("Rel pos: {:?}", position);
+
+            // Now that we have a relative position, checks for buttons?
+            if let Some(i) = self.buttons.iter().position(|btn| {
+                // If position fits there...
+                position.fits_in_rect(btn.offset.get(), btn.button.size)
+            }) {
+                self.focus = Focus::Button(i);
+            } else if position.fits_in_rect((1, 1), self.content.size)
+                && self.content.take_focus(Direction::none())
+            {
+                // Or did we click the content?
+                self.focus = Focus::Content;
+            }
+        }
+    }
+}
+
+impl View for Dialog {
+    fn draw(&self, printer: &Printer) {
+        // This will be the buttons_height used by the buttons.
+        let buttons_height = match self.draw_buttons(printer) {
+            Some(height) => height,
+            None => return,
+        };
+
+        self.draw_content(printer, buttons_height);
+
+        // Print the borders
+        printer.print_box(Vec2::new(0, 0), printer.size, false);
+
+        self.draw_title(printer);
     }
 
     fn required_size(&mut self, req: Vec2) -> Vec2 {
@@ -256,7 +420,7 @@ impl View for Dialog {
         buttons_size.x += self.buttons.len().saturating_sub(1);
 
         for button in &mut self.buttons {
-            let s = button.view.required_size(req);
+            let s = button.button.view.required_size(req);
             buttons_size.x += s.x;
             buttons_size.y = max(buttons_size.y, s.y + 1);
         }
@@ -274,10 +438,11 @@ impl View for Dialog {
 
         // On the Y axis, we add buttons and content.
         // On the X axis, we take the max.
-        let mut inner_size = Vec2::new(max(content_size.x, buttons_size.x),
-                                       content_size.y + buttons_size.y) +
-                             self.padding.combined() +
-                             self.borders.combined();
+        let mut inner_size = Vec2::new(
+            max(content_size.x, buttons_size.x),
+            content_size.y + buttons_size.y,
+        ) + self.padding.combined()
+            + self.borders.combined();
 
         if !self.title.is_empty() {
             // If we have a title, we have to fit it too!
@@ -296,85 +461,29 @@ impl View for Dialog {
         // Buttons are kings, we give them everything they want.
         let mut buttons_height = 0;
         for button in self.buttons.iter_mut().rev() {
-            let size = button.required_size(size);
+            let size = button.button.required_size(size);
             buttons_height = max(buttons_height, size.y + 1);
-            button.layout(size);
+            button.button.layout(size);
         }
 
         // Poor content will have to make do with what's left.
         if buttons_height > size.y {
             buttons_height = size.y;
         }
-        self.content.layout(size.saturating_sub((0, buttons_height)));
+        self.content
+            .layout(size.saturating_sub((0, buttons_height)));
     }
 
     fn on_event(&mut self, event: Event) -> EventResult {
+        // First: some mouse events can instantly change the focus.
+        self.check_focus_grab(&event);
+
         match self.focus {
             // If we are on the content, we can only go down.
-            Focus::Content => {
-                match self.content.on_event(event.clone()) {
-                    EventResult::Ignored if !self.buttons.is_empty() => {
-                        match event {
-                            Event::Key(Key::Down) |
-                            Event::Key(Key::Tab) |
-                            Event::Shift(Key::Tab) => {
-                                // Default to leftmost button when going down.
-                                self.focus = Focus::Button(0);
-                                EventResult::Consumed(None)
-                            }
-                            _ => EventResult::Ignored,
-                        }
-                    }
-                    res => res,
-                }
-            }
+            // TODO: Careful if/when we add buttons elsewhere on the dialog!
+            Focus::Content => self.on_event_content(event),
             // If we are on a button, we have more choice
-            Focus::Button(i) => {
-                match self.buttons[i].on_event(event.clone()) {
-                    EventResult::Ignored => {
-                        match event {
-                            // Up goes back to the content
-                            Event::Key(Key::Up) => {
-                                if self.content.take_focus(Direction::down()) {
-                                    self.focus = Focus::Content;
-                                    EventResult::Consumed(None)
-                                } else {
-                                    EventResult::Ignored
-                                }
-                            }
-                            Event::Shift(Key::Tab) => {
-                                if self.content.take_focus(Direction::back()) {
-                                    self.focus = Focus::Content;
-                                    EventResult::Consumed(None)
-                                } else {
-                                    EventResult::Ignored
-                                }
-                            }
-                            Event::Key(Key::Tab) => {
-                                if self.content
-                                       .take_focus(Direction::front()) {
-                                    self.focus = Focus::Content;
-                                    EventResult::Consumed(None)
-                                } else {
-                                    EventResult::Ignored
-                                }
-                            }
-                            // Left and Right move to other buttons
-                            Event::Key(Key::Right) if i + 1 <
-                                                      self.buttons.len() => {
-                                self.focus = Focus::Button(i + 1);
-                                EventResult::Consumed(None)
-                            }
-                            Event::Key(Key::Left) if i > 0 => {
-                                self.focus = Focus::Button(i - 1);
-                                EventResult::Consumed(None)
-                            }
-                            _ => EventResult::Ignored,
-                        }
-                    }
-                    res => res,
-                }
-            }
+            Focus::Button(i) => self.on_event_button(event, i),
         }
     }
 
@@ -392,8 +501,9 @@ impl View for Dialog {
         }
     }
 
-    fn call_on_any<'a>(&mut self, selector: &Selector,
-                       callback: Box<FnMut(&mut Any) + 'a>) {
+    fn call_on_any<'a>(
+        &mut self, selector: &Selector, callback: Box<FnMut(&mut Any) + 'a>
+    ) {
         self.content.call_on_any(selector, callback);
     }