mjdev-desktop/compositor/native/shim.c

/*
 * Copyright (c) Milan Jurkulák 2026.
 *  Contact:
 *  e: mimoccc@gmail.com
 *  e: mj@mjdev.org
 *  w: https://mjdev.org
 *
 * wlroots 0.18 based compositor core, exposed to Kotlin/Native
 * through the flat API in shim.h. Policy (layers, focus order,
 * IPC) lives on the Kotlin side.
 */

#define _POSIX_C_SOURCE 200809L

#include <assert.h>
#include <signal.h>
#include <stdbool.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/wait.h>
#include <time.h>
#include <unistd.h>

#include <wayland-server-core.h>
#include <wlr/backend.h>
#include <wlr/backend/session.h>
#include <wlr/render/allocator.h>
#include <wlr/render/wlr_renderer.h>
#include <wlr/types/wlr_compositor.h>
#include <wlr/types/wlr_subcompositor.h>
#include <wlr/types/wlr_cursor.h>
#include <wlr/types/wlr_data_device.h>
#include <wlr/types/wlr_input_device.h>
#include <wlr/types/wlr_keyboard.h>
#include <wlr/types/wlr_output.h>
#include <wlr/types/wlr_output_layout.h>
#include <wlr/types/wlr_pointer.h>
#include <wlr/types/wlr_scene.h>
#include <wlr/types/wlr_seat.h>
#include <wlr/types/wlr_xcursor_manager.h>
#include <wlr/types/wlr_xdg_decoration_v1.h>
#include <wlr/types/wlr_xdg_shell.h>
#include <wlr/util/log.h>
#include <wlr/xwayland.h>
#include <xkbcommon/xkbcommon.h>

#include "shim.h"

enum mjc_cursor_mode {
    MJC_CURSOR_PASSTHROUGH,
    MJC_CURSOR_MOVE,
    MJC_CURSOR_RESIZE,
};

struct mjc_server {
    struct wl_display *display;
    struct wl_event_loop *loop;
    struct wlr_backend *backend;
    struct wlr_session *session;
    struct wlr_renderer *renderer;
    struct wlr_allocator *allocator;
    struct wlr_compositor *compositor;
    struct wlr_scene *scene;
    struct wlr_scene_output_layout *scene_layout;
    struct wlr_output_layout *output_layout;
    struct wlr_scene_tree *layers[5];

    struct wlr_xdg_shell *xdg_shell;
    struct wlr_xdg_decoration_manager_v1 *xdg_decoration_mgr;
    struct wlr_xwayland *xwayland;

    struct wlr_cursor *cursor;
    struct wlr_xcursor_manager *cursor_mgr;
    struct wlr_seat *seat;

    struct wl_list views;     /* mjc_view.link */
    struct wl_list keyboards; /* mjc_keyboard.link */
    struct wl_list fd_sources;/* mjc_fd_source.link */

    struct mjc_view *focused_view;
    mjc_callbacks cbs;
    void *ud;
    const char *socket;
    uint64_t next_view_id;

    /* nested-output size (16:9 default); used when the backend has no mode list
     * (wayland/headless) and to lock the window against host-driven resizes */
    int output_w, output_h;
    /* throttle timestamp (msec) for the pointer callback */
    uint32_t last_pointer_ms;

    enum mjc_cursor_mode cursor_mode;
    struct mjc_view *grabbed_view;
    double grab_x, grab_y;
    struct wlr_box grab_geobox;
    uint32_t resize_edges;

    struct wl_event_source *sigchld_source;

    /* pids spawned via mjc_spawn; sigchld must reap only these,
     * other children (Xwayland) belong to wlroots */
    pid_t spawned_pids[64];
    int spawned_count;

    struct wl_listener new_output;
    struct wl_listener new_xdg_toplevel;
    struct wl_listener new_xdg_popup;
    struct wl_listener new_toplevel_decoration;
    struct wl_listener new_xwayland_surface;
    struct wl_listener xwayland_ready;
    struct wl_listener cursor_motion;
    struct wl_listener cursor_motion_absolute;
    struct wl_listener cursor_button;
    struct wl_listener cursor_axis;
    struct wl_listener cursor_frame;
    struct wl_listener new_input;
    struct wl_listener request_cursor;
    struct wl_listener request_set_selection;
};

struct mjc_view {
    struct wl_list link;
    struct mjc_server *server;
    uint64_t id;
    bool is_xwayland;
    bool mapped;
    bool minimized;
    bool maximized;
    /* re-center once the client commits its post-unmaximize size */
    bool pending_center;
    bool focusable;
    mjc_layer layer;
    struct wlr_scene_tree *scene_tree;
    /* floating geometry remembered while maximized (x/y are scene coords) */
    struct wlr_box saved_geo;

    /* xdg shell */
    struct wlr_xdg_toplevel *xdg_toplevel;

    /* xwayland */
    struct wlr_xwayland_surface *xsurface;

    struct wl_listener map;
    struct wl_listener unmap;
    struct wl_listener commit;
    struct wl_listener destroy;
    struct wl_listener request_move;
    struct wl_listener request_resize;
    struct wl_listener request_maximize;
    struct wl_listener request_minimize;
    struct wl_listener request_fullscreen;
    struct wl_listener set_title;
    struct wl_listener set_app_id; /* set_class for xwayland */
    struct wl_listener associate;
    struct wl_listener dissociate;
    struct wl_listener request_configure;
    struct wl_listener set_geometry;
};

struct mjc_popup {
    struct wlr_xdg_popup *xdg_popup;
    struct wl_listener commit;
    struct wl_listener destroy;
};

struct mjc_decoration {
    struct wlr_xdg_toplevel_decoration_v1 *decoration;
    struct wl_listener request_mode;
    struct wl_listener destroy;
};

struct mjc_keyboard {
    struct wl_list link;
    struct mjc_server *server;
    struct wlr_keyboard *wlr_keyboard;
    struct wl_listener modifiers;
    struct wl_listener key;
    struct wl_listener destroy;
};

struct mjc_output {
    struct wl_list link;
    struct mjc_server *server;
    struct wlr_output *wlr_output;
    struct wl_listener frame;
    struct wl_listener request_state;
    struct wl_listener destroy;
};

struct mjc_fd_source {
    struct wl_list link;
    struct mjc_server *server;
    int fd;
    struct wl_event_source *source;
};

/* ------------------------------------------------------------------ */
/* helpers                                                             */
/* ------------------------------------------------------------------ */

static struct wlr_surface *view_surface(struct mjc_view *view) {
    if (view->is_xwayland) {
        return view->xsurface != NULL ? view->xsurface->surface : NULL;
    }
    return view->xdg_toplevel->base->surface;
}

static void emit_focus_change(struct mjc_server *server) {
    if (server->cbs.focus_change != NULL) {
        server->cbs.focus_change(server->ud, server->focused_view);
    }
}

static void view_deactivate(struct mjc_view *view) {
    if (view->is_xwayland) {
        if (view->xsurface != NULL) {
            wlr_xwayland_surface_activate(view->xsurface, false);
        }
    } else {
        wlr_xdg_toplevel_set_activated(view->xdg_toplevel, false);
    }
}

