/* Copyright ©2008 Kris Maglione <fbsdaemon@gmail.com>
 * See LICENSE file for license details.
 */
#include "hack.h"
#include <dlfcn.h>
#include <stdlib.h>
#include <stdio.h>
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>

#include "util.c"
#include "x11.c"

enum {
	Timeout = 10000,
};

static void*	xlib;

static long	transient;
static Atom	types[32];
static long	ntypes;
static char*	tags[32];
static long	ntags;
static long	pid;
static char	hostname[256];
static long	nmsec;

typedef Window (*mapfn)(Display*, Window);

static Window (*mapwindow)(Display*, Window);
static Window (*mapraised)(Display*, Window);

static long
msec(void) {
	struct timeval tv;

	if(!gettimeofday(&tv, 0))
		return 0;
	return tv.tv_sec*1000 + tv.tv_usec/1000;
}

static void
init(Display *d) { /* Hrm... assumes one display... */
	char *toks[nelem(types)];
	char *s, *p;
	long n;
	int i;

	xlib = dlopen("libX11.so", RTLD_GLOBAL | RTLD_LAZY);
	if(xlib == nil)
		return;
	mapwindow = (mapfn)(uintptr_t)dlsym(xlib, "XMapWindow");
	mapraised = (mapfn)(uintptr_t)dlsym(xlib, "XMapRaised");

	unsetenv("LD_PRELOAD");

	if((s = getenv("WMII_HACK_TRANSIENT"))) {
		if(getlong(s, &n))
			transient = n;
		unsetenv("WMII_HACK_TRANSIENT");
	}
	if((s = getenv("WMII_HACK_TYPE"))) {
		s = strdup(s);
		unsetenv("WMII_HACK_TYPE");

		n = tokenize(toks, nelem(toks), s, ',');
		for(i=0; i < n; i++) {
			for(p=toks[i]; *p; p++)
				if(*p >= 'a' && *p <= 'z')
					*p += 'A' - 'a';
			toks[i] = smprint("_NET_WM_WINDOW_TYPE_%s", toks[i]);
		}
		XInternAtoms(d, toks, n, False, types);
		ntypes = n;
		for(i=0; i < n; i++)
			free(toks[i]);
		free(s);
	}
	if((s = getenv("WMII_HACK_TAGS"))) {
		s = strdup(s);
		unsetenv("WMII_HACK_TAGS");

		n = tokenize(toks, nelem(toks)-1, s, '+');
		for(i=0; i < n; i++)
			tags[i] = strdup(toks[i]);
		ntags = n;
		free(s);
	}

	pid = getpid();
	gethostname(hostname, sizeof hostname);
}

static void
setprops(Display *d, Window w) {

	if(!xlib)
		init(d);

	changeprop_long(d, w, "_NET_WM_PID", "CARDINAL", &pid, 1);
	changeprop_string(d, w, "_NET_WM_CLIENT_MACHINE", hostname);

	/* Kludge. */
	if(nmsec == 0)
		nmsec = msec();
	if(msec() > nmsec + Timeout)
		return;

	if(transient)
		changeprop_long(d, w, "WM_TRANSIENT_FOR", "WINDOW", &transient, 1);
	if(ntypes)
		changeprop_long(d, w, "_NET_WM_WINDOW_TYPE", "ATOM", (long*)types, ntypes);
	if(ntags)
		changeprop_textlist(d, w, "_WMII_TAGS", "UTF8_STRING", tags);
}

int
XMapWindow(Display *d, Window w) {

	setprops(d, w);
	return mapwindow(d, w);
}

int
XMapRaised(Display *d, Window w) {

	setprops(d, w);
	return mapraised(d, w);
}