#include <SDL3/SDL_hints.h>
#include <SDL3/SDL_keycode.h>
#include <SDL3/SDL_oldnames.h>
#include <clay/clay.h>
#include "renderer/clay_renderer_SDL3.h"

#include "application.h"
#include "defs.h"
#include "style.h"

#define SDL_MAIN_HANDLED
#include <SDL3/SDL.h>
#include <SDL3/SDL_error.h>
#include <SDL3/SDL_events.h>
#include <SDL3/SDL_init.h>
#include <SDL3/SDL_log.h>
#include <SDL3/SDL_mouse.h>
#include <SDL3/SDL_render.h>
#include <SDL3/SDL_video.h>
#include <SDL3_ttf/SDL_ttf.h>

#include <stddef.h>
#include <stdint.h>
#include <stdbool.h>
#include <stdlib.h>

constexpr SDL_InitFlags sdlInitFlags = SDL_WINDOW_RESIZABLE | SDL_WINDOW_HIGH_PIXEL_DENSITY;

SDL_Window *window = nullptr;
SDL_Renderer *renderer = nullptr;
int screenWidth = 1920, screenHeight = 1080;
bool running = true;
uint64_t clayMemorySize = 0;

Clay_Arena clayPrimaryArena;
TTF_Font *fonts[FONT_MAX];
TTF_TextEngine *textEngine = nullptr;

Clay_SDL3RendererData backendData = {
	.renderer = nullptr,
	.fonts = nullptr,
	.textEngine = nullptr
};

static inline
void LogOutputResolution() {
	int w, h;
	SDL_GetCurrentRenderOutputSize(renderer, &w, &h);
	SDL_Log("output size: %i, %d", w, h);
}

static
Clay_Dimensions MeasureText(Clay_StringSlice text, Clay_TextElementConfig *config, void *userData) {
	TTF_Font **fonts = userData;
	TTF_Font *font = fonts[config->fontId];
	int width, height;
	TTF_SetFontSize(font, config->fontSize);
	if (!TTF_GetStringSize(font, text.chars, text.length, &width, &height)) {
		SDL_LogError(SDL_LOG_CATEGORY_ERROR, "MeasureText failed to measure text %s", SDL_GetError());
	}
	return (Clay_Dimensions) { width, height };
}

static
void HandleClayErrors(Clay_ErrorData data) {
	SDL_LogError(SDL_LOG_CATEGORY_ERROR, "%s", data.errorText.chars);
}

static inline void InitFonts() {
	fonts[FONT_DEFAULT] = TTF_OpenFont("assets/AdwaitaSans-Regular.ttf", baseFontSize * 5);
	if (fonts[FONT_DEFAULT] == nullptr) {
		SDL_LogError(SDL_LOG_CATEGORY_ERROR, "TTF_OpenFont failed: Failed to load adwaita sans: %s", SDL_GetError());
		exit(6);
	}
	TTF_SetFontHinting(fonts[FONT_DEFAULT], TTF_HINTING_LIGHT_SUBPIXEL);
	fonts[FONT_BOLD] = TTF_OpenFont("assets/AdwaitaSans-Regular.ttf", baseFontSize * 5);
	if (fonts[FONT_BOLD] == nullptr) {
		SDL_LogError(SDL_LOG_CATEGORY_ERROR, "TTF_OpenFont failed: Failed to load adwaita sans bold: %s", SDL_GetError());
		exit(7);
	}
	TTF_SetFontHinting(fonts[FONT_BOLD], TTF_HINTING_LIGHT_SUBPIXEL);
	TTF_SetFontStyle(fonts[FONT_BOLD], TTF_STYLE_BOLD);
}

