#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <inttypes.h>
#include <errno.h>
#include <xf86drm.h>
#include <xf86drmMode.h>
#include <drm_mode.h>
#include <EGL/egl.h>
#include <EGL/eglext.h>
#include <gbm.h>
#include <GLES3/gl3.h>
#include <wayland-server.h>

#include "backend/drm/backend.h"
#include "backend/drm/drm.h"
#include "common/log.h"

static const char *conn_name[] = {
	[DRM_MODE_CONNECTOR_Unknown]     = "Unknown",
	[DRM_MODE_CONNECTOR_VGA]         = "VGA",
	[DRM_MODE_CONNECTOR_DVII]        = "DVI-I",
	[DRM_MODE_CONNECTOR_DVID]        = "DVI-D",
	[DRM_MODE_CONNECTOR_DVIA]        = "DVI-A",
	[DRM_MODE_CONNECTOR_Composite]   = "Composite",
	[DRM_MODE_CONNECTOR_SVIDEO]      = "SVIDEO",
	[DRM_MODE_CONNECTOR_LVDS]        = "LVDS",
	[DRM_MODE_CONNECTOR_Component]   = "Component",
	[DRM_MODE_CONNECTOR_9PinDIN]     = "DIN",
	[DRM_MODE_CONNECTOR_DisplayPort] = "DP",
	[DRM_MODE_CONNECTOR_HDMIA]       = "HDMI-A",
	[DRM_MODE_CONNECTOR_HDMIB]       = "HDMI-B",
	[DRM_MODE_CONNECTOR_TV]          = "TV",
	[DRM_MODE_CONNECTOR_eDP]         = "eDP",
	[DRM_MODE_CONNECTOR_VIRTUAL]     = "Virtual",
	[DRM_MODE_CONNECTOR_DSI]         = "DSI",
};

static void page_flip_handler(int fd, unsigned seq, unsigned tv_sec, unsigned tv_usec,
	void *user) {

	struct wlr_drm_output *out = user;
	struct wlr_drm_backend *backend = wl_container_of(out->renderer, backend, renderer);

	out->pageflip_pending = true;
	if (!out->cleanup) {
		wl_signal_emit(&backend->signals.output_render, out);
	}
}

int wlr_drm_event(int fd, uint32_t mask, void *data) {
	drmEventContext event = {
		.version = DRM_EVENT_CONTEXT_VERSION,
		.page_flip_handler = page_flip_handler,
	};

	drmHandleEvent(fd, &event);

	return 1;
}

bool wlr_drm_renderer_init(struct wlr_drm_renderer *renderer, int fd) {

	renderer->gbm = gbm_create_device(fd);
	if (!renderer->gbm) {
		wlr_log(L_ERROR, "Failed to create GBM device: %s", strerror(errno));
		return false;
	}

	if (!wlr_egl_init(&renderer->egl, EGL_PLATFORM_GBM_MESA, renderer->gbm)) {
		gbm_device_destroy(renderer->gbm);
		return false;
	}

	renderer->fd = fd;
	return true;
}

void wlr_drm_renderer_free(struct wlr_drm_renderer *renderer) {
	if (!renderer) {
		return;
	}

	wlr_egl_free(&renderer->egl);
	gbm_device_destroy(renderer->gbm);
}

static int find_id(const void *item, const void *cmp_to) {
	const struct wlr_drm_output *out = item;
	const uint32_t *id = cmp_to;

	if (out->connector < *id) {
		return -1;
	} else if (out->connector > *id) {
		return 1;
	} else {
		return 0;
	}
}

void wlr_drm_scan_connectors(struct wlr_drm_backend *backend) {
	drmModeRes *res = drmModeGetResources(backend->fd);
	if (!res) {
		wlr_log(L_ERROR, "Failed to get DRM resources");
		return;
	}

	for (int i = 0; i < res->count_connectors; ++i) {
		uint32_t id = res->connectors[i];

		drmModeConnector *conn = drmModeGetConnector(backend->fd, id);
		if (!conn) {
			wlr_log(L_ERROR, "Failed to get DRM connector");
			continue;
		}

		struct wlr_drm_output *out;
		int index = list_seq_find(backend->outputs, find_id, &id);

		if (index == -1) {
			out = calloc(1, sizeof *out);
			if (!out) {
				wlr_log(L_ERROR, "Allocation failed: %s", strerror(errno));
				drmModeFreeConnector(conn);
				continue;
			}

			out->renderer = &backend->renderer;
			out->state = DRM_OUTPUT_DISCONNECTED;
			out->connector = res->connectors[i];
			snprintf(out->name, sizeof out->name, "%s-%"PRIu32,
				 conn_name[conn->connector_type],
				 conn->connector_type_id);

			list_add(backend->outputs, out);
			wlr_log(L_INFO, "Found display '%s'", out->name);
		} else {
			out = backend->outputs->items[index];
		}

		if (out->state == DRM_OUTPUT_DISCONNECTED &&
			conn->connection == DRM_MODE_CONNECTED) {

			out->state = DRM_OUTPUT_NEEDS_MODESET;
			wlr_log(L_INFO, "Sending modesetting signal for '%s'", out->name);
			wl_signal_emit(&backend->signals.output_add, out);

		} else if (out->state == DRM_OUTPUT_CONNECTED &&
			conn->connection != DRM_MODE_CONNECTED) {

			out->state = DRM_OUTPUT_DISCONNECTED;
			wlr_drm_output_free(out, false);
			wlr_log(L_INFO, "Sending destruction signal for '%s'", out->name);
			wl_signal_emit(&backend->signals.output_rem, out);
		}

		drmModeFreeConnector(conn);
	}

	drmModeFreeResources(res);
}

