/*
 * (C)opyright MMIV-MMVI Anselm R. Garbe <garbeam at gmail dot com>
 * See LICENSE file for license details.
 */

#include <assert.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <X11/Xlib.h>
#include <X11/Xatom.h>
#include <X11/cursorfont.h>
#include <X11/Xproto.h>
#include <X11/Xutil.h>

#include "wmii.h"

/* array indexes for file pointers */
typedef enum {
    B_CTL,
    B_NEW,
    B_EXPANDABLE,
    B_GEOMETRY,
    B_FONT,
    B_FG_COLOR,
    B_BORDER_COLOR,
    B_BG_COLOR,
    B_EVENT_RESIZE,
    B_LAST
} BarIndexes;

typedef struct Item Item;
struct Item {
    File *root;
    Draw d;
    Item *next;
};

static unsigned int id = 1;
static IXPServer *ixps = 0;
static Display *dpy;
static GC gc;
static Window win;
static XRectangle rect;
static XRectangle brect;
static int screen_num;
static Bool displayed = False;
static char *sockfile = nil;
static File *file[B_LAST];
static Item *items = nil;
static size_t nitems = 0;
static Pixmap pmap;
static XFontStruct *font;
static Align align = SOUTH;

static Draw zero_draw = { 0 };

static void draw_bar(void *obj, char *arg);
static void quit(void *obj, char *arg);
static void display(void *obj, char *arg);
static void reset(void *obj, char *arg);
static void xdestroy(void *obj, char *arg);
static void handle_after_write(IXPServer * s, File * f);

static Action acttbl[] = {
    {"quit", quit},
    {"display", display},
    {"update", draw_bar},
    {"reset", reset},
    {"destroy", xdestroy},
    {0, 0}
};

static char *version[] = {
    "wmiibar - window manager improved bar - " VERSION "\n"
        "  (C)opyright MMIV-MMVI Anselm R. Garbe\n", 0
};

static void
usage()
{
    fprintf(stderr, "%s",
            "usage: wmiibar -s <socket file> [-v]\n"
            "      -s    socket file\n" "      -v    version info\n");
    exit(1);
}

/**
 * <path>/data          "<txt value>"
 * <path>/fgcolor      "#RRGGBBAA"
 * <path>/bgcolor      "#RRGGBBAA"
 * <path>/bordercolor  "#RRGGBBAA"
 * <path>/b1press       "<command>"
 * <path>/b2press       "<command>"
 * <path>/b3press       "<command>"
 * <path>/b4press       "<command>"
 * <path>/b5press       "<command>"
 */
static void
create_label(char *path)
{
    File *f;
    char buf[MAX_BUF];
    int i;

    snprintf(buf, MAX_BUF, "%s/data", path);
    f = ixp_create(ixps, buf);
    f->after_write = handle_after_write;
    snprintf(buf, MAX_BUF, "%s/fgcolor", path);
    wmii_create_ixpfile(ixps, buf, file[B_FG_COLOR]->content);
    snprintf(buf, MAX_BUF, "%s/bgcolor", path);
    wmii_create_ixpfile(ixps, buf, file[B_BG_COLOR]->content);
    snprintf(buf, MAX_BUF, "%s/bordercolor", path);
    wmii_create_ixpfile(ixps, buf, file[B_BORDER_COLOR]->content);
    for(i = 1; i < 6; i++) {    /* 5 buttons events */
        snprintf(buf, MAX_BUF, "%s/b%dpress", path, i);
        ixp_create(ixps, buf);
    }
}

static void
xdestroy(void *obj, char *arg)
{
    char buf[512];
    if(!arg)
        return;
    snprintf(buf, sizeof(buf), "/%s", arg);
    ixps->remove(ixps, buf);
    draw_bar(0, 0);
}

static void
reset(void *obj, char *arg)
{
    int i;
    char buf[512];
    for(i = 1; i <= id; i++) {
        snprintf(buf, sizeof(buf), "/%d", i);
        ixps->remove(ixps, buf);
    }
    id = 1;
    draw_bar(0, 0);
}

static void
quit(void *obj, char *arg)
{
    ixps->runlevel = SHUTDOWN;
}

