]> git.armaanb.net Git - chorizo.git/blobdiff - browser.c
Built-in download feature
[chorizo.git] / browser.c
index 6877977f6b18d43cd6216525e1f31376fe4ff4bb..199ea2238fe5c0ed498a3c4aab74f648bed3d563 100644 (file)
--- a/browser.c
+++ b/browser.c
 static void adblock(WebKitWebView *, WebKitWebFrame *, WebKitWebResource *,
                     WebKitNetworkRequest *, WebKitNetworkResponse *, gpointer);
 static void adblock_load(void);
-static void client_destroy(GtkWidget *, gpointer);
-static void client_new(const gchar *uri);
-static gboolean client_new_request(WebKitWebView *, WebKitWebFrame *,
-                                   WebKitNetworkRequest *,
-                                   WebKitWebNavigationAction *,
-                                   WebKitWebPolicyDecision *, gpointer);
+static void client_destroy(GtkWidget *obj, gpointer data);
+static gboolean client_destroy_request(WebKitWebView *, gpointer);
+static WebKitWebView *client_new(const gchar *uri);
+static WebKitWebView *client_new_request(WebKitWebView *, WebKitWebFrame *,
+                                         gpointer);
 static void cooperation_setup(void);
 static void changed_load_status(GObject *obj, GParamSpec *pspec,
                                 gpointer data);
 static void changed_title(GObject *, GParamSpec *, gpointer);
 static void changed_uri(GObject *, GParamSpec *, gpointer);
+static gboolean download_handle(WebKitWebView *, WebKitDownload *, gpointer);
 static gboolean download_request(WebKitWebView *, WebKitWebFrame *,
                                  WebKitNetworkRequest *, gchar *,
                                  WebKitWebPolicyDecision *, gpointer);
-static gboolean download_wget(WebKitWebView *, WebKitDownload *, gpointer);
 static gchar *ensure_url_scheme(const gchar *);
 static void grab_environment_configuration(void);
 static void hover_web_view(WebKitWebView *, gchar *, gchar *, gpointer);