void mjc_view_focus(struct mjc_view *view) {
    if (view == NULL || !view->focusable || view->minimized) {
        return;
    }
    struct mjc_server *server = view->server;
    struct wlr_seat *seat = server->seat;
    if (server->focused_view == view) {
        return;
    }
    if (server->focused_view != NULL) {
        view_deactivate(server->focused_view);
    }
    if (view->scene_tree != NULL) {
        wlr_scene_node_raise_to_top(&view->scene_tree->node);
    }
    if (view->is_xwayland) {
        wlr_xwayland_surface_activate(view->xsurface, true);
        wlr_xwayland_surface_restack(view->xsurface, NULL, XCB_STACK_MODE_ABOVE);
    } else {
        wlr_xdg_toplevel_set_activated(view->xdg_toplevel, true);
    }
    struct wlr_surface *surface = view_surface(view);
    struct wlr_keyboard *keyboard = wlr_seat_get_keyboard(seat);
    if (surface != NULL) {
        if (keyboard != NULL) {
            wlr_seat_keyboard_notify_enter(seat, surface,
                keyboard->keycodes, keyboard->num_keycodes, &keyboard->modifiers);
        } else {
            wlr_seat_keyboard_notify_enter(seat, surface, NULL, 0, NULL);
        }
    }
    server->focused_view = view;
    emit_focus_change(server);
}

static void clear_focus_if(struct mjc_server *server, struct mjc_view *view) {
    if (server->focused_view == view) {
        server->focused_view = NULL;
        emit_focus_change(server);
    }
    if (server->grabbed_view == view) {
        server->grabbed_view = NULL;
        server->cursor_mode = MJC_CURSOR_PASSTHROUGH;
    }
}

static struct mjc_view *desktop_view_at(struct mjc_server *server,
        double lx, double ly, struct wlr_surface **surface,
        double *sx, double *sy) {
    struct wlr_scene_node *node =
        wlr_scene_node_at(&server->scene->tree.node, lx, ly, sx, sy);
    if (node == NULL || node->type != WLR_SCENE_NODE_BUFFER) {
        return NULL;
    }
    struct wlr_scene_buffer *scene_buffer = wlr_scene_buffer_from_node(node);
    struct wlr_scene_surface *scene_surface =
        wlr_scene_surface_try_from_buffer(scene_buffer);
    if (scene_surface == NULL) {
        return NULL;
    }
    *surface = scene_surface->surface;
    struct wlr_scene_tree *tree = node->parent;
    while (tree != NULL && tree->node.data == NULL) {
        tree = tree->node.parent;
    }
    return tree != NULL ? tree->node.data : NULL;
}

/* ------------------------------------------------------------------ */
/* outputs                                                             */
/* ------------------------------------------------------------------ */

static void output_frame(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_output *output = wl_container_of(listener, output, frame);
    struct wlr_scene_output *scene_output = wlr_scene_get_scene_output(
        output->server->scene, output->wlr_output);
    if (scene_output == NULL) {
        return;
    }
    wlr_scene_output_commit(scene_output, NULL);
    struct timespec now;
    clock_gettime(CLOCK_MONOTONIC, &now);
    wlr_scene_output_send_frame_done(scene_output, &now);
}

static void output_request_state(struct wl_listener *listener, void *data) {
    struct mjc_output *output = wl_container_of(listener, output, request_state);
    (void) data; /* ignore the host-requested size: keep the nested output locked
                  * to its fixed 16:9 mode so the window cannot be resized
                  * (moving and closing still work via the host decoration) */
    struct wlr_output_state state;
    wlr_output_state_init(&state);
    wlr_output_state_set_custom_mode(&state,
        output->server->output_w, output->server->output_h, 0);
    wlr_output_commit_state(output->wlr_output, &state);
    wlr_output_state_finish(&state);
}

static void output_destroy(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_output *output = wl_container_of(listener, output, destroy);
    struct mjc_server *server = output->server;
    wl_list_remove(&output->frame.link);
    wl_list_remove(&output->request_state.link);
    wl_list_remove(&output->destroy.link);
    wl_list_remove(&output->link);
    free(output);
    /* nested backends (x11/wayland) have no session; closing the host window
     * destroys the output, so quit the compositor and let the run action finish.
     * a real (drm) session keeps running when an output is unplugged. */
    if (server->session == NULL) {
        wl_display_terminate(server->display);
    }
}

static void server_new_output(struct wl_listener *listener, void *data) {
    struct mjc_server *server = wl_container_of(listener, server, new_output);
    struct wlr_output *wlr_output = data;

    wlr_output_init_render(wlr_output, server->allocator, server->renderer);

    struct wlr_output_state state;
    wlr_output_state_init(&state);
    wlr_output_state_set_enabled(&state, true);
    struct wlr_output_mode *mode = wlr_output_preferred_mode(wlr_output);
    if (mode != NULL) {
        wlr_output_state_set_mode(&state, mode);
    } else {
        /* wayland / headless backends expose no mode list, so preferred_mode is
         * NULL; without a mode the output has no dimensions and renders black.
         * pin it to the configured 16:9 size. */
        wlr_output_state_set_custom_mode(&state,
            server->output_w, server->output_h, 0);
    }
    wlr_output_commit_state(wlr_output, &state);
    wlr_output_state_finish(&state);

    struct mjc_output *output = calloc(1, sizeof(*output));
    output->wlr_output = wlr_output;
    output->server = server;

    output->frame.notify = output_frame;
    wl_signal_add(&wlr_output->events.frame, &output->frame);
    output->request_state.notify = output_request_state;
    wl_signal_add(&wlr_output->events.request_state, &output->request_state);
    output->destroy.notify = output_destroy;
    wl_signal_add(&wlr_output->events.destroy, &output->destroy);

    wl_list_init(&output->link);

    struct wlr_output_layout_output *l_output =
        wlr_output_layout_add_auto(server->output_layout, wlr_output);
    struct wlr_scene_output *scene_output =
        wlr_scene_output_create(server->scene, wlr_output);
    wlr_scene_output_layout_add_output(server->scene_layout, l_output, scene_output);
}

/* ------------------------------------------------------------------ */
/* keyboard & input                                                    */
/* ------------------------------------------------------------------ */

static void keyboard_handle_modifiers(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_keyboard *keyboard = wl_container_of(listener, keyboard, modifiers);
    wlr_seat_set_keyboard(keyboard->server->seat, keyboard->wlr_keyboard);
    wlr_seat_keyboard_notify_modifiers(keyboard->server->seat,
        &keyboard->wlr_keyboard->modifiers);
}

static void keyboard_handle_key(struct wl_listener *listener, void *data) {
    struct mjc_keyboard *keyboard = wl_container_of(listener, keyboard, key);
    struct mjc_server *server = keyboard->server;
    struct wlr_keyboard_key_event *event = data;

    uint32_t keycode = event->keycode + 8;
    const xkb_keysym_t *syms;
    int nsyms = xkb_state_key_get_syms(
        keyboard->wlr_keyboard->xkb_state, keycode, &syms);
    uint32_t modifiers = wlr_keyboard_get_modifiers(keyboard->wlr_keyboard);
    bool pressed = event->state == WL_KEYBOARD_KEY_STATE_PRESSED;

    bool handled = false;
    for (int i = 0; i < nsyms; i++) {
        /* VT switching always works */
        if (pressed && syms[i] >= XKB_KEY_XF86Switch_VT_1 &&
                syms[i] <= XKB_KEY_XF86Switch_VT_12) {
            if (server->session != NULL) {
                wlr_session_change_vt(server->session,
                    syms[i] - XKB_KEY_XF86Switch_VT_1 + 1);
            }
            handled = true;
            continue;
        }
        if (server->cbs.key != NULL &&
                server->cbs.key(server->ud, syms[i], modifiers, pressed)) {
            handled = true;
        }
    }

    if (!handled) {
        wlr_seat_set_keyboard(server->seat, keyboard->wlr_keyboard);
        wlr_seat_keyboard_notify_key(server->seat,
            event->time_msec, event->keycode, event->state);
    }
}

static void keyboard_handle_destroy(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_keyboard *keyboard = wl_container_of(listener, keyboard, destroy);
    wl_list_remove(&keyboard->modifiers.link);
    wl_list_remove(&keyboard->key.link);
    wl_list_remove(&keyboard->destroy.link);
    wl_list_remove(&keyboard->link);
    free(keyboard);
}