static void
display(void *obj, char *arg)
{
    if(!arg)
        return;
    displayed = blitz_strtonum(arg, 0, 1);
    if(displayed) {
        XMapRaised(dpy, win);
        draw_bar(0, 0);
    } else {
        XUnmapWindow(dpy, win);
        XSync(dpy, False);
    }
}

static void
init_draw_label(char *path, Draw * d)
{
    char buf[MAX_BUF];
    File *f;

    /* text stuff */
    snprintf(buf, MAX_BUF, "%s/data", path);
    f = ixp_walk(ixps, buf);
    d->data = f->content;
    /* style stuff */
    snprintf(buf, MAX_BUF, "%s/fgcolor", path);
    f = ixp_walk(ixps, buf);
    d->fg = blitz_loadcolor(dpy, screen_num, f->content);
    snprintf(buf, MAX_BUF, "%s/bgcolor", path);
    f = ixp_walk(ixps, buf);
    d->bg = blitz_loadcolor(dpy, screen_num, f->content);
    snprintf(buf, MAX_BUF, "%s/bordercolor", path);
    f = ixp_walk(ixps, buf);
    d->border = blitz_loadcolor(dpy, screen_num, f->content);
}

static void
init_item(char *path, Item * i)
{
    i->d = zero_draw;
    i->root = ixp_walk(ixps, path);
    i->d.gc = gc;
    i->d.drawable = pmap;
    i->d.rect = brect;
    i->d.rect.y = 0;
    init_draw_label(path, &i->d);
}

static int
comp_str(const void *s1, const void *s2)
{
    return strcmp(*(char **) s1, *(char **) s2);
}

static void
draw()
{
    Item *i;
    unsigned int idx = 0, w = 0, xoff = 0;
    unsigned expandable = 0;
    char buf[32];

    if(!nitems)
        return;

    expandable = blitz_strtonum(file[B_EXPANDABLE]->content, 1, nitems + 1);
    snprintf(buf, sizeof(buf), "/%d", expandable);
	expandable--;
    if(!ixp_walk(ixps, buf))
        expandable = 0;

    /* precalc */
    for(i = items; i; i = i->next) {
        if(idx != expandable) {
            i->d.rect.width = brect.height;
            if(i->d.data) {
                if(!strncmp(i->d.data, "%m:", 3))
                    /* meter */
                    i->d.rect.width = brect.height / 2;
                else
                    i->d.rect.width +=
                        XTextWidth(font, i->d.data, strlen(i->d.data));
            }
            w += i->d.rect.width;
        }
        idx++;
    }
    if(w > brect.width) {
        /* failsafe mode, give all labels same width */
        w = brect.width / nitems;
        for(i = items; i && i->next; i = i->next)
            i->d.rect.width = w;
        i->d.rect.width = w;
        i->d.rect.width = brect.width - i->d.rect.x;
    } else {
        idx = 0;
        for(i = items; i && idx != expandable; i = i->next)
            idx++;
        i->d.rect.width = brect.width - w;
    }
    for(i = items; i; i = i->next) {
        i->d.font = font;
        i->d.rect.x = xoff;
        xoff += i->d.rect.width;
        if(i->d.data && !strncmp(i->d.data, "%m:", 3))
            blitz_drawmeter(dpy, &i->d);
        else
            blitz_drawlabel(dpy, &i->d);
    }
    XCopyArea(dpy, pmap, win, gc, 0, 0, brect.width, brect.height, 0, 0);
    XSync(dpy, False);
}

static void
draw_bar(void *obj, char *arg)
{
    File *label = 0;
    unsigned int j = 0;
    Item *i, *it;
    char buf[512];

    if(!displayed)
        return;
    while((i = items)) {
        items = items->next;
        free(i);
    }
    it = items = nil;
    snprintf(buf, sizeof(buf), "%s", "/1");
    label = ixp_walk(ixps, buf);
    if(!label) {
        Draw d = { 0 };
        /* default stuff */
        d.gc = gc;
        d.drawable = pmap;
        d.rect.width = brect.width;
        d.rect.height = brect.height;
        d.bg = blitz_loadcolor(dpy, screen_num, file[B_BG_COLOR]->content);
        d.fg = blitz_loadcolor(dpy, screen_num, file[B_FG_COLOR]->content);
        d.border =
            blitz_loadcolor(dpy, screen_num,
                            file[B_BORDER_COLOR]->content);
        blitz_drawlabelnoborder(dpy, &d);
    } else {
        File *f;
        char **paths = 0;
        /*
         * take order into account, directory names are used in
         * alphabetical order
         */
        nitems = 0;
        for(f = label; f; f = f->next)
            nitems++;
        paths = cext_emallocz(sizeof(char *) * nitems);
        j = 0;
        for(f = label; f; f = f->next)
            paths[j++] = f->name;
        qsort(paths, nitems, sizeof(char *), comp_str);
        for(j = 0; j < nitems; j++) {
            snprintf(buf, sizeof(buf), "/%s", paths[j]);
            i = cext_emallocz(sizeof(Item));
            init_item(buf, i);
            if(!it)
                it = items = i;
            else {
                it->next = i;
                it = i;
            }
        }
        free(paths);
        draw();
    }
}