@@ -85,7 +84,7 @@ adblock(WebKitWebView *web_view, WebKitWebFrame *frame,
 
        uri = webkit_network_request_get_uri(request);
        if (show_all_requests)
-               fprintf(stderr, "-> %s\n", uri);
+               fprintf(stderr, "   -> %s\n", uri);
 
        while (it)
        {
@@ -93,7 +92,7 @@ adblock(WebKitWebView *web_view, WebKitWebFrame *frame,
                {
                        webkit_network_request_set_uri(request, "about:blank");
                        if (show_all_requests)
-                               fprintf(stderr, "\tBLOCKED!\n");
+                               fprintf(stderr, "            BLOCKED!\n");
                        return;
                }
                it = g_slist_next(it);
@@ -141,32 +140,38 @@ client_destroy(GtkWidget *obj, gpointer data)
        struct Client *c = (struct Client *)data;
 
        (void)obj;
+       (void)data;
 
-       webkit_web_view_stop_loading(WEBKIT_WEB_VIEW(c->web_view));
-       gtk_widget_destroy(c->web_view);
-       gtk_widget_destroy(c->scroll);
-       gtk_widget_destroy(c->status);
-       gtk_widget_destroy(c->location);
-       gtk_widget_destroy(c->vbox);
-       gtk_widget_destroy(c->win);
        free(c);
-
        clients--;
+
        if (clients == 0)
                gtk_main_quit();
 }
 
-void
+gboolean
+client_destroy_request(WebKitWebView *web_view, gpointer data)
+{
+       struct Client *c = (struct Client *)data;
+
+       (void)web_view;
+
+       gtk_widget_destroy(c->win);
+
+       return TRUE;
+}
+
+WebKitWebView *
 client_new(const gchar *uri)
 {
        struct Client *c;
        gchar *f;
 
-       if (cooperative_instances && !cooperative_alone)
+       if (uri != NULL && cooperative_instances && !cooperative_alone)
        {
                write(cooperative_pipe_fp, uri, strlen(uri));
                write(cooperative_pipe_fp, "\n", 1);
-               return;
+               return NULL;
        }
 
        c = malloc(sizeof(struct Client));
@@ -200,8 +205,7 @@ client_new(const gchar *uri)
         * to work. This is not needed when using Gtk3. */
        gtk_window_set_default_size(GTK_WINDOW(c->win), 1024, 768);
 
-       g_signal_connect(G_OBJECT(c->win), "destroy",
-                        G_CALLBACK(client_destroy), c);
+       g_signal_connect(G_OBJECT(c->win), "destroy", G_CALLBACK(client_destroy), c);
        gtk_window_set_title(GTK_WINDOW(c->win), __NAME__);
 
        c->web_view = webkit_web_view_new();
@@ -220,14 +224,15 @@ client_new(const gchar *uri)
                         G_CALLBACK(changed_uri), c);
        g_signal_connect(G_OBJECT(c->web_view), "notify::load-status",
                         G_CALLBACK(changed_load_status), c);
-       g_signal_connect(G_OBJECT(c->web_view),
-                        "new-window-policy-decision-requested",
+       g_signal_connect(G_OBJECT(c->web_view), "create-web-view",
                         G_CALLBACK(client_new_request), NULL);
+       g_signal_connect(G_OBJECT(c->web_view), "close-web-view",
+                        G_CALLBACK(client_destroy_request), c);
        g_signal_connect(G_OBJECT(c->web_view),
                         "mime-type-policy-decision-requested",
                         G_CALLBACK(download_request), NULL);
        g_signal_connect(G_OBJECT(c->web_view), "download-requested",
-                        G_CALLBACK(download_wget), NULL);
+                        G_CALLBACK(download_handle), NULL);
        g_signal_connect(G_OBJECT(c->web_view), "key-press-event",
                         G_CALLBACK(key_web_view), c);
        g_signal_connect(G_OBJECT(c->web_view), "button-press-event",
@@ -265,28 +270,28 @@ client_new(const gchar *uri)
        gtk_widget_grab_focus(c->web_view);
        gtk_widget_show_all(c->win);
 
-       f = ensure_url_scheme(uri);
-       webkit_web_view_load_uri(WEBKIT_WEB_VIEW(c->web_view), f);
-       g_free(f);
+       if (uri != NULL)
+       {
+               f = ensure_url_scheme(uri);
+               if (show_all_requests)
+                       fprintf(stderr, "====> %s\n", uri);
+               webkit_web_view_load_uri(WEBKIT_WEB_VIEW(c->web_view), f);
+               g_free(f);
+       }
 
        clients++;
+
+       return WEBKIT_WEB_VIEW(c->web_view);
 }
 
-gboolean
-client_new_request(WebKitWebView *web_view, WebKitWebFrame *frame,
-                   WebKitNetworkRequest *request,
-                   WebKitWebNavigationAction *navigation_action,
-                   WebKitWebPolicyDecision *policy_decision, gpointer user_data)
+WebKitWebView *
+client_new_request(WebKitWebView *web_view, WebKitWebFrame *frame, gpointer data)
 {
        (void)web_view;
        (void)frame;
-       (void)navigation_action;
-       (void)user_data;
-
-       webkit_web_policy_decision_ignore(policy_decision);
-       client_new(webkit_network_request_get_uri(request));
+       (void)data;
 
-       return TRUE;
+       return client_new(NULL);
 }
 
 void
@@ -370,57 +375,60 @@ changed_uri(GObject *obj, GParamSpec *pspec, gpointer data)
 }
 
 gboolean
-download_request(WebKitWebView *web_view, WebKitWebFrame *frame,
-                 WebKitNetworkRequest *request, gchar *mime_type,
-                 WebKitWebPolicyDecision *policy_decision, gpointer data)
+download_handle(WebKitWebView *web_view, WebKitDownload *download, gpointer data)
 {
-       (void)frame;
-       (void)request;
+       gchar *path, *path2 = NULL, *uri;
+       gboolean ret;
+       int suffix = 1;
+
+       (void)web_view;
        (void)data;
 
-       if (!webkit_web_view_can_show_mime_type(web_view, mime_type))
+       path = g_build_filename(download_dir,
+                               webkit_download_get_suggested_filename(download),
+                               NULL);
+       path2 = g_strdup(path);
+       while (g_file_test(path2, G_FILE_TEST_EXISTS) && suffix < 1000)
        {
-               webkit_web_policy_decision_download(policy_decision);
-               return TRUE;
+               g_free(path2);
+
+               path2 = g_strdup_printf("%s.%d", path, suffix);
+               suffix++;
        }
-       return FALSE;
+
+       if (suffix == 1000)
+       {
+               fprintf(stderr, __NAME__": Suffix reached limit for download.\n");
+               ret = FALSE;
+       }
+       else
+       {
+               uri = g_filename_to_uri(path2, NULL, NULL);
+               webkit_download_set_destination_uri(download, uri);
+               ret = TRUE;
+               g_free(uri);
+       }
+
+       g_free(path);
+       g_free(path2);
+
+       return ret;
 }
 
 gboolean
-download_wget(WebKitWebView *web_view, WebKitDownload *download, gpointer data)
+download_request(WebKitWebView *web_view, WebKitWebFrame *frame,
+                 WebKitNetworkRequest *request, gchar *mime_type,
+                 WebKitWebPolicyDecision *policy_decision, gpointer data)
 {
-       const gchar *uri;
-       char id[16] = "";
-       gint ret;
-
-       (void)web_view;
+       (void)frame;
+       (void)request;
        (void)data;
 
-       uri = webkit_download_get_uri(download);
-       if (fork() == 0)
+       if (!webkit_web_view_can_show_mime_type(web_view, mime_type))
        {
-               chdir(download_dir);
-               if (embed == 0)
-                       ret = execlp("xterm", "xterm", "-hold", "-e", "wget", uri, NULL);
-               else
-               {
-                       if (snprintf(id, 16, "%ld", embed) >= 16)
-                       {
-                               fprintf(stderr, __NAME__": id for xterm embed truncated!\n");
-                               exit(EXIT_FAILURE);
-                       }
-                       ret = execlp("xterm", "xterm", "-hold", "-into", id, "-e", "wget",
-                                    uri, NULL);
-               }
-
-               if (ret == -1)
-               {
-                       fprintf(stderr, __NAME__": exec'ing xterm for download");
-                       perror(" failed");
-                       exit(EXIT_FAILURE);
-               }
+               webkit_web_policy_decision_download(policy_decision);
+               return TRUE;
        }
-
        return FALSE;
 }
 
@@ -431,9 +439,9 @@ ensure_url_scheme(const gchar *t)
 
        f = g_ascii_strdown(t, -1);
        if (!g_str_has_prefix(f, "http:") &&
-               !g_str_has_prefix(f, "https:") &&
-               !g_str_has_prefix(f, "file:") &&
-               !g_str_has_prefix(f, "about:"))
+           !g_str_has_prefix(f, "https:") &&
+           !g_str_has_prefix(f, "file:") &&
+           !g_str_has_prefix(f, "about:"))
        {
                g_free(f);
                f = g_strdup_printf("http://%s", t);
@@ -450,26 +458,19 @@ grab_environment_configuration(void)
 
        e = g_getenv(__NAME_UPPERCASE__"_ACCEPTED_LANGUAGE");
        if (e != NULL)
-       {
                accepted_language = g_strdup(e);
-       }
 
        e = g_getenv(__NAME_UPPERCASE__"_DOWNLOAD_DIR");
        if (e != NULL)
-       {
                download_dir = g_strdup(e);
-       }
 
        e = g_getenv(__NAME_UPPERCASE__"_ZOOM");
        if (e != NULL)
-       {
                global_zoom = atof(e);
-       }
 }
 
 void
