#define _POSIX_C_SOURCE 200809L
#include <assert.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <wlr/types/wlr_data_control_v1.h>
#include <wlr/types/wlr_data_device.h>
#include "util/signal.h"
#include "wlr-data-control-unstable-v1-protocol.h"

#define DATA_CONTROL_MANAGER_VERSION 1

struct client_data_source {
	struct wlr_data_source source;
	struct wl_resource *resource;
	bool finalized;
};

static const struct wlr_data_source_impl client_source_impl;

static struct client_data_source *
		client_data_source_from_source(struct wlr_data_source *wlr_source) {
	assert(wlr_source->impl == &client_source_impl);
	return (struct client_data_source *)wlr_source;
}

static void client_source_send(struct wlr_data_source *wlr_source,
		const char *mime_type, int fd) {
	struct client_data_source *source =
		client_data_source_from_source(wlr_source);
	zwlr_data_control_source_v1_send_send(source->resource, mime_type, fd);
	close(fd);
}

static void client_source_cancel(struct wlr_data_source *wlr_source) {
	struct client_data_source *source =
		client_data_source_from_source(wlr_source);
	zwlr_data_control_source_v1_send_cancelled(source->resource);
	wlr_data_source_finish(wlr_source);
	// Make the resource inert
	wl_resource_set_user_data(source->resource, NULL);
	free(source);
}

static const struct wlr_data_source_impl client_source_impl = {
	.send = client_source_send,
	.cancel = client_source_cancel,
};

static const struct zwlr_data_control_source_v1_interface source_impl;

static struct client_data_source *source_from_resource(
		struct wl_resource *resource) {
	assert(wl_resource_instance_of(resource,
		&zwlr_data_control_source_v1_interface, &source_impl));
	return wl_resource_get_user_data(resource);
}

static void source_handle_offer(struct wl_client *client,
		struct wl_resource *resource, const char *mime_type) {
	struct client_data_source *source = source_from_resource(resource);
	if (source == NULL) {
		return;
	}

	if (source->finalized) {
		wl_resource_post_error(resource,
			ZWLR_DATA_CONTROL_SOURCE_V1_ERROR_INVALID_OFFER,
			"cannot mutate offer after set_selection");
		return;
	}

	char *dup_mime_type = strdup(mime_type);
	if (dup_mime_type == NULL) {
		wl_resource_post_no_memory(resource);
		return;
	}

	char **p = wl_array_add(&source->source.mime_types, sizeof(char *));
	if (p == NULL) {
		free(dup_mime_type);
		wl_resource_post_no_memory(resource);
		return;
	}

	*p = dup_mime_type;
}

static void source_handle_destroy(struct wl_client *client,
		struct wl_resource *resource) {
	wl_resource_destroy(resource);
}

static const struct zwlr_data_control_source_v1_interface source_impl = {
	.offer = source_handle_offer,
	.destroy = source_handle_destroy,
};

static void source_handle_resource_destroy(struct wl_resource *resource) {
	struct client_data_source *source = source_from_resource(resource);
	if (source == NULL) {
		return;
	}
	wlr_data_source_cancel(&source->source);
}


static const struct zwlr_data_control_offer_v1_interface offer_impl;

static struct wlr_data_control_v1 *control_from_offer_resource(
		struct wl_resource *resource) {
	assert(wl_resource_instance_of(resource,
		&zwlr_data_control_offer_v1_interface, &offer_impl));
	return wl_resource_get_user_data(resource);
}

static void offer_handle_receive(struct wl_client *client,
		struct wl_resource *resource, const char *mime_type, int fd) {
	struct wlr_data_control_v1 *control = control_from_offer_resource(resource);
	if (control == NULL || control->seat->selection_source == NULL) {
		close(fd);
		return;
	}
	wlr_data_source_send(control->seat->selection_source, mime_type, fd);
}

static void offer_handle_destroy(struct wl_client *client,
		struct wl_resource *resource) {
	wl_resource_destroy(resource);
}

static const struct zwlr_data_control_offer_v1_interface offer_impl = {
	.receive = offer_handle_receive,
	.destroy = offer_handle_destroy,
};

static void offer_handle_resource_destroy(struct wl_resource *resource) {
	struct wlr_data_control_v1 *control = control_from_offer_resource(resource);
	if (control != NULL) {
		control->selection_offer_resource = NULL;
	}
}

