#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>

#include <X11/Xlib.h>
#include <X11/Xutil.h>
#include <X11/Xresource.h>
#include <X11/Xft/Xft.h>

#include <unistd.h>
#include <assert.h>

#include "drw.h"
#include "util.h"

enum {
	SchemeNorm,
	SchemeSel,
	SchemeOut,
	SchemeLast,
}; /* color schemes */

static Window root, parentwin, win;
static int screen;
static Display *display;
static int sep_width;
static int bh, mw, mh;
static XIC xic;

#include "cfg.h"


static void
grabkeyboard(void)
{
	struct timespec ts = { .tv_sec = 0, .tv_nsec = 1000000  };
	int i;

	// XXXembed
	/* if (embed) */
	/* 	return; */
	/* try to grab keyboard, we may have to wait for another process to ungrab */
	for (i = 0; i < 1000; i++) {
		if (XGrabKeyboard(display, DefaultRootWindow(display), True,
				  GrabModeAsync, GrabModeAsync, CurrentTime)
		    == GrabSuccess)
			return;
		nanosleep(&ts, NULL);
	}
	die("cannot grab keyboard");
}

static void
setup(Drw *drw)
{
	int x, y;
	XSetWindowAttributes swa;
	XWindowAttributes wa;
	XIM xim;
	XClassHint ch = {"cmdtree", "cmdtree"};

	bh = drw->fonts->h + 2;
	/* init appearance */
	drw_scm_create(drw, schemes, LENGTH(schemes));

	if (!XGetWindowAttributes(display, parentwin, &wa))
		die("could not get embedding window attributes: 0x%lx",
		    parentwin);
	x = 0;
	y = topbar ? 0 : wa.height - mh;
	mw = wa.width;
	lines = 3;
	lines = MAX(lines, 0);
	mh = (lines + 1) * bh;
	sep_width = drw_fontset_getwidth(drw, separator);

	swa.override_redirect = True;
	swa.background_pixel = schemes[SchemeNorm].bg_clr.pixel;
	swa.event_mask = ExposureMask | KeyPressMask | VisibilityChangeMask;

	win = XCreateWindow(display, parentwin, x, y, mw, mh, 0,
	                    CopyFromParent, CopyFromParent, CopyFromParent,
	                    CWOverrideRedirect | CWBackPixel | CWEventMask, &swa);

	XSetClassHint(display, win, &ch);

	// what do?
	XMapRaised(display, win);
	XSetInputFocus(display, win, RevertToParent, CurrentTime);

	xim = XOpenIM(display, NULL, NULL, NULL);
	xic = XCreateIC(xim, XNInputStyle, XIMPreeditNothing | XIMStatusNothing,
			XNClientWindow, win, XNFocusWindow, win, NULL);
	XSelectInput(display, win, ExposureMask | KeyPressMask);
	XMapWindow(display, win);


	XSetInputFocus(display, win, RevertToParent, CurrentTime);

	// XXXembed
	/* if (embed) { */
	/* 	XSelectInput(dpy, parentwin, FocusChangeMask); */
	/* 	if (XQueryTree(dpy, parentwin, &dw, &w, &dws, &du) && dws) { */
	/* 		for (i = 0; i < du && dws[i] != win; ++i) */
	/* 			XSelectInput(dpy, dws[i], FocusChangeMask); */
	/* 		XFree(dws); */
	/* 	} */
	/* 	grabfocus(); */
	/* } */

	/* drw_resize(drw, mw, mh); */
	/* drawmenu(); */
}

static int
draw_command(Drw *drw, int x, int y, const char *name, const char *binding) {
	int lpad = 0;
	int pad = 0;
	unsigned int w = 50;
	int invert = 0;

	//drw_fontset_getwidth(drw, binding) + 1;

	drw_setscheme(drw, &schemes[SchemeNorm].bind_clr,
		      &schemes[SchemeNorm].bg_clr);

	w = drw_fontset_getwidth(drw, binding);
	x = drw_text(drw, x+pad, y, w, bh, lpad, binding, invert);

	w = sep_width;
	drw_setscheme(drw, &schemes[SchemeNorm].arrow_clr,
		      &schemes[SchemeNorm].bg_clr);
	x = drw_text(drw, x+pad, y, w, bh, lpad, separator, invert);

	drw_setscheme(drw, &schemes[SchemeNorm].name_clr,
		      &schemes[SchemeNorm].bg_clr);

	w = drw_fontset_getwidth(drw, name);
	x = drw_text(drw, x+pad, y, w, bh, lpad, name, invert);

	return x;
}

/* static void */
/* calc_tree_exts(struct node *nodes, int num_nodes, int *rows, int *cols) { */
/* } */

static void
draw_tree(Drw *drw, int x, int y, int w, int h) {
	int i, dx = x, dy = y;
	char buf[512];
	char smallbuf[32];

	if (!drw)
		return;

	drw_setscheme(drw, &schemes[SchemeNorm].bg_clr,
		      &schemes[SchemeNorm].bg_clr);
	drw_rect(drw, 0, 0, w, h, 1, 1);

	int c = '0';
	for (i = 0; i < 50; ++i, ++c) {
		if (i % 2 == 0)
			snprintf(buf, 512, "item-long-%d", i);
		else if (i % 6 == 0)
			snprintf(buf, 512, "herpderp-%d", i);
		else if (i % 7 == 0)
			snprintf(buf, 512, "ksdfsdjhfsdf-%d", i);
		else
			snprintf(buf, 512, "hi-%d", i);
		if (c > '~') c = '0';
		sprintf(smallbuf, "%c", c);
		dx = draw_command(drw, dx, dy, buf, smallbuf) + 20;
		if (dx >= mw) {
			dx = 0;
			dy += bh;
		}
	}

	XCopyArea(drw->dpy, drw->drawable, win, drw->gc, x, y, w, h, x, y);
	XSync(drw->dpy, False);
}


static void
cleanup(Drw *drw, int code) {
	drw_free(drw);
	exit(0);
}

static void
run(Drw *drw) {
	XEvent e;
	int done = 0;
	char buf[32];
	KeySym ksym = NoSymbol;
	Status status;
	int code = 0;

	while (!done) {
		XNextEvent(display, &e);
		switch (e.type) {
		case Expose:
			draw_tree(drw, 0, 0, mw, mh);
			break;
		case KeyPress:
			XmbLookupString(xic, (XKeyEvent*)&e, buf, sizeof buf,
					&ksym, &status);
			if (ksym == XK_q || ksym == XK_Escape) {
				code = 1;
				done = 1;
			}
			break;
		}
	}

	cleanup(drw, code);
}


int main(void) {
	XWindowAttributes wa;
	Drw *drw;

	display = XOpenDisplay(NULL);
	if (display == NULL) {
		fprintf(stderr, "Cannot open display\n");
		exit(1);
	}

	screen = DefaultScreen(display);
	root = RootWindow(display, screen);

	// XXXembed
	/* if (!embed || !(parentwin = strtol(embed, NULL, 0))) */
	/* 	parentwin = root; */

	parentwin = root;

	if (!XGetWindowAttributes(display, parentwin, &wa))
		die("could not get embedding window attributes: 0x%lx",
		    parentwin);

	drw = drw_create(display, screen, root, wa.width, wa.height);

	if (!drw_fontset_create(drw, fonts, LENGTH(fonts)))
		die("no fonts could be loaded.");

	/* grabkeyboard(); */
	setup(drw);
	run(drw);

	XCloseDisplay(display);
	return 0;
}