static void server_new_keyboard(struct mjc_server *server,
        struct wlr_input_device *device) {
    struct wlr_keyboard *wlr_keyboard = wlr_keyboard_from_input_device(device);

    struct mjc_keyboard *keyboard = calloc(1, sizeof(*keyboard));
    keyboard->server = server;
    keyboard->wlr_keyboard = wlr_keyboard;

    struct xkb_context *context = xkb_context_new(XKB_CONTEXT_NO_FLAGS);
    struct xkb_keymap *keymap = xkb_keymap_new_from_names(context, NULL,
        XKB_KEYMAP_COMPILE_NO_FLAGS);
    wlr_keyboard_set_keymap(wlr_keyboard, keymap);
    xkb_keymap_unref(keymap);
    xkb_context_unref(context);
    wlr_keyboard_set_repeat_info(wlr_keyboard, 25, 600);

    keyboard->modifiers.notify = keyboard_handle_modifiers;
    wl_signal_add(&wlr_keyboard->events.modifiers, &keyboard->modifiers);
    keyboard->key.notify = keyboard_handle_key;
    wl_signal_add(&wlr_keyboard->events.key, &keyboard->key);
    keyboard->destroy.notify = keyboard_handle_destroy;
    wl_signal_add(&device->events.destroy, &keyboard->destroy);

    wlr_seat_set_keyboard(server->seat, wlr_keyboard);
    wl_list_insert(&server->keyboards, &keyboard->link);
}

static void server_new_pointer(struct mjc_server *server,
        struct wlr_input_device *device) {
    wlr_cursor_attach_input_device(server->cursor, device);
}

static void server_new_input(struct wl_listener *listener, void *data) {
    struct mjc_server *server = wl_container_of(listener, server, new_input);
    struct wlr_input_device *device = data;
    switch (device->type) {
    case WLR_INPUT_DEVICE_KEYBOARD:
        server_new_keyboard(server, device);
        break;
    case WLR_INPUT_DEVICE_POINTER:
        server_new_pointer(server, device);
        break;
    default:
        break;
    }
    uint32_t caps = WL_SEAT_CAPABILITY_POINTER;
    if (!wl_list_empty(&server->keyboards)) {
        caps |= WL_SEAT_CAPABILITY_KEYBOARD;
    }
    wlr_seat_set_capabilities(server->seat, caps);
}

static void seat_request_cursor(struct wl_listener *listener, void *data) {
    struct mjc_server *server = wl_container_of(listener, server, request_cursor);
    struct wlr_seat_pointer_request_set_cursor_event *event = data;
    struct wlr_seat_client *focused_client =
        server->seat->pointer_state.focused_client;
    if (focused_client == event->seat_client) {
        wlr_cursor_set_surface(server->cursor, event->surface,
            event->hotspot_x, event->hotspot_y);
    }
}

static void seat_request_set_selection(struct wl_listener *listener, void *data) {
    struct mjc_server *server =
        wl_container_of(listener, server, request_set_selection);
    struct wlr_seat_request_set_selection_event *event = data;
    wlr_seat_set_selection(server->seat, event->source, event->serial);
}

/* ------------------------------------------------------------------ */
/* cursor                                                              */
/* ------------------------------------------------------------------ */

static void view_current_geometry(struct mjc_view *view, struct wlr_box *box);

static void process_cursor_move(struct mjc_server *server) {
    struct mjc_view *view = server->grabbed_view;
    if (view == NULL || view->scene_tree == NULL) {
        return;
    }
    int x = (int)(server->cursor->x - server->grab_x);
    int y = (int)(server->cursor->y - server->grab_y);
    wlr_scene_node_set_position(&view->scene_tree->node, x, y);
    if (view->is_xwayland && view->xsurface != NULL) {
        wlr_xwayland_surface_configure(view->xsurface,
            (int16_t) x, (int16_t) y,
            (uint16_t) view->xsurface->width, (uint16_t) view->xsurface->height);
    }
}

static void process_cursor_resize(struct mjc_server *server) {
    struct mjc_view *view = server->grabbed_view;
    if (view == NULL || view->scene_tree == NULL) {
        return;
    }
    double border_x = server->cursor->x - server->grab_x;
    double border_y = server->cursor->y - server->grab_y;
    int new_left = server->grab_geobox.x;
    int new_right = server->grab_geobox.x + server->grab_geobox.width;
    int new_top = server->grab_geobox.y;
    int new_bottom = server->grab_geobox.y + server->grab_geobox.height;

    if (server->resize_edges & WLR_EDGE_TOP) {
        new_top = (int) border_y;
        if (new_top >= new_bottom) {
            new_top = new_bottom - 1;
        }
    } else if (server->resize_edges & WLR_EDGE_BOTTOM) {
        new_bottom = (int) border_y;
        if (new_bottom <= new_top) {
            new_bottom = new_top + 1;
        }
    }
    if (server->resize_edges & WLR_EDGE_LEFT) {
        new_left = (int) border_x;
        if (new_left >= new_right) {
            new_left = new_right - 1;
        }
    } else if (server->resize_edges & WLR_EDGE_RIGHT) {
        new_right = (int) border_x;
        if (new_right <= new_left) {
            new_right = new_left + 1;
        }
    }

    int new_width = new_right - new_left;
    int new_height = new_bottom - new_top;

    if (view->is_xwayland && view->xsurface != NULL) {
        wlr_scene_node_set_position(&view->scene_tree->node, new_left, new_top);
        wlr_xwayland_surface_configure(view->xsurface,
            (int16_t) new_left, (int16_t) new_top,
            (uint16_t) new_width, (uint16_t) new_height);
    } else {
        struct wlr_box geo_box;
        wlr_xdg_surface_get_geometry(view->xdg_toplevel->base, &geo_box);
        wlr_scene_node_set_position(&view->scene_tree->node,
            new_left - geo_box.x, new_top - geo_box.y);
        wlr_xdg_toplevel_set_size(view->xdg_toplevel, new_width, new_height);
    }
}

static void process_cursor_motion(struct mjc_server *server, uint32_t time) {
    if (server->cursor_mode == MJC_CURSOR_MOVE) {
        process_cursor_move(server);
        return;
    } else if (server->cursor_mode == MJC_CURSOR_RESIZE) {
        process_cursor_resize(server);
        return;
    }

    double sx, sy;
    struct wlr_seat *seat = server->seat;
    struct wlr_surface *surface = NULL;
    desktop_view_at(server, server->cursor->x, server->cursor->y,
        &surface, &sx, &sy);
    if (surface == NULL) {
        wlr_cursor_set_xcursor(server->cursor, server->cursor_mgr, "default");
    }
    if (surface != NULL) {
        wlr_seat_pointer_notify_enter(seat, surface, sx, sy);
        wlr_seat_pointer_notify_motion(seat, time, sx, sy);
    } else {
        wlr_seat_pointer_clear_focus(seat);
    }

    // broadcast the pointer position to ipc clients (throttled), so the shell can do
    // edge detection without the AWT global pointer which is unreliable under nested XWayland
    if (server->cbs.pointer != NULL && (time - server->last_pointer_ms) >= 30u) {
        server->last_pointer_ms = time;
        server->cbs.pointer(server->ud,
            (int) server->cursor->x, (int) server->cursor->y);
    }
}

static void server_cursor_motion(struct wl_listener *listener, void *data) {
    struct mjc_server *server = wl_container_of(listener, server, cursor_motion);
    struct wlr_pointer_motion_event *event = data;
    wlr_cursor_move(server->cursor, &event->pointer->base,
        event->delta_x, event->delta_y);
    process_cursor_motion(server, event->time_msec);
}

