3 files changed, 50 insertions, 1 deletions

diff --git a/sway/desktop/wl_shell.c b/sway/desktop/wl_shell.c
index 9641b911..b2e026ef 100644
--- a/sway/desktop/wl_shell.c
+++ b/sway/desktop/wl_shell.c
@@ -37,6 +37,14 @@ static void set_size(struct sway_view *view, int width, int height) {
         wlr_wl_shell_surface_configure(view->wlr_wl_shell_surface, 0, width, height);
 }
 
+static void set_position(struct sway_view *view, double ox, double oy) {
+        if (!assert_wl_shell(view)) {
+                return;
+        }
+        view->swayc->x = ox;
+        view->swayc->y = oy;
+}
+
 static void handle_commit(struct wl_listener *listener, void *data) {
         struct sway_wl_shell_surface *sway_surface =
                 wl_container_of(listener, sway_surface, commit);
@@ -87,6 +95,7 @@ void handle_wl_shell_surface(struct wl_listener *listener, void *data) {
         sway_view->type = SWAY_WL_SHELL_VIEW;
         sway_view->iface.get_prop = get_prop;
         sway_view->iface.set_size = set_size;
+        sway_view->iface.set_position = set_position;
         sway_view->wlr_wl_shell_surface = shell_surface;
         sway_view->sway_wl_shell_surface = sway_surface;
         sway_view->surface = shell_surface->surface;
diff --git a/sway/desktop/xdg_shell_v6.c b/sway/desktop/xdg_shell_v6.c
index 2545d1a6..37e39f37 100644
--- a/sway/desktop/xdg_shell_v6.c
+++ b/sway/desktop/xdg_shell_v6.c
@@ -37,6 +37,14 @@ static void set_size(struct sway_view *view, int width, int height) {
         wlr_xdg_toplevel_v6_set_size(view->wlr_xdg_surface_v6, width, height);
 }
 
+static void set_position(struct sway_view *view, double ox, double oy) {
+        if (!assert_xdg(view)) {
+                return;
+        }
+        view->swayc->x = ox;
+        view->swayc->y = oy;
+}
+
 static void handle_commit(struct wl_listener *listener, void *data) {
         struct sway_xdg_surface_v6 *sway_surface =
                 wl_container_of(listener, sway_surface, commit);
@@ -87,6 +95,7 @@ void handle_xdg_shell_v6_surface(struct wl_listener *listener, void *data) {
         sway_view->type = SWAY_XDG_SHELL_V6_VIEW;
         sway_view->iface.get_prop = get_prop;
         sway_view->iface.set_size = set_size;
+        sway_view->iface.set_position = set_position;
         sway_view->wlr_xdg_surface_v6 = xdg_surface;
         sway_view->sway_xdg_surface_v6 = sway_surface;
         sway_view->surface = xdg_surface->surface;
diff --git a/sway/desktop/xwayland.c b/sway/desktop/xwayland.c
index 04ec118d..266a5869 100644
--- a/sway/desktop/xwayland.c
+++ b/sway/desktop/xwayland.c
@@ -3,14 +3,17 @@
 #include <stdlib.h>
 #include <wayland-server.h>
 #include <wlr/xwayland.h>
+#include <wlr/types/wlr_output_layout.h>
+#include <wlr/types/wlr_output.h>
 #include "sway/container.h"
 #include "sway/layout.h"
 #include "sway/server.h"
 #include "sway/view.h"
+#include "sway/output.h"
 #include "log.h"
 
  static bool assert_xwayland(struct sway_view *view) {
-         return sway_assert(view->type == SWAY_XWAYLAND_VIEW,
+         return sway_assert(view->type == SWAY_XWAYLAND_VIEW && view->wlr_xwayland_surface,
                  "Expected xwayland view!");
  }
 
@@ -40,6 +43,33 @@ static void set_size(struct sway_view *view, int width, int height) {
                 width, height);
 }
 
+static void set_position(struct sway_view *view, double ox, double oy) {
+        if (!assert_xwayland(view)) {
+                return;
+        }
+        swayc_t *output = swayc_parent_by_type(view->swayc, C_OUTPUT);
+        if (!sway_assert(output, "view must be within tree to set position")) {
+                return;
+        }
+        swayc_t *root = swayc_parent_by_type(output, C_ROOT);
+        if (!sway_assert(root, "output must be within tree to set position")) {
+                return;
+        }
+        struct wlr_output_layout *layout = root->output_layout;
+        struct wlr_output_layout_output *loutput =
+                wlr_output_layout_get(layout, output->sway_output->wlr_output);
+        if (!sway_assert(loutput, "output must be within layout to set position")) {
+                return;
+        }
+
+        view->swayc->x = ox;
+        view->swayc->y = oy;
+
+        wlr_xwayland_surface_configure(view->wlr_xwayland_surface,
+                ox + loutput->x, oy + loutput->y,
+                view->width, view->height);
+}
+
 static void handle_commit(struct wl_listener *listener, void *data) {
         struct sway_xwayland_surface *sway_surface =
                 wl_container_of(listener, sway_surface, commit);
@@ -102,6 +132,7 @@ void handle_xwayland_surface(struct wl_listener *listener, void *data) {
         sway_view->type = SWAY_XWAYLAND_VIEW;
         sway_view->iface.get_prop = get_prop;
         sway_view->iface.set_size = set_size;
+        sway_view->iface.set_position = set_position;
         sway_view->wlr_xwayland_surface = xsurface;
         sway_view->sway_xwayland_surface = sway_surface;
         // TODO remove from the tree when the surface goes away (unmapped)