/*
 * layout.c - layout management
 *
 * Copyright © 2007 Julien Danjou <julien@danjou.info>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 */

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

#include "screen.h"
#include "layout.h"
#include "tag.h"
#include "util.h"
#include "statusbar.h"
#include "layouts/floating.h"

/** Arrange windows following current selected layout
 * \param disp display ref
 * \param awesomeconf awesome config
 */
void
arrange(Display * disp, awesome_config *awesomeconf)
{
    Client *c;

    for(c = *awesomeconf->clients; c; c = c->next)
    {
        if(isvisible(c, awesomeconf->screen, awesomeconf->tags, awesomeconf->ntags))
            unban(c);
        /* we don't touch other screens windows */
        else if(c->screen == awesomeconf->screen)
            ban(c);
    }
    awesomeconf->current_layout->arrange(disp, awesomeconf);
    focus(disp, NULL, True, awesomeconf);
    restack(disp, awesomeconf);
}

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

    if(!*awesomeconf->client_sel)
        return;
    for(c = (*awesomeconf->client_sel)->next; c && !isvisible(c, awesomeconf->screen, awesomeconf->tags, awesomeconf->ntags); c = c->next);
    if(!c)
        for(c = *awesomeconf->clients; c && !isvisible(c, awesomeconf->screen, awesomeconf->tags, awesomeconf->ntags); c = c->next);
    if(c)
    {
        focus(c->display, c, True, awesomeconf);
        restack(c->display, awesomeconf);
    }
}

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

    if(!*awesomeconf->client_sel)
        return;
    for(c = (*awesomeconf->client_sel)->prev; c && !isvisible(c, awesomeconf->screen, awesomeconf->tags, awesomeconf->ntags); c = c->prev);
    if(!c)
    {
        for(c = *awesomeconf->clients; c && c->next; c = c->next);
        for(; c && !isvisible(c, awesomeconf->screen, awesomeconf->tags, awesomeconf->ntags); c = c->prev);
    }
    if(c)
    {
        focus(c->display, c, True, awesomeconf);
        restack(c->display, awesomeconf);
    }
}

void
loadawesomeprops(Display *disp, awesome_config * awesomeconf)
{
    int i;
    char *prop;

    prop = p_new(char, awesomeconf->ntags + 1);

    if(xgettextprop(disp, RootWindow(disp, awesomeconf->phys_screen), AWESOMEPROPS_ATOM(disp), prop, awesomeconf->ntags + 1))
        for(i = 0; i < awesomeconf->ntags && prop[i]; i++)
            if(prop[i] == '1')
            {
                awesomeconf->tags[i].selected = True;
                awesomeconf->current_layout = awesomeconf->tags[i].layout;
            }
            else
                awesomeconf->tags[i].selected = False;

    p_delete(&prop);
}

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

    drawstatusbar(disp, awesomeconf);
    if(!*awesomeconf->client_sel)
        return;
    if((*awesomeconf->client_sel)->isfloating || IS_ARRANGE(0, layout_floating))
        XRaiseWindow(disp, (*awesomeconf->client_sel)->win);
    if(!IS_ARRANGE(0, layout_floating))
    {
        wc.stack_mode = Below;
        wc.sibling = awesomeconf->statusbar.window;
        if(!(*awesomeconf->client_sel)->isfloating)
        {
            XConfigureWindow(disp, (*awesomeconf->client_sel)->win, CWSibling | CWStackMode, &wc);
            wc.sibling = (*awesomeconf->client_sel)->win;
        }
        for(c = *awesomeconf->clients; c; c = c->next)
        {
            if(!IS_TILED(c, awesomeconf->screen, awesomeconf->tags, awesomeconf->ntags) || c == *awesomeconf->client_sel)
                continue;
            XConfigureWindow(disp, c->win, CWSibling | CWStackMode, &wc);
            wc.sibling = c->win;
        }
    }
    if(awesomeconf->focus_move_pointer)
        XWarpPointer(disp, None, (*awesomeconf->client_sel)->win, 0, 0, 0, 0, (*awesomeconf->client_sel)->w / 2, (*awesomeconf->client_sel)->h / 2);
    XSync(disp, False);
    while(XCheckMaskEvent(disp, EnterWindowMask, &ev));
}