static inline
void InitSDL() {
	SDL_SetHint(SDL_HINT_RENDER_LINE_METHOD, "3");
	if (!SDL_Init(SDL_INIT_VIDEO | SDL_INIT_EVENTS)) {
		SDL_LogError(SDL_LOG_CATEGORY_ERROR, "SDL_Init failed: %s", SDL_GetError());
		exit(1);
	}
	if ((window = SDL_CreateWindow("Window", screenWidth, screenHeight, sdlInitFlags)) == nullptr) {
		SDL_LogError(SDL_LOG_CATEGORY_ERROR, "SDL_CreateWindow failed: %s", SDL_GetError());
		exit(2);
	}
	if ((renderer = SDL_CreateRenderer(window, NULL)) == nullptr) {
		SDL_LogError(SDL_LOG_CATEGORY_ERROR, "SDL_CreateRenderer failed: %s", SDL_GetError());
		exit(3);
	}
	if (!TTF_Init()) {
		SDL_LogError(SDL_LOG_CATEGORY_ERROR, "TTF_Init failed: %s", SDL_GetError());
		exit(4);
	}
	if ((textEngine = TTF_CreateRendererTextEngine(renderer)) == nullptr) {
		SDL_LogError(SDL_LOG_CATEGORY_ERROR, "TTF_CreateRendererTextEngine failed: %s", SDL_GetError());
		exit(5);
	}
}

static
void InitClay() {
	clayMemorySize = Clay_MinMemorySize();
	clayPrimaryArena = Clay_CreateArenaWithCapacityAndMemory(clayMemorySize, SDL_malloc(clayMemorySize));
	Clay_Initialize(clayPrimaryArena, (Clay_Dimensions) { screenWidth, screenHeight }, (Clay_ErrorHandler) { HandleClayErrors });
	Clay_SetMeasureTextFunction(MeasureText, fonts);
	Clay_SetLayoutDimensions((Clay_Dimensions) { screenWidth, screenHeight });
	float x, y;
	SDL_GetMouseState(&x, &y);
	Clay_SetPointerState((Clay_Vector2) { x, y }, false);
}

extern Clay_RenderCommandArray RenderApplication();
extern void HandleEvent(SDL_Event event);

int main(int argc, char *argv[]) {
	InitSDL();
	InitFonts();
	LogOutputResolution();
	InitClay();
	backendData = (Clay_SDL3RendererData) {
		.renderer = renderer,
		.fonts = fonts,
		.textEngine = textEngine,
	};
	SDL_Event event;
	uint64_t startFrameTime = SDL_GetTicksNS();
	double deltaTime = 0.0;
	bool mouseButtonDown = false;
	bool shiftDown = false;
	while (running) {
		deltaTime = SDL_GetTicksNS() - startFrameTime;
		startFrameTime = SDL_GetTicksNS();
		Clay_Vector2 scrollMotion = { 0, 0 };
		while (SDL_PollEvent(&event)) {
			HandleEvent(event);
			switch (event.type) {
				case SDL_EVENT_QUIT:
					running = false;
					break;
				case SDL_EVENT_WINDOW_RESIZED:
					Clay_SetLayoutDimensions((Clay_Dimensions){
						event.window.data1,
						event.window.data2
					});
					LogOutputResolution();
					break;
				case SDL_EVENT_MOUSE_WHEEL:
					if (shiftDown) {
						scrollMotion = (Clay_Vector2) { event.wheel.y * 5.f, event.wheel.x * 5.f };
					} else {
						scrollMotion = (Clay_Vector2) { event.wheel.x * 5.f, event.wheel.y * 5.f };
					}
					break;
				case SDL_EVENT_MOUSE_MOTION:
					Clay_SetPointerState((Clay_Vector2) { event.motion.x, event.motion.y }, mouseButtonDown);
					break;
				case SDL_EVENT_MOUSE_BUTTON_DOWN:
				case SDL_EVENT_MOUSE_BUTTON_UP:
					if (event.button.button == SDL_BUTTON_LEFT) {
						mouseButtonDown = event.button.down;
						Clay_SetPointerState((Clay_Vector2) { event.button.x, event.button.y }, mouseButtonDown);
					}
					break;
				case SDL_EVENT_KEY_DOWN:
				case SDL_EVENT_KEY_UP:
					if (event.key.key == SDLK_LSHIFT || event.key.key == SDLK_RSHIFT) {
						shiftDown = event.key.down;
					}
					break;
				default: break;
			}
		}
		Clay_UpdateScrollContainers(true, scrollMotion, deltaTime);
		SDL_SetRenderDrawColor(renderer, 10, 10, 10, 255);
		SDL_RenderClear(renderer);
		Clay_RenderCommandArray array = RenderApplication();
		SDL_Clay_RenderClayCommands(&backendData, &array);
		SDL_RenderPresent(renderer);
	}
	SDL_DestroyRenderer(renderer);
	SDL_DestroyWindow(window);
	SDL_Quit();
	return 0;
}