applied patch from arg. thanks :)
[surf.git] / surf.c
diff --git a/surf.c b/surf.c
index a36af8f..367ba1c 100644 (file)
--- a/surf.c
+++ b/surf.c
 #include <gdk/gdkkeysyms.h>
 #include <string.h>
 #include <unistd.h>
-#include <getopt.h>
 #include <stdlib.h>
 #include <stdio.h>
 #include <webkit/webkit.h>
 #include <glib/gstdio.h>
+#include <JavaScriptCore/JavaScript.h>
 
 #define LENGTH(x)               (sizeof x / sizeof x[0])
 #define CLEANMASK(mask)         (mask & ~(GDK_MOD2_MASK))
 
-Display *dpy;
-Atom urlprop;
 typedef union Arg Arg;
 union Arg {
        const gboolean b;
        const gint i;
        const void *v;
-} ;
+};
 
 typedef struct Client {
        GtkWidget *win, *scroll, *vbox, *urlbar, *searchbar, *indicator;
@@ -46,10 +44,10 @@ typedef struct Cookie {
 } Cookie;
 
 typedef enum {
-    BROWSER = 0x0001,
-    SEARCHBAR = 0x0010,
-    URLBAR = 0x0100,
-    ALWAYS = ~0,
+       Browser = 0x0001,
+       SearchBar = 0x0010,
+       UrlBar = 0x0100,
+       Any = ~0,
 } KeyFocus;
 
 typedef struct {
@@ -60,16 +58,16 @@ typedef struct {
        KeyFocus focus;
 } Key;
 
-SoupCookieJar *cookiejar;
-SoupSession *session;
-Client *clients = NULL;
-Cookie *cookies = NULL;
-GdkNativeWindow embed = 0;
-gboolean showxid = FALSE;
-gboolean ignore_once = FALSE;
-gchar *workdir;
-extern char *optarg;
-extern gint optind;
+static Display *dpy;
+static Atom urlprop;
+static SoupCookieJar *cookiejar;
+static SoupSession *session;
+static Client *clients = NULL;
+/*static Cookie *cookies = NULL;*/
+static GdkNativeWindow embed = 0;
+static gboolean showxid = FALSE;
+static gboolean ignore_once = FALSE;
+static gchar *workdir;
 
 static void cleanup(void);
 static void clipboard(Client *c, const Arg *arg);
@@ -103,8 +101,7 @@ static void reload(Client *c, const Arg *arg);
 static void rereadcookies(void);
 static void setcookie(char *name, char *val, char *dom, char *path, long exp);
 static void setup(void);
-static void titlechange(WebKitWebView* view, WebKitWebFrame* frame,
-               const gchar* title, Client *c);
+static void titlechange(WebKitWebView* view, WebKitWebFrame* frame, const gchar* title, Client *c);
 static void scroll(Client *c, const Arg *arg);
 static void searchtext(Client *c, const Arg *arg);
 static void source(Client *c, const Arg *arg);
@@ -115,6 +112,7 @@ static void titlechange(WebKitWebView* view, WebKitWebFrame* frame, const gchar*
 static gboolean unfocusbar(GtkWidget *w, GdkEventFocus *e, Client *c);
 static void usage(void);
 static void update(Client *c);
+static void windowobjectcleared(GtkWidget *w, WebKitWebFrame *frame, JSContextRef js, JSObjectRef win, Client *c);
 static void zoom(Client *c, const Arg *arg);
 
 #include "config.h"
@@ -123,6 +121,7 @@ void
 cleanup(void) {
        while(clients)
                destroyclient(clients);
+       g_free(workdir);
 }
 
 void
@@ -274,11 +273,11 @@ keypress(GtkWidget* w, GdkEventKey *ev, Client *c) {
        if(ev->type != GDK_KEY_PRESS)
                return FALSE;
        if(GTK_WIDGET_HAS_FOCUS(c->searchbar))
-               focus = SEARCHBAR;
+               focus = SearchBar;
        else if(GTK_WIDGET_HAS_FOCUS(c->urlbar))
-               focus = URLBAR;
+               focus = UrlBar;
        else
-               focus = BROWSER;
+               focus = Browser;
        for(i = 0; i < LENGTH(keys); i++) {
                if(focus & keys[i].focus
                                && gdk_keyval_to_lower(ev->keyval) == keys[i].keyval
@@ -407,6 +406,7 @@ newclient(void) {
        g_signal_connect(G_OBJECT(c->view), "hovering-over-link", G_CALLBACK(linkhover), c);
        g_signal_connect(G_OBJECT(c->view), "create-web-view", G_CALLBACK(newwindow), c);
        g_signal_connect(G_OBJECT(c->view), "download-requested", G_CALLBACK(initdownload), c);
+       g_signal_connect(G_OBJECT(c->view), "window-object-cleared", G_CALLBACK(windowobjectcleared), c);
        g_signal_connect_after(session, "request-started", G_CALLBACK(request), c);
 
        /* urlbar */
@@ -425,8 +425,6 @@ newclient(void) {
        g_signal_connect (G_OBJECT (c->indicator), "expose_event",
                        G_CALLBACK (exposeindicator), c);
 
-       /* downloadbar */
-
        /* Arranging */
        gtk_container_add(GTK_CONTAINER(c->scroll), GTK_WIDGET(c->view));
        gtk_container_add(GTK_CONTAINER(c->win), c->vbox);
@@ -583,6 +581,10 @@ setup(void) {
        SoupSession *s;
        FILE *tmp;
 
+       gtk_init(NULL, NULL);
+       if (!g_thread_supported())
+               g_thread_init(NULL);
+
        dpy = GDK_DISPLAY();
        session = webkit_get_default_session();
        urlprop = XInternAtom(dpy, "_SURF_URL", False);
@@ -590,13 +592,16 @@ setup(void) {
 
        /* create dirs and files */
        home = g_get_home_dir();
-       workdir = g_build_filename(home, ".surf", NULL);
+       workdir = g_strdup(g_build_filename(home, ".surf", NULL));
        g_mkdir_with_parents(workdir, 0755);
        name = g_build_filename(workdir, "dl", NULL);
        g_mkdir(name, 0755);
        name = g_build_filename(workdir, "style.css", NULL);
        if((tmp = g_fopen(name, "a")));
                fclose(tmp);
+       name = g_build_filename(workdir, "script.js", NULL);
+       if((tmp = g_fopen(name, "a")));
+               fclose(tmp);
 
 
        /* cookie persistance */
@@ -686,6 +691,23 @@ update(Client *c) {
 
 }
 
+void
+windowobjectcleared(GtkWidget *w, WebKitWebFrame *frame, JSContextRef js, JSObjectRef win, Client *c) {
+       JSStringRef jsscript;
+       gchar *script, *filename;
+       JSValueRef exception = NULL;
+       GError *error;
+       
+       filename = g_build_filename(workdir, "script.js", NULL);
+       if(g_file_get_contents(filename, &script, NULL, &error)) {
+               script = g_strdup_printf("window.addEventListener"
+                               "('DOMContentLoaded', function () { %s }, true);",
+                               script);
+               jsscript = JSStringCreateWithUTF8CString (script);
+               JSEvaluateScript (js, jsscript, JSContextGetGlobalObject(js), NULL, 0, &exception);
+       }
+}
+
 void
 zoom(Client *c, const Arg *arg) {
        if(arg->i < 0)          /* zoom out */
@@ -697,45 +719,35 @@ zoom(Client *c, const Arg *arg) {
 }
 
 int main(int argc, char *argv[]) {
-       Client *c;
-       gint o, a;
+       int i;
        Arg arg;
+       Client *c;
 
-       gtk_init(NULL, NULL);
-       if (!g_thread_supported())
-               g_thread_init(NULL);
-       while((o = getopt(argc, argv, "vhxe:")) != -1)
-               switch(o) {
-               case 'x':
+       /* command line args */
+       for(i = 1, arg.v = NULL; i < argc; i++) {
+               if(!strcmp(argv[i], "-x"))
                        showxid = TRUE;
-                       break;
-               case 'e':
-                       if(!(a = atoi(optarg)))
+               else if(!strcmp(argv[i], "-e")) {
+                       if(++i < argc)
+                               embed = atoi(argv[i]);
+                       else
                                usage();
-                       embed = a;
-                       break;
-               case 'v':
+               }
+               else if(!strcmp(argv[i], "-v"))
                        die("surf-"VERSION", © 2009 surf engineers, see LICENSE for details\n");
-                       break;
-               default:
+               else if(argv[i][0] == '-')
                        usage();
-               }
-       if(optind + 1 == argc) {
-               c = newclient();
-               arg.v = argv[optind];
-               if(strchr("./", argv[optind][0]) || strcmp("-", argv[optind]) == 0)
-                       loadfile(c, argv[optind]);
                else
-                       loaduri(c, &arg);
-
+                       arg.v = argv[i];
        }
-       else if(optind != argc)
-               usage();
        setup();
-       if(!clients)
-               newclient();
-
-
+       c = newclient();
+       if(arg.v) {
+               if(strchr("./", ((char *)arg.v)[0]) || strcmp("-", (char *)arg.v) == 0)
+                       loadfile(c, (char *)arg.v);
+               else
+                       loaduri(c, &arg);
+       }
        gtk_main();
        cleanup();
        return EXIT_SUCCESS;