Implement compositor support for swaylock

This makes swaylock more or less work.

1 files changed, 11 insertions, 2 deletions

diff --git a/swaylock/main.c b/swaylock/main.c
index 4073da99..82b88731 100644
--- a/swaylock/main.c
+++ b/swaylock/main.c
@@ -108,7 +108,6 @@ int main(int argc, char **argv) {
                 if (!window) {
                         sway_abort("Failed to create surfaces.");
                 }
-                lock_set_lock_surface(registry->swaylock, output->output, window->surface);
                 list_add(surfaces, window);
         }
 
@@ -217,7 +216,17 @@ int main(int argc, char **argv) {
 
         cairo_surface_destroy(image);
 
-        while (wl_display_dispatch(registry->display) != -1);
+        bool locked = false;
+        while (wl_display_dispatch(registry->display) != -1) {
+                if (!locked) {
+                        for (i = 0; i < registry->outputs->length; ++i) {
+                                struct output_state *output = registry->outputs->items[i];
+                                struct window *window = surfaces->items[i];
+                                lock_set_lock_surface(registry->swaylock, output->output, window->surface);
+                        }
+                        locked = true;
+                }
+        }
 
         for (i = 0; i < surfaces->length; ++i) {
                 struct window *window = surfaces->items[i];