static void server_cursor_motion_absolute(struct wl_listener *listener, void *data) {
    struct mjc_server *server =
        wl_container_of(listener, server, cursor_motion_absolute);
    struct wlr_pointer_motion_absolute_event *event = data;
    wlr_cursor_warp_absolute(server->cursor, &event->pointer->base,
        event->x, event->y);
    process_cursor_motion(server, event->time_msec);
}

static void server_cursor_button(struct wl_listener *listener, void *data) {
    struct mjc_server *server = wl_container_of(listener, server, cursor_button);
    struct wlr_pointer_button_event *event = data;
    wlr_seat_pointer_notify_button(server->seat,
        event->time_msec, event->button, event->state);

    if (event->state == WL_POINTER_BUTTON_STATE_RELEASED) {
        server->cursor_mode = MJC_CURSOR_PASSTHROUGH;
        server->grabbed_view = NULL;
    } else {
        double sx, sy;
        struct wlr_surface *surface = NULL;
        struct mjc_view *view = desktop_view_at(server,
            server->cursor->x, server->cursor->y, &surface, &sx, &sy);
        if (view != NULL && view->focusable) {
            mjc_view_focus(view);
        }
    }
}

static void server_cursor_axis(struct wl_listener *listener, void *data) {
    struct mjc_server *server = wl_container_of(listener, server, cursor_axis);
    struct wlr_pointer_axis_event *event = data;
    wlr_seat_pointer_notify_axis(server->seat,
        event->time_msec, event->orientation, event->delta,
        event->delta_discrete, event->source, event->relative_direction);
}

static void server_cursor_frame(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_server *server = wl_container_of(listener, server, cursor_frame);
    wlr_seat_pointer_notify_frame(server->seat);
}

/* ------------------------------------------------------------------ */
/* interactive move/resize                                             */
/* ------------------------------------------------------------------ */

static void view_current_geometry(struct mjc_view *view, struct wlr_box *box) {
    if (view->is_xwayland && view->xsurface != NULL) {
        box->x = view->scene_tree != NULL ? view->scene_tree->node.x : view->xsurface->x;
        box->y = view->scene_tree != NULL ? view->scene_tree->node.y : view->xsurface->y;
        box->width = view->xsurface->width;
        box->height = view->xsurface->height;
    } else {
        struct wlr_box geo;
        wlr_xdg_surface_get_geometry(view->xdg_toplevel->base, &geo);
        box->x = view->scene_tree->node.x + geo.x;
        box->y = view->scene_tree->node.y + geo.y;
        box->width = geo.width;
        box->height = geo.height;
    }
}

static void begin_interactive(struct mjc_view *view,
        enum mjc_cursor_mode mode, uint32_t edges) {
    struct mjc_server *server = view->server;
    if (view_surface(view) !=
            server->seat->pointer_state.focused_surface) {
        /* only react to requests from the focused client */
        return;
    }
    server->grabbed_view = view;
    server->cursor_mode = mode;

    struct wlr_box geo_box;
    view_current_geometry(view, &geo_box);

    if (mode == MJC_CURSOR_MOVE) {
        server->grab_x = server->cursor->x - view->scene_tree->node.x;
        server->grab_y = server->cursor->y - view->scene_tree->node.y;
    } else {
        double border_x = geo_box.x +
            ((edges & WLR_EDGE_RIGHT) ? geo_box.width : 0);
        double border_y = geo_box.y +
            ((edges & WLR_EDGE_BOTTOM) ? geo_box.height : 0);
        server->grab_x = server->cursor->x - border_x;
        server->grab_y = server->cursor->y - border_y;
        server->grab_geobox = geo_box;
        server->resize_edges = edges;
    }
}

/* ------------------------------------------------------------------ */
/* view common                                                         */
/* ------------------------------------------------------------------ */

/* places a freshly mapped floating view in the middle of the output */
static void view_center(struct mjc_view *view) {
    struct mjc_server *server = view->server;
    if (view->scene_tree == NULL || view->maximized) {
        return;
    }
    struct wlr_box out;
    wlr_output_layout_get_box(server->output_layout, NULL, &out);
    struct wlr_box geo;
    view_current_geometry(view, &geo);
    if (out.width <= 0 || geo.width <= 0) {
        return;
    }
    int x = out.x + (out.width - geo.width) / 2;
    int y = out.y + (out.height - geo.height) / 2;
    if (x < out.x) x = out.x;
    if (y < out.y) y = out.y;
    if (view->is_xwayland) {
        mjc_view_set_position(view, x, y);
    } else {
        struct wlr_box off;
        wlr_xdg_surface_get_geometry(view->xdg_toplevel->base, &off);
        wlr_scene_node_set_position(&view->scene_tree->node, x - off.x, y - off.y);
    }
}

static void view_emit_new(struct mjc_view *view) {
    if (view->server->cbs.view_new != NULL) {
        view->server->cbs.view_new(view->server->ud, view, view->is_xwayland);
    }
}

static void view_emit_map(struct mjc_view *view) {
    if (view->server->cbs.view_map != NULL) {
        view->server->cbs.view_map(view->server->ud, view);
    }
}

static void view_emit_unmap(struct mjc_view *view) {
    if (view->server->cbs.view_unmap != NULL) {
        view->server->cbs.view_unmap(view->server->ud, view);
    }
}

static void view_emit_destroy(struct mjc_view *view) {
    if (view->server->cbs.view_destroy != NULL) {
        view->server->cbs.view_destroy(view->server->ud, view);
    }
}

static struct mjc_view *view_create(struct mjc_server *server, bool is_xwayland) {
    struct mjc_view *view = calloc(1, sizeof(*view));
    view->server = server;
    view->id = ++server->next_view_id;
    view->is_xwayland = is_xwayland;
    view->focusable = true;
    view->layer = MJC_LAYER_NORMAL;
    wl_list_insert(&server->views, &view->link);
    return view;
}

static void view_free(struct mjc_view *view) {
    clear_focus_if(view->server, view);
    view_emit_destroy(view);
    wl_list_remove(&view->link);
    free(view);
}

/* ------------------------------------------------------------------ */
/* xdg shell views                                                     */
/* ------------------------------------------------------------------ */

static void xdg_toplevel_map(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, map);
    view->mapped = true;
    view_center(view);
    view_emit_map(view);
    mjc_view_focus(view);
}

static void xdg_toplevel_unmap(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, unmap);
    view->mapped = false;
    clear_focus_if(view->server, view);
    view_emit_unmap(view);
}

static void xdg_toplevel_commit(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, commit);
    if (view->xdg_toplevel->base->initial_commit) {
        if (view->xdg_toplevel->requested.maximized) {
            mjc_view_set_maximized(view, true);
        } else {
            /* 0x0 lets the client pick its own (or remembered) size */
            wlr_xdg_toplevel_set_size(view->xdg_toplevel, 0, 0);
        }
        return;
    }
    if (view->pending_center && view->mapped && !view->maximized &&
            !view->xdg_toplevel->current.maximized) {
        struct wlr_box geo;
        wlr_xdg_surface_get_geometry(view->xdg_toplevel->base, &geo);
        if (geo.width > 0) {
            view->pending_center = false;
            if (view->saved_geo.width > 0 && view->scene_tree != NULL) {
                /* saved coords are visible coords, geo.x/y the offsets */
                wlr_scene_node_set_position(&view->scene_tree->node,
                    view->saved_geo.x - geo.x, view->saved_geo.y - geo.y);
            } else {
                view_center(view);
            }
        }
    }
}

static void xdg_toplevel_destroy(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, destroy);
    wl_list_remove(&view->map.link);
    wl_list_remove(&view->unmap.link);
    wl_list_remove(&view->commit.link);
    wl_list_remove(&view->destroy.link);
    wl_list_remove(&view->request_move.link);
    wl_list_remove(&view->request_resize.link);
    wl_list_remove(&view->request_maximize.link);
    wl_list_remove(&view->request_minimize.link);
    wl_list_remove(&view->request_fullscreen.link);
    wl_list_remove(&view->set_title.link);
    wl_list_remove(&view->set_app_id.link);
    view_free(view);
}