static Item *
item_for_file(File * f)
{
    Item *i;
    for(i = items; i; i = i->next) {
        if(f == i->root)
            return i;
    }
    return nil;
}

static void
handle_buttonpress(XButtonPressedEvent * e)
{
    Item *i;
    File *p;
    char buf[MAX_BUF];
    char path[512];
    for(i = items; i; i = i->next)
        if(blitz_ispointinrect(e->x, e->y, &i->d.rect)) {
            path[0] = 0;
            wmii_get_ixppath(i->root, path, sizeof(path));
            snprintf(buf, MAX_BUF, "%s/b%upress", path, e->button);
            if((p = ixp_walk(ixps, buf)))
                if(p->content)
                    wmii_spawn(dpy, p->content);
            return;
        }
}

static void
check_event(Connection * e)
{
    XEvent ev;

    while(XPending(dpy)) {
        XNextEvent(dpy, &ev);
        switch (ev.type) {
        case ButtonPress:
            handle_buttonpress(&ev.xbutton);
            break;
        case Expose:
            if(ev.xexpose.count == 0) {
                /* XRaiseWindow(dpy, win); */
                draw_bar(0, 0);
            }
            break;
        default:
            break;
        }
    }
}

static void
update_geometry()
{
    brect = rect;
    brect.height = font->ascent + font->descent + 4;
    if(align == SOUTH)
        brect.y = rect.height - brect.height;
    XMoveResizeWindow(dpy, win, brect.x, brect.y, brect.width,
                      brect.height);
    XSync(dpy, False);
    XFreePixmap(dpy, pmap);
    pmap =
        XCreatePixmap(dpy, win, brect.width, brect.height,
                      DefaultDepth(dpy, screen_num));
    XSync(dpy, False);
    if(file[B_EVENT_RESIZE]->content)
        wmii_spawn(dpy, file[B_EVENT_RESIZE]->content);
}

static void
handle_after_write(IXPServer * s, File * f)
{
    int i;
    size_t len;
    Item *item;
    char buf[512];

    buf[0] = 0;
    if(!strncmp(f->name, "data", 5)) {
        if((item = item_for_file(f->parent))) {
            wmii_get_ixppath(f->parent, buf, sizeof(buf));
            init_draw_label(buf, &item->d);
            draw();
        }
    } else if(file[B_GEOMETRY] == f) {
        if(f->content) {
            if(!strncmp(f->content, "south", 6))
                align = SOUTH;
            else if(!strncmp(f->content, "north", 6))
                align = NORTH;
            update_geometry();
            draw_bar(0, 0);
        }
    } else if(file[B_CTL] == f) {
        for(i = 0; acttbl[i].name; i++) {
            len = strlen(acttbl[i].name);
            if(!strncmp(acttbl[i].name, (char *) f->content, len)) {
                if(strlen(f->content) > len) {
                    acttbl[i].func(0, &((char *) f->content)[len + 1]);
                } else {
                    acttbl[i].func(0, 0);
                }
                break;
            }
        }
    } else if(file[B_FONT] == f) {
        XFreeFont(dpy, font);
        font = blitz_getfont(dpy, file[B_FONT]->content);
        update_geometry();
        draw_bar(0, 0);
    }
    check_event(0);
}

