#include <wayland-server-core.h>
#include <wlr/util/log.h>

#include "types.h"

bool absinthe_toplevel_is_x11(struct absinthe_toplevel *toplevel)
{
#ifdef XWAYLAND
    return toplevel->type == ABSINTHE_TOPLEVEL_X11;
#endif
    return false;
}

bool absinthe_toplevel_is_unmanaged(struct absinthe_toplevel *toplevel)
{
#ifdef XWAYLAND
    if (absinthe_toplevel_is_x11(toplevel))
        return toplevel->toplevel.x11->override_redirect;
#endif
    return false;
}

struct absinthe_toplevel *absinthe_toplevel_at(struct absinthe_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) {
        return NULL;
    }

    switch (node->type) {
    case WLR_SCENE_NODE_BUFFER:
        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) {
            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->node.data;
        break;
    case WLR_SCENE_NODE_RECT:
        return node->data;
        break;
    default:
        return NULL;
    }
}

void absinthe_toplevel_set_position(struct absinthe_toplevel *toplevel, int32_t x, int32_t y)
{
    wlr_scene_node_set_position(&toplevel->scene_tree->node, x, y);
}

void absinthe_toplevel_set_size(struct absinthe_toplevel *toplevel, int32_t width, int32_t height)
{
    if (width < 0 || height < 0)
        return;

    wlr_xdg_toplevel_set_size(toplevel->toplevel.xdg, width, height);
}

void absinthe_toplevel_set_fullscreen(struct absinthe_toplevel *toplevel, bool fullscreen)
{
    if (!toplevel || toplevel->fullscreen == fullscreen)
        return;

    struct absinthe_output *output = toplevel->server->focused_output;
    toplevel->fullscreen = fullscreen;
    wlr_xdg_toplevel_set_fullscreen(toplevel->toplevel.xdg, fullscreen);

    if (fullscreen) {
        toplevel->prev_geometry = toplevel->geometry;
        toplevel->geometry = output->geometry;
        toplevel->border_width = 0;
        absinthe_toplevel_set_size(toplevel, toplevel->geometry.width, toplevel->geometry.height);
    } else {
        toplevel->geometry = toplevel->prev_geometry;
        toplevel->border_width = ABSINTHE_TOPLEVEL_BORDER_WIDTH;
        absinthe_toplevel_set_size(toplevel, toplevel->geometry.width, toplevel->geometry.height);
    }
}

void absinthe_toplevel_set_border_color(struct absinthe_toplevel *toplevel, const float color[4])
{
    for (int i = 0; i < 4; ++i) {
        wlr_scene_rect_set_color(toplevel->border[i], color);
    }
}

void absinthe_toplevel_update_borders_geometry(struct absinthe_toplevel *toplevel)
{
    int32_t border_width = toplevel->border_width;

    if (toplevel->geometry.width - 2 * border_width < 0 || toplevel->geometry.height - 2 * border_width < 0)
        return;

	wlr_scene_node_set_position(&toplevel->scene_tree->node, toplevel->geometry.x, toplevel->geometry.y);
	wlr_scene_node_set_position(&toplevel->scene_surface->node, border_width, border_width);

    wlr_scene_rect_set_size(toplevel->border[0], toplevel->geometry.width - 2 * border_width, border_width);
    wlr_scene_rect_set_size(toplevel->border[1], toplevel->geometry.width - 2 * border_width, border_width);
    wlr_scene_rect_set_size(toplevel->border[2], border_width, toplevel->geometry.height);
    wlr_scene_rect_set_size(toplevel->border[3], border_width, toplevel->geometry.height);

	wlr_scene_node_set_position(&toplevel->border[0]->node, border_width, 0);
	wlr_scene_node_set_position(&toplevel->border[1]->node, border_width, toplevel->geometry.height - border_width);
	wlr_scene_node_set_position(&toplevel->border[2]->node, 0, 0);
	wlr_scene_node_set_position(&toplevel->border[3]->node, toplevel->geometry.width - border_width, 0);
}