#include "event.h"
#include "openbox.h"
#include "client.h"
#include "frame.h"
#include "screen.h"
#include "prop.h"
#include "dispatch.h"
#include "focus.h"
#include "parse.h"
#include "engine.h"

#include <X11/Xlib.h>
#include <glib.h>

Client *focus_client = NULL;
GList **focus_order = NULL; /* these lists are created when screen_startup
                               sets the number of desktops */

Window focus_backup = None;
gboolean focus_new = TRUE;
gboolean focus_follow = FALSE;
static gboolean focus_last = TRUE;
static gboolean focus_last_on_desktop = TRUE;

static gboolean noreorder = 0;

static void parse_assign(char *name, ParseToken *value)
{
    if (!g_ascii_strcasecmp(name, "focusnew")) {
        if (value->type != TOKEN_BOOL)
            yyerror("invalid value");
        else {
            focus_new = value->data.bool;
        }
    } else if (!g_ascii_strcasecmp(name, "followmouse")) {
        if (value->type != TOKEN_BOOL)
            yyerror("invalid value");
        else {
            focus_follow = value->data.bool;
        }
    } else if (!g_ascii_strcasecmp(name, "focuslast")) {
        if (value->type != TOKEN_BOOL)
            yyerror("invalid value");
        else {
            focus_last = value->data.bool;
        }
    } else if (!g_ascii_strcasecmp(name, "focuslastondesktop")) {
        if (value->type != TOKEN_BOOL)
            yyerror("invalid value");
        else {
            focus_last_on_desktop = value->data.bool;
        }
    } else
        yyerror("invalid option");
    parse_free_token(value);
}

void focus_startup()
{
    /* create the window which gets focus when no clients get it. Have to
       make it override-redirect so we don't try manage it, since it is
       mapped. */
    XSetWindowAttributes attrib;

    focus_client = NULL;
    focus_new = TRUE;
    focus_follow = FALSE;
    focus_last = TRUE;
    focus_last_on_desktop = TRUE;

    attrib.override_redirect = TRUE;
    focus_backup = XCreateWindow(ob_display, ob_root,
				 -100, -100, 1, 1, 0,
                                 CopyFromParent, InputOutput, CopyFromParent,
                                 CWOverrideRedirect, &attrib);
    XMapRaised(ob_display, focus_backup);

    /* start with nothing focused */
    focus_set_client(NULL);

    parse_reg_section("focus", NULL, parse_assign);
}

void focus_shutdown()
{
    guint i;

    for (i = 0; i < screen_num_desktops; ++i)
        g_list_free(focus_order[i]);
    g_free(focus_order);
    focus_order = NULL;

    XDestroyWindow(ob_display, focus_backup);

    /* reset focus to root */
    XSetInputFocus(ob_display, PointerRoot, RevertToPointerRoot,
                   event_lasttime);
}

static void push_to_top(Client *client)
{
    guint desktop;

    desktop = client->desktop;
    if (desktop == DESKTOP_ALL) desktop = screen_desktop;
    focus_order[desktop] = g_list_remove(focus_order[desktop], client);
    focus_order[desktop] = g_list_prepend(focus_order[desktop], client);
}

void focus_set_client(Client *client)
{
    Window active;
    Client *old;

    /* uninstall the old colormap, and install the new one */
    screen_install_colormap(focus_client, FALSE);
    screen_install_colormap(client, TRUE);

    if (client == NULL) {
	/* when nothing will be focused, send focus to the backup target */
	XSetInputFocus(ob_display, focus_backup, RevertToPointerRoot,
                       event_lasttime);
        XSync(ob_display, FALSE);
    }

    old = focus_client;
    focus_client = client;

    /* move to the top of the list */
    if (noreorder)
        --noreorder;
    else if (client != NULL)
        push_to_top(client);

    /* set the NET_ACTIVE_WINDOW hint */
    active = client ? client->window : None;
    PROP_SET32(ob_root, net_active_window, window, active);

    if (focus_client != NULL)
        dispatch_client(Event_Client_Focus, focus_client, 0, 0);
    if (old != NULL)
        dispatch_client(Event_Client_Unfocus, old, 0, 0);
}

static gboolean focus_under_pointer()
{
    Window w;
    int i, x, y;
    guint u;
    GList *it;

    if (XQueryPointer(ob_display, ob_root, &w, &w, &x, &y, &i, &i, &u)) {
        for (it = stacking_list; it != NULL; it = it->next) {
            Client *c = it->data;
            if (c->desktop == screen_desktop &&
                RECT_CONTAINS(c->frame->area, x, y))
                break;
        }
        if (it != NULL)
            return client_normal(it->data) && client_focus(it->data);
    }
    return FALSE;
}

void focus_fallback(gboolean switching_desks)
{
    GList *it;
    Client *old = NULL;

    old = focus_client;

    /* unfocus any focused clients.. they can be focused by Pointer events
       and such, and then when I try focus them, I won't get a FocusIn event
       at all for them.
    */
    focus_set_client(NULL);

    if (switching_desks) {
        /* don't skip any windows when switching desktops */
        old = NULL;
    }

    if (!(switching_desks ? focus_last_on_desktop : focus_last)) {
        if (focus_follow) focus_under_pointer();
        return;
    }

    for (it = focus_order[screen_desktop]; it != NULL; it = it->next)
        if (it->data != old && client_normal(it->data))
            if (client_focus(it->data))
                return;

    /* nothing to focus */
    focus_set_client(NULL);
}

void focus_cycle(gboolean forward, gboolean linear, gboolean done,
                 gboolean cancel)
{
    static Client *first = NULL;
    static Client *t = NULL;
    static GList *order = NULL;
    GList *it, *start, *list;
    Client *ft;

    if (cancel) {
        if (first) client_focus(first);
        goto done_cycle;
    } else if (done) {
        if (focus_client) {
            push_to_top(focus_client); /* move to top of focus_order */
            stacking_raise(focus_client);
        }
        goto done_cycle;
    }
    if (!first) first = focus_client;

    if (linear) list = client_list;
    else        list = focus_order[screen_desktop];

    start = it = g_list_find(list, focus_client);
    if (!start) /* switched desktops or something? */
        start = it = forward ? g_list_last(list) : g_list_first(list);
    if (!start) goto done_cycle;

    do {
        if (forward) {
            it = it->next;
            if (it == NULL) it = list;
        } else {
            it = it->prev;
            if (it == NULL) it = g_list_last(list);
        }
        ft = client_focus_target(it->data);
        if (ft == it->data && focus_client != ft && client_focusable(ft)) {
            if (client_focus(ft)) {
                noreorder++; /* avoid reordering the focus_order */
                break;
            }
        }
    } while (it != start);
    return;

done_cycle:
    t = NULL;
    first = NULL;
    g_list_free(order);
    order = NULL;
}