void
saveawesomeprops(Display *disp, awesome_config *awesomeconf)
{
    int i;
    char *prop;

    prop = p_new(char, awesomeconf->ntags + 1);
    for(i = 0; i < awesomeconf->ntags; i++)
        prop[i] = awesomeconf->tags[i].selected ? '1' : '0';
    prop[i] = '\0';
    XChangeProperty(disp, RootWindow(disp, awesomeconf->phys_screen),
                    AWESOMEPROPS_ATOM(disp), XA_STRING, 8,
                    PropModeReplace, (unsigned char *) prop, i);
    p_delete(&prop);
}

void
uicb_setlayout(Display *disp,
               awesome_config * awesomeconf,
               const char *arg)
{
    int i;
    Client *c;

    if(arg)
    {
        for(i = 0; i < awesomeconf->nlayouts && &awesomeconf->layouts[i] != awesomeconf->current_layout; i++);
        i = compute_new_value_from_arg(arg, (double) i);
        if(i >= awesomeconf->nlayouts)
            i = 0;
        else if(i < 0)
            i = awesomeconf->nlayouts - 1;
    }
    else
        i = 0;

    awesomeconf->current_layout = &awesomeconf->layouts[i];

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

    if(*awesomeconf->client_sel)
        arrange(disp, awesomeconf);
    else
        drawstatusbar(disp, awesomeconf);

    saveawesomeprops(disp, awesomeconf);

    for(i = 0; i < awesomeconf->ntags; i++)
        if (awesomeconf->tags[i].selected)
            awesomeconf->tags[i].layout = awesomeconf->current_layout;
}

static void
maximize(int x, int y, int w, int h, awesome_config *awesomeconf)
{
    if(!*awesomeconf->client_sel)
        return;

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

    arrange(awesomeconf->display, awesomeconf);
}

void
uicb_togglemax(Display *disp,
               awesome_config *awesomeconf,
               const char *arg __attribute__ ((unused)))
{
    ScreenInfo *si = get_screen_info(disp, awesomeconf->screen, &awesomeconf->statusbar);

    maximize(si[awesomeconf->screen].x_org, si[awesomeconf->screen].y_org,
             si[awesomeconf->screen].width - 2 * awesomeconf->borderpx,
             si[awesomeconf->screen].height - 2 * awesomeconf->borderpx,
             awesomeconf);
    XFree(si);
}

void
uicb_toggleverticalmax(Display *disp,
                       awesome_config *awesomeconf,
                       const char *arg __attribute__ ((unused)))
{
    ScreenInfo *si = get_screen_info(disp, awesomeconf->screen, &awesomeconf->statusbar);

    if(*awesomeconf->client_sel)
        maximize((*awesomeconf->client_sel)->x,
                 si[awesomeconf->screen].y_org,
                 (*awesomeconf->client_sel)->w,
                 si[awesomeconf->screen].height - 2 * awesomeconf->borderpx,
                 awesomeconf);
    XFree(si);
}


void
uicb_togglehorizontalmax(Display *disp,
                         awesome_config *awesomeconf,
                         const char *arg __attribute__ ((unused)))
{
    ScreenInfo *si = get_screen_info(disp, awesomeconf->screen, &awesomeconf->statusbar);

    if(*awesomeconf->client_sel)
        maximize(si[awesomeconf->screen].x_org,
                 (*awesomeconf->client_sel)->y,
                 si[awesomeconf->screen].height - 2 * awesomeconf->borderpx,
                 (*awesomeconf->client_sel)->h,
                 awesomeconf);
    XFree(si);
}

void
uicb_zoom(Display *disp __attribute__ ((unused)),
          awesome_config *awesomeconf,
          const char *arg __attribute__ ((unused)))
{
    if(!*awesomeconf->client_sel)
        return;
    detach(awesomeconf->clients, *awesomeconf->client_sel);
    attach(awesomeconf->clients, *awesomeconf->client_sel);
    focus(awesomeconf->display, *awesomeconf->client_sel, True, awesomeconf);
    arrange(awesomeconf->display, awesomeconf);
}