#include <wayland-client.h>
#include <sys/mman.h>
#include <unistd.h>

#include "drw.h"
#include "shm_open.h"

void
drwsurf_resize(struct drwsurf *ds, uint32_t w, uint32_t h, uint32_t s) {
	if (ds->buf) {
		munmap(ds->pool_data, ds->size);
		wl_buffer_destroy(ds->buf);
		ds->buf = NULL;
	}

	ds->scale = s;
	ds->width = w * s;
	ds->height = h * s;

	setup_buffer(ds);
}

static void surface_frame_callback(void *data, struct wl_callback *cb,
                                   uint32_t time);

static struct wl_callback_listener frame_listener = {
    .done = surface_frame_callback
};

void
drwsurf_flip(struct drwsurf *ds) {
	ds->cb = wl_surface_frame(ds->surf);
	wl_callback_add_listener(ds->cb, &frame_listener, (void *)ds);

	wl_surface_attach(ds->surf, ds->buf, 0, 0);
	wl_surface_commit(ds->surf);
}

void
surface_frame_callback(void *data, struct wl_callback *cb, uint32_t time) {
	struct drwsurf *ds = (struct drwsurf *)data;
	wl_callback_destroy(cb);
	ds->cb = NULL;

	drwsurf_flip(ds);
}

void
drw_draw_text(struct drwsurf *d, Color color,
	uint32_t x, uint32_t y,
	uint32_t w, uint32_t h,
	const char *label) {

	cairo_save(d->cairo);

    cairo_set_source_rgba (
		d->cairo,
		color.bgra[2] / (double)255,
		color.bgra[1] / (double)255,
		color.bgra[0] / (double)255,
		color.bgra[3] / (double)255
    );
	cairo_move_to(d->cairo, x + (double)w / 2.0, y + (double)h / 2.0);

	pango_layout_set_text(d->layout, label, -1);

	int width, height;
	pango_layout_get_size(d->layout, &width, &height);

	cairo_rel_move_to(d->cairo, - ((double)width / PANGO_SCALE) / 2, - ((double)height / PANGO_SCALE) / 2);
	pango_cairo_show_layout(d->cairo, d->layout);
	cairo_restore(d->cairo);

	wl_surface_damage(d->surf, x, y, w, h);
}

void
drw_fill_rectangle(struct drwsurf *d, Color color, uint32_t x, uint32_t y,
	uint32_t w, uint32_t h) {
	cairo_save(d->cairo);

	cairo_set_operator(d->cairo, CAIRO_OPERATOR_SOURCE);

	cairo_rectangle(d->cairo, x, y, w, h);
	cairo_set_source_rgba(
		d->cairo,
		color.bgra[2] / (double)255,
		color.bgra[1] / (double)255,
		color.bgra[0] / (double)255,
		color.bgra[3] / (double)255
	);
	cairo_fill(d->cairo);

	cairo_restore(d->cairo);

	wl_surface_damage(d->surf, x, y, w, h);
}

uint32_t
setup_buffer(struct drwsurf *drwsurf)
{
	int stride = drwsurf->width * 4;
	drwsurf->size = stride * drwsurf->height;

	int fd = allocate_shm_file(drwsurf->size);
	if (fd == -1) {
		return 1;
	}

	drwsurf->pool_data = mmap(NULL, drwsurf->size,
	PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
	if (drwsurf->pool_data == MAP_FAILED) {
		close(fd);
		return 1;
	}

	struct wl_shm_pool *pool = wl_shm_create_pool(drwsurf->ctx->shm, fd, drwsurf->size);
	drwsurf->buf = wl_shm_pool_create_buffer(pool, 0,
		drwsurf->width, drwsurf->height, stride, WL_SHM_FORMAT_ARGB8888);
	wl_shm_pool_destroy(pool);
	close(fd);

	cairo_surface_t *s = cairo_image_surface_create_for_data(drwsurf->pool_data,
		CAIRO_FORMAT_ARGB32,
	drwsurf->width, drwsurf->height, stride);

	drwsurf->cairo = cairo_create(s);
	cairo_scale(drwsurf->cairo, drwsurf->scale, drwsurf->scale);
	drwsurf->layout = pango_cairo_create_layout(drwsurf->cairo);
	pango_layout_set_font_description(drwsurf->layout, drwsurf->ctx->font_description);
	cairo_save(drwsurf->cairo);

	wl_surface_set_buffer_scale(drwsurf->surf, drwsurf->scale);

	return 0;
}