static void xdg_toplevel_request_move(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, request_move);
    begin_interactive(view, MJC_CURSOR_MOVE, 0);
}

static void xdg_toplevel_request_resize(struct wl_listener *listener, void *data) {
    struct wlr_xdg_toplevel_resize_event *event = data;
    struct mjc_view *view = wl_container_of(listener, view, request_resize);
    begin_interactive(view, MJC_CURSOR_RESIZE, event->edges);
}

static void xdg_toplevel_request_maximize(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, request_maximize);
    if (view->xdg_toplevel->base->initialized) {
        mjc_view_set_maximized(view, view->xdg_toplevel->requested.maximized);
        wlr_xdg_surface_schedule_configure(view->xdg_toplevel->base);
    }
}

static void xdg_toplevel_request_minimize(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, request_minimize);
    mjc_view_set_minimized(view, true);
}

static void xdg_toplevel_request_fullscreen(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, request_fullscreen);
    if (view->xdg_toplevel->base->initialized) {
        wlr_xdg_surface_schedule_configure(view->xdg_toplevel->base);
    }
}

static void xdg_toplevel_set_title(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, set_title);
    if (view->server->cbs.view_title != NULL) {
        view->server->cbs.view_title(view->server->ud, view,
            view->xdg_toplevel->title);
    }
}

static void xdg_toplevel_set_app_id(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, set_app_id);
    if (view->server->cbs.view_app_id != NULL) {
        view->server->cbs.view_app_id(view->server->ud, view,
            view->xdg_toplevel->app_id);
    }
}

static void server_new_xdg_toplevel(struct wl_listener *listener, void *data) {
    struct mjc_server *server =
        wl_container_of(listener, server, new_xdg_toplevel);
    struct wlr_xdg_toplevel *xdg_toplevel = data;

    struct mjc_view *view = view_create(server, false);
    view->xdg_toplevel = xdg_toplevel;
    view->scene_tree = wlr_scene_xdg_surface_create(
        server->layers[MJC_LAYER_NORMAL], xdg_toplevel->base);
    view->scene_tree->node.data = view;
    xdg_toplevel->base->data = view->scene_tree;

    view->map.notify = xdg_toplevel_map;
    wl_signal_add(&xdg_toplevel->base->surface->events.map, &view->map);
    view->unmap.notify = xdg_toplevel_unmap;
    wl_signal_add(&xdg_toplevel->base->surface->events.unmap, &view->unmap);
    view->commit.notify = xdg_toplevel_commit;
    wl_signal_add(&xdg_toplevel->base->surface->events.commit, &view->commit);
    view->destroy.notify = xdg_toplevel_destroy;
    wl_signal_add(&xdg_toplevel->events.destroy, &view->destroy);

    view->request_move.notify = xdg_toplevel_request_move;
    wl_signal_add(&xdg_toplevel->events.request_move, &view->request_move);
    view->request_resize.notify = xdg_toplevel_request_resize;
    wl_signal_add(&xdg_toplevel->events.request_resize, &view->request_resize);
    view->request_maximize.notify = xdg_toplevel_request_maximize;
    wl_signal_add(&xdg_toplevel->events.request_maximize, &view->request_maximize);
    view->request_minimize.notify = xdg_toplevel_request_minimize;
    wl_signal_add(&xdg_toplevel->events.request_minimize, &view->request_minimize);
    view->request_fullscreen.notify = xdg_toplevel_request_fullscreen;
    wl_signal_add(&xdg_toplevel->events.request_fullscreen, &view->request_fullscreen);
    view->set_title.notify = xdg_toplevel_set_title;
    wl_signal_add(&xdg_toplevel->events.set_title, &view->set_title);
    view->set_app_id.notify = xdg_toplevel_set_app_id;
    wl_signal_add(&xdg_toplevel->events.set_app_id, &view->set_app_id);

    view_emit_new(view);
}

/* popups */

static void xdg_popup_commit(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_popup *popup = wl_container_of(listener, popup, commit);
    if (popup->xdg_popup->base->initial_commit) {
        wlr_xdg_surface_schedule_configure(popup->xdg_popup->base);
    }
}

static void xdg_popup_destroy(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_popup *popup = wl_container_of(listener, popup, destroy);
    wl_list_remove(&popup->commit.link);
    wl_list_remove(&popup->destroy.link);
    free(popup);
}

static void server_new_xdg_popup(struct wl_listener *listener, void *data) {
    (void) listener;
    struct wlr_xdg_popup *xdg_popup = data;

    struct wlr_xdg_surface *parent =
        wlr_xdg_surface_try_from_wlr_surface(xdg_popup->parent);
    if (parent == NULL || parent->data == NULL) {
        return;
    }
    struct wlr_scene_tree *parent_tree = parent->data;
    xdg_popup->base->data =
        wlr_scene_xdg_surface_create(parent_tree, xdg_popup->base);

    struct mjc_popup *popup = calloc(1, sizeof(*popup));
    popup->xdg_popup = xdg_popup;
    popup->commit.notify = xdg_popup_commit;
    wl_signal_add(&xdg_popup->base->surface->events.commit, &popup->commit);
    popup->destroy.notify = xdg_popup_destroy;
    wl_signal_add(&xdg_popup->events.destroy, &popup->destroy);
}

/* decorations - clients draw their own buttons (CSD), the compositor
 * renders no frames, so server side mode is never granted */

static void decoration_request_mode(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_decoration *deco = wl_container_of(listener, deco, request_mode);
    if (deco->decoration->toplevel->base->initialized) {
        wlr_xdg_toplevel_decoration_v1_set_mode(deco->decoration,
            WLR_XDG_TOPLEVEL_DECORATION_V1_MODE_CLIENT_SIDE);
    }
}

static void decoration_destroy(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_decoration *deco = wl_container_of(listener, deco, destroy);
    wl_list_remove(&deco->request_mode.link);
    wl_list_remove(&deco->destroy.link);
    free(deco);
}

static void server_new_toplevel_decoration(struct wl_listener *listener,
        void *data) {
    (void) listener;
    struct wlr_xdg_toplevel_decoration_v1 *decoration = data;
    struct mjc_decoration *deco = calloc(1, sizeof(*deco));
    deco->decoration = decoration;
    deco->request_mode.notify = decoration_request_mode;
    wl_signal_add(&decoration->events.request_mode, &deco->request_mode);
    deco->destroy.notify = decoration_destroy;
    wl_signal_add(&decoration->events.destroy, &deco->destroy);
    decoration_request_mode(&deco->request_mode, NULL);
}

/* ------------------------------------------------------------------ */
/* xwayland views                                                      */
/* ------------------------------------------------------------------ */

static void xwayland_surface_map(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, map);
    struct mjc_server *server = view->server;
    struct wlr_xwayland_surface *xsurface = view->xsurface;

    mjc_layer layer = xsurface->override_redirect
        ? MJC_LAYER_OVERLAY : view->layer;
    view->scene_tree = wlr_scene_tree_create(server->layers[layer]);
    view->scene_tree->node.data = view;
    wlr_scene_surface_create(view->scene_tree, xsurface->surface);
    wlr_scene_node_set_position(&view->scene_tree->node,
        xsurface->x, xsurface->y);

    view->mapped = true;
    /* X clients that did not place themselves land at 0,0 - center those */
    if (!xsurface->override_redirect && xsurface->x == 0 && xsurface->y == 0) {
        view_center(view);
    }
    view_emit_map(view);
    if (!xsurface->override_redirect) {
        mjc_view_focus(view);
    }
}

