Adding small fix by Alex Puterbaugh. Thanks.
[surf.git] / surf.c
diff --git a/surf.c b/surf.c
index 197de44..448c762 100644 (file)
--- a/surf.c
+++ b/surf.c
@@ -56,13 +56,13 @@ typedef struct {
 
 static Display *dpy;
 static Atom uriprop, findprop;
-static SoupSession *session;
 static Client *clients = NULL;
 static GdkNativeWindow embed = 0;
 static gboolean showxid = FALSE;
 static int ignorexprop = 0;
 static char winid[64];
 static char *progname;
+static gboolean loadimage = 1, plugin = 1, script = 1;
 
 static char *buildpath(const char *path);
 static void cleanup(void);
@@ -80,7 +80,9 @@ static void drawindicator(Client *c);
 static gboolean exposeindicator(GtkWidget *w, GdkEventExpose *e, Client *c);
 static void find(Client *c, const Arg *arg);
 static const char *getatom(Client *c, Atom a);
+static const char *getcookies(SoupURI *uri);
 static char *geturi(Client *c);
+void gotheaders(SoupMessage *msg, gpointer user_data);
 static gboolean initdownload(WebKitWebView *v, WebKitDownload *o, Client *c);
 static void itemclick(GtkMenuItem *mi, Client *c);
 static gboolean keypress(GtkWidget *w, GdkEventKey *ev, Client *c);
@@ -90,6 +92,7 @@ static void loaduri(Client *c, const Arg *arg);
 static void navigate(Client *c, const Arg *arg);
 static Client *newclient(void);
 static void newwindow(Client *c, const Arg *arg);
+static void newrequest(SoupSession *s, SoupMessage *msg, gpointer v);
 static void pasteuri(GtkClipboard *clipboard, const char *text, gpointer d);
 static void print(Client *c, const Arg *arg);
 static GdkFilterReturn processx(GdkXEvent *xevent, GdkEvent *event, gpointer d);
@@ -98,6 +101,7 @@ static void reload(Client *c, const Arg *arg);
 static void resize(GtkWidget *w, GtkAllocation *a, Client *c);
 static void scroll(Client *c, const Arg *arg);
 static void setatom(Client *c, Atom a, const char *v);
+static void setcookie(SoupCookie *c);
 static void setup(void);
 static void sigchld(int unused);
 static void source(Client *c, const Arg *arg);
@@ -250,7 +254,7 @@ void
 download(Client *c, const Arg *arg) {
        char *uri;
        WebKitNetworkRequest *r;
-       WebKitDownload       *dl;
+       WebKitDownload *dl;
 
        if(arg->v)
                uri = (char *)arg->v;
@@ -259,6 +263,7 @@ download(Client *c, const Arg *arg) {
        r = webkit_network_request_new(uri);
        dl = webkit_download_new(r);
        initdownload(c->view, dl, c);
+       webkit_download_start(c->download);
 }
 
 void
@@ -299,6 +304,15 @@ find(Client *c, const Arg *arg) {
        webkit_web_view_search_text(c->view, s, FALSE, forward, TRUE);
 }
 
+const char *
+getcookies(SoupURI *uri) {
+       const char *c;
+       SoupCookieJar *j = soup_cookie_jar_text_new(cookiefile, TRUE);
+       c = soup_cookie_jar_get_cookies(j, uri, TRUE);
+       g_object_unref(j);
+       return c;
+}
+
 const char *
 getatom(Client *c, Atom a) {
        static char buf[BUFSIZ];
@@ -327,12 +341,26 @@ geturi(Client *c) {
        return uri;
 }
 
+void
+gotheaders(SoupMessage *msg, gpointer v) {
+       SoupURI *uri;
+       GSList *l, *p;
+
+       uri = soup_message_get_uri(msg);
+       for(p = l = soup_cookies_from_response(msg); p;
+               p = g_slist_next(p))  {
+               setcookie((SoupCookie *)p->data);
+       }
+       soup_cookies_free(l);
+}
+
 gboolean
 initdownload(WebKitWebView *view, WebKitDownload *o, Client *c) {
        const char *filename;
-       char *uri, *html;
+       char *uri;
+       WebKitWebBackForwardList *h;
+       WebKitWebHistoryItem *i;
 
-       stop(c, NULL);
        c->download = o;
        filename = webkit_download_get_suggested_filename(o);
        if(!strcmp("", filename))
@@ -340,17 +368,14 @@ initdownload(WebKitWebView *view, WebKitDownload *o, Client *c) {
        uri = g_strconcat("file://", dldir, "/", filename, NULL);
        webkit_download_set_destination_uri(c->download, uri);
        c->progress = 0;
-       g_free(uri);
-       html = g_strdup_printf("Download <b>%s</b>...", filename);
-       webkit_web_view_load_html_string(c->view, html,
-                       webkit_download_get_uri(c->download));
+       h = webkit_web_view_get_back_forward_list(c->view);
+       i = webkit_web_history_item_new_with_data(webkit_download_get_uri(c->download), filename);
+       webkit_web_back_forward_list_add_item(h, i);
        g_signal_connect(c->download, "notify::progress", G_CALLBACK(updatedownload), c);
        g_signal_connect(c->download, "notify::status", G_CALLBACK(updatedownload), c);
-       webkit_download_start(c->download);
        
        c->title = copystr(&c->title, filename);
        update(c);
-       g_free(html);
        return TRUE;
 }
 
@@ -397,14 +422,15 @@ void
 loadstatuschange(WebKitWebView *view, GParamSpec *pspec, Client *c) {
        switch(webkit_web_view_get_load_status (c->view)) {
        case WEBKIT_LOAD_COMMITTED:
-               setatom(c, uriprop, geturi(c));
+       if(c->download)
+               stop(c, NULL);
+       setatom(c, uriprop, geturi(c));
                break;
        case WEBKIT_LOAD_FINISHED:
                c->progress = 0;
                update(c);
                break;
-       case WEBKIT_LOAD_PROVISIONAL:
-       case WEBKIT_LOAD_FIRST_VISUALLY_NON_EMPTY_LAYOUT:
+       default:
                break;
        }
 }
@@ -535,6 +561,9 @@ newclient(void) {
        g_object_set(G_OBJECT(settings), "user-agent", ua, NULL);
        uri = g_strconcat("file://", stylefile, NULL);
        g_object_set(G_OBJECT(settings), "user-stylesheet-uri", uri, NULL);
+       g_object_set(G_OBJECT(settings), "auto-load-images", loadimage, NULL);
+       g_object_set(G_OBJECT(settings), "enable-plugins", plugin, NULL);
+       g_object_set(G_OBJECT(settings), "enable-scripts", script, NULL);
        g_free(uri);
        setatom(c, findprop, "");
        setatom(c, uriprop, "");
@@ -551,10 +580,23 @@ newclient(void) {
        return c;
 }
 
+void
+newrequest(SoupSession *s, SoupMessage *msg, gpointer v) {
+       SoupMessageHeaders *h = msg->request_headers;
+       SoupURI *uri;
+       const char *c;
+
+       soup_message_headers_remove(h, "Cookie");
+       uri = soup_message_get_uri(msg);
+       if((c = getcookies(uri)))
+               soup_message_headers_append(h, "Cookie", c);
+       g_signal_connect_after(G_OBJECT(msg), "got-headers", G_CALLBACK(gotheaders), NULL);
+}
+
 void
 newwindow(Client *c, const Arg *arg) {
        guint i = 0;
-       const char *cmd[7], *uri;
+       const char *cmd[10], *uri;
        const Arg a = { .v = (void *)cmd };
        char tmp[64];
 
@@ -564,9 +606,14 @@ newwindow(Client *c, const Arg *arg) {
                snprintf(tmp, LENGTH(tmp), "%u\n", (int)embed);
                cmd[i++] = tmp;
        }
-       if(showxid) {
+       if(!script)
+               cmd[i++] = "-s";
+       if(!plugin)
+               cmd[i++] = "-p";
+       if(!loadimage)
+               cmd[i++] = "-i";
+       if(showxid)
                cmd[i++] = "-x";
-       }
        cmd[i++] = "--";
        uri = arg->v ? (char *)arg->v : c->linkhover;
        if(uri)
@@ -614,7 +661,7 @@ processx(GdkXEvent *e, GdkEvent *event, gpointer d) {
 
 void
 progresschange(WebKitWebView *view, GParamSpec *pspec, Client *c) {
-       c->progress = webkit_web_view_get_progress(c->view);;
+       c->progress = webkit_web_view_get_progress(c->view) * 100;
        update(c);
 }
 
@@ -653,6 +700,19 @@ scroll(Client *c, const Arg *arg) {
        gtk_adjustment_set_value(a, v);
 }
 
+void
+setcookie(SoupCookie *c) {
+       SoupDate *e;
+       SoupCookieJar *j = soup_cookie_jar_text_new(cookiefile, FALSE);
+       c = soup_cookie_copy(c);
+       if(c->expires == NULL) {
+               e = soup_date_new_from_time_t(time(NULL) + sessiontime);
+               soup_cookie_set_expires(c, e);
+       }
+       soup_cookie_jar_add_cookie(j, c);
+       g_object_unref(j);
+}
+
 void
 setatom(Client *c, Atom a, const char *v) {
        XSync(dpy, False);
@@ -664,10 +724,10 @@ setatom(Client *c, Atom a, const char *v) {
 
 void
 setup(void) {
-       SoupSession *s;
        char *proxy;
        char *new_proxy;
        SoupURI *puri;
+       SoupSession *s;
 
        /* clean up any zombies immediately */
        sigchld(0);
@@ -676,16 +736,22 @@ setup(void) {
                g_thread_init(NULL);
 
        dpy = GDK_DISPLAY();
-       session = webkit_get_default_session();
+       s = webkit_get_default_session();
        uriprop = XInternAtom(dpy, "_SURF_URI", False);
        findprop = XInternAtom(dpy, "_SURF_FIND", False);
 
-       /* create dirs and files */
+       /* dirs and files */
        cookiefile = buildpath(cookiefile);
        dldir = buildpath(dldir);
        scriptfile = buildpath(scriptfile);
        stylefile = buildpath(stylefile);
 
+       /* request handler */
+       s = webkit_get_default_session();
+       soup_session_remove_feature_by_type(s, soup_cookie_get_type());
+       soup_session_remove_feature_by_type(s, soup_cookie_jar_get_type());
+       g_signal_connect_after(G_OBJECT(s), "request-started", G_CALLBACK(newrequest), NULL);
+
        /* proxy */
        if((proxy = getenv("http_proxy")) && strcmp(proxy, "")) {
                new_proxy = g_strrstr(proxy, "http://") ? g_strdup(proxy) :
@@ -766,6 +832,8 @@ updatedownload(WebKitDownload *o, GParamSpec *pspec, Client *c) {
        if(status == WEBKIT_DOWNLOAD_STATUS_STARTED || status == WEBKIT_DOWNLOAD_STATUS_CREATED) {
                c->progress = (gint)(webkit_download_get_progress(c->download)*100);
        }
+       else
+               stop(c, NULL);
        update(c);
 }
 
@@ -778,7 +846,7 @@ updatewinid(Client *c) {
 void
 usage(void) {
        fputs("surf - simple browser\n", stderr);
-       die("usage: surf [-e Window] [-x] [uri]\n");
+       die("usage: surf [-e xid] [-i] [-p] [-s] [-v] [-x] [uri]\n");
 }
 
 void
@@ -807,29 +875,43 @@ zoom(Client *c, const Arg *arg) {
        }
 }
 
-int main(int argc, char *argv[]) {
+int
+main(int argc, char *argv[]) {
        int i;
        Arg arg;
 
        progname = argv[0];
        /* command line args */
-       for(i = 1, arg.v = NULL; i < argc && argv[i][0] == '-'; i++) {
-               if(!strcmp(argv[i], "-x"))
-                       showxid = TRUE;
-               else if(!strcmp(argv[i], "-e")) {
+       for(i = 1, arg.v = NULL; i < argc && argv[i][0] == '-' &&
+                       argv[i][1] != '\0' && argv[i][2] == '\0'; i++) {
+               if(!strcmp(argv[i], "--")) {
+                       i++;
+                       break;
+               }
+               switch(argv[i][1]) {
+               case 'e':
                        if(++i < argc)
                                embed = atoi(argv[i]);
                        else
                                usage();
-               }
-               else if(!strcmp(argv[i], "--")) {
-                       i++;
                        break;
-               }
-               else if(!strcmp(argv[i], "-v"))
+               case 'i':
+                       loadimage = 0;
+                       break;
+               case 'p':
+                       plugin = 0;
+                       break;
+               case 's':
+                       script = 0;
+                       break;
+               case 'x':
+                       showxid = TRUE;
+                       break;
+               case 'v':
                        die("surf-"VERSION", © 2009 surf engineers, see LICENSE for details\n");
-               else
+               default:
                        usage();
+               }
        }
        if(i < argc)
                arg.v = argv[i];