#include <X11/X.h>
#include <X11/Xatom.h>
#include <gtk/gtk.h>
#include <gdk/gdkx.h>
#include <gdk/gdk.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <webkit/webkit.h>

#define LENGTH(x) (sizeof x / sizeof x[0])
/* Plan9-style Argument parsing */
/* Vars: _c -> count; _b -> break; _a -> argument */
#define ARG int _c, _b; char *_a; \
	for(_c = 1; _c < argc && argv[_c][0] == '-' && argv[_c][1] && \
			(strcmp(argv[_c], "--") != 0); _c++) \
		for(_a = &argv[_c][1], _b = 0; !_b && *_a; _a++ ) \
			switch(*_a)
#define ARGVAL()	(!_b && _a[1] && (_b = 1) ? &_a[1] : _c + 1 == argc ? \
		0 : argv[++_c])
#define ARGCHR()	(*_a)
#define ARGC()		_c

Display *dpy;
Atom urlprop;
GtkWidget *win;
GtkWidget *browser;
WebKitWebView *view;
gchar *title;
gint progress = 100;
gboolean embed = FALSE;
gboolean showxid = FALSE;
gboolean ignore_once = FALSE;

static void setup(void);
static void cleanup(void);
static void updatetitle(void);
static void windestroy(GtkWidget* w, gpointer d);
static gboolean keypress(GtkWidget* w, GdkEventKey *ev);
static void titlechange(WebKitWebView* view, WebKitWebFrame* frame, const gchar* title, gpointer d);
static void progresschange(WebKitWebView *view, gint p, gpointer d);
static void loadcommit(WebKitWebView *view, WebKitWebFrame *f, gpointer d);
static void linkhover(WebKitWebView* page, const gchar* t, const gchar* l, gpointer d);
static gboolean newwindow(WebKitWebView *view, WebKitWebFrame *f,
		WebKitNetworkRequest *r, WebKitWebNavigationAction *n,
		WebKitWebPolicyDecision *p, gpointer d);
static gboolean download(WebKitWebView *view, GObject *o, gpointer d);
static void loaduri(gchar *uri);
static void loadfile(gchar *f);
static void setupx();
GdkFilterReturn processx(GdkXEvent *xevent, GdkEvent *event, gpointer data);

GdkFilterReturn
processx(GdkXEvent *e, GdkEvent *event, gpointer data) {
	XPropertyEvent *ev;
	Atom adummy;
	int idummy;
	unsigned long ldummy;
	unsigned char *buf = NULL;
	if(((XEvent *)e)->type == PropertyNotify) {
		ev = &((XEvent *)e)->xproperty;
		if(ignore_once == FALSE && ev->atom == urlprop && ev->state == PropertyNewValue) {
			XGetWindowProperty(dpy, ev->window, urlprop, 0L, BUFSIZ, False, XA_STRING, 
			&adummy, &idummy, &ldummy, &ldummy, &buf);
			loaduri((gchar *)buf);
			XFree(buf);
			return GDK_FILTER_REMOVE;
		}
		ignore_once = FALSE;
	}
	return GDK_FILTER_CONTINUE;
}

void
setupx() {
	dpy = GDK_WINDOW_XDISPLAY(GTK_WIDGET(win)->window);
	urlprop = XInternAtom(dpy, "_SURF_URL", False);
	gdk_window_add_filter(GTK_WIDGET(win)->window, processx, NULL);
	gdk_window_set_events(GTK_WIDGET(win)->window, GDK_ALL_EVENTS_MASK);
}

void
loadfile(gchar *f) {
	GIOChannel *c = NULL;
	GError *e = NULL;
	GString *code = g_string_new("");
	GString *uri = g_string_new(f);
	gchar *line;

	if(strcmp(f, "-") == 0) {
		c = g_io_channel_unix_new(STDIN_FILENO);
		if (c) {
			while(g_io_channel_read_line(c, &line, NULL, NULL, &e) == G_IO_STATUS_NORMAL) {
				g_string_append(code, line);
				g_free(line);
			}
			webkit_web_view_load_html_string(view, code->str, NULL);
			g_free(code);
			g_io_channel_shutdown(c, FALSE, NULL);
		}
	}
	else {
		g_string_prepend(uri, "file://");
		loaduri(uri->str);
	}
	
}

static void loaduri(gchar *uri) {
	GString* u = g_string_new(uri);
	if(g_strrstr(u->str, "://") == NULL)
		g_string_prepend(u, "http://");
	webkit_web_view_load_uri(view, u->str);
	g_string_free(u, TRUE);
}

gboolean
download(WebKitWebView *view, GObject *o, gpointer d) {
	/* TODO */
	return FALSE;
}