static void xwayland_surface_unmap(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, unmap);
    view->mapped = false;
    clear_focus_if(view->server, view);
    /* emit while the scene node still exists so the policy side can
     * read the last window geometry */
    view_emit_unmap(view);
    if (view->scene_tree != NULL) {
        wlr_scene_node_destroy(&view->scene_tree->node);
        view->scene_tree = NULL;
    }
}

static void xwayland_surface_associate(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, associate);
    view->map.notify = xwayland_surface_map;
    wl_signal_add(&view->xsurface->surface->events.map, &view->map);
    view->unmap.notify = xwayland_surface_unmap;
    wl_signal_add(&view->xsurface->surface->events.unmap, &view->unmap);
}

static void xwayland_surface_dissociate(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, dissociate);
    wl_list_remove(&view->map.link);
    wl_list_remove(&view->unmap.link);
}

static void xwayland_surface_destroy(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, destroy);
    wl_list_remove(&view->associate.link);
    wl_list_remove(&view->dissociate.link);
    wl_list_remove(&view->destroy.link);
    wl_list_remove(&view->request_configure.link);
    wl_list_remove(&view->request_move.link);
    wl_list_remove(&view->request_resize.link);
    wl_list_remove(&view->request_minimize.link);
    wl_list_remove(&view->request_maximize.link);
    wl_list_remove(&view->set_title.link);
    wl_list_remove(&view->set_app_id.link);
    wl_list_remove(&view->set_geometry.link);
    if (view->scene_tree != NULL) {
        wlr_scene_node_destroy(&view->scene_tree->node);
        view->scene_tree = NULL;
    }
    view_free(view);
}

static void xwayland_surface_request_configure(struct wl_listener *listener,
        void *data) {
    struct mjc_view *view = wl_container_of(listener, view, request_configure);
    struct wlr_xwayland_surface_configure_event *event = data;
    wlr_xwayland_surface_configure(view->xsurface,
        event->x, event->y, event->width, event->height);
    if (view->scene_tree != NULL) {
        wlr_scene_node_set_position(&view->scene_tree->node, event->x, event->y);
    }
}

static void xwayland_surface_set_geometry(struct wl_listener *listener,
        void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, set_geometry);
    if (view->scene_tree != NULL && view->xsurface != NULL) {
        wlr_scene_node_set_position(&view->scene_tree->node,
            view->xsurface->x, view->xsurface->y);
    }
}

static void xwayland_surface_request_move(struct wl_listener *listener,
        void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, request_move);
    begin_interactive(view, MJC_CURSOR_MOVE, 0);
}

static void xwayland_surface_request_resize(struct wl_listener *listener,
        void *data) {
    struct mjc_view *view = wl_container_of(listener, view, request_resize);
    struct wlr_xwayland_resize_event *event = data;
    begin_interactive(view, MJC_CURSOR_RESIZE, event->edges);
}

static void xwayland_surface_request_minimize(struct wl_listener *listener,
        void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, request_minimize);
    mjc_view_set_minimized(view, true);
}

static void xwayland_surface_request_maximize(struct wl_listener *listener,
        void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, request_maximize);
    struct wlr_xwayland_surface *xsurface = view->xsurface;
    if (xsurface == NULL) {
        return;
    }
    /* the client announces the state it wants via _NET_WM_STATE */
    mjc_view_set_maximized(view,
        xsurface->maximized_vert && xsurface->maximized_horz);
}

static void xwayland_surface_set_title(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, set_title);
    if (view->server->cbs.view_title != NULL) {
        view->server->cbs.view_title(view->server->ud, view,
            view->xsurface->title);
    }
}

static void xwayland_surface_set_class(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_view *view = wl_container_of(listener, view, set_app_id);
    if (view->server->cbs.view_app_id != NULL) {
        view->server->cbs.view_app_id(view->server->ud, view,
            view->xsurface->class);
    }
}

static void server_new_xwayland_surface(struct wl_listener *listener, void *data) {
    struct mjc_server *server =
        wl_container_of(listener, server, new_xwayland_surface);
    struct wlr_xwayland_surface *xsurface = data;

    struct mjc_view *view = view_create(server, true);
    view->xsurface = xsurface;
    if (xsurface->override_redirect) {
        view->focusable = false;
    }
    xsurface->data = view;

    view->associate.notify = xwayland_surface_associate;
    wl_signal_add(&xsurface->events.associate, &view->associate);
    view->dissociate.notify = xwayland_surface_dissociate;
    wl_signal_add(&xsurface->events.dissociate, &view->dissociate);
    view->destroy.notify = xwayland_surface_destroy;
    wl_signal_add(&xsurface->events.destroy, &view->destroy);
    view->request_configure.notify = xwayland_surface_request_configure;
    wl_signal_add(&xsurface->events.request_configure, &view->request_configure);
    view->request_move.notify = xwayland_surface_request_move;
    wl_signal_add(&xsurface->events.request_move, &view->request_move);
    view->request_resize.notify = xwayland_surface_request_resize;
    wl_signal_add(&xsurface->events.request_resize, &view->request_resize);
    view->request_minimize.notify = xwayland_surface_request_minimize;
    wl_signal_add(&xsurface->events.request_minimize, &view->request_minimize);
    view->request_maximize.notify = xwayland_surface_request_maximize;
    wl_signal_add(&xsurface->events.request_maximize, &view->request_maximize);
    view->set_title.notify = xwayland_surface_set_title;
    wl_signal_add(&xsurface->events.set_title, &view->set_title);
    view->set_app_id.notify = xwayland_surface_set_class;
    wl_signal_add(&xsurface->events.set_class, &view->set_app_id);
    view->set_geometry.notify = xwayland_surface_set_geometry;
    wl_signal_add(&xsurface->events.set_geometry, &view->set_geometry);

    view_emit_new(view);
}

static void server_xwayland_ready(struct wl_listener *listener, void *data) {
    (void) data;
    struct mjc_server *server = wl_container_of(listener, server, xwayland_ready);
    wlr_xwayland_set_seat(server->xwayland, server->seat);
}

/* ------------------------------------------------------------------ */
/* sigchld                                                             */
/* ------------------------------------------------------------------ */

static int handle_sigchld(int signal_number, void *data) {
    (void) signal_number;
    struct mjc_server *server = data;
    for (int i = 0; i < server->spawned_count;) {
        int status;
        pid_t pid = waitpid(server->spawned_pids[i], &status, WNOHANG);
        if (pid > 0) {
            server->spawned_pids[i] =
                server->spawned_pids[--server->spawned_count];
            if (server->cbs.child_exit != NULL) {
                server->cbs.child_exit(server->ud, (int) pid, status);
            }
        } else {
            i++;
        }
    }
    return 0;
}

/* ------------------------------------------------------------------ */
/* public API                                                          */
/* ------------------------------------------------------------------ */

mjc_server *mjc_create(void) {
    struct mjc_server *server = calloc(1, sizeof(*server));
    wl_list_init(&server->views);
    wl_list_init(&server->keyboards);
    wl_list_init(&server->fd_sources);
    return server;
}

