package io.gitlab.jfronny.glaunch;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import imgui.ImGui;
import imgui.ImGuiIO;
import imgui.flag.ImGuiConfigFlags;
import imgui.flag.ImGuiWindowFlags;
import imgui.gl3.ImGuiImplGl3;
import imgui.glfw.ImGuiImplGlfw;
import io.gitlab.jfronny.glaunch.gson.MinecraftArgumentDeserializer;
import io.gitlab.jfronny.glaunch.gson.RulesDeserializer;
import io.gitlab.jfronny.glaunch.model.MinecraftArgument;
import io.gitlab.jfronny.glaunch.model.Rules;
import io.gitlab.jfronny.glaunch.util.Utils;
import io.gitlab.jfronny.glaunch.windows.MainWindow;
import io.gitlab.jfronny.glaunch.windows.Window;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.lwjgl.glfw.*;
import org.lwjgl.opengl.GL;
import org.lwjgl.opengl.GL32;
import org.lwjgl.system.MemoryStack;
import org.lwjgl.system.MemoryUtil;

import java.io.IOException;
import java.io.InputStream;
import java.nio.IntBuffer;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.LinkedHashSet;
import java.util.Objects;
import java.util.Set;

//TODO generate gitignore
//TODO mods browser
//TODO load and launch instance from metadata
//TODO allow instance sync through metadata
//TODO update checker
public class GLaunch {
    public static final Set<Window> WINDOWS = new LinkedHashSet<>();
    public static final Logger LOGGER = LogManager.getFormatterLogger("GLaunch");
    public static final Gson GSON = new GsonBuilder()
            .registerTypeAdapter(MinecraftArgument.class, new MinecraftArgumentDeserializer())
            .registerTypeAdapter(Rules.class, new RulesDeserializer())
            .create();
    //TODO allow moving to .config or elsewhere
    public static final Path CACHE_DIR = Path.of("run/cache");
    public static final Path INSTANCE_DIR = Path.of("run/instances");
    public static final Path ASSETS_DIR = Path.of("run/assets");
    public static final Path LIBRARIES_DIR = Path.of("run/libraries");
    private static final Path CONFIG_PATH = Path.of("run/glaunch2.json");
    public static Config CONFIG;

    public static void main(String[] args) throws IOException {
        LOGGER.info("Setting up cache dir");
        if (!Files.exists(CONFIG_PATH)) Utils.writeObject(CONFIG_PATH, new Config());
        CONFIG = Utils.loadObject(CONFIG_PATH, Config.class);
        if (!Files.exists(CACHE_DIR)) Files.createDirectories(CACHE_DIR);
        if (!Files.exists(INSTANCE_DIR)) Files.createDirectories(INSTANCE_DIR);
        if (!Files.exists(ASSETS_DIR)) Files.createDirectories(ASSETS_DIR);
        if (!Files.exists(LIBRARIES_DIR)) Files.createDirectories(LIBRARIES_DIR);
        LOGGER.info("Initializing UI");
        WINDOWS.add(new MainWindow());
        GLaunch gLaunch = new GLaunch();
        gLaunch.init();
        gLaunch.run();
        gLaunch.dispose();
    }

    public static void open(Window window) {
        WINDOWS.add(window);
    }

    private final ImGuiImplGlfw imGuiGlfw = new ImGuiImplGlfw();
    private final ImGuiImplGl3 imGuiGl3 = new ImGuiImplGl3();

    private String glslVersion = null;

    /**
     * Pointer to the native GLFW window.
     */
    protected long handle;

    /**
     * Method to initialize application.
     */
    protected void init() {
        initWindow();
        initImGui();
        imGuiGlfw.init(handle, true);
        imGuiGl3.init(glslVersion);
    }

    /**
     * Method to dispose all used application resources and destroy its window.
     */
    protected void dispose() {
        imGuiGl3.dispose();
        imGuiGlfw.dispose();
        ImGui.destroyContext();
        Callbacks.glfwFreeCallbacks(handle);
        GLFW.glfwDestroyWindow(handle);
        GLFW.glfwTerminate();
        Objects.requireNonNull(GLFW.glfwSetErrorCallback(null)).free();
    }