static void
handle_before_read(IXPServer * s, File * f)
{
    char buf[64];
    if(f == file[B_GEOMETRY]) {
        snprintf(buf, 64, "%d %d %d %d", brect.x, brect.y, brect.width,
                 brect.height);
        if(f->content)
            free(f->content);
        f->content = cext_estrdup(buf);
        f->size = strlen(buf);
    } else if(f == file[B_NEW]) {
        snprintf(buf, sizeof(buf), "%d", id++);
        if(f->content)
            free(f->content);
        f->content = strdup(buf);
        f->size = strlen(buf);
        create_label(buf);
        draw_bar(0, 0);
    }
}

static int
dummy_error_handler(Display * dpy, XErrorEvent * err)
{
    return 0;
}

int
main(int argc, char *argv[])
{
    int i;
    XSetWindowAttributes wa;
    XGCValues gcv;

    /* command line args */
    for(i = 1; (i < argc) && (argv[i][0] == '-'); i++) {
        switch (argv[i][1]) {
        case 'v':
            fprintf(stdout, "%s", version[0]);
            exit(0);
            break;
        case 's':
            if(i + 1 < argc)
                sockfile = argv[++i];
            else
                usage();
            break;
        default:
            usage();
            break;
        }
    }

    dpy = XOpenDisplay(0);
    if(!dpy) {
        fprintf(stderr, "%s", "wmiibar: cannot open display\n");
        exit(1);
    }
    XSetErrorHandler(dummy_error_handler);
    screen_num = DefaultScreen(dpy);

    ixps = wmii_setup_server(sockfile);


    /* init */
    if(!(file[B_CTL] = ixp_create(ixps, "/ctl"))) {
        perror("wmiibar: cannot connect IXP server");
        exit(1);
    }
    file[B_CTL]->after_write = handle_after_write;
    file[B_NEW] = ixp_create(ixps, "/new");
    file[B_NEW]->before_read = handle_before_read;
    file[B_FONT] = wmii_create_ixpfile(ixps, "/font", BLITZ_FONT);
    file[B_FONT]->after_write = handle_after_write;
    font = blitz_getfont(dpy, file[B_FONT]->content);
    file[B_BG_COLOR] =
        wmii_create_ixpfile(ixps, "/bgcolor", BLITZ_NORM_BG_COLOR);
    file[B_FG_COLOR] =
        wmii_create_ixpfile(ixps, "/fgcolor", BLITZ_NORM_FG_COLOR);
    file[B_BORDER_COLOR] =
        wmii_create_ixpfile(ixps, "/bordercolor", BLITZ_NORM_BORDER_COLOR);
    file[B_GEOMETRY] = ixp_create(ixps, "/geometry");
    file[B_GEOMETRY]->before_read = handle_before_read;
    file[B_GEOMETRY]->after_write = handle_after_write;
    file[B_EXPANDABLE] = ixp_create(ixps, "/expandable");
    file[B_EVENT_RESIZE] = ixp_create(ixps, "/event/resize");

    wa.override_redirect = 1;
    wa.background_pixmap = ParentRelative;
    wa.event_mask =
        ExposureMask | ButtonPressMask | SubstructureRedirectMask |
        SubstructureNotifyMask;

    rect.x = rect.y = 0;
    rect.width = DisplayWidth(dpy, screen_num);
    rect.height = DisplayHeight(dpy, screen_num);
    brect = rect;
    brect.height = font->ascent + font->descent + 4;
    brect.y = rect.height - brect.height;

    win = XCreateWindow(dpy, RootWindow(dpy, screen_num), brect.x, brect.y,
                        brect.width, brect.height, 0, DefaultDepth(dpy,
                                                                   screen_num),
                        CopyFromParent, DefaultVisual(dpy, screen_num),
                        CWOverrideRedirect | CWBackPixmap | CWEventMask,
                        &wa);
    XDefineCursor(dpy, win, XCreateFontCursor(dpy, XC_left_ptr));
    XSync(dpy, False);

    gcv.function = GXcopy;
    gcv.graphics_exposures = False;
    gc = XCreateGC(dpy, win, 0, 0);

    pmap =
        XCreatePixmap(dpy, win, brect.width, brect.height,
                      DefaultDepth(dpy, screen_num));

    /* main event loop */
    run_server_with_fd_support(ixps, ConnectionNumber(dpy), check_event,
                               0);
    deinit_server(ixps);
    XFreePixmap(dpy, pmap);
    XFreeGC(dpy, gc);
    XCloseDisplay(dpy);

    return 0;
}