bool mjc_start(mjc_server *server, const mjc_callbacks *callbacks, void *userdata) {
    server->cbs = *callbacks;
    server->ud = userdata;

    wlr_log_init(WLR_INFO, NULL);

    /* nested-output size, 16:9 default; override with MJDEVC_OUTPUT=WIDTHxHEIGHT */
    server->output_w = 1280;
    server->output_h = 720;
    const char *out_env = getenv("MJDEVC_OUTPUT");
    if (out_env != NULL) {
        int w = 0, h = 0;
        if (sscanf(out_env, "%dx%d", &w, &h) == 2 && w > 0 && h > 0) {
            server->output_w = w;
            server->output_h = h;
        }
    }

    server->display = wl_display_create();
    server->loop = wl_display_get_event_loop(server->display);
    server->backend = wlr_backend_autocreate(server->loop, &server->session);
    if (server->backend == NULL) {
        fprintf(stderr, "mjdevc: failed to create wlr_backend\n");
        return false;
    }

    server->renderer = wlr_renderer_autocreate(server->backend);
    if (server->renderer == NULL) {
        fprintf(stderr, "mjdevc: failed to create wlr_renderer\n");
        return false;
    }
    wlr_renderer_init_wl_display(server->renderer, server->display);

    server->allocator = wlr_allocator_autocreate(server->backend, server->renderer);
    if (server->allocator == NULL) {
        fprintf(stderr, "mjdevc: failed to create wlr_allocator\n");
        return false;
    }

    server->compositor = wlr_compositor_create(server->display, 5, server->renderer);
    wlr_subcompositor_create(server->display);
    wlr_data_device_manager_create(server->display);

    server->output_layout = wlr_output_layout_create(server->display);
    server->new_output.notify = server_new_output;
    wl_signal_add(&server->backend->events.new_output, &server->new_output);

    server->scene = wlr_scene_create();
    server->scene_layout =
        wlr_scene_attach_output_layout(server->scene, server->output_layout);
    for (int i = 0; i < 5; i++) {
        server->layers[i] = wlr_scene_tree_create(&server->scene->tree);
    }

    server->xdg_shell = wlr_xdg_shell_create(server->display, 3);
    server->new_xdg_toplevel.notify = server_new_xdg_toplevel;
    wl_signal_add(&server->xdg_shell->events.new_toplevel, &server->new_xdg_toplevel);
    server->new_xdg_popup.notify = server_new_xdg_popup;
    wl_signal_add(&server->xdg_shell->events.new_popup, &server->new_xdg_popup);

    server->xdg_decoration_mgr = wlr_xdg_decoration_manager_v1_create(server->display);
    server->new_toplevel_decoration.notify = server_new_toplevel_decoration;
    wl_signal_add(&server->xdg_decoration_mgr->events.new_toplevel_decoration,
        &server->new_toplevel_decoration);

    server->cursor = wlr_cursor_create();
    wlr_cursor_attach_output_layout(server->cursor, server->output_layout);
    server->cursor_mgr = wlr_xcursor_manager_create(NULL, 24);
    server->cursor_mode = MJC_CURSOR_PASSTHROUGH;
    server->cursor_motion.notify = server_cursor_motion;
    wl_signal_add(&server->cursor->events.motion, &server->cursor_motion);
    server->cursor_motion_absolute.notify = server_cursor_motion_absolute;
    wl_signal_add(&server->cursor->events.motion_absolute,
        &server->cursor_motion_absolute);
    server->cursor_button.notify = server_cursor_button;
    wl_signal_add(&server->cursor->events.button, &server->cursor_button);
    server->cursor_axis.notify = server_cursor_axis;
    wl_signal_add(&server->cursor->events.axis, &server->cursor_axis);
    server->cursor_frame.notify = server_cursor_frame;
    wl_signal_add(&server->cursor->events.frame, &server->cursor_frame);

    server->new_input.notify = server_new_input;
    wl_signal_add(&server->backend->events.new_input, &server->new_input);
    server->seat = wlr_seat_create(server->display, "seat0");
    server->request_cursor.notify = seat_request_cursor;
    wl_signal_add(&server->seat->events.request_set_cursor, &server->request_cursor);
    server->request_set_selection.notify = seat_request_set_selection;
    wl_signal_add(&server->seat->events.request_set_selection,
        &server->request_set_selection);

    const char *socket = wl_display_add_socket_auto(server->display);
    if (socket == NULL) {
        fprintf(stderr, "mjdevc: failed to create wayland socket\n");
        return false;
    }
    server->socket = socket;

    server->xwayland = wlr_xwayland_create(server->display, server->compositor, true);
    if (server->xwayland != NULL) {
        server->new_xwayland_surface.notify = server_new_xwayland_surface;
        wl_signal_add(&server->xwayland->events.new_surface,
            &server->new_xwayland_surface);
        server->xwayland_ready.notify = server_xwayland_ready;
        wl_signal_add(&server->xwayland->events.ready, &server->xwayland_ready);
    }

    server->sigchld_source = wl_event_loop_add_signal(server->loop,
        SIGCHLD, handle_sigchld, server);

    if (!wlr_backend_start(server->backend)) {
        fprintf(stderr, "mjdevc: failed to start backend\n");
        return false;
    }

    setenv("WAYLAND_DISPLAY", socket, true);
    if (server->xwayland != NULL) {
        setenv("DISPLAY", server->xwayland->display_name, true);
    }

    if (server->cbs.ready != NULL) {
        server->cbs.ready(server->ud);
    }
    return true;
}

void mjc_run(mjc_server *server) {
    wl_display_run(server->display);
}

void mjc_terminate(mjc_server *server) {
    wl_display_terminate(server->display);
}

void mjc_destroy(mjc_server *server) {
    if (server->xwayland != NULL) {
        wlr_xwayland_destroy(server->xwayland);
    }
    wl_display_destroy_clients(server->display);
    wlr_scene_node_destroy(&server->scene->tree.node);
    wlr_xcursor_manager_destroy(server->cursor_mgr);
    wlr_cursor_destroy(server->cursor);
    wlr_allocator_destroy(server->allocator);
    wlr_renderer_destroy(server->renderer);
    wlr_backend_destroy(server->backend);
    wl_display_destroy(server->display);
    free(server);
}

const char *mjc_socket_name(mjc_server *server) {
    return server->socket;
}

const char *mjc_xwayland_display(mjc_server *server) {
    return server->xwayland != NULL ? server->xwayland->display_name : NULL;
}

void mjc_output_size(mjc_server *server, int *width, int *height) {
    struct wlr_box box;
    wlr_output_layout_get_box(server->output_layout, NULL, &box);
    *width = box.width;
    *height = box.height;
}

int mjc_spawn(mjc_server *server, const char *command) {
    pid_t pid = fork();
    if (pid < 0) {
        return -1;
    }
    if (pid == 0) {
        setsid();
        sigset_t set;
        sigemptyset(&set);
        sigprocmask(SIG_SETMASK, &set, NULL);
        execl("/bin/sh", "/bin/sh", "-c", command, (void *) NULL);
        _exit(127);
    }
    if (server->spawned_count <
            (int) (sizeof(server->spawned_pids) / sizeof(server->spawned_pids[0]))) {
        server->spawned_pids[server->spawned_count++] = pid;
    }
    return (int) pid;
}

/* unix socket helpers */

#include <fcntl.h>
#include <sys/socket.h>
#include <sys/un.h>

int mjc_unix_listen(const char *path, int backlog) {
    int fd = socket(AF_UNIX, SOCK_STREAM | SOCK_NONBLOCK | SOCK_CLOEXEC, 0);
    if (fd < 0) {
        return -1;
    }
    struct sockaddr_un addr = {0};
    addr.sun_family = AF_UNIX;
    if (strlen(path) >= sizeof(addr.sun_path)) {
        close(fd);
        return -1;
    }
    strncpy(addr.sun_path, path, sizeof(addr.sun_path) - 1);
    unlink(path);
    if (bind(fd, (struct sockaddr *) &addr, sizeof(addr)) != 0 ||
            listen(fd, backlog) != 0) {
        close(fd);
        return -1;
    }
    return fd;
}

int mjc_unix_accept(int server_fd) {
    int fd = accept(server_fd, NULL, NULL);
    if (fd >= 0) {
        fcntl(fd, F_SETFL, O_NONBLOCK);
    }
    return fd;
}

/* fd integration */

