/* See LICENSE file for copyright and license details. */

#include <stdlib.h>
#include <string.h>
#include <X11/Xatom.h>
#include <X11/Xutil.h>

#include "jdwm.h"
#include "layout.h"
#include "tag.h"
#include "layouts/floating.h"
#include "util.h"

int blw = 0;

/* static */

static char prop[128];

/* extern */
extern Layout ** taglayouts;
extern int wax, way, wah, waw;  /* windowarea geometry */
extern Window barwin;
extern Client *clients, *sel;   /* global client list */
extern Atom jdwmprops;
extern DC dc;

void
arrange(Display * disp, jdwm_config *jdwmconf)
{
    Client *c;

    for(c = clients; c; c = c->next)
        if(isvisible(c, jdwmconf->selected_tags, jdwmconf->ntags))
            unban(c);
        else
            ban(c);
    jdwmconf->current_layout->arrange(disp, jdwmconf);
    focus(disp, &dc, NULL, jdwmconf);
    restack(disp, jdwmconf);
}

void
uicb_focusnext(Display *disp __attribute__ ((unused)),
               jdwm_config * jdwmconf,
               const char *arg __attribute__ ((unused)))
{
    Client *c;

    if(!sel)
        return;
    for(c = sel->next; c && !isvisible(c, jdwmconf->selected_tags, jdwmconf->ntags); c = c->next);
    if(!c)
        for(c = clients; c && !isvisible(c, jdwmconf->selected_tags, jdwmconf->ntags); c = c->next);
    if(c)
    {
        focus(c->display, &dc, c, jdwmconf);
        restack(c->display, jdwmconf);
    }
}

void
uicb_focusprev(Display *disp __attribute__ ((unused)),
               jdwm_config *jdwmconf,
               const char *arg __attribute__ ((unused)))
{
    Client *c;

    if(!sel)
        return;
    for(c = sel->prev; c && !isvisible(c, jdwmconf->selected_tags, jdwmconf->ntags); c = c->prev);
    if(!c)
    {
        for(c = clients; c && c->next; c = c->next);
        for(; c && !isvisible(c, jdwmconf->selected_tags, jdwmconf->ntags); c = c->prev);
    }
    if(c)
    {
        focus(c->display, &dc, c, jdwmconf);
        restack(c->display, jdwmconf);
    }
}

void
initlayouts(jdwm_config * jdwmconf)
{
    int w, i;

    for(blw = i = 0; i < jdwmconf->nlayouts; i++)
    {
        w = textw(jdwmconf->layouts[i].symbol);
        if(w > blw)
            blw = w;
    }

    taglayouts = p_new(Layout *, jdwmconf->ntags);
    for(i = 0; i < jdwmconf->ntags; i++)
        taglayouts[i] = jdwmconf->layouts;
}

void
loadjdwmprops(Display *disp, jdwm_config * jdwmconf)
{
    int i;

    if(gettextprop(disp, DefaultRootWindow(disp), jdwmprops, prop, sizeof(prop)))
    {
        for(i = 0; i < jdwmconf->ntags && i < ssizeof(prop) - 1 && prop[i] != '\0'; i++)
            jdwmconf->selected_tags[i] = prop[i] == '1';
    }
}

inline Client *
nexttiled(Client * c, Bool * tags, int ntags)
{
    for(; c && (c->isfloating || !isvisible(c, tags, ntags)); c = c->next);
    return c;
}

void
restack(Display * disp, jdwm_config *jdwmconf)
{
    Client *c;
    XEvent ev;
    XWindowChanges wc;

    drawstatus(disp, jdwmconf);
    if(!sel)
        return;
    if(sel->isfloating || IS_ARRANGE(floating))
        XRaiseWindow(disp, sel->win);
    if(!IS_ARRANGE(floating))
    {
        wc.stack_mode = Below;
        wc.sibling = barwin;
        if(!sel->isfloating)
        {
            XConfigureWindow(disp, sel->win, CWSibling | CWStackMode, &wc);
            wc.sibling = sel->win;
        }
        for(c = nexttiled(clients, jdwmconf->selected_tags, jdwmconf->ntags); c; c = nexttiled(c->next, jdwmconf->selected_tags, jdwmconf->ntags))
        {
            if(c == sel)
                continue;
            XConfigureWindow(disp, c->win, CWSibling | CWStackMode, &wc);
            wc.sibling = c->win;
        }
    }
    XSync(disp, False);
    while(XCheckMaskEvent(disp, EnterWindowMask, &ev));
}

