// SPDX-License-Identifier: GPL-2.0-only

#include <assert.h>
#include "labwc.h"
#include "ssd.h"

struct ssd_part *
add_scene_part(struct wl_list *part_list, enum ssd_part_type type)
{
	struct ssd_part *part = calloc(1, sizeof(struct ssd_part));
	part->type = type;
	wl_list_insert(part_list->prev, &part->link);
	return part;
}

struct ssd_part *
add_scene_rect(struct wl_list *list, enum ssd_part_type type,
	struct wlr_scene_node *parent, int width, int height,
	int x, int y, float color[4])
{
	struct ssd_part *part = add_scene_part(list, type);
	part->node = &wlr_scene_rect_create(
		parent, width, height, color)->node;
	wlr_scene_node_set_position(part->node, x, y);
	return part;
}

struct ssd_part *
add_scene_buffer(struct wl_list *list, enum ssd_part_type type,
	struct wlr_scene_node *parent, struct wlr_buffer *buffer,
	int x, int y)
{
	struct ssd_part *part = add_scene_part(list, type);
	part->node = &wlr_scene_buffer_create(parent, buffer)->node;
	wlr_scene_node_set_position(part->node, x, y);
	return part;
}

static void
finish_scene_button(struct wl_list *part_list, enum ssd_part_type type,
	struct wlr_scene_node *parent, struct wlr_buffer *icon_buffer)
{
	float hover_bg[4] = {0.15f, 0.15f, 0.15f, 0.3f};

	/* Icon */
	add_scene_buffer(part_list, type, parent, icon_buffer,
		(BUTTON_WIDTH - icon_buffer->width) / 2,
		(SSD_HEIGHT - icon_buffer->height) / 2);

	/* Hover overlay */
	struct ssd_part *hover_part;
	hover_part = add_scene_rect(part_list, type, parent,
		BUTTON_WIDTH, SSD_HEIGHT, 0, 0, hover_bg);
	wlr_scene_node_set_enabled(hover_part->node, false);
}

struct ssd_part *
add_scene_button_corner(struct wl_list *part_list, enum ssd_part_type type,
	struct wlr_scene_node *parent, struct wlr_buffer *corner_buffer,
	struct wlr_buffer *icon_buffer, int x)
{
	struct ssd_part *part;
	/* Background */
	part = add_scene_buffer(part_list, type, parent, corner_buffer, x, 0);
	finish_scene_button(part_list, type, part->node, icon_buffer);
	return part;
}

struct ssd_part *
add_scene_button(struct wl_list *part_list, enum ssd_part_type type,
	struct wlr_scene_node *parent, float *bg_color,
	struct wlr_buffer *icon_buffer, int x)
{
	struct ssd_part *part;
	/* Background */
	part = add_scene_rect(part_list, type, parent,
		BUTTON_WIDTH, SSD_HEIGHT, x, 0, bg_color);
	finish_scene_button(part_list, type, part->node, icon_buffer);
	return part;
}

struct ssd_part *
ssd_get_part(struct wl_list *part_list, enum ssd_part_type type)
{
	struct ssd_part *part;
	wl_list_for_each(part, part_list, link) {
		if (part->type == type) {
			return part;
		}
	}
	return NULL;
}

void
ssd_destroy_parts(struct wl_list *list)
{
	struct ssd_part *part, *tmp;
	wl_list_for_each_reverse_safe(part, tmp, list, link) {
		if (part->node) {
			wlr_scene_node_destroy(part->node);
		}
		if (part->buffer) {
			wlr_buffer_drop(&part->buffer->base);
		}
		wl_list_remove(&part->link);
		free(part);
	}
	assert(wl_list_empty(list));
}