    /**
     * Method to create and initialize GLFW window.
     */
    protected void initWindow() {
        GLFWErrorCallback.createPrint(System.err).set();

        if (!GLFW.glfwInit()) {
            throw new IllegalStateException("Unable to initialize GLFW");
        }

        decideGlGlslVersions();

        GLFW.glfwWindowHint(GLFW.GLFW_VISIBLE, GLFW.GLFW_FALSE);
        GLFW.glfwWindowHint(GLFW.GLFW_RESIZABLE, GLFW.GLFW_FALSE);
        handle = GLFW.glfwCreateWindow(10, 10, "GLaunch", MemoryUtil.NULL, MemoryUtil.NULL);

        if (handle == MemoryUtil.NULL) {
            throw new RuntimeException("Failed to create the GLFW window");
        }

        try (MemoryStack stack = MemoryStack.stackPush()) {
            final IntBuffer pWidth = stack.mallocInt(1); // int*
            final IntBuffer pHeight = stack.mallocInt(1); // int*

            GLFW.glfwGetWindowSize(handle, pWidth, pHeight);
            final GLFWVidMode vidmode = Objects.requireNonNull(GLFW.glfwGetVideoMode(GLFW.glfwGetPrimaryMonitor()));
            GLFW.glfwSetWindowPos(handle, (vidmode.width() - pWidth.get(0)) / 2, (vidmode.height() - pHeight.get(0)) / 2);
        }

        GLFW.glfwMakeContextCurrent(handle);

        GL.createCapabilities();

        GLFW.glfwSwapInterval(GLFW.GLFW_TRUE);

        clearBuffer();
        renderBuffer();
    }

    private void decideGlGlslVersions() {
        final boolean isMac = System.getProperty("os.name").toLowerCase().contains("mac");
        if (isMac) {
            glslVersion = "#version 150";
            GLFW.glfwWindowHint(GLFW.GLFW_CONTEXT_VERSION_MAJOR, 3);
            GLFW.glfwWindowHint(GLFW.GLFW_CONTEXT_VERSION_MINOR, 2);
            GLFW.glfwWindowHint(GLFW.GLFW_OPENGL_PROFILE, GLFW.GLFW_OPENGL_CORE_PROFILE);  // 3.2+ only
            GLFW.glfwWindowHint(GLFW.GLFW_OPENGL_FORWARD_COMPAT, GLFW.GLFW_TRUE);          // Required on Mac
        } else {
            glslVersion = "#version 130";
            GLFW.glfwWindowHint(GLFW.GLFW_CONTEXT_VERSION_MAJOR, 3);
            GLFW.glfwWindowHint(GLFW.GLFW_CONTEXT_VERSION_MINOR, 0);
        }
    }

    /**
     * Method to initialize Dear ImGui context. Could be overridden to do custom Dear ImGui setup before application start.
     */
    protected void initImGui() {
        ImGui.createContext();
        ImGuiIO io = ImGui.getIO();
        io.addConfigFlags(ImGuiConfigFlags.ViewportsEnable);
        //io.setConfigViewportsNoDecoration(false);
        io.setConfigViewportsNoAutoMerge(true);
        try (InputStream is = GLaunch.class.getClassLoader().getResourceAsStream("font.ttf")) {
            assert is != null;
            io.setFontDefault(io.getFonts().addFontFromMemoryTTF(is.readAllBytes(), 16f));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * Main application loop.
     */
    protected void run() {
        while (!GLFW.glfwWindowShouldClose(handle)) {
            //frame
            clearBuffer();
            imGuiGlfw.newFrame();
            if (CONFIG.darkTheme) ImGui.styleColorsDark();
            else ImGui.styleColorsLight();
            ImGui.newFrame();
            //render
            if (GLaunch.WINDOWS.isEmpty()) GLFW.glfwSetWindowShouldClose(handle, true); //TODO fix hs_error_pid
            else {
                for (Window window : GLaunch.WINDOWS.toArray(new Window[0])) {
                    if (window.isNew()) window.preFirstDraw();
                    ImGui.begin(window.getName(), ImGuiWindowFlags.MenuBar);
                    window.draw();
                    ImGui.end();
                }
            }
            //end frame
            ImGui.render();
            imGuiGl3.renderDrawData(ImGui.getDrawData());

            if (ImGui.getIO().hasConfigFlags(ImGuiConfigFlags.ViewportsEnable)) {
                final long backupWindowPtr = GLFW.glfwGetCurrentContext();
                ImGui.updatePlatformWindows();
                ImGui.renderPlatformWindowsDefault();
                GLFW.glfwMakeContextCurrent(backupWindowPtr);
            }

            renderBuffer();
        }
    }

    private void clearBuffer() {
        GL32.glClear(GL32.GL_COLOR_BUFFER_BIT | GL32.GL_DEPTH_BUFFER_BIT);
    }

    private void renderBuffer() {
        GLFW.glfwSwapBuffers(handle);
        GLFW.glfwPollEvents();
    }
}