#include "screen.h"

#include <assert.h>
#include <glib.h>
#include <locale.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <X11/extensions/randr.h>
#include <X11/extensions/Xinerama.h>
#include <X11/extensions/Xrandr.h>
#include <X11/Xatom.h>
#include <X11/X.h>
#include <X11/Xlib.h>
#include <X11/Xresource.h>

#include "../log.h"
#include "../settings.h"
#include "../utils.h"
#include "x.h"

struct screen_info *screens;
int screens_len;

bool dunst_follow_errored = false;

static int randr_major_version = 0;
static int randr_minor_version = 0;

void randr_init(void);
void randr_update(void);
void xinerama_update(void);
void screen_update_fallback(void);
static void x_follow_setup_error_handler(void);
static int x_follow_tear_down_error_handler(void);
static int FollowXErrorHandler(Display *display, XErrorEvent *e);
static Window get_focused_window(void);


/**
 * A cache variable to cache the Xft.dpi xrdb values.
 * We do not expect to change xrdb often, but there's much
 * overhead to query it once.
 *
 * @retval -DBL_MAX: uncached
 * @retval     <=0: Invalid and unusable value
 * @retval      >0: valid
 */
double screen_dpi_xft_cache = -DBL_MAX;

void screen_dpi_xft_cache_purge(void)
{
        screen_dpi_xft_cache = -DBL_MAX;
}

static double screen_dpi_get_from_xft(void)
{
        if (screen_dpi_xft_cache == -DBL_MAX) {
                screen_dpi_xft_cache = 0;

                char *xrmType;
                XrmValue xrmValue;
                XrmDatabase db = XrmGetDatabase(xctx.dpy);
                ASSERT_OR_RET(db, screen_dpi_xft_cache);
                if (XrmGetResource(db, "Xft.dpi", "Xft.dpi", &xrmType, &xrmValue))
                        screen_dpi_xft_cache = strtod(xrmValue.addr, NULL);
        }
        return screen_dpi_xft_cache;
}

static double screen_dpi_get_from_monitor(const struct screen_info *scr)
{
        return (double)scr->h * 25.4 / (double)scr->mmh;
}

double screen_dpi_get(const struct screen_info *scr)
{
        if (  ! settings.force_xinerama
             && settings.per_monitor_dpi)
                return screen_dpi_get_from_monitor(scr);

        if (screen_dpi_get_from_xft() > 0)
                return screen_dpi_get_from_xft();

        // Calculate the DPI on the overall screen size.
        // xrandr --dpi <DPI> does only change the overall screen's millimeters,
        // but not the physical screen's sizes.
        //
        // The screen parameter is XDefaultScreen(), as our scr->id references
        // the xrandr monitor and not the xrandr screen
        return ((((double)DisplayWidth  (xctx.dpy, XDefaultScreen(xctx.dpy))) * 25.4) /
                 ((double)DisplayWidthMM(xctx.dpy, XDefaultScreen(xctx.dpy))));
}

void init_screens(void)
{
        if (settings.force_xinerama) {
                xinerama_update();
        } else {
                randr_init();
                randr_update();
        }
}

void alloc_screen_ar(int n)
{
        assert(n > 0);
        g_free(screens);
        screens = g_malloc0(n * sizeof(struct screen_info));
        screens_len = n;
}

void randr_init(void)
{
        int ignored;
        if (!XRRQueryExtension(xctx.dpy, &ignored, &ignored)) {
                LOG_W("Could not initialize the RandR extension. "
                      "Falling back to single monitor mode.");
                return;
        }
        XRRQueryVersion(xctx.dpy, &randr_major_version, &randr_minor_version);
        XRRSelectInput(xctx.dpy, RootWindow(xctx.dpy, DefaultScreen(xctx.dpy)), RRScreenChangeNotifyMask);
}