gboolean
newwindow(WebKitWebView *view, WebKitWebFrame *f,
		WebKitNetworkRequest *r, WebKitWebNavigationAction *n,
		WebKitWebPolicyDecision *p, gpointer d) {
	/* TODO */
	return FALSE;
}
void
linkhover(WebKitWebView* page, const gchar* t, const gchar* l, gpointer d) {
	/* TODO */
}

void
loadcommit(WebKitWebView *view, WebKitWebFrame *f, gpointer d) {
	gchar *uri;

	if(!(uri = (gchar *)webkit_web_view_get_uri(view)))
		uri = "(null)";
	ignore_once = TRUE;
	XChangeProperty(dpy, GDK_WINDOW_XID(GTK_WIDGET(win)->window), urlprop,
			XA_STRING, 8, PropModeReplace, (unsigned char *)uri,
			strlen(uri) + 1);
}

void
progresschange(WebKitWebView* view, gint p, gpointer d) {
	progress = p;
	updatetitle();
}

void
updatetitle() {
	char t[512];
	if(progress == 100)
		snprintf(t, LENGTH(t), "%s", title);
	else
		snprintf(t, LENGTH(t), "%s [%i%%]", title, progress);
	gtk_window_set_title(GTK_WINDOW(win), t);
}

void
titlechange(WebKitWebView *v, WebKitWebFrame *f, const gchar *t, gpointer d) {
	if(title)
		g_free(title);
	title = g_strdup(t);
	updatetitle();
}

void
windestroy(GtkWidget* w, gpointer d) {
	gtk_main_quit();
}

gboolean
keypress(GtkWidget* w, GdkEventKey *ev) {
	/* TODO */
	return FALSE;
}

void setup(void) {
	if(embed) {
		win = gtk_plug_new(0);
	}
	else {
		win = gtk_window_new(GTK_WINDOW_TOPLEVEL);
		gtk_window_set_wmclass(GTK_WINDOW(win), "surf", "surf");
	}
	gtk_window_set_default_size(GTK_WINDOW(win), 800, 600);
	browser = gtk_scrolled_window_new(NULL, NULL);
	g_signal_connect (G_OBJECT(win), "destroy", G_CALLBACK(windestroy), NULL);
	g_signal_connect (G_OBJECT(win), "key-press-event", G_CALLBACK(keypress), NULL);

	gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(browser),
			GTK_POLICY_NEVER, GTK_POLICY_NEVER);
	view = WEBKIT_WEB_VIEW(webkit_web_view_new());
	gtk_container_add(GTK_CONTAINER(browser), GTK_WIDGET(view));

	g_signal_connect(G_OBJECT(view), "title-changed", G_CALLBACK(titlechange), view);
	g_signal_connect(G_OBJECT(view), "load-progress-changed", G_CALLBACK(progresschange), view);
	g_signal_connect(G_OBJECT(view), "load-committed", G_CALLBACK(loadcommit), view);
	g_signal_connect(G_OBJECT(view), "hovering-over-link", G_CALLBACK(linkhover), view);
	g_signal_connect(G_OBJECT(view), "new-window-policy-decision-requested", G_CALLBACK(newwindow), view);
	g_signal_connect(G_OBJECT(view), "download-requested", G_CALLBACK(download), view);
	/* g_signal_connect(G_OBJECT(view), "create-web-view", G_CALLBACK(createwebview), view); */

	gtk_container_add(GTK_CONTAINER(win), browser);
	gtk_widget_grab_focus(GTK_WIDGET(view));
	gtk_widget_show_all(win);
	if(showxid)
		printf("%u\n", (unsigned int)GDK_WINDOW_XID(GTK_WIDGET(win)->window));
}

void cleanup() {
}

int main(int argc, char *argv[]) {
	gchar *uri = NULL, *file = NULL;

	ARG {
	case 'x':
		showxid = TRUE;
		break;
	case 'e':
		showxid = TRUE;
		embed = TRUE;
		break;
	case 'u':
		if(!(uri = ARGVAL()))
			goto argerr;
		break;
	case 'f':
		if(!(file = ARGVAL()))
			goto argerr;
		break;
	argerr:
	default:
		puts("surf - simple browser");
		printf("usage: %s [-e] [-u uri] [-f file]", argv[0]);
		return EXIT_FAILURE;
	}
	if(argc != ARGC())
		goto argerr;
	gtk_init(NULL, NULL);
	if (!g_thread_supported())
		g_thread_init(NULL);
	setup();
	setupx();
	if(uri)
		loaduri(uri);
	else if(file)
		loadfile(file);
	updatetitle();
	gtk_main();
	cleanup();
	return EXIT_SUCCESS;
}