Improve handling of layer surface output assignment

2 files changed, 20 insertions, 12 deletions

diff --git a/src/output.zig b/src/output.zig
index 74a0c2a..c88f726 100644
--- a/src/output.zig
+++ b/src/output.zig
@@ -58,6 +58,7 @@ pub const Output = struct {
 
         self.root = root;
         self.wlr_output = wlr_output;
+        wlr_output.data = self;
 
         for (self.layers) |*layer| {
             layer.* = std.TailQueue(LayerSurface).init();
diff --git a/src/server.zig b/src/server.zig
index abb6b13..0bb2118 100644
--- a/src/server.zig
+++ b/src/server.zig
@@ -172,20 +172,27 @@ pub const Server = struct {
             },
         );
 
-        // TODO: this is insufficent for multi output support
-        if (server.root.outputs.first) |node| {
-            const output = &node.data;
-            if (wlr_layer_surface.output == null) {
+        // If the new layer surface does not have an output assigned to it, use the
+        // first output or close the surface if none are available.
+        if (wlr_layer_surface.output == null) {
+            if (server.root.outputs.first) |node| {
+                const output = &node.data;
+                Log.Debug.log(
+                    "New layer surface had null output, assigning it to output {}",
+                    .{output.wlr_output.name},
+                );
                 wlr_layer_surface.output = output.wlr_output;
+            } else {
+                Log.Error.log(
+                    "No output available for layer surface '{}'",
+                    .{wlr_layer_surface.namespace},
+                );
+                c.wlr_layer_surface_v1_close(wlr_layer_surface);
+                return;
             }
-
-            output.addLayerSurface(wlr_layer_surface) catch unreachable;
-        } else {
-            Log.Error.log(
-                "No output available for layer surface '{}' autoassign",
-                .{wlr_layer_surface.namespace},
-            );
-            c.wlr_layer_surface_v1_close(wlr_layer_surface);
         }
+
+        const output = @ptrCast(*Output, @alignCast(@alignOf(*Output), wlr_layer_surface.output.*.data));
+        output.addLayerSurface(wlr_layer_surface) catch unreachable;
     }
 };