-hover_web_view(WebKitWebView *web_view, gchar *title, gchar *uri,
-                   gpointer data)
+hover_web_view(WebKitWebView *web_view, gchar *title, gchar *uri, gpointer data)
 {
        struct Client *c = (struct Client *)data;
 
@@ -507,6 +508,8 @@ key_location(GtkWidget *widget, GdkEvent *event, gpointer data)
                                else
                                {
                                        f = ensure_url_scheme(t);
+                                       if (show_all_requests)
+                                               fprintf(stderr, "====> %s\n", f);
                                        webkit_web_view_load_uri(WEBKIT_WEB_VIEW(c->web_view), f);
                                        g_free(f);
                                }
@@ -528,7 +531,7 @@ key_web_view(GtkWidget *widget, GdkEvent *event, gpointer data)
        struct Client *c = (struct Client *)data;
        WebKitHitTestResultContext ht_context;
        WebKitHitTestResult *ht_result = NULL;
-       char *ht_uri =  NULL;
+       gchar *ht_uri = NULL, *f;
 
        (void)widget;
 
@@ -572,8 +575,11 @@ key_web_view(GtkWidget *widget, GdkEvent *event, gpointer data)
                                        search(c, -1);
                                        return TRUE;
                                case GDK_KEY_g:
-                                       webkit_web_view_load_uri(WEBKIT_WEB_VIEW(c->web_view),
-                                                                first_uri);
+                                       f = ensure_url_scheme(first_uri);
+                                       if (show_all_requests)
+                                               fprintf(stderr, "====> %s\n", f);
+                                       webkit_web_view_load_uri(WEBKIT_WEB_VIEW(c->web_view), f);
+                                       g_free(f);
                                        return TRUE;
                        }
                }