static struct wl_resource *create_offer(struct wlr_data_control_v1 *control,
		struct wlr_data_source *source) {
	struct wl_client *client = wl_resource_get_client(control->resource);
	uint32_t version = wl_resource_get_version(control->resource);
	struct wl_resource *resource = wl_resource_create(client,
		&zwlr_data_control_offer_v1_interface, version, 0);
	if (resource == NULL) {
		return NULL;
	}
	wl_resource_set_implementation(resource, &offer_impl, control,
		offer_handle_resource_destroy);

	zwlr_data_control_v1_send_data_offer(control->resource, resource);

	char **p;
	wl_array_for_each(p, &source->mime_types) {
		zwlr_data_control_offer_v1_send_offer(resource, *p);
	}

	return resource;
}


static const struct zwlr_data_control_v1_interface control_impl;

static struct wlr_data_control_v1 *control_from_resource(
		struct wl_resource *resource) {
	assert(wl_resource_instance_of(resource,
		&zwlr_data_control_v1_interface, &control_impl));
	return wl_resource_get_user_data(resource);
}

static void control_handle_set_selection(struct wl_client *client,
		struct wl_resource *control_resource,
		struct wl_resource *source_resource) {
	struct wlr_data_control_v1 *control =
		control_from_resource(control_resource);
	struct client_data_source *source = source_from_resource(source_resource);
	if (control == NULL) {
		return;
	}

	struct wlr_data_source *wlr_source = source ? &source->source : NULL;
	struct wl_display *display = wl_client_get_display(client);
	wlr_seat_set_selection(control->seat, wlr_source,
		wl_display_next_serial(display));
}

static void control_handle_destroy(struct wl_client *client,
		struct wl_resource *control_resource) {
	wl_resource_destroy(control_resource);
}

static const struct zwlr_data_control_v1_interface control_impl = {
	.set_selection = control_handle_set_selection,
	.destroy = control_handle_destroy,
};

static void control_send_selection(struct wlr_data_control_v1 *control) {
	struct wlr_data_source *source = control->seat->selection_source;

	if (control->selection_offer_resource != NULL) {
		// Make the offer inert
		wl_resource_set_user_data(control->selection_offer_resource, NULL);
	}

	control->selection_offer_resource = NULL;
	if (source != NULL) {
		control->selection_offer_resource = create_offer(control, source);
		if (control->selection_offer_resource == NULL) {
			wl_resource_post_no_memory(control->resource);
			return;
		}
	}

	zwlr_data_control_v1_send_selection(control->resource,
		control->selection_offer_resource);
}

static void control_handle_resource_destroy(struct wl_resource *resource) {
	struct wlr_data_control_v1 *control = control_from_resource(resource);
	wlr_data_control_v1_destroy(control);
}

static void control_handle_seat_destroy(struct wl_listener *listener,
		void *data) {
	struct wlr_data_control_v1 *control =
		wl_container_of(listener, control, seat_destroy);
	wlr_data_control_v1_destroy(control);
}

static void control_handle_seat_selection(struct wl_listener *listener,
		void *data) {
	struct wlr_data_control_v1 *control =
		wl_container_of(listener, control, seat_selection);
	control_send_selection(control);
}

void wlr_data_control_v1_destroy(struct wlr_data_control_v1 *control) {
	if (control == NULL) {
		return;
	}
	zwlr_data_control_v1_send_finished(control->resource);
	// Make the resources inert
	wl_resource_set_user_data(control->resource, NULL);
	if (control->selection_offer_resource != NULL) {
		wl_resource_set_user_data(control->selection_offer_resource, NULL);
	}
	wl_list_remove(&control->seat_destroy.link);
	wl_list_remove(&control->seat_selection.link);
	wl_list_remove(&control->link);
	free(control);
}


static const struct zwlr_data_control_manager_v1_interface manager_impl;

static struct wlr_data_control_manager_v1 *manager_from_resource(
		struct wl_resource *resource) {
	assert(wl_resource_instance_of(resource,
		&zwlr_data_control_manager_v1_interface, &manager_impl));
	return wl_resource_get_user_data(resource);
}

static void manager_handle_create_data_source(struct wl_client *client,
		struct wl_resource *manager_resource, uint32_t id) {
	struct client_data_source *source =
		calloc(1, sizeof(struct client_data_source));
	if (source == NULL) {
		wl_resource_post_no_memory(manager_resource);
		return;
	}
	wlr_data_source_init(&source->source, &client_source_impl);

	uint32_t version = wl_resource_get_version(manager_resource);
	source->resource = wl_resource_create(client,
		&zwlr_data_control_source_v1_interface, version, id);
	if (source->resource == NULL) {
		wl_resource_post_no_memory(manager_resource);
		free(source);
		return;
	}
	wl_resource_set_implementation(source->resource, &source_impl, source,
		source_handle_resource_destroy);
}

