#include <PreCompiled.h>

#include <Core/Application.h>
#include <Core/SDLManager.h>
#include <Renderer/RenderCore.h>
#include <mlx.h>
#include <Core/Memory.h>
#include <Core/Handles.h>

static mlx::Application* __internal_application_ptr = nullptr;

#ifndef DISABLE_ALL_SAFETIES
	#define MLX_CHECK_APPLICATION_POINTER(ptr) \
		if(ptr == NULL || ptr->app.Get() != __internal_application_ptr) \
			mlx::FatalError("invalid mlx handle passed to '%'", MLX_FUNC_SIG); \
		else {} // just to avoid issues with possible if-else statements outside this macro
#else
	#define MLX_CHECK_APPLICATION_POINTER(ptr)
#endif

extern "C"
{
	mlx_context mlx_init()
	{
		if(__internal_application_ptr != nullptr)
		{
			mlx::Error("MLX cannot be initialized multiple times");
			return nullptr;
		}

		mlx::MemManager::Get(); // just to initialize the C garbage collector

		try { __internal_application_ptr = new mlx::Application; }
		catch(...) { mlx::FatalError("internal application memory allocation failed"); }

		mlx_context_handler* context;
		try { context = new mlx_context_handler; }
		catch(...) { mlx::FatalError("mlx_context memory allocation failed"); }
		context->app = __internal_application_ptr;
		return context;
	}

	void mlx_set_fps_goal(mlx_context mlx, int fps)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		if(fps < 0)
			mlx::Error("You cannot set a negative FPS cap (nice try)");
		else if(fps == 0)
			mlx::Error("You cannot set a FPS cap to 0 (nice try)");
		else
			mlx->app->SetFPSCap(static_cast<std::uint32_t>(fps));
	}

	void mlx_destroy_context(mlx_context mlx)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		delete mlx;
		delete __internal_application_ptr;
		__internal_application_ptr = nullptr;
	}

	mlx_window mlx_new_window(mlx_context mlx, const mlx_window_create_info* info)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		return mlx->app->NewGraphicsSuport(info);
	}

	void mlx_destroy_window(mlx_context mlx, mlx_window win)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx->app->DestroyGraphicsSupport(win);
	}

	void mlx_set_window_position(mlx_context mlx, mlx_window win, int x, int y)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx::NonOwningPtr<mlx::GraphicsSupport> gs = mlx->app->GetGraphicsSupport(win);
		if(!gs && !gs->HasWindow())
			return;
		gs->GetWindow()->SetPosition(x, y);
	}

	void mlx_clear_window(mlx_context mlx, mlx_window win, int color)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx::NonOwningPtr<mlx::GraphicsSupport> gs = mlx->app->GetGraphicsSupport(win);
		if(!gs)
			return;
		gs->ResetRenderData(color);
	}

	void mlx_get_screen_size(mlx_context mlx, mlx_window win, int* w, int* h)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx::NonOwningPtr<mlx::GraphicsSupport> gs = mlx->app->GetGraphicsSupport(win);
		if(!gs)
			return;
		if(!gs->HasWindow())
		{
			*w = 0;
			*h = 0;
		}
		else
			gs->GetWindow()->GetScreenSizeWindowIsOn(w, h);
	}

	void mlx_loop_hook(mlx_context mlx, int (*f)(void*), void* param)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx->app->LoopHook(f, param);
	}

	void mlx_loop(mlx_context mlx)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx->app->Run();
	}

	void mlx_loop_end(mlx_context mlx)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx->app->LoopEnd();
	}

	void mlx_mouse_show(mlx_context mlx)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx::SDLManager::ShowCursor();
	}

	void mlx_mouse_hide(mlx_context mlx)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx::SDLManager::HideCursor();
	}

	void mlx_mouse_move(mlx_context mlx, mlx_window win, int x, int y)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx::NonOwningPtr<mlx::GraphicsSupport> gs = mlx->app->GetGraphicsSupport(win);
		if(!gs && !gs->HasWindow())
			return;
		gs->GetWindow()->MoveMouse(x, y);
	}

	void mlx_mouse_get_pos(mlx_context mlx, int* x, int* y)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx->app->GetMousePos(x, y);
	}

	void mlx_on_event(mlx_context mlx, mlx_window win, mlx_event_type event, int (*funct_ptr)(int, void*), void* param)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx->app->OnEvent(win, static_cast<int>(event), funct_ptr, param);
	}

	void mlx_pixel_put(mlx_context mlx, mlx_window win, int x, int y, int color)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx::NonOwningPtr<mlx::GraphicsSupport> gs = mlx->app->GetGraphicsSupport(win);
		if(!gs)
			return;
		gs->PixelPut(x, y, color);
	}

	mlx_image mlx_new_image(mlx_context mlx, int width, int height)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		if(width <= 0 || height <= 0)
		{
			mlx::Error("invalid image size (% x %)", width, height);
			return nullptr;
		}
		return mlx->app->NewTexture(width, height);
	}

	mlx_image mlx_new_image_from_file(mlx_context mlx, char* filename, int* width, int* height)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		if (filename == nullptr)
		{
			mlx::Error("PNG loader: filename is NULL");
			return nullptr;
		}
		std::filesystem::path file(filename);
		if( file.extension() != ".png" &&
			file.extension() != ".jpg" &&
			file.extension() != ".jpeg" &&
			file.extension() != ".bmp" &&
			file.extension() != ".dib")
		{
			mlx::Error("PNG loader: not a valid file format '%'", filename);
			return nullptr;
		}
		return mlx->app->NewStbTexture(filename, width, height);
	}

	void mlx_destroy_image(mlx_context mlx, mlx_image image)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx->app->DestroyTexture(image);
	}

	int mlx_get_image_pixel(mlx_context mlx, mlx_image image, int x, int y)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx::NonOwningPtr<mlx::Texture> texture = mlx->app->GetTexture(image);
		if(!texture)
			return 0;
		return texture->GetPixel(x, y);
	}

	void mlx_set_image_pixel(mlx_context mlx, mlx_image image, int x, int y, int color)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx::NonOwningPtr<mlx::Texture> texture = mlx->app->GetTexture(image);
		if(!texture)
			return;
		texture->SetPixel(x, y, color);
	}

	void mlx_put_image_to_window(mlx_context mlx, mlx_window win, mlx_image image, int x, int y)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx::NonOwningPtr<mlx::GraphicsSupport> gs = mlx->app->GetGraphicsSupport(win);
		if(!gs)
			return;
		mlx::NonOwningPtr<mlx::Texture> texture = mlx->app->GetTexture(image);
		if(!texture)
			return;
		gs->TexturePut(texture, x, y, 1.0f, 0.0f);
	}

	void mlx_string_put(mlx_context mlx, mlx_window win, int x, int y, int color, char* str)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		mlx::NonOwningPtr<mlx::GraphicsSupport> gs = mlx->app->GetGraphicsSupport(win);
		if(!gs)
			return;
		if(str == nullptr)
		{
			mlx::Error("invalid text (NULL)");
			return;
		}
		if(std::strlen(str) == 0)
		{
			mlx::Warning("trying to put an empty text");
			return;
		}
		gs->StringPut(x, y, color, str);
	}

	void mlx_set_font(mlx_context mlx, char* filepath)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		if (filepath == nullptr)
		{
			mlx::Error("Font loader: filepath is NULL");
			return;
		}
		std::filesystem::path file(filepath);
		if(std::strcmp(filepath, "default") != 0 && file.extension() != ".ttf" && file.extension() != ".tte")
		{
			mlx::Error("TTF loader: not a truetype font file '%'", filepath);
			return;
		}
		if(std::strcmp(filepath, "default") == 0)
			mlx->app->LoadFont(file, 6.f);
		else
			mlx->app->LoadFont(file, 16.f);
	}

	void mlx_set_font_scale(mlx_context mlx, char* filepath, float scale)
	{
		MLX_CHECK_APPLICATION_POINTER(mlx);
		if (filepath == nullptr)
		{
			mlx::Error("Font loader: filepath is NULL");
			return;
		}
		std::filesystem::path file(filepath);
		if(std::strcmp(filepath, "default") != 0 && file.extension() != ".ttf" && file.extension() != ".tte")
		{
			mlx::Error("TTF loader: not a truetype font file '%'", filepath);
			return;
		}
		mlx->app->LoadFont(file, scale);
	}
}