void
savejdwmprops(Display *disp, jdwm_config *jdwmconf)
{
    int i;
    for(i = 0; i < jdwmconf->ntags && i < ssizeof(prop) - 1; i++)
        prop[i] = jdwmconf->selected_tags[i] ? '1' : '0';
    prop[i] = '\0';
    XChangeProperty(disp, DefaultRootWindow(disp), jdwmprops, XA_STRING, 8, PropModeReplace, (unsigned char *) prop, i);
}

void
uicb_setlayout(Display *disp, jdwm_config * jdwmconf, const char *arg)
{
    int i, j;
    Client *c;

    if(!arg)
    {
        if(!(++jdwmconf->current_layout)->symbol)
            jdwmconf->current_layout = &jdwmconf->layouts[0];
    }
    else
    {
        i = strtol(arg, NULL, 10);
        if(i < 0 || i >= jdwmconf->nlayouts)
            return;
        jdwmconf->current_layout = &jdwmconf->layouts[i];
    }

    for(c = clients; c; c = c->next)
        c->ftview = True;

    if(sel)
        arrange(disp, jdwmconf);
    else
        drawstatus(disp, jdwmconf);

    savejdwmprops(disp, jdwmconf);

    for(j = 0; j < jdwmconf->ntags; j++)
        if (jdwmconf->selected_tags[j])
            taglayouts[j] = jdwmconf->current_layout;
}

void
uicb_togglebar(Display *disp,
               jdwm_config *jdwmconf,
               const char *arg __attribute__ ((unused)))
{
    if(jdwmconf->current_bpos == BarOff)
        jdwmconf->current_bpos = (jdwmconf->bpos == BarOff) ? BarTop : jdwmconf->bpos;
    else
        jdwmconf->current_bpos = BarOff;
    updatebarpos(disp, jdwmconf->current_bpos);
    arrange(disp, jdwmconf);
}

static void
maximize(int x, int y, int w, int h, jdwm_config *jdwmconf)
{
    XEvent ev;

    if(!sel)
        return;

    if((sel->ismax = !sel->ismax))
    {
        sel->wasfloating = sel->isfloating;
        sel->isfloating = True;
        sel->rx = sel->x;
        sel->ry = sel->y;
        sel->rw = sel->w;
        sel->rh = sel->h;
        resize(sel, x, y, w, h, True);
    }
    else if(sel->isfloating)
        resize(sel, sel->rx, sel->ry, sel->rw, sel->rh, True);
    else
        sel->isfloating = False;

    drawstatus(sel->display, jdwmconf);

    while(XCheckMaskEvent(sel->display, EnterWindowMask, &ev));
}

void
uicb_togglemax(Display *disp __attribute__ ((unused)),
               jdwm_config *jdwmconf,
               const char *arg __attribute__ ((unused)))
{
    maximize(wax, way, waw - 2 * jdwmconf->borderpx, wah - 2 * jdwmconf->borderpx, jdwmconf);
}

void
uicb_toggleverticalmax(Display *disp __attribute__ ((unused)),
                       jdwm_config *jdwmconf,
                       const char *arg __attribute__ ((unused)))
{
    if(sel)
        maximize(sel->x, way, sel->w, wah - 2 * jdwmconf->borderpx, jdwmconf);
}


void
uicb_togglehorizontalmax(Display *disp __attribute__ ((unused)),
                         jdwm_config *jdwmconf,
                         const char *arg __attribute__ ((unused)))
{
    if(sel)
        maximize(wax, sel->y, waw - 2 * jdwmconf->borderpx, sel->h, jdwmconf);
}

void 
uicb_zoom(Display *disp __attribute__ ((unused)), 
          jdwm_config *jdwmconf,
          const char *arg __attribute__ ((unused))) 
{ 
    Client *c;
    if(!sel || ((c = sel) == nexttiled(clients, jdwmconf->selected_tags, jdwmconf->ntags) && !(c = nexttiled(c->next, jdwmconf->selected_tags, jdwmconf->ntags))))
        return;
    detach(c);
    attach(c);
    focus(c->display, &dc, c, jdwmconf);
    arrange(c->display, jdwmconf);
}