Fix `swaylock -f`

This commit makes swaylock poll the registry the first time in its
fork in order for sway to recognize it and grant it lock privileges

1 files changed, 29 insertions, 25 deletions

diff --git a/swaylock/main.c b/swaylock/main.c
index c2615951..1ed28c2b 100644
--- a/swaylock/main.c
+++ b/swaylock/main.c
@@ -343,6 +343,7 @@ cairo_surface_t *load_image(char *image_path) {
 
 int main(int argc, char **argv) {
         const char *scaling_mode_str = "fit", *socket_path = NULL;
+        char *image_path = NULL;
         int i;
         void *images = NULL;
         config = init_config();
@@ -400,8 +401,6 @@ int main(int argc, char **argv) {
                 "  For more information see `man swaylock`\n";
 
 
-        registry = registry_poll();
-
         int c;
         while (1) {
                 int option_index = 0;
@@ -418,29 +417,7 @@ int main(int argc, char **argv) {
                 }
                 case 'i':
                 {
-                        char *image_path = strchr(optarg, ':');
-                        if (image_path == NULL) {
-                                if (render_data.num_images == 0) {
-                                        // Provided image without output
-                                        render_data.image = load_image(optarg);
-                                        render_data.num_images = -1;
-                                } else {
-                                        sway_log(L_ERROR, "output must be defined for all --images or no --images");
-                                        exit(EXIT_FAILURE);
-                                }
-                        } else {
-                                // Provided image for all outputs
-                                if (render_data.num_images == 0) {
-                                        images = calloc(registry->outputs->length, sizeof(char*) * 2);
-                                } else if (render_data.num_images == -1) {
-                                        sway_log(L_ERROR, "output must be defined for all --images or no --images");
-                                        exit(EXIT_FAILURE);
-                                }
-
-                                image_path[0] = '\0';
-                                ((char**) images)[render_data.num_images * 2] = optarg;
-                                ((char**) images)[render_data.num_images++ * 2 + 1] = ++image_path;
-                        }
+                        image_path = optarg;
                         break;
                 }
                 case 't':
@@ -519,6 +496,33 @@ int main(int argc, char **argv) {
                         exit(EXIT_FAILURE);
                 }
         }
+        registry = registry_poll();
+
+        if (image_path) {
+                char *path = strchr(image_path, ':');
+                if (path == NULL) {
+                        if (render_data.num_images == 0) {
+                                // Provided image without output
+                                render_data.image = load_image(image_path);
+                                render_data.num_images = -1;
+                        } else {
+                                sway_log(L_ERROR, "output must be defined for all --images or no --images");
+                                exit(EXIT_FAILURE);
+                        }
+                } else {
+                        // Provided image for all outputs
+                        if (render_data.num_images == 0) {
+                                images = calloc(registry->outputs->length, sizeof(char*) * 2);
+                        } else if (render_data.num_images == -1) {
+                                sway_log(L_ERROR, "output must be defined for all --images or no --images");
+                                exit(EXIT_FAILURE);
+                        }
+
+                        path[0] = '\0';
+                        ((char**) images)[render_data.num_images * 2] = image_path;
+                        ((char**) images)[render_data.num_images++ * 2 + 1] = ++path;
+                }
+        }
 
         render_data.scaling_mode = SCALING_MODE_STRETCH;
         if (strcmp(scaling_mode_str, "stretch") == 0) {