#include "../include/mgl/mgl.h"
#include <X11/Xutil.h>
#include <X11/XKBlib.h>
#include <stdio.h>
#ifndef NDEBUG
#include <stdlib.h>
#endif

static mgl_context context;
static int init_count = 0;
static XErrorHandler prev_xerror = NULL;
static XIOErrorHandler prev_xioerror = NULL;

static int ignore_xerror(Display *display, XErrorEvent *ee) {
    (void)display;
    (void)ee;
    return 0;
}

static int ignore_xioerror(Display *display) {
    (void)display;
    return 0;
}

int mgl_init(void) {
    ++init_count;
    if(init_count == 1) {
        context.connection = XOpenDisplay(NULL);
        if(!context.connection) {
            fprintf(stderr, "mgl error: XOpenDisplay failed\n");
            mgl_deinit();
            return -1;
        }

        prev_xerror = XSetErrorHandler(ignore_xerror);
        prev_xioerror = XSetIOErrorHandler(ignore_xioerror);
        XInitThreads();
        XkbSetDetectableAutoRepeat(context.connection, True, NULL);

        context.wm_delete_window_atom = XInternAtom(context.connection, "WM_DELETE_WINDOW", False);
        context.net_wm_ping_atom = XInternAtom(context.connection, "_NET_WM_PING", False);
        context.net_wm_pid_atom = XInternAtom(context.connection, "_NET_WM_PID", False);

        if(mgl_gl_load(&context.gl) != 0) {
            mgl_deinit();
            return -1;
        }
    }
    return 0;
}

void mgl_deinit(void) {
    if(init_count == 1) {
        if(context.connection) {
            XSetIOErrorHandler(prev_xioerror);
            prev_xioerror = NULL;

            XSetErrorHandler(prev_xerror);
            prev_xerror = NULL;

            XCloseDisplay(context.connection);
            context.connection = NULL;

            /*
                GLX needs to be unloaded after closing the display on nvidia because
                nvidia registers cleanup callbacks on exit, that uses the x11 display.
            */
            mgl_gl_unload(&context.gl);
        }
    }

    if(init_count > 0)
        --init_count;
}

mgl_context* mgl_get_context(void) {
#ifndef NDEBUG
    if(init_count == 0) {
        fprintf(stderr, "mgl error: mgl_get_context was called before mgl_init\n");
        abort();
    }
#endif
    return &context;
}