Fix various toggle-visible issues

- Track "start" command internally as a special mode.
- Ensure that DoubleMemPool is in good state if draw bails.
- Ensure that Launcher does not terminate daemonized session.
- Ensure that size changes cause full force redraw.

src/app.rs

@@ -21,6 +21,7 @@ use crate::widget::{DrawContext, Widget};
 use crate::cmd::Cmd;
 use crate::doublemempool::DoubleMemPool;
 
+#[derive(Debug)]
 pub enum OutputMode {
     Active,
     All,
@@ -123,12 +124,18 @@ impl AppInner {
 
         self.configured_surfaces = Arc::new(Mutex::new(0));
 
-        if self.visible {
+        if !self.visible {
+            self.surfaces = Vec::new();
+            self.shell_surfaces = Vec::new();
+            return;
+        }
+
         match self.output_mode {
             OutputMode::Active => {
                 if self.shell_surfaces.len() > 0 {
                     return;
                 }
+
                 let (surface, shell_surface) = AppInner::add_shell_surface(
                     &compositor,
                     &shell,
@@ -160,10 +167,6 @@ impl AppInner {
             }
         }
         self.draw_tx.send(Cmd::ForceDraw).unwrap();
-        } else {
-            self.surfaces = Vec::new();
-            self.shell_surfaces = Vec::new();
-        }
     }
 
     fn add_output(&mut self, id: u32, output: wl_output::WlOutput) {
@@ -210,7 +213,7 @@ pub struct App {
 }
 
 impl App {
-    pub fn redraw(&mut self, mut force: bool) -> Result<(), ::std::io::Error> {
+    pub fn redraw(&mut self, force: bool) -> Result<(), ::std::io::Error> {
         let widget = match self.widget {
             Some(ref mut widget) => widget,
             None => return Ok(()),
@@ -219,7 +222,9 @@ impl App {
         let inner = self.inner.lock().unwrap();
         let time = Local::now();
 
-        if inner.shell_surfaces.len() != *inner.configured_surfaces.lock().unwrap() {
+        if !inner.visible ||
+            inner.shell_surfaces.len() != *inner.configured_surfaces.lock().unwrap() ||
+            inner.surfaces.len() == 0 {
             // Not ready yet
             return Ok(());
         }
@@ -231,6 +236,7 @@ impl App {
 
         let size = widget.size();
         let size_changed = self.last_dim != size;
+        let force = force | size_changed;
 
         // resize the pool if relevant
         pool.resize((4 * size.0 * size.1) as usize)
@@ -239,8 +245,8 @@ impl App {
         let mut buf = Buffer::new(mmap, size);
 
         // Copy old damage
-        if let Some(d) = &self.last_damage {
+        let force = match (force, &self.last_damage) {
-            if !size_changed {
+            (false, Some(d)) => {
                 let lastmmap = last.mmap();
                 let last = Buffer::new(lastmmap, size);
 
@@ -250,12 +256,10 @@ impl App {
                 for d in d {
                     last.copy_to(&mut buf, d.clone());
                 }
-            } else {
+                false
-                force = true;
-            }
-        } else {
-            force = true;
             }
+            _ => true
+        };
 
         if force {
             buf.memset(&self.bg);
@@ -272,8 +276,9 @@ impl App {
 
         mmap.flush().unwrap();
 
-        if !size_changed && !report.full_damage && report.damage.len() == 0 {
+        if !force && !report.full_damage && report.damage.len() == 0 {
             // Nothing to do
+            self.pools.never_mind();
             return Ok(());
         }
 
@@ -319,6 +324,20 @@ impl App {
         inner.outputs_changed();
     }
 
+    pub fn hide(&mut self) {
+        let mut inner = self.inner.lock().unwrap();
+        inner.visible = false;
+        self.last_dim = (0, 0);
+        inner.outputs_changed();
+    }
+
+    pub fn show(&mut self) {
+        let mut inner = self.inner.lock().unwrap();
+        inner.visible = true;
+        self.last_dim = (0, 0);
+        inner.outputs_changed();
+    }
+
     pub fn cmd_queue(&self) -> Arc<Mutex<VecDeque<Cmd>>> {
         self.cmd_queue.clone()
     }
@@ -470,7 +489,6 @@ impl App {
             },
         ));
 
-        inner.lock().unwrap().outputs_changed();
         event_queue.sync_roundtrip().unwrap();
 
         //

src/doublemempool.rs

@@ -19,7 +19,7 @@ impl DoubleMemPool {
     pub fn pool(&mut self) -> Option<(&mut MemPool, &mut MemPool)> {
         let switch = self.switch;
         self.switch = !self.switch;
-        let (last, cur) = if switch {
+        let (last, cur) = if !switch {
             (&mut self.pool2, &mut self.pool1)
         } else {
             (&mut self.pool1, &mut self.pool2)
@@ -31,4 +31,8 @@ impl DoubleMemPool {
             Some((last, cur))
         }
     }
+
+    pub fn never_mind(&mut self) {
+        self.switch = !self.switch;
+    }
 }

src/main.rs

@@ -25,6 +25,7 @@ use cmd::Cmd;
 
 enum Mode {
     Start,
+    Daemonize,
     StartOrKill,
     ToggleVisible,
     PrintConfig(bool),
@@ -63,7 +64,7 @@ fn main() {
     let mode = match args.len() {
         1 => Mode::Start,
         2 => match args[1].as_str() {
-            "start" => Mode::Start,
+            "start" => Mode::Daemonize,
             "start-or-kill" => Mode::StartOrKill,
             "toggle-visible" => Mode::ToggleVisible,
             "print-config" => Mode::PrintConfig(!is_yaml),
@@ -80,6 +81,8 @@ fn main() {
         }
     };
 
+    let mut daemon = false;
+
     match mode {
         Mode::ToggleVisible => {
             if let Ok(mut socket) = UnixStream::connect(socket_path.clone()) {
@@ -101,6 +104,13 @@ fn main() {
                 std::process::exit(1);
             };
         }
+        Mode::Daemonize => {
+            if let Ok(_) = UnixStream::connect(socket_path.clone()) {
+                eprintln!("wldash is already running");
+                std::process::exit(1);
+            };
+            daemon = true;
+        }
         Mode::PrintConfig(json) => {
             if json {
                 println!("{}", serde_json::to_string_pretty(&config).unwrap());
@@ -133,6 +143,11 @@ fn main() {
     });
 
     let mut app = App::new(tx_draw, output_mode, background, scale);
+    if daemon {
+        app.hide();
+    } else {
+        app.show();
+    }
     let widget = mod_rx.recv().unwrap();
     app.set_widget(widget).unwrap();
 
@@ -224,11 +239,17 @@ fn main() {
                 }
                 Cmd::ToggleVisible => {
                     app.toggle_visible();
+                    app.flush_display();
                 }
                 Cmd::Exit => {
+                    if daemon {
+                        app.hide();
+                        app.flush_display();
+                    } else {
                         let _ = std::fs::remove_file(socket_path);
                         return;
                     }
+                }
             },
             None => {
                 app.flush_display();