static void free_fb(struct gbm_bo *bo, void *data) {
	uint32_t *id = data;

	if (id && *id) {
		drmModeRmFB(gbm_bo_get_fd(bo), *id);
	}

	free(id);
}

static uint32_t get_fb_for_bo(int fd, struct gbm_bo *bo) {
	uint32_t *id = gbm_bo_get_user_data(bo);

	if (id) {
		return *id;
	}

	id = calloc(1, sizeof *id);
	if (!id) {
		wlr_log(L_ERROR, "Allocation failed: %s", strerror(errno));
		return 0;
	}

	drmModeAddFB(fd, gbm_bo_get_width(bo), gbm_bo_get_height(bo), 24, 32,
		     gbm_bo_get_stride(bo), gbm_bo_get_handle(bo).u32, id);

	gbm_bo_set_user_data(bo, id, free_fb);

	return *id;
}

static bool display_init_renderer(struct wlr_drm_renderer *renderer,
	struct wlr_drm_output *out) {

	out->renderer = renderer;

	out->gbm = gbm_surface_create(renderer->gbm, out->width, out->height,
		GBM_FORMAT_XRGB8888, GBM_BO_USE_SCANOUT | GBM_BO_USE_RENDERING);
	if (!out->gbm) {
		wlr_log(L_ERROR, "Failed to create GBM surface for %s: %s", out->name,
			strerror(errno));
		return false;
	}

	out->egl = wlr_egl_create_surface(&renderer->egl, out->gbm);
	if (out->egl == EGL_NO_SURFACE) {
		wlr_log(L_ERROR, "Failed to create EGL surface for %s", out->name);
		return false;
	}

	// Render black frame

	eglMakeCurrent(renderer->egl.display, out->egl, out->egl, renderer->egl.context);

	glViewport(0, 0, out->width, out->height);
	glClearColor(0.0, 0.0, 0.0, 1.0);
	glClear(GL_COLOR_BUFFER_BIT);

	eglSwapBuffers(renderer->egl.display, out->egl);

	struct gbm_bo *bo = gbm_surface_lock_front_buffer(out->gbm);
	uint32_t fb_id = get_fb_for_bo(renderer->fd, bo);

	drmModeSetCrtc(renderer->fd, out->crtc, fb_id, 0, 0,
		       &out->connector, 1, out->active_mode);
	drmModePageFlip(renderer->fd, out->crtc, fb_id, DRM_MODE_PAGE_FLIP_EVENT, out);

	gbm_surface_release_buffer(out->gbm, bo);

	return true;
}

static drmModeModeInfo *select_mode(size_t num_modes, drmModeModeInfo modes[static num_modes],
	drmModeCrtc *old_crtc, const char *str) {

	if (strcmp(str, "preferred") == 0) {
		return &modes[0];
	}

	if (strcmp(str, "current") == 0) {
		if (!old_crtc) {
			wlr_log(L_ERROR, "Display does not have currently configured mode");
			return NULL;
		}

		for (size_t i = 0; i < num_modes; ++i) {
			if (memcmp(&modes[i], &old_crtc->mode, sizeof modes[0]) == 0) {
				return &modes[i];
			}
		}

		// We should never get here
		assert(0);
	}

	unsigned width = 0;
	unsigned height = 0;
	unsigned rate = 0;
	int ret;

	if ((ret = sscanf(str, "%ux%u@%u", &width, &height, &rate)) != 2 && ret != 3) {
		wlr_log(L_ERROR, "Invalid modesetting string '%s'", str);
		return NULL;
	}

	for (size_t i = 0; i < num_modes; ++i) {
		if (modes[i].hdisplay == width && modes[i].vdisplay == height && 
			(!rate || modes[i].vrefresh == rate)) {

			return &modes[i];
		}
	}

	wlr_log(L_ERROR, "Unable to find mode %ux%u@%u", width, height, rate);
	return NULL;
}