void randr_update(void)
{
        if (randr_major_version < 1
            || (randr_major_version == 1 && randr_minor_version < 5)) {
                LOG_W("Server RandR version too low (%i.%i). "
                      "Falling back to single monitor mode.",
                      randr_major_version,
                      randr_minor_version);
                screen_update_fallback();
                return;
        }

        int n = 0;
        XRRMonitorInfo *m = XRRGetMonitors(xctx.dpy, RootWindow(xctx.dpy, DefaultScreen(xctx.dpy)), true, &n);

        if (n < 1) {
                LOG_C("Get monitors reported %i monitors. "
                      "Falling back to single monitor mode.", n);
                screen_update_fallback();
                return;
        }

        assert(m);

        alloc_screen_ar(n);

        for (int i = 0; i < n; i++) {
                screens[i].id = i;
                screens[i].x = m[i].x;
                screens[i].y = m[i].y;
                screens[i].w = m[i].width;
                screens[i].h = m[i].height;
                screens[i].mmh = m[i].mheight;
                screens[i].dpi = screen_dpi_get(&screens[i]);
        }

        XRRFreeMonitors(m);
}

bool screen_check_event(XEvent *ev)
{
        if (XRRUpdateConfiguration(ev)) {
                LOG_D("XEvent: processing 'RRScreenChangeNotify'");
                randr_update();

                return true;
        }
        return false;
}

void xinerama_update(void)
{
        int n;
        XineramaScreenInfo *info = XineramaQueryScreens(xctx.dpy, &n);

        if (!info) {
                LOG_W("Could not get xinerama screen info. "
                      "Falling back to single monitor mode.");
                screen_update_fallback();
                return;
        }

        alloc_screen_ar(n);

        for (int i = 0; i < n; i++) {
                screens[i].id = i;
                screens[i].x = info[i].x_org;
                screens[i].y = info[i].y_org;
                screens[i].h = info[i].height;
                screens[i].w = info[i].width;
        }
        XFree(info);
}

void screen_update_fallback(void)
{
        alloc_screen_ar(1);

        int screen;
        if (settings.monitor >= 0)
                screen = settings.monitor;
        else
                screen = DefaultScreen(xctx.dpy);

        screens[0].w = DisplayWidth(xctx.dpy, screen);
        screens[0].h = DisplayHeight(xctx.dpy, screen);
}

/* see screen.h */
bool have_fullscreen_window(void)
{
        return window_is_fullscreen(get_focused_window());
}

/**
 * X11 ErrorHandler to mainly discard BadWindow parameter error
 */
static int XErrorHandlerFullscreen(Display *display, XErrorEvent *e)
{
        /* Ignore BadWindow errors. Window may have been gone */
        if (e->error_code == BadWindow) {
                return 0;
        }

        char err_buf[BUFSIZ];
        XGetErrorText(display, e->error_code, err_buf, BUFSIZ);
        fputs(err_buf, stderr);
        fputs("\n", stderr);

        return 0;
}

/* see screen.h */
bool window_is_fullscreen(Window window)
{
        bool fs = false;

        ASSERT_OR_RET(window, false);

        Atom has_wm_state = XInternAtom(xctx.dpy, "_NET_WM_STATE", True);
        if (has_wm_state == None){
                return false;
        }

        XFlush(xctx.dpy);
        XSetErrorHandler(XErrorHandlerFullscreen);

        Atom actual_type_return;
        int actual_format_return;
        unsigned long bytes_after_return;
        unsigned char *prop_to_return;
        unsigned long n_items;
        int result = XGetWindowProperty(
                        xctx.dpy,
                        window,
                        has_wm_state,
                        0,                     /* long_offset */
                        sizeof(window),        /* long_length */
                        false,                 /* delete */
                        AnyPropertyType,       /* req_type */
                        &actual_type_return,
                        &actual_format_return,
                        &n_items,
                        &bytes_after_return,
                        &prop_to_return);

        XFlush(xctx.dpy);
        XSync(xctx.dpy, false);
        XSetErrorHandler(NULL);

        if (result == Success) {
                for(int i = 0; i < n_items; i++) {
                        Atom atom = ((Atom*) prop_to_return)[i];
                        if (!atom)
                                continue;

                        char *s = XGetAtomName(xctx.dpy, atom);
                        if (!s)
                                continue;

                        if (STR_EQ(s, "_NET_WM_STATE_FULLSCREEN"))
                                fs = true;
                        XFree(s);
                        if (fs)
                                break;
                }
        }

        if (prop_to_return)
                XFree(prop_to_return);

        return fs;
}