static void manager_handle_get_data_control(struct wl_client *client,
		struct wl_resource *manager_resource, uint32_t id,
		struct wl_resource *seat_resource) {
	struct wlr_data_control_manager_v1 *manager =
		manager_from_resource(manager_resource);
	struct wlr_seat_client *seat_client =
		wlr_seat_client_from_resource(seat_resource);

	struct wlr_data_control_v1 *control =
		calloc(1, sizeof(struct wlr_data_control_v1));
	if (control == NULL) {
		wl_resource_post_no_memory(manager_resource);
		return;
	}
	control->manager = manager;
	control->seat = seat_client->seat;

	uint32_t version = wl_resource_get_version(manager_resource);
	control->resource = wl_resource_create(client,
		&zwlr_data_control_v1_interface, version, id);
	if (control->resource == NULL) {
		wl_resource_post_no_memory(manager_resource);
		free(control);
		return;
	}
	wl_resource_set_implementation(control->resource, &control_impl, control,
		control_handle_resource_destroy);
	struct wl_resource *resource = control->resource;

	control->seat_destroy.notify = control_handle_seat_destroy;
	wl_signal_add(&control->seat->events.destroy, &control->seat_destroy);

	control->seat_selection.notify = control_handle_seat_selection;
	wl_signal_add(&control->seat->events.selection, &control->seat_selection);

	wl_list_insert(&manager->controls, &control->link);
	wlr_signal_emit_safe(&manager->events.new_control, control);

	// At this point maybe the compositor decided to destroy the control. If
	// it's the case then the resource will be inert.
	control = control_from_resource(resource);
	if (control != NULL) {
		control_send_selection(control);
	}
}

static void manager_handle_destroy(struct wl_client *client,
		struct wl_resource *manager_resource) {
	wl_resource_destroy(manager_resource);
}

static const struct zwlr_data_control_manager_v1_interface manager_impl = {
	.create_data_source = manager_handle_create_data_source,
	.get_data_control = manager_handle_get_data_control,
	.destroy = manager_handle_destroy,
};

static void manager_handle_resource_destroy(struct wl_resource *resource) {
	wl_list_remove(wl_resource_get_link(resource));
}

static void manager_bind(struct wl_client *client, void *data, uint32_t version,
		uint32_t id) {
	struct wlr_data_control_manager_v1 *manager = data;

	struct wl_resource *resource = wl_resource_create(client,
		&zwlr_data_control_manager_v1_interface, version, id);
	if (resource == NULL) {
		wl_client_post_no_memory(client);
		return;
	}
	wl_resource_set_implementation(resource, &manager_impl, manager,
		manager_handle_resource_destroy);

	wl_list_insert(&manager->resources, wl_resource_get_link(resource));
}

static void handle_display_destroy(struct wl_listener *listener, void *data) {
	struct wlr_data_control_manager_v1 *manager =
		wl_container_of(listener, manager, display_destroy);
	wlr_data_control_manager_v1_destroy(manager);
}

struct wlr_data_control_manager_v1 *wlr_data_control_manager_v1_create(
		struct wl_display *display) {
	struct wlr_data_control_manager_v1 *manager =
		calloc(1, sizeof(struct wlr_data_control_manager_v1));
	if (manager == NULL) {
		return NULL;
	}
	wl_list_init(&manager->resources);
	wl_list_init(&manager->controls);
	wl_signal_init(&manager->events.destroy);
	wl_signal_init(&manager->events.new_control);

	manager->global = wl_global_create(display,
		&zwlr_data_control_manager_v1_interface, DATA_CONTROL_MANAGER_VERSION,
		manager, manager_bind);
	if (manager->global == NULL) {
		free(manager);
		return NULL;
	}

	manager->display_destroy.notify = handle_display_destroy;
	wl_display_add_destroy_listener(display, &manager->display_destroy);

	return manager;
}

void wlr_data_control_manager_v1_destroy(
		struct wlr_data_control_manager_v1 *manager) {
	if (manager == NULL) {
		return;
	}

	wlr_signal_emit_safe(&manager->events.destroy, manager);

	struct wlr_data_control_v1 *control, *control_tmp;
	wl_list_for_each_safe(control, control_tmp, &manager->controls, link) {
		wl_resource_destroy(control->resource);
	}

	struct wl_resource *resource, *resource_tmp;
	wl_resource_for_each_safe(resource, resource_tmp, &manager->resources) {
		wl_resource_destroy(resource);
	}

	wl_list_remove(&manager->display_destroy.link);
	free(manager);
}