bool wlr_drm_output_modeset(struct wlr_drm_output *out, const char *str) {
	struct wlr_drm_backend *backend = wl_container_of(out->renderer, backend, renderer);
	wlr_log(L_INFO, "Modesetting %s with '%s'", out->name, str);

	drmModeConnector *conn = drmModeGetConnector(backend->fd, out->connector);
	if (!conn) {
		wlr_log(L_ERROR, "Failed to get DRM connector");
		goto error;
	}

	if (conn->connection != DRM_MODE_CONNECTED || conn->count_modes == 0) {
		wlr_log(L_ERROR, "%s is not connected", out->name);
		goto error;
	}

	out->num_modes = conn->count_modes;
	out->modes = malloc(sizeof *out->modes * out->num_modes);
	if (!out->modes) {
		wlr_log(L_ERROR, "Allocation failed: %s", strerror(errno));
		goto error;
	}
	memcpy(out->modes, conn->modes, sizeof *out->modes * out->num_modes);

	wlr_log(L_INFO, "Detected modes:");
	for (size_t i = 0; i < out->num_modes; ++i) {
		wlr_log(L_INFO, "  %"PRIu16"@%"PRIu16"@%"PRIu32,
			out->modes[i].hdisplay, out->modes[i].vdisplay,
			out->modes[i].vrefresh);
	}

	drmModeEncoder *curr_enc = drmModeGetEncoder(backend->fd, conn->encoder_id);
	if (curr_enc) {
		out->old_crtc = drmModeGetCrtc(backend->fd, curr_enc->crtc_id);
		free(curr_enc);
	}

	out->active_mode = select_mode(out->num_modes, out->modes,
		out->old_crtc, str);
	if (!out->active_mode) {
		wlr_log(L_ERROR, "Failed to configure %s", out->name);
		goto error;
	}

	drmModeRes *res = drmModeGetResources(backend->fd);
	if (!res) {
		wlr_log(L_ERROR, "Failed to get DRM resources");
		goto error;
	}

	bool success = false;
	for (int i = 0; !success && i < conn->count_encoders; ++i) {
		drmModeEncoder *enc = drmModeGetEncoder(backend->fd, conn->encoders[i]);
		if (!enc)
			continue;

		for (int j = 0; j < res->count_crtcs; ++j) {
			if ((enc->possible_crtcs & (1 << j)) == 0) {
				continue;
			}

			if ((backend->taken_crtcs & (1 << j)) == 0) {
				backend->taken_crtcs |= 1 << j;
				out->crtc = res->crtcs[j];

				success = true;
				break;
			}
		}

		drmModeFreeEncoder(enc);
	}

	drmModeFreeResources(res);

	if (!success) {
		wlr_log(L_ERROR, "Failed to find CRTC for %s", out->name);
		goto error;
	}

	out->state = DRM_OUTPUT_CONNECTED;

	out->width = out->active_mode->hdisplay;
	out->height = out->active_mode->vdisplay;

	if (!display_init_renderer(&backend->renderer, out)) {
		wlr_log(L_ERROR, "Failed to initalise renderer for %s", out->name);
		goto error;
	}

	drmModeFreeConnector(conn);

	wlr_log(L_INFO, "Configuring %s with mode %"PRIu16"x%"PRIu16"@%"PRIu32"",
		out->name, out->active_mode->hdisplay, out->active_mode->vdisplay,
		out->active_mode->vrefresh);

	return true;

error:
	out->state = DRM_OUTPUT_DISCONNECTED;
	drmModeFreeConnector(conn);
	free(out->modes);

	wl_signal_emit(&backend->signals.output_rem, out);

	return false;
}

void wlr_drm_output_free(struct wlr_drm_output *out, bool restore) {
	if (!out || out->state != DRM_OUTPUT_CONNECTED) {
		return;
	}

	struct wlr_drm_renderer *renderer = out->renderer;

	eglDestroySurface(renderer->egl.display, out->egl);
	gbm_surface_destroy(out->gbm);

	free(out->modes);
	out->state = DRM_OUTPUT_DISCONNECTED;

	if (!restore) {
		return;
	}

	drmModeCrtc *crtc = out->old_crtc;
	if (crtc) {
		// Wait for exising page flips to finish

		drmEventContext event = {
			.version = DRM_EVENT_CONTEXT_VERSION,
			.page_flip_handler = page_flip_handler,
		};

		out->cleanup = true;
		while (out->pageflip_pending)
			drmHandleEvent(renderer->fd, &event);

		drmModeSetCrtc(renderer->fd, crtc->crtc_id, crtc->buffer_id,
			       crtc->x, crtc->y, &out->connector,
			       1, &crtc->mode);
		drmModeFreeCrtc(crtc);
	}
}

void wlr_drm_output_begin(struct wlr_drm_output *out) {
	struct wlr_drm_renderer *renderer = out->renderer;
	eglMakeCurrent(renderer->egl.display, out->egl, out->egl, renderer->egl.context);
}

void wlr_drm_output_end(struct wlr_drm_output *out) {
	struct wlr_drm_renderer *renderer = out->renderer;
	eglSwapBuffers(renderer->egl.display, out->egl);

	struct gbm_bo *bo = gbm_surface_lock_front_buffer(out->gbm);
	uint32_t fb_id = get_fb_for_bo(renderer->fd, bo);

	drmModePageFlip(renderer->fd, out->crtc, fb_id, DRM_MODE_PAGE_FLIP_EVENT, out);

	gbm_surface_release_buffer(out->gbm, bo);

	out->pageflip_pending = false;
}