/*
 * Select the screen on which the Window
 * should be displayed.
 */
const struct screen_info *get_active_screen(void)
{
        int ret = 0;
        bool force_follow_mouse = false;

        if (settings.f_mode == FOLLOW_NONE) {
                if (settings.monitor >= 0 && settings.monitor < screens_len) {
                        ret = settings.monitor;
                }
                goto sc_cleanup;
        } else {
                int x, y;
                assert(settings.f_mode == FOLLOW_MOUSE
                                || settings.f_mode == FOLLOW_KEYBOARD);

                x_follow_setup_error_handler();

                Window root =
                        RootWindow(xctx.dpy, DefaultScreen(xctx.dpy));

                if (settings.f_mode == FOLLOW_KEYBOARD) {
                        Window focused = get_focused_window();

                        if (!focused) {
                                /*
                                 * If no window is focused, or the focus is set
                                 * to dynamically change to the root window of
                                 * the screen the pointer is on, force following
                                 * the mouse.
                                 */
                                force_follow_mouse = true;
                        } else {
                                Window child_return;
                                /*
                                 * The window with input focus might be on a
                                 * different X screen. Use the mouse location
                                 * in that case.
                                 */
                                force_follow_mouse = !XTranslateCoordinates(
                                                        xctx.dpy, focused,root,
                                                        0, 0, &x, &y,
                                                        &child_return);
                        }
                }

                if (settings.f_mode == FOLLOW_MOUSE || force_follow_mouse) {
                        int dummy;
                        unsigned int dummy_ui;
                        Window dummy_win;

                        XQueryPointer(xctx.dpy,
                                      root,
                                      &dummy_win,
                                      &dummy_win,
                                      &x,
                                      &y,
                                      &dummy,
                                      &dummy,
                                      &dummy_ui);
                }

                for (int i = 0; i < screens_len; i++) {
                        if (INRECT(x, y, screens[i].x, screens[i].y,
                                         screens[i].w, screens[i].h)) {
                                ret = i;
                        }
                }

                if (ret > 0)
                        goto sc_cleanup;

                /* something seems to be wrong. Fall back to default */
                ret = 0;
                goto sc_cleanup;
        }
sc_cleanup:
        x_follow_tear_down_error_handler();
        assert(screens);
        assert(ret >= 0 && ret < screens_len);
        return &screens[ret];
}

/*
 * Return the window that currently has
 * the keyboard focus.
 */
static Window get_focused_window(void)
{
        Window focused;
        int ignored;

        XGetInputFocus(xctx.dpy, &focused, &ignored);

        if (focused == None || focused == PointerRoot)
                focused = 0;
        return focused;
}

static void x_follow_setup_error_handler(void)
{
        dunst_follow_errored = false;

        XFlush(xctx.dpy);
        XSetErrorHandler(FollowXErrorHandler);
}

static int x_follow_tear_down_error_handler(void)
{
        XFlush(xctx.dpy);
        XSync(xctx.dpy, false);
        XSetErrorHandler(NULL);
        return dunst_follow_errored;
}

static int FollowXErrorHandler(Display *display, XErrorEvent *e)
{
        dunst_follow_errored = true;
        char err_buf[BUFSIZ];
        XGetErrorText(display, e->error_code, err_buf, BUFSIZ);
        LOG_W("%s", err_buf);

        return 0;
}
/* vim: set ft=c tabstop=8 shiftwidth=8 expandtab textwidth=0: */