static int handle_fd_event(int fd, uint32_t mask, void *data) {
    struct mjc_fd_source *source = data;
    if (source->server->cbs.fd_event != NULL) {
        return source->server->cbs.fd_event(source->server->ud, fd, mask);
    }
    return 0;
}

bool mjc_loop_add_fd(mjc_server *server, int fd, uint32_t mask) {
    struct mjc_fd_source *source = calloc(1, sizeof(*source));
    source->server = server;
    source->fd = fd;
    source->source = wl_event_loop_add_fd(server->loop, fd, mask,
        handle_fd_event, source);
    if (source->source == NULL) {
        free(source);
        return false;
    }
    wl_list_insert(&server->fd_sources, &source->link);
    return true;
}

void mjc_loop_remove_fd(mjc_server *server, int fd) {
    struct mjc_fd_source *source, *tmp;
    wl_list_for_each_safe(source, tmp, &server->fd_sources, link) {
        if (source->fd == fd) {
            wl_event_source_remove(source->source);
            wl_list_remove(&source->link);
            free(source);
            return;
        }
    }
}

/* views */

uint64_t mjc_view_id(mjc_view *view) {
    return view->id;
}

int mjc_view_pid(mjc_view *view) {
    if (view->is_xwayland) {
        return view->xsurface != NULL ? (int) view->xsurface->pid : 0;
    }
    pid_t pid = 0;
    struct wlr_surface *surface = view_surface(view);
    if (surface != NULL) {
        struct wl_client *client = wl_resource_get_client(surface->resource);
        wl_client_get_credentials(client, &pid, NULL, NULL);
    }
    return (int) pid;
}

const char *mjc_view_title(mjc_view *view) {
    if (view->is_xwayland) {
        return view->xsurface != NULL ? view->xsurface->title : NULL;
    }
    return view->xdg_toplevel->title;
}

const char *mjc_view_app_id(mjc_view *view) {
    if (view->is_xwayland) {
        return view->xsurface != NULL ? view->xsurface->class : NULL;
    }
    return view->xdg_toplevel->app_id;
}

bool mjc_view_is_xwayland(mjc_view *view) {
    return view->is_xwayland;
}

bool mjc_view_is_mapped(mjc_view *view) {
    return view->mapped;
}

bool mjc_view_is_minimized(mjc_view *view) {
    return view->minimized;
}

bool mjc_view_is_maximized(mjc_view *view) {
    return view->maximized;
}

bool mjc_view_is_focused(mjc_view *view) {
    return view->server->focused_view == view;
}

void mjc_view_get_geometry(mjc_view *view, int *x, int *y, int *width, int *height) {
    struct wlr_box box = {0};
    if (view->scene_tree != NULL || !view->is_xwayland) {
        view_current_geometry(view, &box);
    }
    *x = box.x;
    *y = box.y;
    *width = box.width;
    *height = box.height;
}

void mjc_view_close(mjc_view *view) {
    if (view->is_xwayland) {
        if (view->xsurface != NULL) {
            wlr_xwayland_surface_close(view->xsurface);
        }
    } else {
        wlr_xdg_toplevel_send_close(view->xdg_toplevel);
    }
}

void mjc_view_set_minimized(mjc_view *view, bool minimized) {
    if (view->minimized == minimized) {
        return;
    }
    view->minimized = minimized;
    if (view->scene_tree != NULL) {
        wlr_scene_node_set_enabled(&view->scene_tree->node, !minimized);
    }
    if (view->is_xwayland && view->xsurface != NULL) {
        wlr_xwayland_surface_set_minimized(view->xsurface, minimized);
    }
    if (minimized) {
        clear_focus_if(view->server, view);
    } else {
        mjc_view_focus(view);
    }
}

void mjc_view_set_maximized(mjc_view *view, bool maximized) {
    struct mjc_server *server = view->server;
    if (view->maximized == maximized) {
        return;
    }
    struct wlr_box box;
    wlr_output_layout_get_box(server->output_layout, NULL, &box);
    if (maximized && view->mapped && view->scene_tree != NULL) {
        /* remember the floating geometry (visible coords) for restore */
        view_current_geometry(view, &view->saved_geo);
    }
    view->maximized = maximized;
    if (view->is_xwayland) {
        if (view->xsurface == NULL) {
            return;
        }
        wlr_xwayland_surface_set_maximized(view->xsurface, maximized);
        if (maximized) {
            wlr_xwayland_surface_configure(view->xsurface,
                (int16_t) box.x, (int16_t) box.y,
                (uint16_t) box.width, (uint16_t) box.height);
            if (view->scene_tree != NULL) {
                wlr_scene_node_set_position(&view->scene_tree->node, box.x, box.y);
            }
        } else if (view->saved_geo.width > 0) {
            wlr_xwayland_surface_configure(view->xsurface,
                (int16_t) view->saved_geo.x, (int16_t) view->saved_geo.y,
                (uint16_t) view->saved_geo.width, (uint16_t) view->saved_geo.height);
            if (view->scene_tree != NULL) {
                wlr_scene_node_set_position(&view->scene_tree->node,
                    view->saved_geo.x, view->saved_geo.y);
            }
        }
    } else {
        wlr_xdg_toplevel_set_maximized(view->xdg_toplevel, maximized);
        if (maximized) {
            wlr_xdg_toplevel_set_size(view->xdg_toplevel, box.width, box.height);
            if (view->scene_tree != NULL) {
                wlr_scene_node_set_position(&view->scene_tree->node, box.x, box.y);
            }
        } else {
            /* 0x0 when no remembered size - the client picks its own;
             * placement happens on the commit that carries the new size,
             * when the decoration offsets are known again */
            wlr_xdg_toplevel_set_size(view->xdg_toplevel,
                view->saved_geo.width, view->saved_geo.height);
            view->pending_center = true;
        }
    }
}

void mjc_view_set_layer(mjc_view *view, mjc_layer layer) {
    view->layer = layer;
    if (view->scene_tree != NULL) {
        wlr_scene_node_reparent(&view->scene_tree->node,
            view->server->layers[layer]);
    }
}

void mjc_view_set_focusable(mjc_view *view, bool focusable) {
    view->focusable = focusable;
    if (!focusable) {
        clear_focus_if(view->server, view);
    }
}

void mjc_view_set_position(mjc_view *view, int x, int y) {
    if (view->scene_tree != NULL) {
        wlr_scene_node_set_position(&view->scene_tree->node, x, y);
    }
    if (view->is_xwayland && view->xsurface != NULL) {
        wlr_xwayland_surface_configure(view->xsurface,
            (int16_t) x, (int16_t) y,
            (uint16_t) view->xsurface->width, (uint16_t) view->xsurface->height);
    }
}

void mjc_view_set_geometry(mjc_view *view, int x, int y, int width, int height) {
    if (width <= 0 || height <= 0) {
        return;
    }
    if (view->maximized) {
        /* keep it as the restore target for the next unmaximize */
        view->saved_geo.x = x;
        view->saved_geo.y = y;
        view->saved_geo.width = width;
        view->saved_geo.height = height;
        return;
    }
    if (view->is_xwayland) {
        if (view->xsurface != NULL) {
            wlr_xwayland_surface_configure(view->xsurface,
                (int16_t) x, (int16_t) y, (uint16_t) width, (uint16_t) height);
        }
        if (view->scene_tree != NULL) {
            wlr_scene_node_set_position(&view->scene_tree->node, x, y);
        }
    } else {
        wlr_xdg_toplevel_set_size(view->xdg_toplevel, width, height);
        if (view->scene_tree != NULL) {
            struct wlr_box off;
            wlr_xdg_surface_get_geometry(view->xdg_toplevel->base, &off);
            wlr_scene_node_set_position(&view->scene_tree->node,
                x - off.x, y - off.y);
        }
    }
}